2 import sys, os, re, stat, readline, fnmatch
3 from bup import options, git, shquote, vfs
4 from bup.helpers import *
6 def print_node(text, n):
7 if stat.S_ISDIR(n.mode):
9 elif stat.S_ISLNK(n.mode):
16 if stat.S_ISDIR(n.mode):
18 print_node(sub.name, sub)
23 def write_to_file(inf, outf):
24 for blob in chunkyreader(inf):
29 if os.isatty(sys.stdin.fileno()):
32 yield raw_input('bup> ')
36 for line in sys.stdin:
40 def _completer_get_subs(line):
41 (qtype, lastword) = shquote.unfinished_word(line)
42 (dir,name) = os.path.split(lastword)
43 #log('\ncompleter: %r %r %r\n' % (qtype, lastword, text))
45 subs = list(filter(lambda x: x.name.startswith(name),
47 return (dir, name, qtype, lastword, subs)
52 def completer(text, state):
56 line = readline.get_line_buffer()[:readline.get_endidx()]
57 if _last_line != line:
58 _last_res = _completer_get_subs(line)
60 (dir, name, qtype, lastword, subs) = _last_res
63 sn1 = sn.resolve('') # deref symlinks
64 fullname = os.path.join(dir, sn.name)
65 if stat.S_ISDIR(sn1.mode):
66 ret = shquote.what_to_add(qtype, lastword, fullname+'/',
69 ret = shquote.what_to_add(qtype, lastword, fullname,
73 log('\nerror in completion: %s\n' % e)
79 o = options.Options('bup ftp', optspec)
80 (opt, flags, extra) = o.parse(sys.argv[1:])
82 git.check_repo_or_die()
84 top = vfs.RefList(None)
90 readline.set_completer_delims(' \t\n\r/')
91 readline.set_completer(completer)
92 readline.parse_and_bind("tab: complete")
98 words = [word for (wordstart,word) in shquote.quotesplit(line)]
99 cmd = words[0].lower()
100 #log('execute: %r %r\n' % (cmd, parm))
103 for parm in (words[1:] or ['.']):
104 do_ls(parm, pwd.resolve(parm))
106 for parm in words[1:]:
107 pwd = pwd.resolve(parm)
111 for parm in words[1:]:
112 write_to_file(pwd.resolve(parm).open(), sys.stdout)
114 if len(words) not in [2,3]:
115 raise Exception('Usage: get <filename> [localname]')
117 (dir,base) = os.path.split(rname)
118 lname = len(words)>2 and words[2] or base
119 log('Saving %r\n' % lname)
120 inf = pwd.resolve(rname).open()
121 write_to_file(inf, open(lname, 'wb'))
123 for parm in words[1:]:
124 (dir,base) = os.path.split(parm)
125 for n in pwd.resolve(dir).subs():
126 if fnmatch.fnmatch(n.name, base):
128 log('Saving %r\n' % n.name)
130 outf = open(n.name, 'wb')
131 write_to_file(inf, outf)
134 log(' error: %s\n' % e)
135 elif cmd == 'help' or cmd == '?':
136 log('Commands: ls cd pwd cat get mget help quit\n')
137 elif cmd == 'quit' or cmd == 'exit' or cmd == 'bye':
140 raise Exception('no such command %r' % cmd)
142 log('error: %s\n' % e)