Update VSTS PR Jenkins build status back to VSTS

tfs/src/main/java/hudson/plugins/tfs/JenkinsEventNotifier.java

@@ -1,7 +1,11 @@
 package hudson.plugins.tfs;
 
 import hudson.plugins.tfs.model.ConnectionParameters;
+import hudson.plugins.tfs.model.GitStatusContext;
+import hudson.plugins.tfs.model.GitStatusState;
 import hudson.plugins.tfs.model.JobCompletionEventArgs;
+import hudson.plugins.tfs.model.PullRequestMergeCommitCreatedEventArgs;
+import hudson.plugins.tfs.model.TeamGitStatus;
 import hudson.plugins.tfs.util.TeamRestClient;
 import jenkins.model.Jenkins;
 import net.sf.json.JSONObject;
@@ -17,6 +21,8 @@
 import java.io.IOException;
 import java.io.InputStreamReader;
 import java.io.UnsupportedEncodingException;
+import java.net.HttpURLConnection;
+import java.net.MalformedURLException;
 import java.net.URI;
 import java.security.InvalidKeyException;
 import java.security.NoSuchAlgorithmException;
@@ -61,6 +67,46 @@ public static void sendJobCompletionEvent(final JSONObject payload) {
         }
     }
 
+    /**
+     * Send the build status of VSTS pull request back to connected TFS/VSTS servers only for those jobs kicked off by PRs.
+     */
+    public static void sendPullRequestBuildStatusEvent(final CommitParameterAction commitParameter, final GitStatusState buildState, final String buildDescription, final String targetUrl, final String jobFullPath) {
+        try {
+            if (commitParameter != null) {
+                if (commitParameter instanceof PullRequestParameterAction) {
+                    final PullRequestParameterAction prpa = (PullRequestParameterAction) commitParameter;
+                    final PullRequestMergeCommitCreatedEventArgs pullRequestMergeCommitCreatedEventArgs = prpa.getPullRequestMergeCommitCreatedEventArgs();
+                    sendPullRequestBuildStatusEvent(pullRequestMergeCommitCreatedEventArgs, buildState, buildDescription, targetUrl, jobFullPath);
+                }
+            }
+        } catch (final Exception e) {
+            log.warning("ERROR: sendPullRequestBuildStatusEvent fails due to exception: " + e.getMessage());
+        }
+        return;
+    }
+
+    /**
+     * Send the build status of VSTS pull request back to connected TFS/VSTS servers.
+     */
+    public static void sendPullRequestBuildStatusEvent(final PullRequestMergeCommitCreatedEventArgs gitCodePushedEventArgs, final GitStatusState buildState, final String buildDescription, final String targetUrl, final String jobFullPath) {
+        try {
+            final TeamGitStatus status = new TeamGitStatus();
+            status.state = buildState;
+            status.description = buildDescription;
+            status.targetUrl = targetUrl;
+            status.context = new GitStatusContext(" ", StringUtils.stripEnd(jobFullPath, "/"));
+
+            final URI collectionUri = gitCodePushedEventArgs.collectionUri;
+            final TeamRestClient client = new TeamRestClient(collectionUri);
+            client.addPullRequestStatus(gitCodePushedEventArgs, status);
+        } catch (MalformedURLException e) {
+            log.warning("ERROR: sendPullRequestBuildStatusEvent fails due to MalformedURLException: " + e.getMessage());
+        } catch (IOException e) {
+            log.warning("ERROR: sendPullRequestBuildStatusEvent fails due to IOException: " + e.getMessage());
+        }
+        return;
+    }
+
     /**
      * This is a helper method to get the JSON for a Jenkins object.
      * @param url
@@ -81,17 +127,22 @@ public static String getApiJson(final String url) {
             request.addHeader("User-Agent", "Jenkins-Self");
 
             final HttpResponse response = client.execute(request);
-
-            try (BufferedReader reader = new BufferedReader(
-                    new InputStreamReader(response.getEntity().getContent(), ENCODING))) {
-
-                final StringBuilder result = new StringBuilder();
-                String line;
-                while ((line = reader.readLine()) != null) {
-                    result.append(line);
-                    result.append("\n");
+            final int statusCode = response.getStatusLine().getStatusCode();
+
+            if (statusCode <= HttpURLConnection.HTTP_ACCEPTED) {
+                try (BufferedReader reader = new BufferedReader(
+                        new InputStreamReader(response.getEntity().getContent(), ENCODING))) {
+                    final StringBuilder result = new StringBuilder();
+                    String line;
+                    while ((line = reader.readLine()) != null) {
+                        result.append(line);
+                        result.append("\n");
+                    }
+                    return result.toString();
                 }
-                return result.toString();
+            } else {
+                log.warning("ERROR: getApiJson: (url=" + url + ") failed due to Http error #" + statusCode);
+                return null;
             }
         } catch (final IOException e) {
             log.warning("ERROR: getApiJson: (url=" + url + ") " + e.getMessage());

tfs/src/main/java/hudson/plugins/tfs/SafeParametersAction.java

@@ -0,0 +1,66 @@
+package hudson.plugins.tfs;
+
+import hudson.EnvVars;
+import hudson.Extension;
+import hudson.model.EnvironmentContributor;
+import hudson.model.ParameterValue;
+import hudson.model.ParametersAction;
+import hudson.model.Run;
+import hudson.model.TaskListener;
+import org.kohsuke.accmod.Restricted;
+import org.kohsuke.accmod.restrictions.NoExternalUse;
+
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.List;
+
+/**
+ * Extension class of ParametersAction to pass in parameters as safe parameters..
+ */
+@Restricted(NoExternalUse.class)
+public class SafeParametersAction extends ParametersAction {
+
+    private List<ParameterValue> parameters;
+
+    public SafeParametersAction(final List<ParameterValue> parameters) {
+        this.parameters = parameters;
+    }
+
+    public SafeParametersAction(final ParameterValue... parameters) {
+        this(Arrays.asList(parameters));
+    }
+
+    @Override
+    public List<ParameterValue> getParameters() {
+        return Collections.unmodifiableList(parameters);
+    }
+
+    @Override
+    public ParameterValue getParameter(final String name) {
+        for (ParameterValue parameter : parameters) {
+            if (parameter != null && parameter.getName().equals(name)) {
+                return parameter;
+            }
+        }
+
+        return null;
+    }
+
+    /**
+    * Environment contributor for SafeParametersAction.
+    */
+    @Extension
+    public static final class SafeParametersActionEnvironmentContributor extends EnvironmentContributor {
+
+        @Override
+        public void buildEnvironmentFor(final Run r, final EnvVars envs, final TaskListener listener) throws IOException, InterruptedException {
+            SafeParametersAction action = r.getAction(SafeParametersAction.class);
+            if (action != null) {
+                for (ParameterValue p : action.getParameters()) {
+                    envs.putIfNotNull(p.getName(), String.valueOf(p.getValue()));
+                }
+            }
+        }
+    }
+}

tfs/src/main/java/hudson/plugins/tfs/TeamEventsEndpoint.java

@@ -34,7 +34,9 @@
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.PrintWriter;
+import java.util.ArrayList;
 import java.util.Collections;
+import java.util.List;
 import java.util.Map;
 import java.util.TreeMap;
 import java.util.logging.Level;
@@ -235,6 +237,20 @@ public static <T extends Trigger> T findTrigger(final Job<?, ?> job, final Class
         return null;
     }
 
+    // A job may have multiple triggers of the same type. For example, both TeamPRPushTrigger and TeamPushTrigger are TeamPushTrigger type.
+    public static <T extends Trigger> List<T> findTriggers(final Job<?, ?> job, final Class<T> tClass) {
+        List<T> triggerList = new ArrayList<>();
+        if (job instanceof ParameterizedJobMixIn.ParameterizedJob) {
+            final ParameterizedJobMixIn.ParameterizedJob pJob = (ParameterizedJobMixIn.ParameterizedJob) job;
+            for (final Trigger trigger : pJob.getTriggers().values()) {
+                if (tClass.isInstance(trigger)) {
+                    triggerList.add(tClass.cast(trigger));
+                }
+            }
+        }
+        return triggerList;
+    }
+
     /**
      * A response contributor for triggering polling of a project.
      *

tfs/src/main/java/hudson/plugins/tfs/TeamPRPushTrigger.java

@@ -0,0 +1,46 @@
+package hudson.plugins.tfs;
+
+import hudson.Extension;
+import hudson.model.Job;
+import org.kohsuke.stapler.DataBoundConstructor;
+import org.kohsuke.stapler.DataBoundSetter;
+
+/**
+ * Triggers a build when we receive a TFS/Team Services Git code push event in a TFS pull request.
+ */
+public class TeamPRPushTrigger extends TeamPushTrigger {
+
+    private String targetBranches;
+
+    @DataBoundConstructor
+    public TeamPRPushTrigger() {
+    }
+
+    public TeamPRPushTrigger(final Job<?, ?> job, final String targetBranches, final String jobContext) {
+        this.job = job;
+        this.targetBranches = targetBranches;
+
+        setJobContext(jobContext);
+    }
+
+    public String getTargetBranches() {
+        return targetBranches;
+    }
+
+    @DataBoundSetter
+    public void setTargetBranches(final String targetBranches) {
+        this.targetBranches = targetBranches;
+    }
+
+    /**
+     * This class extends trigger descriptor class from TeamPushTrigger, creating a separate check box for TeamPRPushTrigger.
+     */
+    @Extension
+    public static class DescriptorImpl extends TeamPushTrigger.DescriptorImpl {
+
+        @Override
+        public String getDisplayName() {
+            return "Build when a change is pushed to a TFS pull request";
+        }
+    }
+}

tfs/src/main/java/hudson/plugins/tfs/TeamPushCause.java

@@ -12,19 +12,22 @@
 public class TeamPushCause extends SCMTriggerCause {
 
     private final String pushedBy;
+    private final String context;
 
-    public TeamPushCause(final String pushedBy) {
-        this("", pushedBy);
+    public TeamPushCause(final String pushedBy, final String context) {
+        this("", pushedBy, context);
     }
 
-    public TeamPushCause(final File logFile, final String pushedBy) throws IOException {
+    public TeamPushCause(final File logFile, final String pushedBy, final String context) throws IOException {
         super(logFile);
         this.pushedBy = pushedBy;
+        this.context = context;
     }
 
-    public TeamPushCause(final String pollingLog, final String pushedBy) {
+    public TeamPushCause(final String pollingLog, final String pushedBy, final String context) {
         super(pollingLog);
         this.pushedBy = pushedBy;
+        this.context = context;
     }
 
     @Override
@@ -43,4 +46,8 @@ public boolean equals(final Object o) {
     public int hashCode() {
         return super.hashCode();
     }
+
+    public String getRunContext() {
+        return this.context;
+    }
 }