write python scripts to filesystem

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

@@ -0,0 +1,125 @@
+/**
+ * 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;
+
+//~--- non-JDK imports --------------------------------------------------------
+
+import sonia.scm.SCMContextProvider;
+
+//~--- JDK imports ------------------------------------------------------------
+
+import java.io.File;
+
+/**
+ *
+ * @author Sebastian Sdorra
+ */
+public enum HgPythonScript
+{
+  BLAME("blame.py"), CHANGELOG("changelog.py"), FILELOG("filelog.py"),
+  UTIL("util.py"), HOOK("hook.py"), HGWEB("hgweb.py");
+
+  /** Field description */
+  private static final String BASE_DIRECTORY =
+    "lib".concat(File.separator).concat("python");
+
+  /** Field description */
+  private static final String BASE_RESOURCE = "/sonia/scm/python/";
+
+  //~--- constructors ---------------------------------------------------------
+
+  /**
+   * Constructs ...
+   *
+   *
+   * @param name
+   */
+  private HgPythonScript(String name)
+  {
+    this.name = name;
+  }
+
+  //~--- get methods ----------------------------------------------------------
+
+  /**
+   * Method description
+   *
+   *
+   * @param context
+   *
+   * @return
+   */
+  public static File getScriptDirectory(SCMContextProvider context)
+  {
+    return new File(context.getBaseDirectory(), BASE_DIRECTORY);
+  }
+
+  /**
+   * Method description
+   *
+   *
+   * @param context
+   *
+   * @return
+   */
+  public File getFile(SCMContextProvider context)
+  {
+    return new File(getScriptDirectory(context), name);
+  }
+
+  /**
+   * Method description
+   *
+   *
+   * @return
+   */
+  public String getName()
+  {
+    return name;
+  }
+
+  /**
+   * Method description
+   *
+   *
+   * @return
+   */
+  public String getResourcePath()
+  {
+    return BASE_RESOURCE.concat(name);
+  }
+
+  //~--- fields ---------------------------------------------------------------
+
+  /** Field description */
+  private String name;
+}

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

@@ -57,13 +57,17 @@ 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.IOUtil;
 import sonia.scm.util.Util;
 import sonia.scm.web.HgWebConfigWriter;
 
 //~--- JDK imports ------------------------------------------------------------
 
 import java.io.File;
+import java.io.FileOutputStream;
 import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
 
 import javax.xml.bind.JAXBContext;
 import javax.xml.bind.JAXBException;
@@ -176,6 +180,7 @@ public class HgRepositoryHandler
   {
     super.init(context);
     registerMissingHooks();
+    writePythonScripts(context);
   }
 
   /**
@@ -678,6 +683,45 @@ public class HgRepositoryHandler
     }
   }
 
+  /**
+   * Method description
+   *
+   *
+   * @param context
+   */
+  private void writePythonScripts(SCMContextProvider context)
+  {
+    IOUtil.mkdirs(HgPythonScript.getScriptDirectory(context));
+
+    for (HgPythonScript script : HgPythonScript.values())
+    {
+      if (logger.isDebugEnabled())
+      {
+        logger.debug("write python script {}", script.getName());
+      }
+
+      InputStream content = null;
+      OutputStream output = null;
+
+      try
+      {
+        content = HgRepositoryHandler.class.getResourceAsStream(
+          script.getResourcePath());
+        output = new FileOutputStream(script.getFile(context));
+        IOUtil.copy(content, output);
+      }
+      catch (IOException ex)
+      {
+        logger.error("could not write script", ex);
+      }
+      finally
+      {
+        IOUtil.close(content);
+        IOUtil.close(output);
+      }
+    }
+  }
+
   //~--- fields ---------------------------------------------------------------
 
   /** Field description */