feat: added infura bundler client

Author: hanzel98

packages/delegation-toolkit/src/actions/infuraBundlerClient.ts

@@ -0,0 +1,114 @@
+import type { Transport, Chain, Hex } from 'viem';
+import {
+  createBundlerClient,
+  type BundlerClientConfig,
+  type SmartAccount,
+} from 'viem/account-abstraction';
+
+/**
+ * Gas price tiers returned by pimlico_getUserOperationGasPrice
+ */
+export type GasPriceTier = {
+  /** Maximum fee per gas in hex format */
+  maxFeePerGas: Hex;
+  /** Maximum priority fee per gas in hex format */
+  maxPriorityFeePerGas: Hex;
+};
+
+/**
+ * Response from pimlico_getUserOperationGasPrice RPC method
+ */
+export type UserOperationGasPriceResponse = {
+  /** Slow gas price tier */
+  slow: GasPriceTier;
+  /** Standard gas price tier */
+  standard: GasPriceTier;
+  /** Fast gas price tier */
+  fast: GasPriceTier;
+};
+
+/**
+ * Infura bundler actions for extending bundler clients.
+ *
+ * @returns A function that takes a bundler client and returns the client extension with Infura bundler actions.
+ */
+const infuraBundlerActions = () => (client: any) => ({
+  /**
+   * Get user operation gas prices from Infura bundler.
+   * Calls the pimlico_getUserOperationGasPrice RPC method.
+   *
+   * @returns Promise resolving to gas price tiers (slow, standard, fast).
+   * @example
+   * ```typescript
+   * const gasPrices = await bundlerClient.getUserOperationGasPrice();
+   * console.log(gasPrices.standard.maxFeePerGas);
+   * ```
+   */
+  async getUserOperationGasPrice(): Promise<UserOperationGasPriceResponse> {
+    const response = await client.request({
+      method: 'pimlico_getUserOperationGasPrice' as any,
+      params: [] as any,
+    });
+
+    return response as unknown as UserOperationGasPriceResponse;
+  },
+});
+
+/**
+ * Type for bundler client extended with Infura bundler actions.
+ */
+export type InfuraBundlerClient<
+  TTransport extends Transport = Transport,
+  TChain extends Chain | undefined = Chain | undefined,
+  TAccount extends SmartAccount | undefined = SmartAccount | undefined,
+> = ReturnType<typeof createBundlerClient<TTransport, TChain, TAccount>> &
+  ReturnType<ReturnType<typeof infuraBundlerActions>>;
+
+/**
+ * Creates an Infura bundler client extended with Infura bundler actions.
+ *
+ * This is a wrapper around viem's createBundlerClient that extends it with
+ * the getUserOperationGasPrice method for retrieving gas prices from Pimlico's
+ * bundler infrastructure via Infura's proxy.
+ *
+ * @param config - Configuration for the bundler client.
+ * @returns Extended bundler client with Infura bundler actions.
+ * @example
+ * ```typescript
+ * import { createPublicClient, http } from 'viem';
+ * import { sepolia } from 'viem/chains';
+ * import { createInfuraBundlerClient } from '@metamask/delegation-toolkit';
+ *
+ * const publicClient = createPublicClient({
+ *   chain: sepolia,
+ *   transport: http('https://sepolia.infura.io/v3/YOUR_API_KEY'),
+ * });
+ *
+ * const bundlerClient = createInfuraBundlerClient({
+ *   client: publicClient,
+ *   transport: http('https://sepolia.infura.io/v3/YOUR_API_KEY'),
+ *   chain: sepolia,
+ * });
+ *
+ * // Use standard bundler methods
+ * const userOpHash = await bundlerClient.sendUserOperation({...});
+ *
+ * // Use Infura specific methods
+ * const gasPrices = await bundlerClient.getUserOperationGasPrice();
+ * ```
+ */
+export function createInfuraBundlerClient<
+  TTransport extends Transport,
+  TChain extends Chain | undefined = undefined,
+  TAccount extends SmartAccount | undefined = undefined,
+>(
+  config: BundlerClientConfig<TTransport, TChain, TAccount>,
+): InfuraBundlerClient<TTransport, TChain, TAccount> {
+  // Create the base bundler client using viem's function
+  const baseBundlerClient = createBundlerClient(config);
+
+  // Extend the client with Infura bundler actions
+  return baseBundlerClient.extend(
+    infuraBundlerActions(),
+  ) as unknown as InfuraBundlerClient<TTransport, TChain, TAccount>;
+}

packages/delegation-toolkit/src/index.ts

@@ -66,3 +66,10 @@ export {
   createCaveatEnforcerClient,
   type CaveatEnforcerClient,
 } from './actions/caveatEnforcerClient';
+
+export {
+  createInfuraBundlerClient,
+  type InfuraBundlerClient,
+  type GasPriceTier,
+  type UserOperationGasPriceResponse,
+} from './actions/infuraBundlerClient';