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