lib/bup/cmd/cat_file.py

   1
   2 from __future__ import absolute_import
   3
   4 import re, stat, sys
   5
   6 from bup import options, git, vfs
   7 from bup.compat import argv_bytes
   8 from bup.helpers import chunkyreader, handle_ctrl_c, log, saved_errors
   9 from bup.io import byte_stream
  10 from bup.repo import LocalRepo
  11
  12 optspec = """
  13 bup cat-file [--meta|--bupm] /branch/revision/[path]
  14 --
  15 meta        print the target's metadata entry (decoded then reencoded) to stdout
  16 bupm        print the target directory's .bupm file directly to stdout
  17 """
  18
  19 def main(argv):
  20     o = options.Options(optspec)
  21     opt, flags, extra = o.parse_bytes(argv[1:])
  22
  23     git.check_repo_or_die()
  24
  25     if not extra:
  26         o.fatal('must specify a target')
  27     if len(extra) > 1:
  28         o.fatal('only one target file allowed')
  29     if opt.bupm and opt.meta:
  30         o.fatal('--meta and --bupm are incompatible')
  31
  32     target = argv_bytes(extra[0])
  33
  34     if not re.match(br'/*[^/]+/[^/]+', target):
  35         o.fatal("path %r doesn't include a branch and revision" % target)
  36
  37     repo = LocalRepo()
  38     resolved = vfs.resolve(repo, target, follow=False)
  39     leaf_name, leaf_item = resolved[-1]
  40     if not leaf_item:
  41         log('error: cannot access %r in %r\n'
  42             % ('/'.join(name for name, item in resolved), path))
  43         sys.exit(1)
  44
  45     mode = vfs.item_mode(leaf_item)
  46
  47     sys.stdout.flush()
  48     out = byte_stream(sys.stdout)
  49
  50     if opt.bupm:
  51         if not stat.S_ISDIR(mode):
  52             o.fatal('%r is not a directory' % target)
  53         _, bupm_oid = vfs.tree_data_and_bupm(repo, leaf_item.oid)
  54         if bupm_oid:
  55             with vfs.tree_data_reader(repo, bupm_oid) as meta_stream:
  56                 out.write(meta_stream.read())
  57     elif opt.meta:
  58         augmented = vfs.augment_item_meta(repo, leaf_item, include_size=True)
  59         out.write(augmented.meta.encode())
  60     else:
  61         if stat.S_ISREG(mode):
  62             with vfs.fopen(repo, leaf_item) as f:
  63                 for b in chunkyreader(f):
  64                     out.write(b)
  65         else:
  66             o.fatal('%r is not a plain file' % target)
  67
  68     if saved_errors:
  69         log('warning: %d errors encountered\n' % len(saved_errors))
  70         sys.exit(1)