#!/usr/bin/env -S tsx

import { ethers } from "ethers";

import { DiamondProxyResolver } from '../src/proxies.js';
import { disasm } from '../src/disasm.js';
import { withCache } from "../src/internal/filecache.js";
import { opcodes } from "../src/opcodes.js";
import { CompatibleProvider } from "../src/providers.js";

const { INFURA_API_KEY, PROVIDER_RPC_URL } = process.env;
const provider = CompatibleProvider(INFURA_API_KEY ? (new ethers.InfuraProvider("homestead", INFURA_API_KEY)) : (PROVIDER_RPC_URL ? (new ethers.JsonRpcProvider(PROVIDER_RPC_URL)) : ethers.getDefaultProvider("homestead")));

async function main() {
    const address = process.env["ADDRESS"] || process.argv[2];
    const selector = process.env["SELECTOR"] || process.argv[3];

    console.debug("Loading code for address:", address);
    const code = await withCache(
        `${address}_abi`,
        async () => {
            return await provider.getCode(address)
        },
    );

    const program = disasm(code);

    let hasDelegateCall = false;
    for (const fn of Object.values(program.dests)) {
        if (fn.opTags.has(opcodes.DELEGATECALL)) {
            hasDelegateCall = true;
            break;
        }
    }

    console.log("Detected proxies: ", program.proxies);

    for (const resolver of program.proxies) {
        console.log("Resolving proxy:", resolver.toString());

        if (!selector && resolver instanceof DiamondProxyResolver) {
            const facets = await (resolver as DiamondProxyResolver).facets(provider, address);
            console.log("Resolved to facets: ", facets);
        } else {
            const addr = await resolver.resolve(provider, address, selector);
            if (addr === "0x0000000000000000000000000000000000000000") continue;
            console.log("Resolved to address:", addr);
        }

        return;
    }

    if (hasDelegateCall && program.proxies.length === 0) {
        console.log("DELEGATECALL detected but no proxies found");
    } else {
        console.log("No DELEGATECALL detected");
        return;
    }

}

main().then().catch(err => {
    console.error("Failed:", err)
    process.exit(2);
})