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, 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     with LocalRepo() as repo:
  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                 % (b'/'.join(name for name, item in resolved), target))
  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)