Function main

encrypt.py:71–121

()

def main():
    plugins = load_plugins()
    plugin_names = [getattr(p, 'name', None) for p in plugins]
    plugin_names = [n for n in plugin_names if n]

    parser = build_base_parser(plugin_names)
    # parse initial to know selected plugin
    args, _ = parser.parse_known_args()
    if args.method is None:
        parser.print_help()
        raise SystemExit(1)
    # find plugin object
    selected = None
    for p in plugins:
        if getattr(p, 'name', None) == args.method:
            selected = p
            break
    if selected is None:
        raise SystemExit(f"Unsupported --method: {args.method}")
    # let plugin add arguments if needed
    if hasattr(selected, 'add_arguments') and callable(selected.add_arguments):
        selected.add_arguments(parser)
    # final parse
    args = parser.parse_args()

    data = read_binary_file(args.input)
    if data is None:
        return

    # call plugin to produce final bytes (pre-encode)
    if hasattr(selected, 'process') and callable(selected.process):
        final = selected.process(data, args)
    elif hasattr(selected, 'process_data') and callable(selected.process_data):
        final = selected.process_data(data, args)
    else:
        raise SystemExit(f"Plugin for {args.method} does not expose a process function")

    # Apply encoding based on --encode option
    if args.encode == "base64":
        final = base64.b64encode(final)
    elif args.encode == "base32":
        final = base64.b32encode(final)
    elif args.encode == "hex":
        final = binascii.hexlify(final)
    elif args.encode == "urlsafe_base64":
        final = base64.urlsafe_b64encode(final)
    elif args.encode == "none":
        pass  # final remains as bytes

    save(args.output, final)
    print(f"Encrypted data (method={args.method}, encode={args.encode}) saved to {args.output}")


if __name__ == '__main__':