import type { RenegadeConfig } from "../createConfig.js";
import { BaseError } from "../errors/base.js";
import type { Wallet } from "../types/wallet.js";
import { getWalletFromRelayer } from "./getWalletFromRelayer.js";

export type WaitForWalletIndexParameters = {
    onComplete?: (wallet: Wallet) => void;
    onFailure?: () => void;
    timeout?: number;
    isLookup?: boolean;
};

export type WaitForWalletIndexReturnType = Promise<void>;

export async function waitForWalletIndexing(
    config: RenegadeConfig,
    parameters: WaitForWalletIndexParameters,
): WaitForWalletIndexReturnType {
    const { onComplete, onFailure, timeout = 60000, isLookup } = parameters;
    const pollingInterval = config.renegadeKeyType === "internal" ? config.pollingInterval : 5000;

    const startTime = Date.now();

    while (true) {
        if (Date.now() - startTime >= timeout) {
            onFailure?.();
            throw new BaseError(`Timed out while ${isLookup ? "looking up" : "creating"} wallet`);
        }

        try {
            const wallet = await getWalletFromRelayer(config);
            if (wallet) {
                onComplete?.(wallet);
                break;
            }
        } catch (_) {
            // Do nothing, just continue the loop
        }

        // Sleep for a bit before polling again
        await new Promise((resolve) => setTimeout(resolve, pollingInterval / 2));
    }
}