rise-wallet 0.1.4-beta.2 → 0.2.29

package/src/viem/ContractActions.ts

@@ -0,0 +1,624 @@
+import * as AbiError from 'ox/AbiError'
+import * as AbiParameters from 'ox/AbiParameters'
+import type * as Address from 'ox/Address'
+import * as Authorization from 'ox/Authorization'
+import * as Errors from 'ox/Errors'
+import * as Hex from 'ox/Hex'
+import * as Signature from 'ox/Signature'
+import * as TypedData from 'ox/TypedData'
+import {
+  type Account as Account_viem,
+  type Authorization as Authorization_viem,
+  BaseError,
+  type Chain,
+  type Client,
+  encodeFunctionData,
+  parseAbi,
+  type SendTransactionParameters,
+  type Transport,
+} from 'viem'
+import {
+  call,
+  getEip712Domain as getEip712Domain_viem,
+  prepareAuthorization,
+  readContract,
+  sendTransaction,
+} from 'viem/actions'
+import {
+  type EncodeExecuteDataParameters,
+  encodeExecuteData,
+  getExecuteError as getExecuteError_viem,
+} from 'viem/experimental/erc7821'
+import * as IthacaAccount from '../core/internal/_generated/contracts/IthacaAccount.js'
+import * as Call from '../core/internal/call.js'
+import type { OneOf } from '../core/internal/types.js'
+import type * as Storage from '../core/Storage.js'
+import * as Account from './Account.js'
+import type { GetAccountParameter } from './internal/utils.js'
+import * as Key from './Key.js'
+
+export {
+  abi,
+  code,
+} from '../core/internal/_generated/contracts/IthacaAccount.js'
+
+/**
+ * Executes a set of calls on a delegated account.
+ *
+ * @example
+ * TODO
+ *
+ * @param client - Client.
+ * @param parameters - Execution parameters.
+ * @returns Transaction hash.
+ */
+export async function execute<
+  const calls extends readonly unknown[],
+  chain extends Chain | undefined,
+  account extends Account.Account | undefined,
+>(
+  client: Client<Transport, chain, account>,
+  parameters: execute.Parameters<calls, account>,
+): Promise<execute.ReturnType> {
+  const { account = client.account } = parameters
+
+  const account_ = account ? Account.from(account) : undefined
+  if (!account_) throw new Error('account is required.')
+
+  // Block expression to obtain the execution request and signatures.
+  const { request, signatures } = await (async () => {
+    const { nonce, key, signatures, storage } = parameters
+
+    // If an execution has been prepared, we can early return the request and signatures.
+    if (nonce && signatures) return { request: parameters, signatures }
+
+    // Otherwise, we need to prepare the execution (compute digests and sign over them).
+    const { digests, request } = await prepareExecute(client, {
+      ...parameters,
+      account: account_,
+    })
+
+    const exec = await Account.sign(account_, {
+      key: digests.auth ? null : key,
+      payload: digests.exec,
+      replaySafe: false,
+      storage,
+    })
+    const auth = await (async () => {
+      if (!digests.auth) return undefined
+      if (account_.source !== 'privateKey')
+        throw new Error('cannot sign authorization without root key.')
+      return account_.sign?.({
+        hash: digests.auth,
+      })
+    })()
+
+    return {
+      request,
+      signatures: {
+        auth,
+        exec,
+      },
+    }
+  })()
+
+  const { authorization, calls, executor, nonce } = request
+
+  // If an authorization signature is provided, it means that we will need to designate
+  // the EOA to the delegation contract. We will need to construct an authorization list
+  // to do so.
+  const authorizationList = (() => {
+    if (!signatures.auth) return undefined
+    const signature = Signature.from(signatures.auth)
+    return [
+      {
+        ...authorization,
+        r: Hex.fromNumber(signature.r),
+        s: Hex.fromNumber(signature.s),
+        yParity: signature.yParity,
+      },
+    ]
+  })()
+
+  // Structure the operation data to be passed to EIP-7821 execution.
+  // The operation data contains the nonce of the execution, as well as the
+  // signature.
+  const opData = AbiParameters.encodePacked(
+    ['uint256', 'bytes'],
+    [nonce, signatures.exec],
+  )
+
+  try {
+    return await sendTransaction(client, {
+      account: typeof executor === 'undefined' ? null : executor,
+      authorizationList,
+      data: encodeExecuteData({ calls, opData }),
+      to: account_.address,
+    } as SendTransactionParameters)
+  } catch (e) {
+    parseExecutionError(e, { calls })
+    throw e
+  }
+}
+
+export declare namespace execute {
+  export type Parameters<
+    calls extends readonly unknown[] = readonly unknown[],
+    account extends Account.Account | undefined = Account.Account | undefined,
+  > = Pick<EncodeExecuteDataParameters<calls>, 'calls'> &
+    GetAccountParameter<account> & {
+      /**
+       * Contract address to delegate to.
+       */
+      delegation?: Address.Address | undefined
+      /**
+       * The executor of the execute transaction.
+       *
+       * - `Account`: execution will be attempted with the specified account.
+       * - `undefined`: the transaction will be filled by the JSON-RPC server.
+       */
+      executor?: Account_viem | undefined
+      /**
+       * Storage to use for keytype-specific caching (e.g. WebAuthn user verification).
+       */
+      storage?: Storage.Storage | undefined
+    } & OneOf<
+      | {
+          /**
+           * EIP-7702 Authorization to use for delegation.
+           */
+          authorization?: Authorization_viem | undefined
+          /**
+           * Nonce to use for execution that will be invalidated by the delegated account.
+           */
+          nonce: bigint
+          /**
+           * Signature for execution. Required if the `executor` is not the EOA.
+           */
+          signatures: {
+            auth?: Hex.Hex | undefined
+            exec: Hex.Hex
+          }
+        }
+      | {
+          /**
+           * Key to use for execution.
+           */
+          key?: number | Key.Key | undefined
+        }
+      | {}
+    >
+
+  export type ReturnType = Hex.Hex
+}
+
+/**
+ * Returns the EIP-712 domain for a delegated account. Used for the execution
+ * signing payload.
+ *
+ * @param client - Client.
+ * @param parameters - Parameters.
+ * @returns EIP-712 domain.
+ */
+export async function getEip712Domain<
+  chain extends Chain | undefined,
+  account extends Account.Account | undefined,
+>(
+  client: Client<Transport, chain, account>,
+  parameters: getEip712Domain.Parameters<account>,
+): Promise<TypedData.Domain> {
+  const { account = client.account } = parameters
+  const account_ = account ? Account.from(account) : undefined
+  if (!account_) throw new Error('account is required.')
+
+  const {
+    domain: { name, version },
+  } = await getEip712Domain_viem(client, {
+    address: account_.address,
+  })
+
+  if (!client.chain) throw new Error('client.chain is required')
+  return {
+    chainId: client.chain.id,
+    name,
+    verifyingContract: account_.address,
+    version,
+  }
+}
+
+export declare namespace getEip712Domain {
+  export type Parameters<
+    account extends Account.Account | undefined = Account.Account | undefined,
+  > = GetAccountParameter<account>
+}
+
+/**
+ * Returns the key at the given index.
+ *
+ * @param client - Client.
+ * @param parameters - Parameters.
+ * @returns Key.
+ */
+export async function keyAt<
+  chain extends Chain | undefined,
+  account extends Account.Account | undefined,
+>(
+  client: Client<Transport, chain, account>,
+  parameters: keyAt.Parameters<account>,
+) {
+  const { account = client.account, index } = parameters
+
+  const account_ = account ? Account.from(account) : undefined
+  if (!account_) throw new Error('account is required.')
+
+  const key = await readContract(client, {
+    abi: IthacaAccount.abi,
+    address: account_.address,
+    args: [BigInt(index)],
+    functionName: 'keyAt',
+  })
+
+  return Key.deserialize(key, { chainId: client.chain?.id ?? 0 })
+}
+
+export declare namespace keyAt {
+  export type Parameters<
+    account extends Account.Account | undefined = Account.Account | undefined,
+  > = GetAccountParameter<account> & {
+    /**
+     * Index of the key to extract.
+     */
+    index: number
+  }
+}
+
+/**
+ * Prepares the payloads to sign over and fills the request to execute a set of calls.
+ *
+ * @example
+ * TODO
+ *
+ * @param client - Client.
+ * @param parameters - Parameters.
+ * @returns Prepared properties.
+ */
+export async function prepareExecute<
+  const calls extends readonly unknown[],
+  chain extends Chain | undefined,
+  account extends Account.Account | undefined,
+>(
+  client: Client<Transport, chain, account>,
+  parameters: prepareExecute.Parameters<calls, account>,
+): Promise<prepareExecute.ReturnType<calls>> {
+  const { account = client.account, delegation, executor, ...rest } = parameters
+
+  const account_ = account ? Account.from(account) : undefined
+  if (!account_) throw new Error('account is required.')
+
+  const calls = parameters.calls.map((call: any) => ({
+    data: call.data ?? '0x',
+    to: call.to === Call.selfAddress ? account_.address : call.to,
+    value: call.value ?? 0n,
+  }))
+
+  const nonce = Hex.toBigInt(
+    Hex.concat(
+      // multichain flag (0 = single chain, 0xc1d0 = multi-chain) // TODO: enable multi-chain
+      Hex.fromNumber(0, { size: 2 }),
+      // sequence key
+      Hex.random(22),
+      // sequential nonce
+      Hex.fromNumber(0, { size: 8 }),
+    ),
+  )
+
+  // Compute the signing payloads for execution and EIP-7702 authorization (optional).
+  const [executePayload, [authorization, authorizationPayload]] =
+    await Promise.all([
+      getExecuteDigest(client, {
+        account: account_,
+        calls,
+        delegation,
+        nonce,
+      }),
+
+      // Only need to compute an authorization payload if we are delegating to an EOA.
+      (async () => {
+        if (!delegation) return []
+
+        const authorization = await prepareAuthorization(client, {
+          account: account_.address,
+          // chainId: 0,
+          contractAddress: delegation,
+          executor,
+        })
+        return [
+          authorization,
+          Authorization.getSignPayload({
+            address: authorization.address,
+            chainId: authorization.chainId,
+            nonce: BigInt(authorization.nonce),
+          }),
+        ]
+      })(),
+    ])
+
+  return {
+    digests: {
+      auth: authorizationPayload,
+      exec: executePayload.digest,
+    },
+    request: {
+      ...rest,
+      account: account_,
+      authorization,
+      calls,
+      executor,
+      nonce,
+    },
+    typedData: executePayload.typedData,
+  } as never
+}
+
+export declare namespace prepareExecute {
+  export type Parameters<
+    calls extends readonly unknown[] = readonly unknown[],
+    account extends Account.Account | undefined = Account.Account | undefined,
+  > = Pick<execute.Parameters<calls, account>, 'calls'> &
+    GetAccountParameter<account> & {
+      /**
+       * Contract address to delegate to.
+       */
+      delegation?: Address.Address | undefined
+      /**
+       * The executor of the execute transaction.
+       *
+       * - `Account`: execution will be attempted with the specified account.
+       * - `undefined`: the transaction will be filled by the JSON-RPC server.
+       */
+      executor?: Account_viem | undefined
+    }
+
+  export type ReturnType<
+    calls extends readonly unknown[] = readonly unknown[],
+  > = {
+    digests: {
+      auth?: Hex.Hex | undefined
+      exec: Hex.Hex
+    }
+    request: Omit<Parameters<calls>, 'account' | 'delegation'> & {
+      account: Account.Account
+      authorization?: Authorization_viem | undefined
+      nonce: bigint
+    }
+    typedData: TypedData.Definition
+  }
+}
+
+export function parseExecutionError<const calls extends readonly unknown[]>(
+  e: unknown,
+  { calls }: { calls?: execute.Parameters<calls>['calls'] | undefined } = {},
+) {
+  if (!(e instanceof BaseError)) return
+
+  const getAbiError = (error: BaseError) => {
+    const cause = error.walk((e) => 'data' in (e as BaseError))
+    if (!cause) return undefined
+
+    let data: Hex.Hex | undefined
+    if (cause instanceof BaseError) {
+      const [, match] = cause.details?.match(/"(0x[0-9a-f]{8})"/) || []
+      if (match) data = match as Hex.Hex
+    }
+
+    if (!data) {
+      if (!('data' in cause)) return undefined
+      if (cause.data instanceof BaseError) return getAbiError(cause.data)
+      if (typeof cause.data !== 'string') return undefined
+      if (cause.data === '0x') return undefined
+      data = cause.data as Hex.Hex
+    }
+
+    try {
+      if (data === '0xd0d5039b') return AbiError.from('error Unauthorized()')
+      return AbiError.fromAbi(
+        [...IthacaAccount.abi, AbiError.from('error CallError()')],
+        data,
+      )
+    } catch {
+      return undefined
+    }
+  }
+  const error = getExecuteError_viem(e as BaseError, {
+    calls: (calls ?? []) as any,
+  })
+  const abiError = getAbiError(error)
+  if (error === e && !abiError) return
+  throw new ExecutionError(Object.assign(error, { abiError }))
+}
+
+export declare namespace parseExecutionError {
+  export type ErrorType = ExecutionError | Errors.GlobalErrorType
+}
+
+/** Thrown when the execution fails. */
+export class ExecutionError extends Errors.BaseError<BaseError> {
+  override readonly name = 'AccountContract.ExecutionError'
+
+  abiError?: AbiError.AbiError | undefined
+
+  constructor(cause: BaseError & { abiError?: AbiError.AbiError | undefined }) {
+    super('An error occurred while executing calls.', {
+      cause,
+      metaMessages: [cause.abiError && 'Reason: ' + cause.abiError.name].filter(
+        Boolean,
+      ),
+    })
+
+    this.abiError = cause.abiError
+  }
+}
+
+export type Decorator<
+  account extends Account.Account | undefined = Account.Account | undefined,
+> = {
+  /**
+   * Executes a set of calls on a delegated account.
+   *
+   * @example
+   * TODO
+   *
+   * @param client - Client.
+   * @param parameters - Execution parameters.
+   * @returns Transaction hash.
+   */
+  execute: <const calls extends readonly unknown[]>(
+    parameters: execute.Parameters<calls, account>,
+  ) => Promise<execute.ReturnType>
+  /**
+   * Returns the EIP-712 domain for a delegated account. Used for the execution
+   * signing payload.
+   *
+   * @param client - Client.
+   * @param parameters - Parameters.
+   * @returns EIP-712 domain.
+   */
+  getEip712Domain: (
+    parameters: getEip712Domain.Parameters<account>,
+  ) => Promise<TypedData.Domain>
+  /**
+   * Returns the key at the given index.
+   *
+   * @param client - Client.
+   * @param parameters - Parameters.
+   * @returns Key.
+   */
+  keyAt: (parameters: keyAt.Parameters<account>) => Promise<Key.Key>
+  /**
+   * Prepares the payloads to sign over and fills the request to execute a set of calls.
+   *
+   * @param client - Client.
+   * @param parameters - Parameters.
+   * @returns Prepared properties.
+   */
+  prepareExecute: <calls extends readonly unknown[] = readonly unknown[]>(
+    parameters: prepareExecute.Parameters<calls, account>,
+  ) => Promise<prepareExecute.ReturnType<calls>>
+}
+
+export function decorator<
+  transport extends Transport,
+  chain extends Chain | undefined,
+  account extends Account.Account | undefined,
+>(client: Client<transport, chain, account>): Decorator<account> {
+  return {
+    execute: (parameters) => execute(client, parameters),
+    getEip712Domain: (parameters) => getEip712Domain(client, parameters),
+    keyAt: (parameters) => keyAt(client, parameters),
+    prepareExecute: (parameters) => prepareExecute(client, parameters),
+  }
+}
+
+///////////////////////////////////////////////////////////////////////////
+// Internal
+///////////////////////////////////////////////////////////////////////////
+
+/** @internal */
+async function getExecuteDigest<
+  const calls extends readonly unknown[],
+  chain extends Chain | undefined,
+  account extends Account.Account | undefined,
+>(
+  client: Client<Transport, chain, account>,
+  parameters: getExecuteDigest.Parameters<calls>,
+): Promise<{ digest: Hex.Hex; typedData: TypedData.Definition }> {
+  const { account = client.account, delegation, nonce } = parameters
+
+  const account_ = account ? Account.from(account) : undefined
+  if (!account_) throw new Error('account is required.')
+
+  // Structure calls into EIP-7821 execution format.
+  const calls = parameters.calls.map((call: any) => ({
+    data: call.data ?? '0x',
+    to: call.to === Call.selfAddress ? account_.address : call.to,
+    value: call.value ?? 0n,
+  }))
+
+  const address = await (async () => {
+    if (!delegation) return account_.address
+    const { data } = await call(client, {
+      data: encodeFunctionData({
+        abi: parseAbi(['function implementation() view returns (address)']),
+        functionName: 'implementation',
+      }),
+      to: delegation!,
+    } as never).catch(() => ({ data: undefined }))
+    if (!data) throw new Error('delegation address not found.')
+    return Hex.slice(data, 12)
+  })()
+
+  const domain = await getEip712Domain(client, { account: address })
+
+  const multichain = nonce & 1n
+
+  if (!client.chain) throw new Error('chain is required.')
+
+  const typedData = {
+    domain: {
+      chainId: client.chain.id,
+      name: domain.name,
+      verifyingContract: account_.address,
+      version: domain.version,
+    },
+    message: {
+      calls,
+      multichain: Boolean(multichain),
+      nonce,
+    },
+    primaryType: 'Execute',
+    types: {
+      Call: [
+        { name: 'to', type: 'address' },
+        { name: 'value', type: 'uint256' },
+        { name: 'data', type: 'bytes' },
+      ],
+      Execute: [
+        { name: 'multichain', type: 'bool' },
+        { name: 'calls', type: 'Call[]' },
+        { name: 'nonce', type: 'uint256' },
+      ],
+    },
+  } as const
+
+  return {
+    digest: TypedData.getSignPayload(typedData),
+    typedData,
+  }
+}
+
+export declare namespace getExecuteDigest {
+  export type Parameters<
+    calls extends readonly unknown[] = readonly unknown[],
+  > = {
+    /**
+     * The delegated account to execute the calls on.
+     */
+    account: Account.Account
+    /**
+     * Contract address to delegate to.
+     */
+    delegation?: Address.Address | undefined
+    /**
+     * Calls to execute.
+     */
+    calls: calls
+    /**
+     * Nonce to use for execution that will be invalidated by the delegated account.
+     */
+    nonce: bigint
+    /**
+     * Nonce salt.
+     */
+    nonceSalt?: bigint | undefined
+  }
+}