|
| 1 | +import { Chain, EIP1193RequestFn, Hex, http, Transport } from "viem"; |
| 2 | +import { chainTransport } from "./chainTransport"; |
| 3 | + |
| 4 | +type AltoSendUserOperationResultuest = { |
| 5 | + txHash: Hex; |
| 6 | + userOpHash: Hex; |
| 7 | +}; |
| 8 | + |
| 9 | +type AltoOptions<transport extends Transport> = { |
| 10 | + /** alto-compatible transport */ |
| 11 | + altoTransport: transport; |
| 12 | + /** fallback transport for all other RPC methods */ |
| 13 | + fallbackDefaultTransport?: Transport; |
| 14 | +}; |
| 15 | + |
| 16 | +export function alto<const altoTransport extends Transport>(transports?: AltoOptions<altoTransport>): altoTransport { |
| 17 | + return ((opts) => { |
| 18 | + const { altoTransport } = transports ?? getDefaultTransports(opts.chain); |
| 19 | + |
| 20 | + const { request: altoRequest, ...rest } = altoTransport(opts); |
| 21 | + |
| 22 | + return { |
| 23 | + ...rest, |
| 24 | + // TODO: type `request` so we don't have to cast |
| 25 | + async request(req): ReturnType<EIP1193RequestFn> { |
| 26 | + if (req.method === "eth_sendUserOperation") { |
| 27 | + console.log("sending user operation via pimlico_sendUserOperationNow", req); |
| 28 | + const result = (await altoRequest({ |
| 29 | + ...req, |
| 30 | + method: "pimlico_sendUserOperationNow", |
| 31 | + })) as AltoSendUserOperationResultuest; |
| 32 | + console.log("pimlico_sendUserOperationNow", result); |
| 33 | + return result; |
| 34 | + } |
| 35 | + |
| 36 | + return await altoRequest(req); |
| 37 | + }, |
| 38 | + }; |
| 39 | + }) as altoTransport; |
| 40 | +} |
| 41 | + |
| 42 | +function getDefaultTransports(chain?: Chain): AltoOptions<Transport> { |
| 43 | + if (!chain) { |
| 44 | + throw new Error("No chain or transports provided"); |
| 45 | + } |
| 46 | + |
| 47 | + const altoTransport = chainTransport(chain.rpcUrls.alto); |
| 48 | + if (!altoTransport) { |
| 49 | + throw new Error("Provided chain does not support alto"); |
| 50 | + } |
| 51 | + |
| 52 | + return { |
| 53 | + altoTransport, |
| 54 | + fallbackDefaultTransport: http(), |
| 55 | + }; |
| 56 | +} |
0 commit comments