import { confirm, number, password } from "@inquirer/prompts";
import { type JsonRpcProvider, formatUnits, formatEther } from "ethers";
import { MARKETPLACE_GATEWAY_V_2 } from "@roninbuilders/contracts";
import { getAxieContract, getWETHContract, getUSDCContract } from "./contracts";
import { getAxieIdsFromAccount } from "./axie";

export async function getAccountInfo(
  address: string,
  provider: JsonRpcProvider,
) {
  const axieIds = await getAxieIdsFromAccount(address, provider);
  const wethContract = getWETHContract(provider);
  const usdcContract = getUSDCContract(provider);
  const balance = await provider.getBalance(address);
  const wethBalance = await wethContract.balanceOf(address);
  const usdcBalance = await usdcContract.balanceOf(address);
  const allowance = await wethContract.allowance(
    address,
    MARKETPLACE_GATEWAY_V_2.address,
  );
  const axieContract = getAxieContract(provider);
  const isApprovedForAll = await axieContract.isApprovedForAll(
    address,
    MARKETPLACE_GATEWAY_V_2.address,
  );

  return {
    address,
    ronBalance: formatEther(balance),
    wethBalance: formatEther(wethBalance),
    usdcBalance: formatUnits(usdcBalance, 6),
    allowance,
    isApprovedForAll,
    axieIds,
  };
}

export async function apiRequest<T>(
  url: string,
  body: BodyInit | null = null,
  headers: Record<string, string> = {},
  method: "GET" | "POST" = "POST",
) {
  const response = await fetch(url, {
    method,
    headers: {
      "Content-Type": "application/json",
      ...headers,
    },
    ...(method === "GET" ? {} : { body }),
  });

  const res: T = await response.json();
  return res;
}

export const askToContinue = async () => {
  const continueUsing = await confirm({
    message: "🔄 Would you like to do something else?",
  });
  if (!continueUsing) {
    console.log("👋 Goodbye!");
    process.exit(0);
  }
};

export async function ensureMarketplaceToken(): Promise<string> {
  if (!process.env.MARKETPLACE_ACCESS_TOKEN) {
    const token = await password({
      message: "🔑 Enter your Marketplace access token:",
      validate: (value) => value !== undefined && value !== "",
    });
    process.env.MARKETPLACE_ACCESS_TOKEN = token;
  }
  return process.env.MARKETPLACE_ACCESS_TOKEN;
}

export const getAxieId = async () => {
  const axieId = await number({
    message: "🆔 Enter Axie ID:",
    validate: (value) => value !== undefined && !Number.isNaN(value),
  });
  if (axieId === undefined) {
    console.log("❌ Invalid Axie ID!");
    return null;
  }
  return axieId;
};