// Copyright 2023 @soul-wallet/extension-base
// SPDX-License-Identifier: Apache-2.0

import { getWasmContractGasLimit } from '@soul-wallet/extension-base/koni/api/tokens/wasm/utils';
import { _AZERO_DOMAIN_REGISTRY_ABI, _NEUROGUNS_PSP34_ABI, _PINK_PSP34_ABI, _PSP22_ABI, _PSP34_ABI } from '@soul-wallet/extension-base/services/chain-service/helper';
import { _SubstrateApi } from '@soul-wallet/extension-base/services/chain-service/types';

import { ApiPromise } from '@polkadot/api';
import { ContractPromise } from '@polkadot/api-contract';

export function getPSP22ContractPromise (apiPromise: ApiPromise, contractAddress: string) {
  return new ContractPromise(apiPromise, _PSP22_ABI, contractAddress);
}

export function isPinkRoboNft (contractAddress: string) {
  return ['XoywUxTTtNKPRrRN7V5KXCqz2QLMFeK7DxhpSniqZHps5Xq'].includes(contractAddress);
}

export function isNeurogunNft (contractAddress: string) {
  return ['aZ9bd2tHeGKrs3FnJv5oe7kgVrP5GQvdJMhC2GxjXA2Yqbd'].includes(contractAddress);
}

export function isAzeroDomainNft (contractAddress: string) {
  return ['5FsB91tXSEuMj6akzdPczAtmBaVKToqHmtAwSUzXh49AYzaD', '5CTQBfBC9SfdrCDBJdfLiyW2pg9z5W6C6Es8sK313BLnFgDf'].includes(contractAddress);
}

export function getPSP34ContractPromise (apiPromise: ApiPromise, contractAddress: string) {
  if (isPinkRoboNft(contractAddress)) {
    return new ContractPromise(apiPromise, _PINK_PSP34_ABI, contractAddress);
  }

  if (isNeurogunNft(contractAddress)) {
    return new ContractPromise(apiPromise, _NEUROGUNS_PSP34_ABI, contractAddress);
  }

  if (isAzeroDomainNft(contractAddress)) {
    return new ContractPromise(apiPromise, _AZERO_DOMAIN_REGISTRY_ABI, contractAddress);
  }

  return new ContractPromise(apiPromise, _PSP34_ABI, contractAddress);
}

export function getTokenUriMethod (contractAddress: string): string {
  if (isPinkRoboNft(contractAddress)) {
    return 'pinkMint::tokenUri';
  }

  if (isNeurogunNft(contractAddress)) {
    return 'tokenUri';
  }

  return 'psp34Traits::tokenUri';
}

const mustFormatNumberReg = /^-?[0-9][0-9,.]+$/;

export async function getPSP34TransferExtrinsic (networkKey: string, substrateApi: _SubstrateApi, senderAddress: string, recipientAddress: string, params: Record<string, any>) {
  const contractAddress = params.contractAddress as string;
  const onChainOption = params.onChainOption as Record<string, string>;

  for (const [key, value] of Object.entries(onChainOption)) {
    if (mustFormatNumberReg.test(value)) {
      onChainOption[key] = value.replaceAll(',', '');
    }
  }

  try {
    const contractPromise = getPSP34ContractPromise(substrateApi.api, contractAddress);
    // @ts-ignore
    const gasLimit = await getWasmContractGasLimit(substrateApi.api, senderAddress, 'psp34::transfer', contractPromise, {}, [recipientAddress, onChainOption, {}]);

    // @ts-ignore
    return contractPromise.tx['psp34::transfer']({ gasLimit }, recipientAddress, onChainOption, {});
  } catch (e) {
    console.debug(e);

    return null;
  }
}