cmd/ftp-cmd.py

   1 #!/bin/sh
   2 """": # -*-python-*-
   3 bup_python="$(dirname "$0")/bup-python" || exit $?
   4 exec "$bup_python" "$0" ${1+"$@"}
   5 """
   6 # end of bup preamble
   7
   8 from __future__ import absolute_import, print_function
   9 import sys, os, stat, fnmatch
  10
  11 from bup import options, git, shquote, ls, vfs
  12 from bup.helpers import chunkyreader, handle_ctrl_c, log
  13 from bup.repo import LocalRepo
  14
  15 handle_ctrl_c()
  16
  17
  18 class OptionError(Exception):
  19     pass
  20
  21
  22 def do_ls(repo, args):
  23     try:
  24         opt = ls.opts_from_cmdline(args, onabort=OptionError)
  25     except OptionError as e:
  26         log('error: %s' % e)
  27         return
  28     return ls.within_repo(repo, opt)
  29
  30
  31 def write_to_file(inf, outf):
  32     for blob in chunkyreader(inf):
  33         outf.write(blob)
  34
  35
  36 def inputiter():
  37     if os.isatty(sys.stdin.fileno()):
  38         while 1:
  39             try:
  40                 yield raw_input('bup> ')
  41             except EOFError:
  42                 print()  # Clear the line for the terminal's next prompt
  43                 break
  44     else:
  45         for line in sys.stdin:
  46             yield line
  47
  48
  49 def _completer_get_subs(repo, line):
  50     (qtype, lastword) = shquote.unfinished_word(line)
  51     (dir,name) = os.path.split(lastword)
  52     dir_path = vfs.resolve(repo, dir or '/')
  53     _, dir_item = dir_path[-1]
  54     if not dir_item:
  55         subs = tuple()
  56     else:
  57         subs = tuple(dir_path + (entry,)
  58                      for entry in vfs.contents(repo, dir_item)
  59                      if (entry[0] != '.' and entry[0].startswith(name)))
  60     return dir, name, qtype, lastword, subs
  61
  62
  63 _last_line = None
  64 _last_res = None
  65 def completer(text, iteration):
  66     global repo
  67     global _last_line
  68     global _last_res
  69     try:
  70         line = readline.get_line_buffer()[:readline.get_endidx()]
  71         if _last_line != line:
  72             _last_res = _completer_get_subs(repo, line)
  73             _last_line = line
  74         (dir, name, qtype, lastword, subs) = _last_res
  75         if iteration < len(subs):
  76             path = subs[iteration]
  77             leaf_name, leaf_item = path[-1]
  78             res = vfs.try_resolve(repo, leaf_name, parent=path[:-1])
  79             leaf_name, leaf_item = res[-1]
  80             fullname = os.path.join(*(name for name, item in res))
  81             if stat.S_ISDIR(vfs.item_mode(leaf_item)):
  82                 ret = shquote.what_to_add(qtype, lastword, fullname+'/',
  83                                           terminate=False)
  84             else:
  85                 ret = shquote.what_to_add(qtype, lastword, fullname,
  86                                           terminate=True) + ' '
  87             return text + ret
  88     except Exception as e:
  89         log('\n')
  90         try:
  91             import traceback
  92             traceback.print_tb(sys.exc_traceback)
  93         except Exception as e2:
  94             log('Error printing traceback: %s\n' % e2)
  95         log('\nError in completion: %s\n' % e)
  96
  97
  98 optspec = """
  99 bup ftp [commands...]
 100 """
 101 o = options.Options(optspec)
 102 (opt, flags, extra) = o.parse(sys.argv[1:])
 103
 104 git.check_repo_or_die()
 105
 106 repo = LocalRepo()
 107 pwd = vfs.resolve(repo, '/')
 108 rv = 0
 109
 110 if extra:
 111     lines = extra
 112 else:
 113     try:
 114         import readline
 115     except ImportError:
 116         log('* readline module not available: line editing disabled.\n')
 117         readline = None
 118
 119     if readline:
 120         readline.set_completer_delims(' \t\n\r/')
 121         readline.set_completer(completer)
 122         if sys.platform.startswith('darwin'):
 123             # MacOS uses a slightly incompatible clone of libreadline
 124             readline.parse_and_bind('bind ^I rl_complete')
 125         readline.parse_and_bind('tab: complete')
 126     lines = inputiter()
 127
 128 for line in lines:
 129     if not line.strip():
 130         continue
 131     words = [word for (wordstart,word) in shquote.quotesplit(line)]
 132     cmd = words[0].lower()
 133     #log('execute: %r %r\n' % (cmd, parm))
 134     try:
 135         if cmd == 'ls':
 136             # FIXME: respect pwd (perhaps via ls accepting resolve path/parent)
 137             do_ls(repo, words[1:])
 138         elif cmd == 'cd':
 139             np = pwd
 140             for parm in words[1:]:
 141                 res = vfs.resolve(repo, parm, parent=np)
 142                 _, leaf_item = res[-1]
 143                 if not leaf_item:
 144                     raise Exception('%r does not exist'
 145                                     % '/'.join(name for name, item in res))
 146                 if not stat.S_ISDIR(vfs.item_mode(leaf_item)):
 147                     raise Exception('%r is not a directory' % parm)
 148                 np = res
 149             pwd = np
 150         elif cmd == 'pwd':
 151             if len(pwd) == 1:
 152                 sys.stdout.write('/')
 153             print('/'.join(name for name, item in pwd))
 154         elif cmd == 'cat':
 155             for parm in words[1:]:
 156                 res = vfs.resolve(repo, parm, parent=pwd)
 157                 _, leaf_item = res[-1]
 158                 if not leaf_item:
 159                     raise Exception('%r does not exist' %
 160                                     '/'.join(name for name, item in res))
 161                 with vfs.fopen(repo, leaf_item) as srcfile:
 162                     write_to_file(srcfile, sys.stdout)
 163         elif cmd == 'get':
 164             if len(words) not in [2,3]:
 165                 rv = 1
 166                 raise Exception('Usage: get <filename> [localname]')
 167             rname = words[1]
 168             (dir,base) = os.path.split(rname)
 169             lname = len(words)>2 and words[2] or base
 170             res = vfs.resolve(repo, rname, parent=pwd)
 171             _, leaf_item = res[-1]
 172             if not leaf_item:
 173                 raise Exception('%r does not exist' %
 174                                 '/'.join(name for name, item in res))
 175             with vfs.fopen(repo, leaf_item) as srcfile:
 176                 with open(lname, 'wb') as destfile:
 177                     log('Saving %r\n' % lname)
 178                     write_to_file(srcfile, destfile)
 179         elif cmd == 'mget':
 180             for parm in words[1:]:
 181                 (dir,base) = os.path.split(parm)
 182
 183                 res = vfs.resolve(repo, dir, parent=pwd)
 184                 _, dir_item = res[-1]
 185                 if not dir_item:
 186                     raise Exception('%r does not exist' % dir)
 187                 for name, item in vfs.contents(repo, dir_item):
 188                     if name == '.':
 189                         continue
 190                     if fnmatch.fnmatch(name, base):
 191                         if stat.S_ISLNK(vfs.item_mode(item)):
 192                             deref = vfs.resolve(repo, name, parent=res)
 193                             deref_name, deref_item = deref[-1]
 194                             if not deref_item:
 195                                 raise Exception('%r does not exist' %
 196                                                 '/'.join(name for name, item
 197                                                          in deref))
 198                             item = deref_item
 199                         with vfs.fopen(repo, item) as srcfile:
 200                             with open(name, 'wb') as destfile:
 201                                 log('Saving %r\n' % name)
 202                                 write_to_file(srcfile, destfile)
 203         elif cmd == 'help' or cmd == '?':
 204             # FIXME: move to stdout
 205             log('Commands: ls cd pwd cat get mget help quit\n')
 206         elif cmd in ('quit', 'exit', 'bye'):
 207             break
 208         else:
 209             rv = 1
 210             raise Exception('no such command %r' % cmd)
 211     except Exception as e:
 212         rv = 1
 213         log('error: %s\n' % e)
 214         raise
 215
 216 sys.exit(rv)