aboutsummaryrefslogtreecommitdiffstats
path: root/libbe/hg.py
blob: 35de8e0f7e210c214cbb7bc3231eacc0e638e1ba (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
# Copyright (C) 2007 Steve Borho <steve@borho.org>
#
#    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

import config
from rcs import invoke, CommandError

def invoke_client(*args, **kwargs):
    directory = kwargs['directory']
    expect = kwargs.get('expect', (0, 1))
    cl_args = ["hg"]
    cl_args.extend(args)
    status,output,error = invoke(cl_args, expect, cwd=directory)
    return status, output

def add_id(filename, paranoid=False):
    invoke_client("add", filename, directory='.')

def delete_id(filename):
    invoke_client("rm", filename, directory='.')

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 lookup_revision(revno, directory):
    return invoke_client('log', '--rev', str(revno), '--template={node}',
                         directory=directory)[1].rstrip('\n')

def export(revno, directory, revision_dir):
    invoke_client("archive", "--rev", str(revno), revision_dir,
                  directory=directory)

def find_or_make_export(revno, directory):
    revision_id = lookup_revision(revno, directory)
    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, revision_id)
    if not os.path.exists(revision_dir):
        export(revno, directory, revision_dir)
    return revision_dir

def hg_root(path):
    return invoke_client("root", "-R", path, directory=None)[1].rstrip('\r')

def path_in_reference(bug_dir, spec):
    if spec is None:
        spec = int(invoke_client('tip', '--template="{rev}"',
                   directory=bug_dir)[1])
    rel_bug_dir = bug_dir[len(hg_root(bug_dir)):]
    export_root = find_or_make_export(spec, directory=bug_dir)
    return os.path.join(export_root, rel_bug_dir)


def unlink(path):
    try:
        os.unlink(path)
        delete_id(path)
    except OSError, e:
        if e.errno != 2:
            raise


def detect(path):
    """Detect whether a directory is revision-controlled using Mercurial"""
    path = os.path.realpath(path)
    old_path = None
    while True:
        if os.path.exists(os.path.join(path, ".hg")):
            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', '--logfile', filename, directory=directory)
    finally:
        os.unlink(filename)

def postcommit(directory):
    pass

name = "hg"