[WIP] workflow support?

src/main/java/org/jenkinsci/plugins/github/pullrequest/builders/GitHubPRStatusBuilder.java

@@ -1,15 +1,19 @@
 package org.jenkinsci.plugins.github.pullrequest.builders;
 
 import hudson.Extension;
+import hudson.FilePath;
 import hudson.Launcher;
 import hudson.model.AbstractBuild;
 import hudson.model.AbstractProject;
-import hudson.model.BuildListener;
+import hudson.model.Run;
+import hudson.model.TaskListener;
 import hudson.tasks.BuildStepDescriptor;
 import hudson.tasks.Builder;
+import jenkins.tasks.SimpleBuildStep;
 import org.jenkinsci.plugins.github.pullrequest.GitHubPRCause;
 import org.jenkinsci.plugins.github.pullrequest.GitHubPRMessage;
 import org.jenkinsci.plugins.github.pullrequest.GitHubPRTrigger;
+import org.jenkinsci.plugins.github.util.JobInfoHelpers;
 import org.kohsuke.accmod.Restricted;
 import org.kohsuke.accmod.restrictions.NoExternalUse;
 import org.kohsuke.github.GHCommitState;
@@ -18,14 +22,15 @@
 import org.slf4j.LoggerFactory;
 
 import javax.annotation.CheckForNull;
+import javax.annotation.Nonnull;
 import java.io.IOException;
 
 import static org.apache.commons.lang3.StringUtils.isNotBlank;
 
 /**
  * Sets pr status for build caused by GitHubPRCause
  */
-public class GitHubPRStatusBuilder extends Builder {
+public class GitHubPRStatusBuilder extends Builder implements SimpleBuildStep {
     private static final Logger LOGGER = LoggerFactory.getLogger(GitHubPRStatusBuilder.class);
 
     public static final GitHubPRMessage DEFAULT_MESSAGE = new GitHubPRMessage("${GITHUB_PR_COND_REF} run started");
@@ -52,41 +57,52 @@ public GitHubPRMessage getStatusMessage() {
     }
 
     @Override
-    public boolean perform(AbstractBuild<?, ?> build, Launcher launcher, BuildListener listener)
-            throws InterruptedException, IOException {
-        GitHubPRTrigger trigger = build.getProject().getTrigger(GitHubPRTrigger.class);
+    public void perform(@Nonnull Run<?, ?> run, @Nonnull FilePath workspace, @Nonnull Launcher launcher,
+                        @Nonnull TaskListener listener) throws InterruptedException, IOException {
+        // No triggers in Run class, but we need it
+        final GitHubPRTrigger trigger = JobInfoHelpers.triggerFrom(run.getParent(), GitHubPRTrigger.class);
         if (trigger == null) {
-            return true;
+            // silently skip. TODO implement error handler, like in publishers
+            return;
         }
 
-        GitHubPRCause cause = build.getCause(GitHubPRCause.class);
+        GitHubPRCause cause = run.getCause(GitHubPRCause.class);
         if (cause == null) {
-            return true;
+            return;
         }
 
         // GitHub status for commit
         try {
             if (statusMessage != null) {
-                String url = trigger.getDescriptor().getJenkinsURL() + build.getUrl();
-
-                trigger.getRemoteRepo().createCommitStatus(cause.getHeadSha(),
+                String url = trigger.getDescriptor().getJenkinsURL() + run.getUrl();
+                String expandedMessage;
+                if (run instanceof AbstractBuild<?, ?>) {
+                    final AbstractBuild<?, ?> build = (AbstractBuild<?, ?>) run;
+                    expandedMessage = statusMessage.expandAll(build, listener);
+                } else {
+                    // Workflow doesn't expand variables with token macro!
+                    expandedMessage = statusMessage.getContent();
+                }
+
+                trigger.getRemoteRepo().createCommitStatus(
+                        cause.getHeadSha(),
                         GHCommitState.PENDING,
                         url,
-                        statusMessage.expandAll(build, listener),
-                        build.getProject().getFullName());
+                        expandedMessage,
+                        run.getParent().getFullName()
+                );
             }
         } catch (IOException e) {
             listener.getLogger().println("Can't update build description");
             LOGGER.error("Can't set commit status", e);
         }
-
-        return true;
     }
 
     @Extension
     public static class DescriptorImpl extends BuildStepDescriptor<Builder> {
         @Override
         public boolean isApplicable(Class<? extends AbstractProject> jobType) {
+            // workflow wants Run, but API doesn't support it
             return true;
         }