console

# coding:utf-8
from account import *
from rpc import get_clients, BroadCast, start_server
from transaction import *
from database import *
from block import *
import sys
from miner import mine
import multiprocessing
import rpc 
from node import *
from lib.common import cprint
import inspect

MODULES = ['account','tx','blockchain','miner','node']

def upper_first(string):
    return string[0].upper()+string[1:]

class Node():

    def add(self, args):
        add_node(args[0])
        rpc.BroadCast().add_node(args[0])
        cprint('Allnode',get_nodes())
    
    def run(self, args):
        start_node(args[0])

    def list(self, args):
        for t in NodeDB().find_all():
            cprint('Node',t)

class Miner():
    def start(self, args):
        if get_account() == None:
            cprint('ERROR','Please create account before start miner.')
            exit()
        start_node(args[0])
        while True :
            cprint('Miner new block',mine().to_dict())

class Account():
    def create(self, args):
        ac = new_account()
        cprint('Private Key',ac[0])
        cprint('Public Key',ac[1])
        cprint('Address',ac[2])

    def get(self, args):
        cprint('All Account',AccountDB().read())

    def current(self, args):
        cprint('Current Account', get_account())

class Blockchain():

    def list(self, args):
        for t in BlockChainDB().find_all():
            cprint('Blockchain',str(t))

class Tx():

    def list(self, args):
        for t in TransactionDB().find_all():
            cprint('Transaction',t)

    def transfer(self, args):
        tx = Transaction.transfer(args[0], args[1], args[2])
        print(Transaction.unblock_spread(tx))
        cprint('Transaction tranfer',tx)

def usage(class_name):
    module = globals()[upper_first(class_name)]
    print('  ' + class_name + '\r')
    print('    [action]\r')
    for k,v in module.__dict__.items():
        if callable(v):
            print('      %s' % (k,))
    print('\r')

def help():
    print("Usage: python console.py [module] [action]\r")
    print('[module]\n')
    for m in MODULES:
        usage(m)

if __name__ == '__main__':
    if len(sys.argv) == 1:
        help()
        exit()
    module = sys.argv[1]
    if module == 'help':
        help()
        exit()
    if module not in MODULES:
        cprint('Error', 'First arg shoud in %s' % (str(MODULES,)))
        exit()
    mob = globals()[upper_first(module)]()
    method = sys.argv[2]
    # try:
    getattr(mob, method)(sys.argv[3:])
    # except Exception as e:
    #     cprint('ERROR','/(ㄒoㄒ)/~~, Maybe command params get wrong, please check and try again.')