✨ feat: add lib examples (#155)

* ✨ feat: init examples * ✨ feat: more examples, fix some async things * 🔧 fix: remove unused `err` var
Jon-Becker · Sep 30, 2023 · b78691f · b78691f
1 parent b7a7533
commit b78691f
Show file tree

Hide file tree

Showing 28 changed files with 478 additions and 302 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,3 +1,13 @@
 [workspace]
-members = ["common", "config", "core", "cli"]
+members = [
+    "common",
+    "config",
+    "core",
+    "cli",
+    "examples/cfg",
+    "examples/dump",
+    "examples/decode",
+    "examples/snapshot",
+    "examples/decompile"
+]
 version = "0.6.0"
diff --git a/cli/src/main.rs b/cli/src/main.rs
@@ -109,17 +109,6 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
                 cmd.rpc_url = configuration.rpc_url;
             }
 
-            // set logger environment variable if not already set
-            if std::env::var("RUST_LOG").is_err() {
-                std::env::set_var(
-                    "RUST_LOG",
-                    match cmd.verbose.log_level() {
-                        Some(level) => level.as_str(),
-                        None => "SILENT",
-                    },
-                );
-            }
-
             let assembly = disassemble(cmd.clone()).await?;
 
             // write to file
@@ -137,17 +126,6 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
                 cmd.rpc_url = configuration.rpc_url;
             }
 
-            // set logger environment variable if not already set
-            if std::env::var("RUST_LOG").is_err() {
-                std::env::set_var(
-                    "RUST_LOG",
-                    match cmd.verbose.log_level() {
-                        Some(level) => level.as_str(),
-                        None => "SILENT",
-                    },
-                );
-            }
-
             let result = decompile(cmd.clone()).await?;
 
             // write to file
@@ -209,18 +187,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
                 cmd.openai_api_key = configuration.openai_api_key;
             }
 
-            // set logger environment variable if not already set
-            if std::env::var("RUST_LOG").is_err() {
-                std::env::set_var(
-                    "RUST_LOG",
-                    match cmd.verbose.log_level() {
-                        Some(level) => level.as_str(),
-                        None => "SILENT",
-                    },
-                );
-            }
-
-            decode(cmd).await;
+            let _ = decode(cmd).await;
         }
 
         Subcommands::CFG(mut cmd) => {
@@ -229,17 +196,6 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
                 cmd.rpc_url = configuration.rpc_url;
             }
 
-            // set logger environment variable if not already set
-            if std::env::var("RUST_LOG").is_err() {
-                std::env::set_var(
-                    "RUST_LOG",
-                    match cmd.verbose.log_level() {
-                        Some(level) => level.as_str(),
-                        None => "SILENT",
-                    },
-                );
-            }
-
             let cfg = cfg(cmd.clone()).await?;
 
             // write to file
@@ -263,17 +219,6 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
                 cmd.transpose_api_key = configuration.transpose_api_key;
             }
 
-            // set logger environment variable if not already set
-            if std::env::var("RUST_LOG").is_err() {
-                std::env::set_var(
-                    "RUST_LOG",
-                    match cmd.verbose.log_level() {
-                        Some(level) => level.as_str(),
-                        None => "SILENT",
-                    },
-                );
-            }
-
             let result = dump(cmd.clone()).await?;
             let mut lines = Vec::new();
 
@@ -305,17 +250,6 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
                 cmd.rpc_url = configuration.rpc_url;
             }
 
-            // set logger environment variable if not already set
-            if std::env::var("RUST_LOG").is_err() {
-                std::env::set_var(
-                    "RUST_LOG",
-                    match cmd.verbose.log_level() {
-                        Some(level) => level.as_str(),
-                        None => "SILENT",
-                    },
-                );
-            }
-
             // write to file
             if ADDRESS_REGEX.is_match(&cmd.target).unwrap() {
                 output_path.push_str(&format!("/{}/snapshot.csv", &cmd.target));
@@ -341,7 +275,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
     }
 
     // check if the version is up to date
-    let remote_version = remote_version();
+    let remote_version = remote_version().await;
     let current_version = current_version();
 
     if remote_version.gt(&current_version) {

diff --git a/common/Cargo.toml b/common/Cargo.toml
@@ -24,3 +24,5 @@ serde = {version = "1.0", features = ["derive"]}
 serde_json = "1.0"
 tokio = {version = "1", features = ["full"]}
 strsim = "0.10.0"
+async-recursion = "1.0.5"
+async-trait = "0.1.51"
diff --git a/common/src/ether/selectors.rs b/common/src/ether/selectors.rs
@@ -138,8 +138,8 @@ where
         let resolve_progress = resolve_progress.clone();
 
         // create a new thread for each selector
-        threads.push(thread::spawn(move || {
-            if let Some(function) = T::resolve(&selector) {
+        threads.push(thread::spawn(move || async move {
+            if let Some(function) = T::resolve(&selector).await {
                 let mut _resolved_functions =
                     function_clone.lock().expect("Could not obtain lock on function_clone.");
                 let mut _resolve_progress =