viem/experimental/erc7821/utils/encodeExecuteData.ts

TypeScript Interface for Ethereum

import type { Narrow } from 'abitype'

import type { ErrorType } from '../../../errors/utils.js'
import type { Calls } from '../../../types/calls.js'
import type { Hex } from '../../../types/misc.js'
import {
  type EncodeFunctionDataErrorType,
  encodeFunctionData,
} from '../../../utils/abi/encodeFunctionData.js'
import { abi, executionMode } from '../constants.js'
import { type EncodeCallsErrorType, encodeCalls } from './encodeCalls.js'

export type EncodeExecuteDataParameters<
  calls extends readonly unknown[] = readonly unknown[],
> = {
  /** Calls to execute. */
  calls: Calls<Narrow<calls>>
  /** Additional data to include for execution. */
  opData?: Hex | undefined
}

export type EncodeExecuteDataReturnType = Hex

export type EncodeExecuteDataErrorType =
  | EncodeCallsErrorType
  | EncodeFunctionDataErrorType
  | ErrorType

export function encodeExecuteData<const calls extends readonly unknown[]>(
  parameters: EncodeExecuteDataParameters<calls>,
): EncodeExecuteDataReturnType {
  const { calls, opData } = parameters

  const encodedCalls = encodeCalls(calls, opData)
  const mode = opData ? executionMode.opData : executionMode.default

  return encodeFunctionData({
    abi,
    functionName: 'execute',
    args: [mode, encodedCalls],
  })
}