aboutsummaryrefslogblamecommitdiffstats
path: root/libbe/git.py
blob: 398585f6792119d2e7f6e19f526a8a137717ae39 (plain) (tree)

















































































































































                                                                                      
# 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"