Merge pull request jenkinsci#404 from mmitche/build-comment-extensions

Add job dsl extensions for build status messages

README.md

@@ -157,6 +157,11 @@ job('upstreamJob') {
                     completedStatus('PENDING', 'still in progress...')
                     completedStatus('ERROR', 'Something went really wrong. Investigate!')
                 }
+                buildStatus {
+                    completedStatus('SUCCESS', 'There were no errors, go have a cup of coffee...')
+                    completedStatus('FAILURE', 'There were errors, for info, please see...')
+                    completedStatus('ERROR', 'There was an error in the infrastructure, please contact...')
+                }
             }
         }
     }

src/main/java/org/jenkinsci/plugins/ghprb/jobdsl/GhprbBuildStatusContext.java

@@ -0,0 +1,23 @@
+package org.jenkinsci.plugins.ghprb.jobdsl;
+
+import javaposse.jobdsl.dsl.Context;
+import org.jenkinsci.plugins.ghprb.extensions.comments.GhprbBuildResultMessage;
+import org.kohsuke.github.GHCommitState;
+
+import java.util.ArrayList;
+import java.util.List;
+
+class GhprbBuildStatusContext implements Context {
+    List<GhprbBuildResultMessage> completedStatus = new ArrayList<GhprbBuildResultMessage>();
+
+    /**
+     * Use a custom status for when a build is completed. Can be called multiple times to set messages for different
+     * build results. Valid build results are {@code 'SUCCESS'}, {@code 'FAILURE'}, and {@code 'ERROR'}.
+     */
+    void completedStatus(String buildResult, String message) {
+        completedStatus.add(new GhprbBuildResultMessage(
+                GHCommitState.valueOf(buildResult),
+                message
+        ));
+    }
+}

src/main/java/org/jenkinsci/plugins/ghprb/jobdsl/GhprbExtensionContext.java

@@ -4,6 +4,7 @@
 import javaposse.jobdsl.plugin.ContextExtensionPoint;
 import org.jenkinsci.plugins.ghprb.extensions.GhprbExtension;
 import org.jenkinsci.plugins.ghprb.extensions.status.GhprbSimpleStatus;
+import org.jenkinsci.plugins.ghprb.extensions.comments.GhprbBuildStatus;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -28,4 +29,16 @@ void commitStatus(Runnable closure) {
                 context.completedStatus
         ));
     }
+
+    /**
+     * Adds build result messages
+     */
+    void buildStatus(Runnable closure) {
+        GhprbBuildStatusContext context = new GhprbBuildStatusContext();
+        ContextExtensionPoint.executeInContext(closure, context);
+
+        extensions.add(new GhprbBuildStatus(
+                context.completedStatus
+        ));
+    }
 }