cmd/ftp-cmd.py

   1 #!/usr/bin/env python
   2 import sys, os, re, stat, readline, fnmatch
   3 from bup import options, git, shquote, vfs
   4 from bup.helpers import *
   5
   6 def print_node(text, n):
   7     if stat.S_ISDIR(n.mode):
   8         print '%s/' % text
   9     elif stat.S_ISLNK(n.mode):
  10         print '%s@' % text
  11     else:
  12         print '%s' % text
  13
  14
  15 def do_ls(path, n):
  16     if stat.S_ISDIR(n.mode):
  17         for sub in n:
  18             print_node(sub.name, sub)
  19     else:
  20         print_node(path, n)
  21
  22
  23 def write_to_file(inf, outf):
  24     for blob in chunkyreader(inf):
  25         outf.write(blob)
  26
  27
  28 def inputiter():
  29     if os.isatty(sys.stdin.fileno()):
  30         while 1:
  31             try:
  32                 yield raw_input('bup> ')
  33             except EOFError:
  34                 break
  35     else:
  36         for line in sys.stdin:
  37             yield line
  38
  39
  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))
  44     n = pwd.resolve(dir)
  45     subs = list(filter(lambda x: x.name.startswith(name),
  46                        n.subs()))
  47     return (dir, name, qtype, lastword, subs)
  48
  49
  50 _last_line = None
  51 _last_res = None
  52 def completer(text, state):
  53     global _last_line
  54     global _last_res
  55     try:
  56         line = readline.get_line_buffer()[:readline.get_endidx()]
  57         if _last_line != line:
  58             _last_res = _completer_get_subs(line)
  59             _last_line = line
  60         (dir, name, qtype, lastword, subs) = _last_res
  61         if state < len(subs):
  62             sn = subs[state]
  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+'/',
  67                                           terminate=False)
  68             else:
  69                 ret = shquote.what_to_add(qtype, lastword, fullname,
  70                                           terminate=True) + ' '
  71             return text + ret
  72     except Exception, e:
  73         log('\nerror in completion: %s\n' % e)
  74
  75
  76 optspec = """
  77 bup ftp
  78 """
  79 o = options.Options('bup ftp', optspec)
  80 (opt, flags, extra) = o.parse(sys.argv[1:])
  81
  82 git.check_repo_or_die()
  83
  84 top = vfs.RefList(None)
  85 pwd = top
  86
  87 if extra:
  88     lines = extra
  89 else:
  90     readline.set_completer_delims(' \t\n\r/')
  91     readline.set_completer(completer)
  92     readline.parse_and_bind("tab: complete")
  93     lines = inputiter()
  94
  95 for line in lines:
  96     if not line.strip():
  97         continue
  98     words = [word for (wordstart,word) in shquote.quotesplit(line)]
  99     cmd = words[0].lower()
 100     #log('execute: %r %r\n' % (cmd, parm))
 101     try:
 102         if cmd == 'ls':
 103             for parm in (words[1:] or ['.']):
 104                 do_ls(parm, pwd.resolve(parm))
 105         elif cmd == 'cd':
 106             for parm in words[1:]:
 107                 pwd = pwd.resolve(parm)
 108         elif cmd == 'pwd':
 109             print pwd.fullname()
 110         elif cmd == 'cat':
 111             for parm in words[1:]:
 112                 write_to_file(pwd.resolve(parm).open(), sys.stdout)
 113         elif cmd == 'get':
 114             if len(words) not in [2,3]:
 115                 raise Exception('Usage: get <filename> [localname]')
 116             rname = words[1]
 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'))
 122         elif cmd == 'mget':
 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):
 127                         try:
 128                             log('Saving %r\n' % n.name)
 129                             inf = n.open()
 130                             outf = open(n.name, 'wb')
 131                             write_to_file(inf, outf)
 132                             outf.close()
 133                         except Exception, e:
 134                             log('  error: %s\n' % e)
 135         else:
 136             raise Exception('no such command %r' % cmd)
 137     except Exception, e:
 138         log('error: %s\n' % e)
 139         #raise