Add GIT support, from Chris Ball and Michael Stone.

1 files changed, 146 insertions, 0 deletions

diff --git a/libbe/git.py b/libbe/git.py
new file mode 100644
index 0000000..398585f
--- /dev/null
+++ b/libbe/git.py
@@ -0,0 +1,146 @@
+# Copyright (C) 2007 Chris Ball <chris@printf.net>
+#
+#    This program is free software; you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License as published by
+#    the Free Software Foundation; either version 2 of the License, or
+#    (at your option) any later version.
+#
+#    This program is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with this program; if not, write to the Free Software
+#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+import os
+import tempfile
+
+from rcs import invoke
+
+def strip_git(filename):
+    # Find the base path of the GIT tree, in order to strip that leading
+    # path from arguments to git -- it doesn't like absolute paths.
+    if os.path.isabs(filename):
+        filename = filename[len(git_repo_for_path('.'))+1:]
+    return filename
+
+def invoke_client(*args, **kwargs):
+    directory = kwargs['directory']
+    expect = kwargs.get('expect', (0, 1))
+    cl_args = ["git"]
+    cl_args.extend(args)
+    status,output,error = invoke(cl_args, expect, cwd=directory)
+    return status, output
+
+def add_id(filename, paranoid=False):
+    filename = strip_git(filename)
+    invoke_client("add", filename, directory=git_repo_for_path('.'))
+
+def delete_id(filename):
+    filename = strip_git(filename)
+    invoke_client("rm", filename, directory=git_repo_for_path('.'))
+
+def mkdir(path, paranoid=False):
+    os.mkdir(path)
+
+def set_file_contents(path, contents):
+    add = not os.path.exists(path)
+    file(path, "wb").write(contents)
+    if add:
+        add_id(path)
+
+def detect(path):
+    """Detect whether a directory is revision-controlled using GIT"""
+    path = os.path.realpath(path)
+    old_path = None
+    while True:
+        if os.path.exists(os.path.join(path, ".git")):
+            return True
+        if path == old_path:
+            return False
+        old_path = path
+        path = os.path.dirname(path)
+
+def precommit(directory):
+    pass
+
+def commit(directory, summary, body=None):
+    if body is not None:
+        summary += '\n' + body
+    descriptor, filename = tempfile.mkstemp()
+    try:
+        temp_file = os.fdopen(descriptor, 'wb')
+        temp_file.write(summary)
+        temp_file.close()
+        invoke_client('commit', '-a', '-F', filename, directory=directory)
+    finally:
+        os.unlink(filename)
+
+def postcommit(directory):
+    pass
+
+
+# In order to diff the bug database, you need a way to check out arbitrary
+# previous revisions and a mechanism for locating the bug_dir in the revision
+# you've checked out.
+#
+# Copying the Mercurial implementation, this feature is implemented by four
+# functions:
+#
+# git_dir_for_path : find '.git' for a git tree.
+#
+# export : check out a commit 'spec' from git-repo 'bug_dir' into a dir
+#          'revision_dir'
+#
+# find_or_make_export : check out a commit 'spec' from git repo 'directory' to
+#                       any location you please and return the path to the checkout
+#
+# path_in_reference : return a path to the bug_dir of the commit 'spec'
+
+def git_repo_for_path(path):
+    """Find the root of the deepest repository containing path."""
+    # Assume that nothing funny is going on; in particular, that we aren't
+    # dealing with a bare repo.
+    return os.path.dirname(git_dir_for_path(path))
+
+def git_dir_for_path(path):
+    """Find the git-dir of the deepest repo containing path."""
+    return invoke_client("rev-parse", "--git-dir", directory=path)[1].rstrip()
+
+def export(spec, bug_dir, revision_dir):
+    """Check out commit 'spec' from the git repo containing bug_dir into
+    'revision_dir'."""
+    if not os.path.exists(revision_dir):
+        os.makedirs(revision_dir)
+    invoke_client("init", directory=revision_dir)
+    invoke_client("pull", git_dir_for_path(bug_dir), directory=revision_dir)
+    invoke_client("checkout", '-f', spec, directory=revision_dir)
+
+def find_or_make_export(spec, directory):
+    """Checkout 'spec' from the repo at 'directory' by hook or by crook and
+    return the path to the working copy."""
+    home = os.path.expanduser("~")
+    revision_root = os.path.join(home, ".be_revs")
+    if not os.path.exists(revision_root):
+        os.mkdir(revision_root)
+    revision_dir = os.path.join(revision_root, spec)
+    if not os.path.exists(revision_dir):
+        export(spec, directory, revision_dir)
+    return revision_dir
+
+def path_in_reference(bug_dir, spec):
+    """Check out 'spec' and return the path to its bug_dir."""
+    spec = spec or 'HEAD'
+    spec = invoke_client('rev-parse', spec, directory=bug_dir)[1].rstrip()
+    # This is a really hairy computation.
+    # The theory is that we can't possibly be working out of a bare repo;
+    # hence, we get the rel_bug_dir by chopping off dirname(git_dir_for_path(bug_dir))
+    # + '/'.
+    rel_bug_dir = strip_git(bug_dir)
+    export_root = find_or_make_export(spec, directory=bug_dir)
+    return os.path.join(export_root, rel_bug_dir)
+
+
+name = "git"
+