import {
  type UseQueryResult,
  useQueries,
  useQuery,
} from "@tanstack/react-query";
import { useMemo } from "react";
import type { Chain, ChainMetadata } from "../../../../chains/types.js";
import {
  convertApiChainToChain,
  getChainMetadata,
} from "../../../../chains/utils.js";
import { pLimit } from "../../../../utils/promise/p-limit.js";

export function useChainName(chain?: Chain) {
  // only if we have a chain and no chain name!
  const isEnabled = !!chain && !chain.name;
  const chainQuery = useQuery({
    queryFn: async () => {
      if (!chain) {
        throw new Error("chain is required");
      }
      return convertApiChainToChain(await getChainMetadata(chain));
    },
    ...getQueryOptions(chain),
    enabled: isEnabled,
    retry: false,
  });

  return {
    isLoading: isEnabled && chainQuery.isLoading,
    name: chain?.name ?? chainQuery.data?.name,
  };
}

export function useChainIconUrl(chain?: Chain) {
  // only if we have a chain and no chain icon url!
  const isEnabled = !!chain && !chain.icon?.url;
  const chainQuery = useQuery({
    // only if we have a chain and no chain icon url!
    queryFn: async () => {
      if (!chain) {
        throw new Error("chain is required");
      }
      return convertApiChainToChain(await getChainMetadata(chain));
    },
    ...getQueryOptions(chain),
    enabled: isEnabled,
    retry: false,
  });

  return {
    isLoading: isEnabled && chainQuery.isLoading,
    url: chain?.icon?.url ?? chainQuery.data?.icon?.url,
  };
}

export function useChainFaucets(chain?: Chain) {
  // only if we have a chain and it might be a testnet and no faucets and its not localhost
  const isEnabled =
    !!chain &&
    "testnet" in chain &&
    !chain.faucets?.length &&
    chain.id !== 1337;

  const chainQuery = useQuery({
    queryFn: async () => {
      if (!chain) {
        throw new Error("chain is required");
      }
      return getChainMetadata(chain);
    },
    ...getQueryOptions(chain),
    enabled: isEnabled,
  });

  return {
    faucets: chain?.faucets ?? chainQuery.data?.faucets ?? [],
    isLoading: isEnabled && chainQuery.isLoading,
  };
}

export function useChainSymbol(chain?: Chain) {
  // only if we have a chain and no chain icon url!
  const isEnabled = !!chain && !chain.nativeCurrency?.symbol;
  const chainQuery = useQuery({
    queryFn: async () => {
      if (!chain) {
        throw new Error("chain is required");
      }
      return getChainMetadata(chain);
    },
    ...getQueryOptions(chain),
    enabled: isEnabled,
  });

  return {
    isLoading: isEnabled && chainQuery.isLoading,
    symbol:
      chain?.nativeCurrency?.symbol ?? chainQuery.data?.nativeCurrency?.symbol,
  };
}

export function useChainExplorers(chain?: Chain) {
  // only if we have a chain and it might be a testnet and no faucets and its not localhost
  const isEnabled = !!chain && !chain.blockExplorers?.length;

  const chainQuery = useQuery({
    queryFn: async () => {
      if (!chain) {
        throw new Error("chain is required");
      }
      return getChainMetadata(chain);
    },
    ...getQueryOptions(chain),
    enabled: isEnabled,
  });

  const toChain = chainQuery.data
    ? convertApiChainToChain(chainQuery.data)
    : undefined;
  return {
    explorers:
      chain?.blockExplorers && chain?.blockExplorers?.length > 0
        ? chain?.blockExplorers
        : (toChain?.blockExplorers ?? []),
    isLoading: isEnabled && chainQuery.isLoading,
  };
}

function getQueryOptions(chain?: Chain) {
  return {
    enabled: !!chain,
    queryKey: ["chain", chain?.id],
    staleTime: 1000 * 60 * 60, // 1 hour
  } as const;
}

/**
 * Retrieves metadata for a chain such as name, icon, available faucets, block explorers, etc.
 *
 * @param chain - Chain to retrieve metadata for, see [defineChain](https://portal.thirdweb.com/references/typescript/v5/defineChain) for how to create a chain from a chain ID.
 * @returns A React Query result containing the chain metadata
 *
 * @example
 * ```
 * import { useChainMetadata } from "thirdweb/react";
 *
 * const { data: chainMetadata } = useChainMetadata(defineChain(11155111));
 *
 * console.log("Name:", chainMetadata.name); // Sepolia
 * console.log("Faucets:", chainMetadata.faucets); // ["https://thirdweb.com/sepolia/faucet"]
 * console.log("Explorers:", chainMetadata.explorers); // ["https://sepolia.etherscan.io/"]
 * ```
 *
 * @chain
 */
export function useChainMetadata(chain?: Chain): UseQueryResult<ChainMetadata> {
  return useQuery({
    ...getQueryOptions(chain),
    queryFn: async () => {
      if (!chain) {
        throw new Error("chainId is required");
      }
      return getChainMetadata(chain);
    },
  });
}

/**
 * @param chains - array of `Chains`
 * @param maxConcurrency - maximum number of concurrent requests to make
 * @internal
 */
export function useChainsQuery(chains: Chain[], maxConcurrency: number) {
  const queryList = useMemo(() => {
    const limit = pLimit(maxConcurrency);
    return chains.map((chain) => {
      return {
        ...getQueryOptions(chain),
        queryFn: () => limit(() => getChainMetadata(chain)),
      };
    });
  }, [chains, maxConcurrency]);

  return useQueries({
    queries: queryList,
  });
}