import sys
from slither.slither import Slither
from slither.evm.convert import SourceToEVM

if len(sys.argv) != 2:
    print('python3 function_called.py functions_called.sol')
    exit(-1)

# Init slither
slither = Slither(sys.argv[1])

# Get the contract evm instructions
contract = slither.get_contract_from_name('Test')
contract_ins = SourceToEVM.get_evm_instructions(contract)
print("## Contract evm instructions: {} ##".format(contract.name))
for ins in contract_ins:
    print(str(ins))

# Get the constructor evm instructions
constructor = contract.constructor
print("## Function evm instructions: {} ##".format(constructor.name))
constructor_ins = SourceToEVM.get_evm_instructions(constructor)
for ins in constructor_ins:
    print(str(ins))
    
# Get the function evm instructions
function = contract.get_function_from_signature('foo()')
print("## Function evm instructions: {} ##".format(function.name))
function_ins = SourceToEVM.get_evm_instructions(function)
for ins in function_ins:
    print(str(ins))
    
# Get the node evm instructions
nodes = function.nodes
for node in nodes:
    node_ins = SourceToEVM.get_evm_instructions(node)
    print("Node evm instructions: {}".format(str(node)))
    for ins in node_ins:
        print(str(ins))