[JENKINS-10706] Add env variable MERCURIAL_REVISION_BRANCH

jenkinsci · Mar 31, 2015 · 1140b3b · 1140b3b
1 parent 3abf51e
commit 1140b3b
Show file tree

Hide file tree

Showing 7 changed files with 46 additions and 11 deletions.
diff --git a/src/main/java/hudson/plugins/mercurial/HgExe.java b/src/main/java/hudson/plugins/mercurial/HgExe.java
@@ -345,6 +345,22 @@ private Set<String> heads(FilePath repo, boolean useTimeout, boolean usingHg15Sy
         return id;
     }
 
+    /**
+     * Gets the branch name of given revision number or of the current workspace.
+     * @param rev the revision to identify; defaults to current working copy
+     */
+    public @CheckForNull String branch(FilePath repository, @Nullable String rev) throws IOException, InterruptedException {
+        ArgumentListBuilder builder = new ArgumentListBuilder("id", "--branch");
+        if (rev != null)
+            builder.add("--rev", rev);
+        String branch = popen(repository, listener, false, builder).trim();
+        if (branch.isEmpty()) {
+            listener.error(Messages.HgExe_expected_to_get_a_branch_name_but_got_nothing());
+            return null;
+        }
+        return branch;
+    }
+
     /**
      * Gets the current value of a specified config item.
      */

diff --git a/src/main/java/hudson/plugins/mercurial/MercurialSCM.java b/src/main/java/hudson/plugins/mercurial/MercurialSCM.java
@@ -73,6 +73,7 @@ public class MercurialSCM extends SCM implements Serializable {
     private static final String ENV_MERCURIAL_REVISION = "MERCURIAL_REVISION";
     private static final String ENV_MERCURIAL_REVISION_SHORT = "MERCURIAL_REVISION_SHORT";
     private static final String ENV_MERCURIAL_REVISION_NUMBER = "MERCURIAL_REVISION_NUMBER";
+    private static final String ENV_MERCURIAL_REVISION_BRANCH = "MERCURIAL_REVISION_BRANCH";
     private static final String ENV_MERCURIAL_REPOSITORY_URL = "MERCURIAL_REPOSITORY_URL";
 
     // old fields are left so that old config data can be read in, but
@@ -352,7 +353,8 @@ public SCMRevisionState calcRevisionsFromBuild(Run<?, ?> build, FilePath workspa
         try {
         String tip = hg.tip(workspace2Repo(workspace, env), null);
         String rev = hg.tipNumber(workspace2Repo(workspace, env), null);
-        return tip != null && rev != null ? new MercurialTagAction(tip, rev, getSubdir(env)) : null;
+        String branch = revisionType == RevisionType.TAG ? hg.branch(workspace2Repo(workspace, env), null) : null;
+        return tip != null && rev != null ? new MercurialTagAction(tip, rev, getSubdir(env), branch) : null;
         } finally {
             hg.close();
         }
@@ -420,6 +422,7 @@ PollingResult compare(Launcher launcher, TaskListener listener, MercurialTagActi
         String _revision = getRevisionExpanded(project, env);
         String remote = hg.tip(repository, _revision);
         String rev = hg.tipNumber(repository, _revision);
+        String branch = revisionType == RevisionType.TAG ? hg.branch(repository, _revision) : null;
 
         if (remote == null) {
             throw new IOException("failed to find ID of branch head");
@@ -428,11 +431,11 @@ PollingResult compare(Launcher launcher, TaskListener listener, MercurialTagActi
             throw new IOException("failed to find revision of branch head");
         }
         if (remote.equals(baseline.id)) { // shortcut
-            return new PollingResult(baseline, new MercurialTagAction(remote, rev, getSubdir(env)), Change.NONE);
+            return new PollingResult(baseline, new MercurialTagAction(remote, rev, getSubdir(env), branch), Change.NONE);
         }
         Set<String> changedFileNames = parseStatus(hg.popen(repository, listener, false, new ArgumentListBuilder("status", "--rev", baseline.id, "--rev", remote)));
 
-        MercurialTagAction cur = new MercurialTagAction(remote, rev, getSubdir(env));
+        MercurialTagAction cur = new MercurialTagAction(remote, rev, getSubdir(env), branch);
         return new PollingResult(baseline,cur,computeDegreeOfChanges(changedFileNames,output));
         } finally {
             hg.close();
@@ -709,8 +712,9 @@ private void update(Run<?, ?> build, Launcher launcher, FilePath repository, Nod
 
         String tip = hg.tip(repository, null);
         String rev = hg.tipNumber(repository, null);
+        String branch = revisionType == RevisionType.TAG ? hg.branch(repository, null) : null;
         if (tip != null && rev != null) {
-            build.addAction(new MercurialTagAction(tip, rev, getSubdir(env)));
+            build.addAction(new MercurialTagAction(tip, rev, getSubdir(env), branch));
         }
         } finally {
             hg.close();
@@ -820,8 +824,9 @@ private void clone(Run<?, ?> build, Launcher launcher, FilePath repository, Node
 
         String tip = hg.tip(repository, null);
         String rev = hg.tipNumber(repository, null);
+        String branch = revisionType == RevisionType.TAG ? hg.branch(repository, null) : null;
         if (tip != null && rev != null) {
-            build.addAction(new MercurialTagAction(tip, rev, getSubdir(env)));
+            build.addAction(new MercurialTagAction(tip, rev, getSubdir(env), branch));
         }
         } finally {
             hg.close();
@@ -839,6 +844,8 @@ void buildEnvVarsFromActionable(Actionable build, Map<String, String> env) {
             env.put(ENV_MERCURIAL_REVISION, a.id);
             env.put(ENV_MERCURIAL_REVISION_SHORT, a.getShortId());
             env.put(ENV_MERCURIAL_REVISION_NUMBER, a.rev);
+            if (revisionType == RevisionType.TAG)
+                env.put(ENV_MERCURIAL_REVISION_BRANCH, a.getBranch());
             env.put(ENV_MERCURIAL_REPOSITORY_URL, this.getSource());
         }
     }

diff --git a/src/main/java/hudson/plugins/mercurial/MercurialTagAction.java b/src/main/java/hudson/plugins/mercurial/MercurialTagAction.java
@@ -32,10 +32,16 @@ public class MercurialTagAction extends SCMRevisionState {
      */
     public final String subdir;
 
-    public MercurialTagAction(@NonNull String id, @NonNull String rev, @Nullable String subdir) {
+    /**
+    * Branch name of current revision.
+    */
+    public final String branch;
+
+    public MercurialTagAction(@NonNull String id, @NonNull String rev, @Nullable String subdir, @Nullable String branch) {
         this.id = id;
         this.rev = rev;
         this.subdir = subdir;
+        this.branch = branch;
     }
 
     @Exported(name = "mercurialNodeName")
@@ -53,6 +59,11 @@ public String getSubdir() {
         return subdir;
     }
 
+    @Exported(name = "mercurialRevisionBranch")
+    public String getBranch() {
+        return branch;
+    }
+
     /**
      * Mercurial often uses a short ID form that consists of 12 letters.
      */

diff --git a/src/main/resources/hudson/plugins/mercurial/Messages.properties b/src/main/resources/hudson/plugins/mercurial/Messages.properties
@@ -1,9 +1,10 @@
 HgExe.expected_to_get_a_revision_number_but_got_instead=Expected to get a revision number but got ''{0}'' instead.
+HgExe.expected_to_get_a_branch_name_but_got_nothing=Expected to get a branch name but got an empty name instead.
 MercurialInstallation.mercurial=Mercurial
 MercurialSCM.dependent_changes_detected=Dependent changes detected
 MercurialSCM.failed_to_clone=Failed to clone {0}
 MercurialSCM.failed_to_compare_with_remote_repository=Failed to compare with remote repository because hg could not be found; check that you''ve properly configured your Mercurial installation
 MercurialSCM.non_dependent_changes_detected=Non-dependent changes detected
 MercurialStatus.mercurial=Mercurial
 MercurialTagAction.BuildData=Mercurial Build Data
-MercurialRevisionColumn.DisplayName=Mercurial Branch
+MercurialRevisionColumn.DisplayName=Mercurial Branch
diff --git a/src/test/java/hudson/plugins/mercurial/ChangeComparatorTest.java b/src/test/java/hudson/plugins/mercurial/ChangeComparatorTest.java
@@ -54,7 +54,7 @@ public Change compare(MercurialSCM scm, Launcher launcher, TaskListener listener
         PollingResult pr = scm.compare(
 				launcher, 
 				listener, 
-				new MercurialTagAction("tip","",null), 
+				new MercurialTagAction("tip","",null,"default"),
 				listener.getLogger(), 
 				Computer.currentComputer().getNode(), 
 				new FilePath(tmp.getRoot()), 

diff --git a/src/test/java/hudson/plugins/mercurial/MercurialSCMTest.java b/src/test/java/hudson/plugins/mercurial/MercurialSCMTest.java
@@ -52,7 +52,7 @@ public class MercurialSCMTest {
         final String EXPECTED_SHORT_ID = "123456789012";
         new MercurialSCM("","","", "", "", null, true).buildEnvVarsFromActionable(new Actionable() {
             @Override public List<Action> getActions() {
-                return Collections.<Action>singletonList(new MercurialTagAction(EXPECTED_SHORT_ID + "1627e63489b4096a8858e559a456", "rev", null));
+                return Collections.<Action>singletonList(new MercurialTagAction(EXPECTED_SHORT_ID + "1627e63489b4096a8858e559a456", "rev", null, "default"));
             }
             @Override public String getDisplayName() {return null;}
             @Override public String getSearchUrl() {return null;}
@@ -65,7 +65,7 @@ public class MercurialSCMTest {
         final String EXPECTED_REPOSITORY_URL = "http://mercurialserver/testrepo";
         new MercurialSCM("",EXPECTED_REPOSITORY_URL,"", "", "", null, true).buildEnvVarsFromActionable(new Actionable() {
             @Override public List<Action> getActions() {
-                return Collections.<Action>singletonList(new MercurialTagAction("1627e63489b4096a8858e559a456", "rev", null));            }
+                return Collections.<Action>singletonList(new MercurialTagAction("1627e63489b4096a8858e559a456", "rev", null, "default"));            }
             @Override public String getDisplayName() {return null;}
             @Override public String getSearchUrl() {return null;}
         }, actualEnvironment);

diff --git a/src/test/java/hudson/plugins/mercurial/MercurialTagActionTest.java b/src/test/java/hudson/plugins/mercurial/MercurialTagActionTest.java
@@ -6,7 +6,7 @@
 
 public class MercurialTagActionTest {
     @Test public void getShortIdReturnsFirstTwelveCharactersOfId(){
-        MercurialTagAction action = new MercurialTagAction("1234567890121627e63489b4096a8858e559a456", "", "");
+        MercurialTagAction action = new MercurialTagAction("1234567890121627e63489b4096a8858e559a456", "", "", "default");
 
         assertEquals("123456789012", action.getShortId());
     }