import {
  Address,
  decodeAbiParameters,
  encodeAbiParameters,
  getAbiItem,
} from "viem";
import { poolConfigEncodingABI } from "./generated/wagmi";

import { Hex } from "viem";

/** ABI parameters for performing a SafeTransferFrom based swap when selling on secondary. */
export const safeTransferSwapAbiParameters = [
  { name: "recipient", internalType: "address payable", type: "address" },
  { name: "minEthToAcquire", internalType: "uint256", type: "uint256" },
  { name: "sqrtPriceLimitX96", internalType: "uint160", type: "uint160" },
] as const;

export const buySupplyWithSwapRouterHookAbiParameters = [
  { name: "buyRecipient", internalType: "address", type: "address" },
  { name: "swapRouterCall", internalType: "bytes", type: "bytes" },
] as const;

export const buySupplyWithSwapRouterHookReturnParameters = [
  { name: "amountCurrency", internalType: "uint256", type: "uint256" },
  { name: "coinsPurchased", internalType: "uint256", type: "uint256" },
] as const;

/**
 * Decodes the return data from the BuySupplyWithSwapRouterHook.
 *
 * @param returnData - The return data from the BuySupplyWithSwapRouterHook.
 * @returns The decoded return data.
 */
export const decodeBuySupplyWithSwapRouterHookReturn = (returnData: Hex) => {
  const result = decodeAbiParameters(
    buySupplyWithSwapRouterHookReturnParameters,
    returnData,
  );

  return {
    amountCurrency: result[0],
    coinsPurchased: result[1],
  };
};

const UNISWAP_V4_MULTICURVE_POOL_VERSION = 4;

/**
 * Encodes the pool configuration data for creating and initializing a coin's liquidity pool,
 * using a multi-curve setup.
 *
 * @param params - The pool configuration parameters.
 * @param params.version - The version of the pool configuration.
 *                         2 for a UniswapV3 pool, or 4 for Doppler/Uniswap V4-style pools.
 * @param params.currency - The address of the currency token (e.g., WETH) to be paired with the coin.
 * @param params.tickLower - An array of numbers representing the lower tick boundaries for each liquidity curve.
 * @param params.tickUpper - An array of numbers representing the upper tick boundaries for each liquidity curve.
 * @param params.numDiscoveryPositions - An array of numbers, where each number specifies the quantity of discrete
 *                                     liquidity positions to be created within the corresponding curve's discovery phase.
 * @param params.maxDiscoverySupplyShare - An array of bigints, where each bigint represents the maximum share of the coin's
 *                                        total supply allocated to the discovery phase of the corresponding curve.
 *                                        This is typically a WAD-scaled value (i.e., scaled by 1e18).
 * @returns The ABI-encoded pool configuration for setting
 */
export const encodeMultiCurvePoolConfig = ({
  currency,
  tickLower,
  tickUpper,
  numDiscoveryPositions,
  maxDiscoverySupplyShare,
}: {
  currency: Address;
  tickLower: number[];
  tickUpper: number[];
  numDiscoveryPositions: number[];
  maxDiscoverySupplyShare: bigint[];
}) => {
  const abiItem = getAbiItem({
    abi: poolConfigEncodingABI,
    name: "encodeMultiCurvePoolConfig",
  });

  return encodeAbiParameters(abiItem.inputs, [
    UNISWAP_V4_MULTICURVE_POOL_VERSION,
    currency,
    tickLower,
    tickUpper,
    numDiscoveryPositions,
    maxDiscoverySupplyShare,
  ]);
};