cmd/ftp-cmd.py

   1 #!/usr/bin/env python
   2 import sys, os, re, stat, fnmatch
   3 from bup import options, git, shquote, vfs
   4 from bup.helpers import *
   5
   6 try:
   7     import readline
   8 except ImportError:
   9     log('* readline module not available: line editing disabled.\n')
  10     readline = None
  11
  12
  13 def node_name(text, n):
  14     if stat.S_ISDIR(n.mode):
  15         return '%s/' % text
  16     elif stat.S_ISLNK(n.mode):
  17         return '%s@' % text
  18     else:
  19         return '%s' % text
  20
  21
  22 def do_ls(path, n):
  23     l = []
  24     if stat.S_ISDIR(n.mode):
  25         for sub in n:
  26             l.append(node_name(sub.name, sub))
  27     else:
  28         l.append(node_name(path, n))
  29     print columnate(l, '')
  30
  31
  32 def write_to_file(inf, outf):
  33     for blob in chunkyreader(inf):
  34         outf.write(blob)
  35
  36
  37 def inputiter():
  38     if os.isatty(sys.stdin.fileno()):
  39         while 1:
  40             try:
  41                 yield raw_input('bup> ')
  42             except EOFError:
  43                 break
  44     else:
  45         for line in sys.stdin:
  46             yield line
  47
  48
  49 def _completer_get_subs(line):
  50     (qtype, lastword) = shquote.unfinished_word(line)
  51     (dir,name) = os.path.split(lastword)
  52     #log('\ncompleter: %r %r %r\n' % (qtype, lastword, text))
  53     n = pwd.resolve(dir)
  54     subs = list(filter(lambda x: x.name.startswith(name),
  55                        n.subs()))
  56     return (dir, name, qtype, lastword, subs)
  57
  58
  59 def find_readline_lib():
  60     """Return the name (and possibly the full path) of the readline library
  61     linked to the given readline module.
  62     """
  63     import readline
  64     f = open(readline.__file__, "rb")
  65     try:
  66         data = f.read()
  67     finally:
  68         f.close()
  69     import re
  70     m = re.search('\0([^\0]*libreadline[^\0]*)\0', data)
  71     if m:
  72         return m.group(1)
  73     return None
  74
  75
  76 def init_readline_vars():
  77     """Work around trailing space automatically inserted by readline.
  78     See http://bugs.python.org/issue5833"""
  79     import ctypes
  80     lib_name = find_readline_lib()
  81     if lib_name is not None:
  82         lib = ctypes.cdll.LoadLibrary(lib_name)
  83         global rl_completion_suppress_append
  84         rl_completion_suppress_append = ctypes.c_int.in_dll(lib,
  85                                     "rl_completion_suppress_append")
  86
  87
  88 rl_completion_suppress_append = None
  89 _last_line = None
  90 _last_res = None
  91 def completer(text, state):
  92     global _last_line
  93     global _last_res
  94     global rl_completion_suppress_append
  95     if rl_completion_suppress_append is not None:
  96         rl_completion_suppress_append.value = 1
  97     try:
  98         line = readline.get_line_buffer()[:readline.get_endidx()]
  99         if _last_line != line:
 100             _last_res = _completer_get_subs(line)
 101             _last_line = line
 102         (dir, name, qtype, lastword, subs) = _last_res
 103         if state < len(subs):
 104             sn = subs[state]
 105             sn1 = sn.resolve('')  # deref symlinks
 106             fullname = os.path.join(dir, sn.name)
 107             if stat.S_ISDIR(sn1.mode):
 108                 ret = shquote.what_to_add(qtype, lastword, fullname+'/',
 109                                           terminate=False)
 110             else:
 111                 ret = shquote.what_to_add(qtype, lastword, fullname,
 112                                           terminate=True) + ' '
 113             return text + ret
 114     except Exception, e:
 115         log('\nerror in completion: %s\n' % e)
 116
 117
 118 optspec = """
 119 bup ftp
 120 """
 121 o = options.Options('bup ftp', optspec)
 122 (opt, flags, extra) = o.parse(sys.argv[1:])
 123
 124 git.check_repo_or_die()
 125
 126 top = vfs.RefList(None)
 127 pwd = top
 128 rv = 0
 129
 130 if extra:
 131     lines = extra
 132 else:
 133     if readline:
 134         readline.set_completer_delims(' \t\n\r/')
 135         readline.set_completer(completer)
 136         readline.parse_and_bind("tab: complete")
 137         init_readline_vars()
 138     lines = inputiter()
 139
 140 for line in lines:
 141     if not line.strip():
 142         continue
 143     words = [word for (wordstart,word) in shquote.quotesplit(line)]
 144     cmd = words[0].lower()
 145     #log('execute: %r %r\n' % (cmd, parm))
 146     try:
 147         if cmd == 'ls':
 148             for parm in (words[1:] or ['.']):
 149                 do_ls(parm, pwd.resolve(parm))
 150         elif cmd == 'cd':
 151             for parm in words[1:]:
 152                 pwd = pwd.resolve(parm)
 153         elif cmd == 'pwd':
 154             print pwd.fullname()
 155         elif cmd == 'cat':
 156             for parm in words[1:]:
 157                 write_to_file(pwd.resolve(parm).open(), sys.stdout)
 158         elif cmd == 'get':
 159             if len(words) not in [2,3]:
 160                 rv = 1
 161                 raise Exception('Usage: get <filename> [localname]')
 162             rname = words[1]
 163             (dir,base) = os.path.split(rname)
 164             lname = len(words)>2 and words[2] or base
 165             inf = pwd.resolve(rname).open()
 166             log('Saving %r\n' % lname)
 167             write_to_file(inf, open(lname, 'wb'))
 168         elif cmd == 'mget':
 169             for parm in words[1:]:
 170                 (dir,base) = os.path.split(parm)
 171                 for n in pwd.resolve(dir).subs():
 172                     if fnmatch.fnmatch(n.name, base):
 173                         try:
 174                             log('Saving %r\n' % n.name)
 175                             inf = n.open()
 176                             outf = open(n.name, 'wb')
 177                             write_to_file(inf, outf)
 178                             outf.close()
 179                         except Exception, e:
 180                             rv = 1
 181                             log('  error: %s\n' % e)
 182         elif cmd == 'help' or cmd == '?':
 183             log('Commands: ls cd pwd cat get mget help quit\n')
 184         elif cmd == 'quit' or cmd == 'exit' or cmd == 'bye':
 185             break
 186         else:
 187             rv = 1
 188             raise Exception('no such command %r' % cmd)
 189     except Exception, e:
 190         rv = 1
 191         log('error: %s\n' % e)
 192         #raise
 193
 194 sys.exit(rv)