#!python

import argparse

from plenum.common.keygen_utils import init_bls_keys
from plenum.common.script_helper import buildKeepDirIfNotExists

from indy_common.config_util import getConfig
from indy_common.config_helper import NodeConfigHelper


def parse_args():
    parser = argparse.ArgumentParser(
        description="Generate BLS keys for a node "
                    "by taking the node's name and a seed")

    parser.add_argument('--name', required=True, help='node name')
    parser.add_argument('--seed', required=False, type=str,
                        help='seeds for keypair')

    args = parser.parse_args()

    return args.name, args.seed


def generate_and_store_bls_keys(keys_dir, node_name, seed):
    try:
        init_bls_keys(keys_dir, node_name, seed)
    except Exception as ex:
        print(ex)
        exit()


if __name__ == "__main__":
    node_name, seed = parse_args()

    node_config_helper = NodeConfigHelper(node_name, getConfig())
    keys_dir = node_config_helper.keys_dir
    buildKeepDirIfNotExists(keys_dir)
    generate_and_store_bls_keys(keys_dir, node_name, seed)
