Mimic functions, in TypeScript.

import { DenominationToken, ERC20Token, log, TokenAmount, TransferBuilder, USD } from '@mimicprotocol/lib-ts'

import { ERC20 } from './types/ERC20'
import { inputs } from './types'

export default function main(): void {
  const token = ERC20Token.fromAddress(inputs.token, inputs.chainId)
  const balance = new ERC20(inputs.token, inputs.chainId).balanceOf(inputs.recipient).unwrap()
  const balanceUsd = TokenAmount.fromBigInt(token, balance).toUsd().unwrap()
  log.info('Balance in USD: {}', [balanceUsd.toString()])

  if (balanceUsd.lt(USD.fromStringDecimal(inputs.thresholdUsd))) {
    const amount = TokenAmount.fromStringDecimal(token, inputs.amount)
    const maxFee = TokenAmount.fromStringDecimal(DenominationToken.USD(), inputs.maxFeeUsd)
    TransferBuilder.forChain(inputs.chainId)
      .addTransferFromTokenAmount(amount, inputs.recipient)
      .build()
      .send(maxFee)
  }
}

import { defineTask, z } from '@thyme-labs/sdk'
import { encodeFunctionData, erc20Abi, formatUnits, parseUnits } from 'viem'

export default defineTask({
  schema: z.object({
    token: z.address(),
    recipient: z.address(),
    amount: z.string(),
    minBalance: z.string(),
  }),
  async run(ctx) {
    const { token, recipient } = ctx.args
    const decimals = await ctx.client.readContract({
      address: token,
      abi: erc20Abi,
      functionName: 'decimals',
    })
    const balance = await ctx.client.readContract({
      address: token,
      abi: erc20Abi,
      functionName: 'balanceOf',
      args: [recipient],
    })
    ctx.logger.info('Balance: ' + formatUnits(balance, decimals))

    if (balance >= parseUnits(ctx.args.minBalance, decimals)) {
      return { canExec: false, message: 'Balance is above the minimum' }
    }

    return {
      canExec: true,
      calls: [{
        to: token,
        data: encodeFunctionData({
          abi: erc20Abi,
          functionName: 'transfer',
          args: [recipient, parseUnits(ctx.args.amount, decimals)],
        }),
      }],
    }
  },
})