| #!/usr/bin/env python -tt |
| |
| import argparse |
| import os |
| import sys |
| |
| import dbsign.commands as C |
| import dbsign.logger as L |
| import dbsign.security as S |
| |
| from dbsign.commands import CFG |
| |
| |
| log = L.get_logger(CFG['executable']) |
| |
| |
| def make_parser(): # type: () -> argparse.ArgumentParser |
| parser = argparse.ArgumentParser(prog=CFG['executable'], add_help=False) |
| parser.set_defaults(func=lambda: C.cmd_help(parser)) |
| parser.add_argument('--debug', '-d', '--verbose', '-v', |
| action='count', default=0, |
| help='be more verbose') |
| parser.add_argument('--quiet', '-q', |
| action='count', default=0, |
| help='be quieter') |
| parser.add_argument('--unsafe', action='store_true', |
| help='permit unsafe operations') |
| subparsers = parser.add_subparsers(help='Subcommands') |
| |
| parser_help = subparsers.add_parser('help', help='Display usage info') |
| parser_help.set_defaults(func=lambda: C.cmd_help(parser)) |
| |
| parser_check = subparsers.add_parser( |
| 'check', help='Validate current system settings') |
| parser_check.set_defaults(func=C.cmd_check) |
| |
| parser_clean = subparsers.add_parser( |
| 'clean', help='Remove identity and keychain') |
| parser_clean.set_defaults(func=C.cmd_clean) |
| |
| parser_import = subparsers.add_parser( |
| 'import', help='Import and trust new identity from .P12 file') |
| parser_import.set_defaults(func=C.cmd_import) |
| parser_import.add_argument('id_file', metavar='P12_FILE', |
| help='Identity in .p12 format') |
| |
| parser_setup = subparsers.add_parser( |
| 'setup', help='Initialize keychain and privilege settings') |
| parser_setup.set_defaults(func=C.cmd_setup) |
| |
| parser_unlock = subparsers.add_parser( |
| 'prep', help='Unlock keychain and verify identity is valid') |
| parser_unlock.set_defaults(func=C.cmd_prep) |
| |
| parser_remove = subparsers.add_parser( |
| 'remove', help='Remove the installed identity') |
| parser_remove.set_defaults(func=C.cmd_remove) |
| |
| parser_lint = subparsers.add_parser( |
| 'lint', help='Lint the program source code') |
| parser_lint.set_defaults(func=C.cmd_lint) |
| |
| parser_test = subparsers.add_parser( |
| 'test', help='Test the program') |
| parser_test.set_defaults(func=C.cmd_test) |
| |
| return parser |
| |
| |
| def main(main_args): # type: (list[str]) -> int |
| parser = make_parser() |
| args = parser.parse_args(main_args) |
| |
| if args.unsafe: |
| os.environ[S.UNSAFE_FLAG] = 'YES' |
| |
| # Logging setup |
| level = L.BASE_LOGLEVEL + (args.quiet - args.debug) * 10 |
| L.set_level(L.normalize(level)) |
| |
| CFG['debug'] = (args.debug > 0) |
| CFG['keydb'] = '{}/Library/Keychains/{}.{}'.format( |
| os.environ['HOME'], CFG['keynick'], S.derive_keychain_extension()) |
| if hasattr(args, 'id_file'): |
| CFG['id_file'] = args.id_file |
| |
| log.debug('Command line parameters: %s', sys.argv) |
| log.debug('Arguments to main: %s', main_args) |
| log.debug('Parsed args: %s', args) |
| log.debug('Configuration: %s', CFG) |
| |
| return args.func() |
| |
| |
| if __name__ == '__main__': |
| return_code = main(sys.argv[1:]) |
| sys.exit(return_code) |
