flatten-super-calls.py

from slither import Slither
from slither.slithir.operations import InternalCall, SolidityCall
from slither.core.expressions.super_call_expression import SuperCallExpression
sl = Slither("MyContract.sol")

c = sl.get_contract_from_name("MyContract")[0]


def get_super_calls(x):
    super_call_content = ""
    for node in x.nodes:
        for op in node.irs:
            if isinstance(op, InternalCall) and isinstance(op.expression, SuperCallExpression)
                super_call_content = op.function.source_mapping.content
                tmp = get_super_calls(op.function)
                if tmp != "":
                    super_call_content += tmp
    return super_call_content

for function in c.functions_declared:
    print(function.name, "\n")
    cnt = get_super_calls(function)
    if cnt != "":
        with open(function.source_mapping.filename.absolute) as f:
            content = f.read()
        with open(function.source_mapping.filename.absolute, "w") as f:
            f.write(content[:function.source_mapping.start] + cnt + content[function.source_mapping.end:])

    print("\n")