1 files changed, 83 insertions, 0 deletions
diff --git a/libbe/bzr.py b/libbe/bzr.py
new file mode 100644
index 0000000..b2e0d6b
--- /dev/null
+++ b/libbe/bzr.py
@@ -0,0 +1,83 @@
+from popen2 import Popen3
+import os
+import config
+
+def invoke(args):
+    q=Popen3(args, True)
+    output = q.fromchild.read()
+    error = q.childerr.read()
+    status = q.wait()
+    if os.WIFEXITED(status):
+        return os.WEXITSTATUS(status), output, error
+    raise Exception("Command failed: %s" % error)
+
+def invoke_client(*args, **kwargs):
+    cl_args = ["bzr"]
+    cl_args.extend(args)
+    status,output,error = invoke(cl_args)
+    if status not in (0,):
+        raise Exception("Command failed: %s" % error)
+    return output
+
+def add_id(filename):
+    invoke_client("add", filename)
+
+def delete_id(filename):
+    invoke_client("remove", filename)
+
+def mkdir(path):
+    os.mkdir(path)
+    add_id(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):
+    return invoke_client("lookup-revision", str(revno)).rstrip('\n')
+
+def export(revno, revision_dir):
+    invoke_client("export", str(revno), revision_dir)
+
+def find_or_make_export(revno):
+    revision_id = lookup_revision(revno)
+    home = os.path.expanduser("~")
+    revision_dir = os.path.join(home, ".bzrrevs", revision_id)
+    if not os.path.exists(revision_dir):
+        export(revno, revision_dir)
+    return revision_dir
+
+def bzr_root(path):
+    return invoke_client("root", path).rstrip('\r')
+
+def path_in_reference(bug_dir, spec):
+    if spec is None:
+        spec = int(invoke_client("revno"))
+    rel_bug_dir = bug_dir[len(bzr_root(bug_dir)):]
+    export_root = find_or_make_export(spec)
+    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 bzr"""
+    path = os.path.realpath(path)
+    while True:
+        if os.path.exists(os.path.join(path, ".bzr")):
+            return True
+        if path == "/":
+            return False
+        path = os.path.dirname(path)
+
+
+name = "bzr"