kohaku-userop-kit 0.0.1

EIP-4337 client
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

use alloy::rpc::types::state::StateOverride;

use crate::{
    builder::UserOperationBuilder,
    signable_user_operation::SignableUserOperation,
    signed_user_operation::SignedUserOperation,
    user_operation::{UserOperationGasEstimate, UserOperationHash, UserOperationReceipt},
};

pub mod pimlico;
mod rpc_client;

pub trait BundlerExt: Sized {
    /// Estimates and sets the gas for this UserOperation.
    ///
    /// Simulates against `state_override` if provided.
    fn with_gas_estimate<B: Bundler + ?Sized>(
        self,
        bundler: &B,
        state_override: Option<&StateOverride>,
    ) -> impl Future<Output = Result<Self, BundlerError>>;
}

/// A bundler provider for 4337 UserOperation JSON-RPC methods.
#[cfg_attr(native, async_trait::async_trait)]
#[cfg_attr(wasm, async_trait::async_trait(?Send))]
pub trait Bundler {
    /// Estimates gas for `op`.
    ///
    /// Simulates against `state_override` if provided.
    async fn estimate_gas(
        &self,
        op: &SignableUserOperation,
        state_override: Option<&StateOverride>,
    ) -> Result<UserOperationGasEstimate, BundlerError>;
    async fn send_user_operation(
        &self,
        op: &SignedUserOperation,
    ) -> Result<UserOperationHash, BundlerError>;
    async fn receipt(
        &self,
        hash: UserOperationHash,
    ) -> Result<Option<UserOperationReceipt>, BundlerError>;
    async fn wait_for_receipt(
        &self,
        hash: UserOperationHash,
    ) -> Result<UserOperationReceipt, BundlerError>;
}

#[derive(Debug, thiserror::Error)]
#[non_exhaustive]
pub enum BundlerError {
    #[error("Timeout")]
    Timeout,
    #[error("Other: {0}")]
    Other(#[from] Box<dyn std::error::Error + Send + Sync>),
}

impl<S> BundlerExt for UserOperationBuilder<S> {
    async fn with_gas_estimate<B: Bundler + ?Sized>(
        self,
        bundler: &B,
        state_override: Option<&StateOverride>,
    ) -> Result<Self, BundlerError> {
        let op = self.build();
        let est = bundler.estimate_gas(&op, state_override).await?;

        Ok(self.with_gas(est))
    }
}