Merge pull request #16 from Vid201/feat/wallet

Bundler wallet (#2)

.gitignore

@@ -1 +1,2 @@
-/target
+# Rust
+/target

Cargo.toml

@@ -11,6 +11,7 @@ anyhow = "1"
 async-trait = "0.1"
 bytes = { version = "1", features = ["serde"] }
 clap = { version = "4", features = ["derive"] }
+dirs = "4.0"
 educe = { version = "0.4", features = ["Debug", "Default"] }
 ethereum-interfaces = { git = "https://github.com/ledgerwatch/interfaces" }
 ethereum-types = { version = "0.14", features = ["codec"] }
@@ -45,3 +46,7 @@ name = "bundler-uopool"
 [[bin]]
 path = "bin/bundler-rpc.rs"
 name = "bundler-rpc"
+
+[[bin]]
+path = "bin/create-wallet.rs"
+name = "create-wallet"

Makefile

@@ -1,11 +1,14 @@
 run-bundler:
-	cargo run -- --mnemonic-file ./src/res/bundler/0xD00D3EEc454D05d3d9bB48532BabED0c89941f17
+	cargo run -- --mnemonic-file ${HOME}/.aa-bundler/0x129D197b2a989C6798601A49D89a4AEC822A17a3
 
 run-bundler-uopool:
 	cargo run --bin bundler-uopool
 
 run-bundler-rpc:
 	cargo run --bin bundler-rpc
 
+run-create-wallet:
+	cargo run --bin create-wallet -- --output-path ${HOME}/.aa-bundler
+
 cargo-fmt:
 	cargo fmt --all

README.md

@@ -12,19 +12,25 @@ For more information: https://hackmd.io/@Vid201/aa-bundler-rust
 
 ## How to run?
 
-Bundler: 
+Create wallet for bundler:
 
 ```bash
-cargo run -- --mnemonic-folder ./src/res/bundler
+cargo run --bin create-wallet -- --output-path ${HOME}/.aa-bundler
 ```
 
-User operation pool:
+Run bundler (with user operation pool and JSON-RPC API): 
+
+```bash
+cargo run -- --mnemonic-file ${HOME}/.aa-bundler/0x129D197b2a989C6798601A49D89a4AEC822A17a3
+```
+
+Run only user operation pool:
 
 ```bash
 cargo run --bin bundler-uopool
 ```
 
-Bundler RPC: 
+Run only JSON-RPC API: 
 
 ```bash
 cargo run --bin bundler-rpc

bin/bundler-rpc.rs

@@ -9,7 +9,7 @@ use aa_bundler::rpc::{eth::EthApiServerImpl, eth_api::EthApiServer};
 #[derive(Parser)]
 #[clap(
     name = "aa-bundler-rpc",
-    about = "RPC server for EIP-4337 Account Abstraction Bundler"
+    about = "JSON-RPC server for EIP-4337 Account Abstraction Bundler"
 )]
 pub struct Opt {
     #[clap(long, default_value = "127.0.0.1:4337")]
@@ -36,7 +36,7 @@ async fn main() -> Result<()> {
     .unwrap();
 
     let _jsonrpc_server_handle = jsonrpc_server.start(api.clone())?;
-    info!("JSONRPC server listening on {}", opt.rpc_listen_address);
+    info!("JSON-RPC server listening on {}", opt.rpc_listen_address);
 
     pending().await
 }

bin/bundler.rs

@@ -1,10 +1,10 @@
-use aa_bundler::rpc::{eth::EthApiServerImpl, eth_api::EthApiServer};
+use aa_bundler::{
+    bundler::bundler::Bundler,
+    models::wallet::Wallet,
+    rpc::{eth::EthApiServerImpl, eth_api::EthApiServer},
+};
 use anyhow::Result;
 use clap::Parser;
-use ethers::{
-    prelude::rand,
-    signers::{coins_bip39::English, MnemonicBuilder},
-};
 use expanded_pathbuf::ExpandedPathBuf;
 use jsonrpsee::{core::server::rpc_module::Methods, server::ServerBuilder, tracing::info};
 use std::{future::pending, panic};
@@ -16,10 +16,7 @@ use std::{future::pending, panic};
 )]
 pub struct Opt {
     #[clap(long)]
-    pub mnemonic_file: Option<ExpandedPathBuf>,
-
-    #[clap(long, default_value = "./src/res/bundler")]
-    pub mnemonic_folder: ExpandedPathBuf,
+    pub mnemonic_file: ExpandedPathBuf,
 
     // #[clap(long, default_value = "127.0.0.1:3000")]
     // pub grpc_listen_address: String,
@@ -52,19 +49,10 @@ fn main() -> Result<()> {
             rt.block_on(async move {
                 info!("Starting AA - Bundler");
 
-                // TODO: move this to bundler package
-                let wallet = if let Some(mnemonic_file) = opt.mnemonic_file {
-                    MnemonicBuilder::<English>::default()
-                        .phrase(mnemonic_file.to_path_buf())
-                        .build()?
-                } else {
-                    let mut rng = rand::thread_rng();
-                    MnemonicBuilder::<English>::default()
-                        .write_to(opt.mnemonic_folder.to_path_buf())
-                        .build_random(&mut rng)?
-                };
+                let wallet = Wallet::from_file(opt.mnemonic_file);
+                info!("{:?}", wallet.signer);
 
-                println!("{:?}", wallet);
+                let bundler = Bundler::new(wallet);
 
                 if !opt.no_uopool {
                     aa_bundler::uopool::run(opt.uopool_opts).await?;
@@ -88,7 +76,7 @@ fn main() -> Result<()> {
                             .unwrap();
 
                             let _jsonrpc_server_handle = jsonrpc_server.start(api.clone()).unwrap();
-                            info!("JSONRPC server listening on {}", opt.rpc_listen_address);
+                            info!("JSON-RPC server listening on {}", opt.rpc_listen_address);
 
                             pending::<()>().await
                         }

bin/create-wallet.rs

@@ -0,0 +1,35 @@
+use aa_bundler::models::wallet::Wallet;
+use anyhow::Result;
+use clap::Parser;
+use dirs::home_dir;
+use expanded_pathbuf::ExpandedPathBuf;
+use jsonrpsee::tracing::info;
+use std::str::FromStr;
+
+#[derive(Parser)]
+#[clap(
+    name = "aa-bundler-create-wallet",
+    about = "Bundler's wallet creation for EIP-4337 Account Abstraction"
+)]
+pub struct Opt {
+    #[clap(long)]
+    pub output_path: Option<ExpandedPathBuf>,
+}
+
+fn main() -> Result<()> {
+    let opt: Opt = Opt::parse();
+
+    tracing_subscriber::fmt::init();
+
+    let path = if let Some(output_path) = opt.output_path {
+        output_path
+    } else {
+        ExpandedPathBuf::from_str(home_dir().unwrap().join(".aa-bundler").to_str().unwrap())
+            .unwrap()
+    };
+
+    let wallet = Wallet::new(path);
+    info!("{:?}", wallet.signer);
+
+    Ok(())
+}

src/bundler/bundler.rs

@@ -0,0 +1,11 @@
+use crate::models::wallet::Wallet;
+
+pub struct Bundler {
+    pub wallet: Wallet,
+}
+
+impl Bundler {
+    pub fn new(wallet: Wallet) -> Self {
+        Self { wallet }
+    }
+}

src/bundler/mod.rs

@@ -0,0 +1 @@
+pub mod bundler;

src/lib.rs

@@ -1,3 +1,4 @@
+pub mod bundler;
 pub mod models;
 pub mod rpc;
 pub mod types;

src/models/mod.rs

@@ -1 +1,2 @@
 pub mod chainspec;
+pub mod wallet;

src/models/wallet.rs

@@ -0,0 +1,34 @@
+use ethers::{
+    prelude::{k256::ecdsa::SigningKey, rand},
+    signers::{coins_bip39::English, MnemonicBuilder},
+};
+use expanded_pathbuf::ExpandedPathBuf;
+use std::fs;
+
+pub struct Wallet {
+    pub signer: ethers::signers::Wallet<SigningKey>,
+}
+
+impl Wallet {
+    pub fn new(output_path: ExpandedPathBuf) -> Self {
+        let mut rng = rand::thread_rng();
+
+        fs::create_dir_all(output_path.to_path_buf()).unwrap();
+
+        Self {
+            signer: MnemonicBuilder::<English>::default()
+                .write_to(output_path.to_path_buf())
+                .build_random(&mut rng)
+                .unwrap(),
+        }
+    }
+
+    pub fn from_file(input_path: ExpandedPathBuf) -> Self {
+        Self {
+            signer: MnemonicBuilder::<English>::default()
+                .phrase(input_path.to_path_buf())
+                .build()
+                .unwrap(),
+        }
+    }
+}

src/rpc/eth.rs

@@ -1,7 +1,7 @@
 use crate::{rpc::eth_api::EthApiServer, types::user_operation::UserOperation};
 use async_trait::async_trait;
 use ethereum_types::{Address, U64};
-use jsonrpsee::core::RpcResult;
+use jsonrpsee::{core::RpcResult, tracing::info};
 
 pub struct EthApiServerImpl {
     pub call_gas_limit: u64,
@@ -22,8 +22,8 @@ impl EthApiServer for EthApiServerImpl {
         user_operation: UserOperation,
         entry_point: Address,
     ) -> RpcResult<bool> {
-        println!("{:?}", user_operation);
-        println!("{:?}", entry_point);
+        info!("{:?}", user_operation);
+        info!("{:?}", entry_point);
         Ok(true)
     }
 }