Extremely basic 'bup server' support.

[bup.git] / cmd-split.py

diff --git a/cmd-split.py b/cmd-split.py
index 9d3d4851c4711c786b797b3b955b47e6885aecb1..b013e6d3a08950a9a21e18fd62761d7568a94693 100755 (executable)
--- a/cmd-split.py
+++ b/cmd-split.py
@@ -1,11 +1,12 @@
 #!/usr/bin/env python
-import sys, time
+import sys, time, re
 import hashsplit, git, options
 from helpers import *
 
 optspec = """
 bup split [-tcb] [-n name] [--bench] [filenames...]
 --
+r,remote=  remote repository path
 b,blobs    output a series of blob ids
 t,tree     output a tree id
 c,commit   output a commit id
@@ -16,6 +17,7 @@ bench      print benchmark timings to stderr
 o = options.Options('bup split', optspec)
 (opt, flags, extra) = o.parse(sys.argv[1:])
 
+git.check_repo_or_die()
 if not (opt.blobs or opt.tree or opt.commit or opt.name):
     log("bup split: use one or more of -b, -t, -c, -n\n")
     o.usage()
@@ -27,7 +29,26 @@ if opt.verbose >= 2:
 
 start_time = time.time()
 
-w = git.PackWriter()
+def server_connect(remote):
+    rs = remote.split(':', 1)
+    if len(rs) == 1:
+        p = subprocess.Popen(['bup', 'server', '-d', opt.remote],
+                             stdin=subprocess.PIPE, stdout=subprocess.PIPE)
+    else:
+        (host, dir) = rs
+        p = subprocess.Popen(['ssh', host, '--', 'bup', 'server'],
+                             stdin=subprocess.PIPE, stdout=subprocess.PIPE)
+        dir = re.sub(r'[\r\n]', ' ', dir)
+        p.stdin.write('set-dir\n%s\n' % dir)
+    return p
+
+if opt.remote:
+    p = server_connect(opt.remote)
+    p.stdin.write('receive-objects\n')
+    w = git.PackWriter_Remote(p.stdin)
+else:
+    w = git.PackWriter()
+    
 (shalist,tree) = hashsplit.split_to_tree(w, hashsplit.autofiles(extra))
 
 if opt.verbose:
@@ -44,6 +65,11 @@ if opt.commit or opt.name:
     if opt.commit:
         print commit.encode('hex')
 
+if opt.remote:
+    w.close()
+    p.stdin.write('quit\n')
+    p.wait()
+
 secs = time.time() - start_time
 size = hashsplit.total_split
 if opt.bench: