import invariant from "tiny-invariant";
import { CREATE_WALLET_ROUTE } from "../constants.js";
import type { RenegadeConfig } from "../createConfig.js";
import { BaseError } from "../errors/base.js";
import { postRelayerRaw } from "../utils/http.js";
import { waitForWalletIndexing } from "./waitForWalletIndexing.js";

export type CreateWalletReturnType = ReturnType<typeof waitForWalletIndexing>;

export type CreateWalletParameters = {
    blinderSeed?: string;
    shareSeed?: string;
    skMatch?: string;
};

export async function createWallet(
    config: RenegadeConfig,
    parameters: CreateWalletParameters = {},
): CreateWalletReturnType {
    const { getBaseUrl, utils } = config;
    let body: string;
    const headers = {
        "Content-Type": "application/json",
    };

    if (config.renegadeKeyType === "internal") {
        const { seed } = config.state;
        invariant(seed, "seed is required");
        body = utils.create_wallet(seed);
    } else {
        const { blinderSeed, shareSeed, skMatch } = parameters;
        const { walletId, publicKey, symmetricKey } = config;
        invariant(blinderSeed, "blinderSeed is required");
        invariant(shareSeed, "shareSeed is required");
        invariant(skMatch, "skMatch is required");
        body = await utils.create_external_wallet(
            walletId,
            blinderSeed,
            shareSeed,
            publicKey,
            skMatch,
            symmetricKey,
        );
    }

    const res = await postRelayerRaw(getBaseUrl(CREATE_WALLET_ROUTE), body, headers);
    if (res.task_id) {
        if (config.renegadeKeyType === "internal") {
            config.setState((x) => ({ ...x, status: "creating wallet" }));
        }
        console.log(`task create-wallet(${res.task_id}): ${res.wallet_id}`, {
            status: "creating wallet",
            walletId: res.wallet_id,
        });
        return waitForWalletIndexing(config, {
            isLookup: false,
            onComplete: (wallet) => {
                if (config.renegadeKeyType === "internal") {
                    config.setState((x) => ({ ...x, status: "in relayer" }));
                }
                console.log(`task create-wallet(${res.task_id}) completed: ${wallet.id}`, {
                    status: "in relayer",
                    walletId: wallet.id,
                });
            },
            onFailure: () => {
                console.log(`task create-wallet(${res.task_id}) failed`);
                if (config.renegadeKeyType === "internal") {
                    config.setState({});
                }
            },
        });
    }
    return Promise.reject(new BaseError("Failed to create wallet"));
}