register missing mercurial hooks

scm-core/src/main/java/sonia/scm/io/DirectoryFileFilter.java

@@ -0,0 +1,67 @@
+/**
+ * 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.io;
+
+//~--- JDK imports ------------------------------------------------------------
+
+import java.io.File;
+import java.io.FileFilter;
+
+/**
+ *
+ * @author Sebastian Sdorra
+ * @since 1.6
+ */
+public class DirectoryFileFilter implements FileFilter
+{
+
+  /** Field description */
+  public static final DirectoryFileFilter instance = new DirectoryFileFilter();
+
+  //~--- methods --------------------------------------------------------------
+
+  /**
+   * Method description
+   *
+   *
+   * @param file
+   *
+   * @return
+   */
+  @Override
+  public boolean accept(File file)
+  {
+    return file.isDirectory();
+  }
+}

scm-plugins/scm-hg-plugin/src/main/java/sonia/scm/repository/HgRepositoryHandler.java

@@ -42,17 +42,21 @@ import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 import sonia.scm.ConfigurationException;
+import sonia.scm.SCMContextProvider;
 import sonia.scm.Type;
 import sonia.scm.installer.HgInstaller;
 import sonia.scm.installer.HgInstallerFactory;
+import sonia.scm.io.DirectoryFileFilter;
 import sonia.scm.io.ExtendedCommand;
 import sonia.scm.io.FileSystem;
 import sonia.scm.io.INIConfiguration;
+import sonia.scm.io.INIConfigurationReader;
 import sonia.scm.io.INIConfigurationWriter;
 import sonia.scm.io.INISection;
 import sonia.scm.plugin.ext.Extension;
 import sonia.scm.store.StoreFactory;
 import sonia.scm.util.AssertUtil;
+import sonia.scm.util.Util;
 import sonia.scm.web.HgWebConfigWriter;
 
 //~--- JDK imports ------------------------------------------------------------
@@ -73,6 +77,9 @@ public class HgRepositoryHandler
         extends AbstractSimpleRepositoryHandler<HgConfig>
 {
 
+  /** Field description */
+  public static final String PATH_HOOK1_6 = ".hook-1.6";
+
   /** Field description */
   public static final String TYPE_DISPLAYNAME = "Mercurial";
 
@@ -86,6 +93,10 @@ public class HgRepositoryHandler
   private static final Logger logger =
     LoggerFactory.getLogger(HgRepositoryHandler.class);
 
+  /** Field description */
+  public static final String PATH_HGRC =
+    ".hg".concat(File.separator).concat("hgrc");
+
   //~--- constructors ---------------------------------------------------------
 
   /**
@@ -146,6 +157,19 @@ public class HgRepositoryHandler
     }
   }
 
+  /**
+   * Method description
+   *
+   *
+   * @param context
+   */
+  @Override
+  public void init(SCMContextProvider context)
+  {
+    super.init(context);
+    registerMissingHooks();
+  }
+
   /**
    * Method description
    *
@@ -253,8 +277,7 @@ public class HgRepositoryHandler
   protected void postCreate(Repository repository, File directory)
           throws IOException, RepositoryException
   {
-    File hgrcFile = new File(directory,
+    File hgrcFile = new File(directory, PATH_HGRC);
-                             ".hg".concat(File.separator).concat("hgrc"));
     INIConfiguration hgrc = new INIConfiguration();
     INISection webSection = new INISection("web");
 
@@ -287,6 +310,139 @@ public class HgRepositoryHandler
     return HgConfig.class;
   }
 
+  //~--- methods --------------------------------------------------------------
+
+  /**
+   * Method description
+   *
+   *
+   * @param file
+   */
+  private void createNewFile(File file)
+  {
+    try
+    {
+      if (!file.createNewFile() && logger.isErrorEnabled())
+      {
+        logger.error("could not create file {}", file);
+      }
+    }
+    catch (IOException ex)
+    {
+      logger.error("could not create file {}".concat(file.getPath()), ex);
+    }
+  }
+
+  /**
+   * Method description
+   *
+   *
+   * @param repositoryDir
+   *
+   * @return
+   */
+  private boolean registerMissingHook(File repositoryDir)
+  {
+    boolean result = false;
+    File hgrc = new File(repositoryDir, PATH_HGRC);
+
+    if (hgrc.exists())
+    {
+      try
+      {
+        INIConfigurationReader reader = new INIConfigurationReader();
+        INIConfiguration c = reader.read(hgrc);
+        INISection hooks = c.getSection("hooks");
+
+        if (hooks == null)
+        {
+          hooks = new INISection("hooks");
+          c.addSection(hooks);
+        }
+
+        String hook = hooks.getParameter("changegroup.scm");
+
+        if (Util.isEmpty(hook))
+        {
+          if (logger.isInfoEnabled())
+          {
+            logger.info("register missing hg hook for respository {}",
+                        repositoryDir.getName());
+          }
+
+          hooks.setParameter("changegroup.scm", "python:scmhooks.callback");
+
+          INIConfigurationWriter writer = new INIConfigurationWriter();
+
+          writer.write(c, hgrc);
+        }
+
+        result = true;
+      }
+      catch (IOException ex)
+      {
+        logger.error("could not register missing hook", ex);
+      }
+    }
+
+    return result;
+  }
+
+  /**
+   * Method description
+   *
+   */
+  private void registerMissingHooks()
+  {
+    HgConfig c = getConfig();
+
+    if (c != null)
+    {
+      File repositoryDirectroy = c.getRepositoryDirectory();
+
+      if (repositoryDirectroy.exists())
+      {
+        File lockFile = new File(repositoryDirectroy, PATH_HOOK1_6);
+
+        if (!lockFile.exists())
+        {
+          File[] dirs =
+            repositoryDirectroy.listFiles(DirectoryFileFilter.instance);
+          boolean success = true;
+
+          if (Util.isNotEmpty(dirs))
+          {
+            for (File dir : dirs)
+            {
+              if (!registerMissingHook(dir))
+              {
+                success = false;
+              }
+            }
+          }
+
+          if (success)
+          {
+            createNewFile(lockFile);
+          }
+        }
+        else if (logger.isDebugEnabled())
+        {
+          logger.debug("hooks allready registered");
+        }
+      }
+      else if (logger.isDebugEnabled())
+      {
+        logger.debug(
+            "repository directory does not exists, could not register missing hooks");
+      }
+    }
+    else if (logger.isDebugEnabled())
+    {
+      logger.debug("config is not available, could not register missing hooks");
+    }
+  }
+
   //~--- fields ---------------------------------------------------------------
 
   /** Field description */