From c3b286a7459615e1062ca80d1919af85afb8dab3 Mon Sep 17 00:00:00 2001
From: Sebastian Sdorra <sebastian.sdorra@cloudogu.com>
Date: Tue, 10 Jul 2012 20:51:36 +0200
Subject: [PATCH] implement branches command for hg

---
 .../scm/repository/spi/HgBranchesCommand.java | 100 ++++++++++++++++++
 .../spi/HgRepositoryServiceProvider.java      |  15 ++-
 2 files changed, 114 insertions(+), 1 deletion(-)
 create mode 100644 scm-plugins/scm-hg-plugin/src/main/java/sonia/scm/repository/spi/HgBranchesCommand.java

diff --git a/scm-plugins/scm-hg-plugin/src/main/java/sonia/scm/repository/spi/HgBranchesCommand.java b/scm-plugins/scm-hg-plugin/src/main/java/sonia/scm/repository/spi/HgBranchesCommand.java
new file mode 100644
index 0000000000..55abe2eca2
--- /dev/null
+++ b/scm-plugins/scm-hg-plugin/src/main/java/sonia/scm/repository/spi/HgBranchesCommand.java
@@ -0,0 +1,100 @@
+/**
+ * Copyright (c) 2010, Sebastian Sdorra All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice,
+ * this list of conditions and the following disclaimer. 2. Redistributions in
+ * binary form must reproduce the above copyright notice, this list of
+ * conditions and the following disclaimer in the documentation and/or other
+ * materials provided with the distribution. 3. Neither the name of SCM-Manager;
+ * nor the names of its contributors may be used to endorse or promote products
+ * derived from this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR
+ * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+ * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ * http://bitbucket.org/sdorra/scm-manager
+ *
+ */
+
+
+
+package sonia.scm.repository.spi;
+
+//~--- non-JDK imports --------------------------------------------------------
+
+import com.google.common.base.Function;
+import com.google.common.collect.Lists;
+
+import sonia.scm.repository.Branch;
+import sonia.scm.repository.Repository;
+import sonia.scm.repository.RepositoryException;
+
+//~--- JDK imports ------------------------------------------------------------
+
+import java.io.IOException;
+
+import java.util.List;
+
+/**
+ *
+ * @author Sebastian Sdorra
+ */
+public class HgBranchesCommand extends AbstractCommand
+  implements BranchesCommand
+{
+
+  /**
+   * Constructs ...
+   *
+   *
+   * @param context
+   * @param repository
+   */
+  public HgBranchesCommand(HgCommandContext context, Repository repository)
+  {
+    super(context, repository);
+  }
+
+  //~--- get methods ----------------------------------------------------------
+
+  /**
+   * Method description
+   *
+   *
+   * @return
+   *
+   * @throws IOException
+   * @throws RepositoryException
+   */
+  @Override
+  public List<Branch> getBranches() throws RepositoryException, IOException
+  {
+    List<com.aragost.javahg.commands.Branch> hgBranches =
+      com.aragost.javahg.commands.BranchesCommand.on(open()).execute();
+
+    List<Branch> branches = Lists.transform(hgBranches,
+                              new Function<com.aragost.javahg.commands.Branch,
+                                Branch>()
+    {
+
+      @Override
+      public Branch apply(com.aragost.javahg.commands.Branch hgBranch)
+      {
+        return new Branch(hgBranch.getName());
+      }
+    });
+
+    return branches;
+  }
+}
diff --git a/scm-plugins/scm-hg-plugin/src/main/java/sonia/scm/repository/spi/HgRepositoryServiceProvider.java b/scm-plugins/scm-hg-plugin/src/main/java/sonia/scm/repository/spi/HgRepositoryServiceProvider.java
index f2499d5d2b..0746801340 100644
--- a/scm-plugins/scm-hg-plugin/src/main/java/sonia/scm/repository/spi/HgRepositoryServiceProvider.java
+++ b/scm-plugins/scm-hg-plugin/src/main/java/sonia/scm/repository/spi/HgRepositoryServiceProvider.java
@@ -62,7 +62,8 @@ public class HgRepositoryServiceProvider extends RepositoryServiceProvider
   private static final Set<Command> COMMANDS = ImmutableSet.of(Command.BLAME,
                                                  Command.BROWSE, Command.CAT,
                                                  Command.DIFF, Command.LOG,
-                                                 Command.TAGS);
+                                                 Command.TAGS,
+                                                 Command.BRANCHES);
 
   //~--- constructors ---------------------------------------------------------
 
@@ -114,6 +115,18 @@ public class HgRepositoryServiceProvider extends RepositoryServiceProvider
     return new HgBlameCommand(context, repository);
   }
 
+  /**
+   * Method description
+   *
+   *
+   * @return
+   */
+  @Override
+  public BranchesCommand getBranchesCommand()
+  {
+    return new HgBranchesCommand(context, repository);
+  }
+
   /**
    * Method description
    *