TFS Execute Custom Code on a Work Item Transition

James Orr picture James Orr · Mar 1, 2011 · Viewed 7.4k times · Source

I'd like TFS 2010 to run a bit of custom code whenever a particular workflow transition happens. Is that possible?

I've found documentation about Custom Actions, which seem to be actions that can automatically trigger work item transitions (am I getting that right?) I also found Custom Activities, which are related to Builds. But nothing that serves this particular requirement - am I missing something?

Thanks for your help!

Answer

Vaccano picture Vaccano · Mar 2, 2011

This is very doable.

It is so doable, that there are many ways to do it. One of my favorites is to make a server side plugin. (Note, this only works on TFS 2010)

These blog posts show the basics:

Here is some code that I have modified from my open source project TFS Aggregator:

public class WorkItemChangedEventHandler : ISubscriber
{       
    /// <summary>
    /// This is the one where all the magic starts.  Main() so to speak.
    /// </summary>
    public EventNotificationStatus ProcessEvent(TeamFoundationRequestContext requestContext, NotificationType notificationType, object notificationEventArgs,
                                                out int statusCode, out string statusMessage, out ExceptionPropertyCollection properties)
    {
        statusCode = 0;
        properties = null;
        statusMessage = String.Empty;
        try
        {
            if (notificationType == NotificationType.Notification && notificationEventArgs is WorkItemChangedEvent)
            {
                // Change this object to be a type we can easily get into
                WorkItemChangedEvent ev = notificationEventArgs as WorkItemChangedEvent;
                // Connect to the setting file and load the location of the TFS server
                string tfsUri = TFSAggregatorSettings.TFSUri;
                // Connect to TFS so we are ready to get and send data.
                Store store = new Store(tfsUri);
                // Get the id of the work item that was just changed by the user.
                int workItemId = ev.CoreFields.IntegerFields[0].NewValue;
                // Download the work item so we can update it (if needed)
                WorkItem eventWorkItem = store.Access.GetWorkItem(workItemId);

                if ((string)(eventWorkItem.Fields["State"].Value) == "Done")
                    {
                        // If the estimated work was changed then revert it back.  
                        // We are in done and don't want to allow changes like that.
                        foreach (IntegerField integerField in ev.ChangedFields.IntegerFields)
                        {
                            if (integerField.Name == "Estimated Work")
                            {
                                eventWorkItem.Open();
                                eventWorkItem.Fields["Estimated Work"].Value = integerField.OldValue;
                                eventWorkItem.Save();
                            }
                        }
                    }
                }
            }

        }
        return EventNotificationStatus.ActionPermitted;
    }

    public string Name
    {
        get { return "SomeName"; }
    }

    public SubscriberPriority Priority
    {
        get { return SubscriberPriority.Normal; }
    }

    public WorkItemChangedEventHandler()
    {
        //DON"T ADD ANYTHING HERE UNLESS YOU REALLY KNOW WHAT YOU ARE DOING.
        //TFS DOES NOT LIKE CONSTRUCTORS HERE AND SEEMS TO FREEZE WHEN YOU TRY :(
    }

    public Type[] SubscribedTypes()
    {
        return new Type[1] { typeof(WorkItemChangedEvent) };
    }
}

/// <summary>
/// Singleton Used to access TFS Data.  This keeps us from connecting each and every time we get an update.
/// </summary>
public class Store
{
    private readonly string _tfsServerUrl;
    public Store(string tfsServerUrl)
    {
        _tfsServerUrl = tfsServerUrl;
    }

    private TFSAccess _access;
    public TFSAccess Access
    {
        get { return _access ?? (_access = new TFSAccess(_tfsServerUrl)); }
    }
}

/// <summary>
/// Don't use this class directly.  Use the StoreSingleton.
/// </summary>
public class TFSAccess
{
    private readonly WorkItemStore _store;
    public TFSAccess(string tfsUri)
    {
        TfsTeamProjectCollection tfs = new TfsTeamProjectCollection(new Uri(tfsUri));
        _store = (WorkItemStore)tfs.GetService(typeof(WorkItemStore));
    }

    public WorkItem GetWorkItem(int workItemId)
    {
        return _store.GetWorkItem(workItemId);
    }
}