WithdrawAuthMethod

import {
  WithdrawAuthMethod,
  isWithdrawApprovalAuth,
  isWithdrawAlreadyApprovedAuth,
  type WithdrawAuthorizationResult,
  type WithdrawApprovalAuthorizationResult,
  type WithdrawAlreadyApprovedAuthorizationResult,
} from '@paxoslabs/amplify-sdk'

export const WithdrawAuthMethod = {
  APPROVAL: 'approval',
  ALREADY_APPROVED: 'already_approved',
} as const

export type WithdrawAuthMethod =
  (typeof WithdrawAuthMethod)[keyof typeof WithdrawAuthMethod]

export type WithdrawAuthorizationResult =
  | WithdrawApprovalAuthorizationResult
  | WithdrawAlreadyApprovedAuthorizationResult

interface WithdrawApprovalAuthorizationResult {
  method: 'approval'
  /** Transaction data for vault share approve() call */
  txData: {
    abi: readonly unknown[]
    address: `0x${string}`
    functionName: 'approve'
    args: [`0x${string}`, bigint]
  }
}

interface WithdrawAlreadyApprovedAuthorizationResult {
  method: 'already_approved'
  /** Current allowance (human-readable) */
  allowance: string
  /** Current allowance as BigInt string */
  allowanceAsBigInt: string
}

import {
  prepareWithdrawalAuthorization,
  isWithdrawApprovalAuth,
  isWithdrawAlreadyApprovedAuth,
} from '@paxoslabs/amplify-sdk'

const auth = await prepareWithdrawalAuthorization(params)

if (isWithdrawApprovalAuth(auth)) {
  // TypeScript knows: auth.method === "approval"
  // auth.txData is available with abi, address, functionName, args
  const approvalHash = await walletClient.writeContract(auth.txData)
  await publicClient.waitForTransactionReceipt({ hash: approvalHash })
}

if (isWithdrawAlreadyApprovedAuth(auth)) {
  // TypeScript knows: auth.method === "already_approved"
  // auth.allowance is available
  console.log(`Existing allowance: ${auth.allowance}`)
}

const auth = await prepareWithdrawalAuthorization(params)

switch (auth.method) {
  case 'approval':
    // auth is narrowed to WithdrawApprovalAuthorizationResult
    await writeContract(auth.txData)
    break

  case 'already_approved':
    // auth is narrowed to WithdrawAlreadyApprovedAuthorizationResult
    console.log(`Allowance: ${auth.allowance}`)
    break
}

interface PrepareWithdrawalAuthorizationParams {
  vaultName: string
  wantAsset: Address
  withdrawAmount: string
  userAddress: Address
  chainId: ChainId
  forceMethod?: 'approval' | 'allowance_check'
}