import type { Config } from "../createConfig.js";
import type { BaseErrorType } from "../errors/base.js";
import { createWallet } from "./createWallet.js";
import { getWalletFromRelayer } from "./getWalletFromRelayer.js";
import { getWalletId } from "./getWalletId.js";
import { checkForWalletUpdatesOnChain, lookupWallet } from "./lookupWallet.js";

export type ConnectParameters = {
    isCreateWallet?: boolean;
};

export type ConnectReturnType = {
    isLookup: boolean;
    job: Promise<void>;
} | void;

export type ConnectErrorType = BaseErrorType;

export async function connect(
    config: Config,
    params: ConnectParameters = {},
): Promise<ConnectReturnType> {
    try {
        const { isCreateWallet = false } = params;
        const walletId = getWalletId(config);
        config.setState((x) => ({ ...x, id: walletId }));

        console.log("Attempting to connect wallet", { walletId: config.state.id });

        try {
            const wallet = await getWalletFromRelayer(config);
            if (wallet) {
                config.setState((x) => ({ ...x, status: "in relayer" }));
                console.log("Wallet found in relayer", {
                    status: "in relayer",
                    walletId: config.state.id,
                });
                return;
            }
        } catch (error) {
            console.error("Wallet not found in relayer", {
                error,
                walletId: config.state.id,
            });
        }

        // Create wallet iff no WalletUpdated events found onchain
        const shouldCreateWallet = await checkForWalletUpdatesOnChain(config);
        if (shouldCreateWallet || isCreateWallet) {
            return Promise.resolve({
                isLookup: false,
                job: createWallet(config),
            });
        }
        return Promise.resolve({
            isLookup: true,
            job: lookupWallet(config),
        });
    } catch (error) {
        console.error("Could not connect wallet", {
            error,
            walletId: config.state.id,
        });
        config.setState({});
    }
}