import sys
from slither import Slither

if len(sys.argv) != 2:
    print('python slithIR.py contract.sol')
    exit(-1)

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

# Iterate over all the contracts
for contract in slither.contracts:

    # Iterate over all the functions
    for function in contract.functions:

        # Dont explore inherited functions
        if function.original_contract == contract:

            print('Function: {}'.format(function.name))

            # Iterate over the nodes of the function
            for node in function.nodes:

                # Print the Solidity expression of the nodes
                # And the SlithIR operations
                if node.expression:

                    print('\tSolidity expression: {}'.format(node.expression))
                    print('\tSlithIR:')
                    for ir in node.irs:
                        print('\t\t\t{}'.format(ir))