#!/usr/bin/python
import codecs, grpc, os, sys, base64
from lndlibs import rpc_pb2 as ln
from lndlibs import rpc_pb2_grpc as lnrpc

# display config script info
if len(sys.argv) <= 1 or sys.argv[1] == "-h" or sys.argv[1] == "help":
    print("# creating or recovering the LND wallet")
    print("# lnd.winitwallet.py new [walletpassword] [?seedpassword]")
    print("# lnd.winitwallet.py seed [walletpassword] [seedstring] [?seedpassword]")
    print("# lnd.winitwallet.py scb [walletpassword] [seedstring] [filepathSCB] [?seedpassword]")
    print("err='missing parameters'")
    sys.exit(1)

walletpassword=""
seedwords=""
seedpassword=""
filepathSCB=""

mode=sys.argv[1]

if mode=="new":

    print("# *** CREATING NEW LND WALLET ***")

    if len(sys.argv)>2:
        walletpassword=sys.argv[2]
        if len(walletpassword)<8:
            print("err='wallet password is too short'")
            sys.exit(1)
    else:
        print("err='wallet password is too short'")
        sys.exit(1)

    if len(sys.argv)>3:
        seedpassword=sys.argv[3]

elif mode=="seed":

    if len(sys.argv)>2:
        walletpassword=sys.argv[2]
        if len(walletpassword)<8:
            print("err='wallet password is too short'")
            sys.exit(1)
    else:
        print("err='not correct amount of parameter'")
        sys.exit(1)

    if len(sys.argv)>3:
        seedwordString=sys.argv[3]
        seedwords=seedwordString.split(",")
        if len(seedwords)<24:
            print("err='not 24 seed words seperated by just commas'")
            sys.exit(1)
    else:
        print("err='not correct amount of parameter'")
        sys.exit(1)

    if len(sys.argv)>4:
        seedpassword=sys.argv[4]

elif mode=="scb":

        print("err='TODO: implement creating from seed/scb'")
        sys.exit(1)

else:

    print("err='unkown mode parameter - run without any parameters to see options'")
    sys.exit(1)

os.environ['GRPC_SSL_CIPHER_SUITES'] = 'HIGH+ECDSA'
cert = open('/mnt/hdd/lnd/tls.cert', 'rb').read()
ssl_creds = grpc.ssl_channel_credentials(cert)
channel = grpc.secure_channel('localhost:10009', ssl_creds)
stub = lnrpc.WalletUnlockerStub(channel)
if mode=="new":

    request = ln.GenSeedRequest()
    try:
        response = stub.GenSeed(request)
        seedwords = response.cipher_seed_mnemonic
        seedwordsString=', '.join(seedwords)
        print("seedwords='"+seedwordsString+"'")

        # add a 6x4 formatted version to the output
        seedwords6x4=""
        for i in range(0,len(seedwords)):
            if i % 6 == 0 and i != 0:
                seedwords6x4=seedwords6x4+"\n"
            singleWord=str(i+1)+":"+seedwords[i]
            while len(singleWord)<12:
                singleWord=singleWord+" "
            seedwords6x4=seedwords6x4+singleWord
        print("seedwords6x4='"+seedwords6x4+"'")

    except grpc.RpcError as rpc_error_call:
        code = rpc_error_call.code()
        print >> sys.stderr, code
        details = rpc_error_call.details()  
        print("err='RPCError GenSeedRequest'")
        print("errMore='"+details+"'")
        sys.exit(1)
    except: 
        e = sys.exc_info()[0]
        print >> sys.stderr, e
        print("err='GenSeedRequest'")
        sys.exit(1)

    request = ln.InitWalletRequest(
        wallet_password=walletpassword,
        cipher_seed_mnemonic=seedwords
    )
    try:
      response = stub.InitWallet(request)
    except grpc.RpcError as rpc_error_call:
        code = rpc_error_call.code()
        print >> sys.stderr, code
        details = rpc_error_call.details()  
        print("err='RPCError InitWallet'")
        print("errMore='"+details+"'")
        sys.exit(1)
    except: 
        e = sys.exc_info()[0]
        print >> sys.stderr, e
        print("err='InitWallet'")
        sys.exit(1)

elif mode=="seed":

    request = ln.InitWalletRequest(
        wallet_password=walletpassword,
        cipher_seed_mnemonic=seedwords,
        recovery_window=1000
    )

    if len(seedpassword)>0:
        print("# running with seed passphrase")
        request = ln.InitWalletRequest(
            wallet_password=walletpassword,
            cipher_seed_mnemonic=seedwords,
            aezeed_passphrase=seedpassword,
            recovery_window=1000
        )

    try:
        response = stub.InitWallet(request)
     except grpc.RpcError as rpc_error_call:
        code = rpc_error_call.code()
        print >> sys.stderr, code
        details = rpc_error_call.details()  
        print("err='RPCError InitWallet'")
        print("errMore='"+details+"'")
        sys.exit(1)
    except: 
        e = sys.exc_info()[0]
        print >> sys.stderr, e
        print("err='InitWallet'")
        sys.exit(1)

elif mode=="scb":

    print("err='TODO: implement creating from seed/scb'")
    sys.exit(1)