import type { Hex } from "viem";
import type {
  BaseTransactionOptions,
  WithOverrides,
} from "../../../../transaction/types.js";
import type { ClaimCondition } from "../../../../utils/extensions/drops/types.js";
import {
  isSetClaimConditionsSupported,
  setClaimConditions as setClaimConditionsMultiPhase,
} from "../../__generated__/IDrop1155/write/setClaimConditions.js";
import {
  claimCondition as claimConditionSinglePhase,
  isClaimConditionSupported,
} from "../../__generated__/IDropSinglePhase1155/read/claimCondition.js";
import {
  isSetClaimConditionsSupported as isSetClaimConditionsSupportedGeneratedSinglePhase,
  setClaimConditions as setClaimConditionsSinglePhase,
} from "../../__generated__/IDropSinglePhase1155/write/setClaimConditions.js";
import {
  type GetClaimConditionsParams,
  getClaimConditions,
  isGetClaimConditionsSupported,
} from "../read/getClaimConditions.js";

export type ResetClaimEligibilityParams = GetClaimConditionsParams;

/**
 * Reset the claim eligibility for all users.
 * This method is only available on the `DropERC1155` contract.
 * @param options
 * @returns the prepared transaction
 * @extension ERC1155
 * @example
 * ```ts
 * import { resetClaimEligibility } from "thirdweb/extensions/erc1155";
 * import { sendTransaction } from "thirdweb";
 *
 * const transaction = resetClaimEligibility({
 *  contract,
 * });
 *
 * await sendTransaction({ transaction, account });
 * ```
 */
export function resetClaimEligibility(
  options: BaseTransactionOptions<
    WithOverrides<ResetClaimEligibilityParams>
  > & {
    singlePhaseDrop?: boolean;
  },
) {
  if (options.singlePhaseDrop) {
    return setClaimConditionsSinglePhase({
      asyncParams: async () => {
        // get existing condition
        const existingCondition = await claimConditionSinglePhase(options).then(
          ([
            startTimestamp,
            maxClaimableSupply,
            supplyClaimed,
            quantityLimitPerWallet,
            merkleRoot,
            pricePerToken,
            currency,
            metadata,
          ]) => ({
            currency,
            maxClaimableSupply,
            merkleRoot,
            metadata,
            pricePerToken,
            quantityLimitPerWallet,
            startTimestamp,
            supplyClaimed,
          }),
        );

        // then simply return the exact same ones, but with the resetClaimEligibility flag set to true
        return {
          // type is necessary because of viem hex shenanigans (strict vs non-strict `0x` prefix string)
          phase: existingCondition,
          resetClaimEligibility: true,
          tokenId: options.tokenId,
        };
      },
      contract: options.contract,
      overrides: options.overrides,
    });
  }
  // download existing conditions
  return setClaimConditionsMultiPhase({
    asyncParams: async () => {
      // get existing conditions
      const existingConditions = await getClaimConditions(options);

      // then simply return the exact same ones, but with the resetClaimEligibility flag set to true
      return {
        // type is necessary because of viem hex shenanigans (strict vs non-strict `0x` prefix string)
        phases: existingConditions as Array<
          ClaimCondition & {
            currency: Hex;
            merkleRoot: Hex;
          }
        >,
        resetClaimEligibility: true,
        tokenId: options.tokenId,
      };
    },
    contract: options.contract,
    overrides: options.overrides,
  });
}

/**
 * Checks if the `resetClaimEligibility` method is supported by the given contract.
 * @param availableSelectors An array of 4byte function selectors of the contract. You can get this in various ways, such as using "whatsabi" or if you have the ABI of the contract available you can use it to generate the selectors.
 * @returns A boolean indicating if the `resetClaimEligibility` method is supported.
 * @extension ERC1155
 * @example
 * ```ts
 * import { isResetClaimEligibilitySupported } from "thirdweb/extensions/erc1155";
 *
 * const supported = isResetClaimEligibilitySupported(["0x..."]);
 * ```
 */
export function isResetClaimEligibilitySupported(availableSelectors: string[]) {
  return (
    (isGetClaimConditionsSupported(availableSelectors) &&
      isSetClaimConditionsSupported(availableSelectors)) ||
    isClaimConditionSupported(availableSelectors) ||
    isSetClaimConditionsSupportedGeneratedSinglePhase(availableSelectors)
  );
}