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