[metadata]

name = Shamira

author = Tadeáš Berkman

url = https://repo.19x19.cz/Shamira

project_urls =

    Issue Tracker = https://trac.19x19.cz/shamira/report

[options]

packages =

    shamira

    shamira.tests

package_dir =

    =src

python_requires = >=3.6

[options.entry_points]

console_scripts =

    shamira = shamira.cli:run

# GNU GPLv3, see LICENSE

import sys

from argparse import ArgumentParser

from .core import generate, reconstruct, SException

from .benchmark import build_subparsers as build_benchmark

def run():

	subparsers = parser.add_subparsers()

	build_split_parser(subparsers.add_parser("split"))

	build_join_parser(subparsers.add_parser("join"))

	build_benchmark(subparsers.add_parser("benchmark"))

	parser.set_defaults(func=lambda _: parser.error("missing command"))

	args = parser.parse_args()

	args.func(args)

def build_split_parser(parser):

	parser.add_argument("-k", type=int, required=True, help="number of shares necessary for recovering the secret")

	parser.add_argument("-n", type=int, required=True, help="number of generated shares")

	encoding = parser.add_mutually_exclusive_group()

	encoding.add_argument("--hex", action="store_true", help="encode shares' bytes as a hexadecimal string")

	encoding.add_argument("--b32", action="store_true", help="encode shares' bytes as a base32 string")

	encoding.add_argument("--b64", action="store_true", help="encode shares' bytes as a base64 string")

	parser.add_argument("--label", help="any label to prefix the shares with")

	parser.add_argument("--omit_k_n", action="store_true", help="suppress the default shares prefix")

	parser.add_argument("secret", nargs="?", help="a secret to be split. Can be provided on the command line,"

		" redirected through stdin, or will be asked for interactively.")

	parser.set_defaults(func=_generate)

def build_join_parser(parser):

	encoding = parser.add_mutually_exclusive_group()

	encoding.add_argument("--hex", action="store_true", help="decode shares' bytes from a hexadecimal string")

	encoding.add_argument("--b32", action="store_true", help="decode shares' bytes from a base32 string")

	encoding.add_argument("--b64", action="store_true", help="decode shares' bytes from a base64 string")

	parser.add_argument("-r", "--raw", action="store_true", help="return the secret as raw bytes")

	parser.add_argument("share", nargs="*", help="shares to be joined. Can be provided on the command line,"

		" redirected through stdin, or will be asked for interactively.")

	parser.set_defaults(func=_reconstruct)

def _generate(args):

	encoding = get_encoding(args) or "b32"

	if args.secret:  # provided as a positional argument

		secret = args.secret

	elif sys.stdin.isatty():  # input from terminal

		secret = input("Enter your secret:\n")