import { getContract } from './provider';
import { ChainContract } from 'viem';
import { parseEther, type Address, type TransactionReceipt } from 'viem';

export const addLock = async (
  sellerAddress: Address,
  tokenAddress: Address,
  amount: number,
): Promise<bigint> => {
  const { address, abi, wallet, client, account } = await getContract();
  const parsedAmount = parseEther(amount.toString());

  if (!wallet) {
    throw new Error('Wallet not connected');
  }

  const { result, request } = await client.simulateContract({
    address,
    abi,
    functionName: 'lock',
    args: [sellerAddress, tokenAddress, parsedAmount, [], []],
    account,
  });
  const hash = await wallet.writeContract(request);
  const receipt = await client.waitForTransactionReceipt({ hash });

  if (!receipt.status)
    throw new Error('Transaction failed: ' + receipt.transactionHash);

  return result;
};

export const withdrawDeposit = async (
  amount: string,
  token: Address,
): Promise<boolean> => {
  const { address, abi, wallet, client, account } = await getContract();

  if (!wallet) {
    throw new Error('Wallet not connected');
  }

  const { request } = await client.simulateContract({
    address,
    abi,
    functionName: 'withdraw',
    args: [token, parseEther(amount), []],
    account,
  });

  const hash = await wallet.writeContract(request);
  const receipt = await client.waitForTransactionReceipt({ hash });

  return receipt.status === 'success';
};

export const releaseLock = async (
  lockID: bigint,
  pixTimestamp: `0x${string}` & { lenght: 34 },
  signature: `0x${string}`,
): Promise<TransactionReceipt> => {
  const { address, abi, wallet, client, account } = await getContract();

  if (!wallet) {
    throw new Error('Wallet not connected');
  }

  const { request } = await client.simulateContract({
    address,
    abi,
    functionName: 'release',
    args: [BigInt(lockID), pixTimestamp, signature],
    account,
  });

  const hash = await wallet.writeContract(request);
  return client.waitForTransactionReceipt({ hash });
};