feat(bundler): extending supported bundler operations (#888)

* feat(bundler): extending supported bundler operations * fix: fixing mock usage
reown-com · Jan 8, 2025 · a9bc59b · a9bc59b
1 parent 31d71d3
commit a9bc59b
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/src/providers/mock_alto.rs b/src/providers/mock_alto.rs
@@ -56,7 +56,9 @@ impl BundlerOpsProvider for MockAltoProvider {
             | SupportedBundlerOps::EthGetUserOperationReceipt
             | SupportedBundlerOps::EthEstimateUserOperationGas
             | SupportedBundlerOps::PimlicoGetUserOperationGasPrice => self.bundler_url.clone(),
-            SupportedBundlerOps::PmSponsorUserOperation => self.paymaster_url.clone(),
+            SupportedBundlerOps::PmSponsorUserOperation
+            | SupportedBundlerOps::PmGetPaymasterData
+            | SupportedBundlerOps::PmGetPaymasterStubData => self.paymaster_url.clone(),
         };
         let response = self
             .http_client
@@ -84,6 +86,8 @@ impl BundlerOpsProvider for MockAltoProvider {
                 "eth_estimateUserOperationGas".into()
             }
             SupportedBundlerOps::PmSponsorUserOperation => "pm_sponsorUserOperation".into(),
+            SupportedBundlerOps::PmGetPaymasterData => "pm_getPaymasterData".into(),
+            SupportedBundlerOps::PmGetPaymasterStubData => "pm_getPaymasterStubData".into(),
             SupportedBundlerOps::PimlicoGetUserOperationGasPrice => {
                 "pimlico_getUserOperationGasPrice".into()
             }

diff --git a/src/providers/mod.rs b/src/providers/mod.rs
@@ -806,6 +806,10 @@ pub enum SupportedBundlerOps {
     /// Paymaster sponsor UserOp
     #[serde(rename = "pm_sponsorUserOperation")]
     PmSponsorUserOperation,
+    #[serde(rename = "pm_getPaymasterData")]
+    PmGetPaymasterData,
+    #[serde(rename = "pm_getPaymasterStubData")]
+    PmGetPaymasterStubData,
     #[serde(rename = "pimlico_getUserOperationGasPrice")]
     PimlicoGetUserOperationGasPrice,
 }

diff --git a/src/providers/pimlico.rs b/src/providers/pimlico.rs
@@ -74,6 +74,8 @@ impl BundlerOpsProvider for PimlicoProvider {
                 "eth_estimateUserOperationGas".into()
             }
             SupportedBundlerOps::PmSponsorUserOperation => "pm_sponsorUserOperation".into(),
+            SupportedBundlerOps::PmGetPaymasterData => "pm_getPaymasterData".into(),
+            SupportedBundlerOps::PmGetPaymasterStubData => "pm_getPaymasterStubData".into(),
             SupportedBundlerOps::PimlicoGetUserOperationGasPrice => {
                 "pimlico_getUserOperationGasPrice".into()
             }