fb_bls_derive_key_and_sign.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import argparse
import sys
import os
import argparse
import getpass
import sys
from termcolor import colored

from utils import genver

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("key_file",type=str, help="BLS key verification file")
    parser.add_argument("--index",type=int, help="derivation index", required=True)
    parser.add_argument("--RSA",type=str, help="Private RSA key file")
    parser.add_argument("--sign_msg", type=str, help="Sign test message (with private key share)")
    args = parser.parse_args()
    
    if not os.path.exists(args.key_file): 
        print(colored(f'BLS verificaion file {args.key_file} not found.',"cyan"))
        exit(-1)
    
    passphrase = None
    if args.RSA:
        if not os.path.exists(args.RSA): 
            print(colored(f'RSA private key files {args.RSA} not found.', "cyan"))
        passphrase = getpass.getpass(prompt='Please enter RSA private key passphrase:')
    else:
        if args.sign_msg:
            print(colored(f'To sign a message, must provide RSA private key file', "cyan"))
            exit(-1)

        passphrase = getpass.getpass(prompt='Please enter BLS public key integrity passphrase:')

    print("Generated Address:", colored(genver.derive_address_and_sign(args.key_file, args.index, passphrase, args.RSA, args.sign_msg), "green"))

if __name__ == "__main__":
    main()