From 49054fbb99772a9e9291bec40765e866e381ad4b Mon Sep 17 00:00:00 2001
From: swalter <swalter@swalter-d630.(none)>
Date: Wed, 3 Dec 2008 09:56:17 -0500
Subject: [PATCH] workdir: new plugin

Create a plugin to automatic creating new workdirs for git repositories.
---
 plugins/workdir.py | 73 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 73 insertions(+)
 create mode 100644 plugins/workdir.py

diff --git a/plugins/workdir.py b/plugins/workdir.py
new file mode 100644
index 0000000..0c6f9f5
--- /dev/null
+++ b/plugins/workdir.py
@@ -0,0 +1,73 @@
+
+from yap.yap import YapCore, YapError
+from yap.util import get_output, takes_options, run_safely
+
+import os
+import tempfile
+
+class WorkdirPlugin(YapCore):
+    "Create extra work directories of a repository"
+
+    def __init__(self, *args, **flags):
+	super(WorkdirPlugin, self).__init__(*args, **flags)
+
+    def _lock_branch(self, branch):
+        repo = get_output('git rev-parse --git-dir')[0]
+        dir = os.path.join(repo, 'yap', 'lock')
+        try:
+            os.mkdir(dir)
+        except OSError:
+            pass
+
+        fd, lockfile = tempfile.mkstemp("yap")
+        os.close(fd)
+        try:
+            os.link(lockfile, os.path.join(dir, branch.replace('/', '\/')))
+        except OSError:
+            raise YapError("That branch is being used by an existing workdir")
+
+    def cmd_workdir(self, branch, workdir=None):
+        self._check_git()
+
+        branches = get_output("git for-each-ref --format='%(refname)' 'refs/heads'")
+        if 'refs/heads/%s' % branch not in branches:
+            raise YapError("Not a branch: %s" % branch)
+
+        current = get_output("git symbolic-ref HEAD")[0]
+
+        print "a"
+        # Make sure the current branch is locked
+        try:
+            self._lock_branch(current.replace('refs/heads/', ''))
+        except:
+            pass
+
+        print "b"
+        self._lock_branch(branch)
+
+        repo = get_output('git rev-parse --git-dir')[0]
+        repo = os.path.join(os.getcwd(), repo)
+        repodir = os.path.dirname(repo)
+        if workdir is None:
+            repoparent, reponame = os.path.split(repodir)
+            workdir = os.path.join(repoparent, "%s-%s" % (reponame, branch))
+
+        try:
+            os.mkdir(workdir)
+        except OSError, e:
+            raise YapError("Can't create new workdir: %s (%s)" % (workdir, e))
+
+        os.chdir(workdir)
+        os.mkdir(".git")
+        os.chdir(".git")
+
+        for x in ["config", "refs", "logs/refs", "objects", "info",
+                  "hooks", "packed-refs", "remotes", "svn"]:
+            if os.path.dirname(x):
+                os.makedirs(os.path.dirname(x))
+            os.symlink(os.path.join(repo, x), x)
+
+        run_safely("cp %s HEAD" % os.path.join(repo, 'HEAD'))
+        os.chdir("..")
+        run_safely("git symbolic-ref HEAD refs/heads/%s" % branch)
+        self.cmd_revert(**{'-a': 1})
-- 
2.11.4.GIT