add JDClient CLI start commands

stratum-mining · Aug 19, 2024 · 3cf2618 · 3cf2618
1 parent edad21c
commit 3cf2618
Show file tree

Hide file tree

Showing 3 changed files with 55 additions and 8 deletions.
diff --git a/stratum-cli/Cargo.lock b/stratum-cli/Cargo.lock
diff --git a/stratum-cli/Cargo.toml b/stratum-cli/Cargo.toml
@@ -8,5 +8,7 @@ clap = { version = "4.5.16", features = ["derive"] }
 ext-config = { version = "0.14.0", features = ["toml"], package = "config" }
 serde = { version = "1.0.89", default-features = false }
 tokio = { version = "1", features = ["full"] }
+tracing-subscriber = { version = "0.3" }
 
 translator_sv2 = { path = "../roles/translator" }
+jd_client = { path = "../roles/jd-client" }
diff --git a/stratum-cli/src/main.rs b/stratum-cli/src/main.rs
@@ -15,6 +15,8 @@ struct Cli {
 enum Commands {
     /// Stratum V2 Translator
     Translator(RoleArgs),
+    /// Stratum V2 Job Declarator Client
+    JDClient(RoleArgs),
 }
 
 #[derive(Debug, Args)]
@@ -32,19 +34,14 @@ enum RoleCommands {
 }
 
 fn parse_file<'a, T: serde::Deserialize<'a>>(config_path: &str) -> Option<T> {
-    let settings = Config::builder()
+    Config::builder()
         .add_source(File::new(config_path, FileFormat::Toml))
-        .build().ok();
-    if let Some(settings) = settings {
-        settings.try_deserialize::<T>().ok()
-    } else {
-        None
-    };
-    None
+        .build().ok().and_then(|s| s.try_deserialize::<T>().ok())
 }
 
 #[tokio::main]
 async fn main() {
+    tracing_subscriber::fmt::init();
     let cli = Cli::parse();
     match &cli.command {
         Some(Commands::Translator (role_args)) => {
@@ -68,6 +65,27 @@ async fn main() {
                 }
             }
         }
+        Some(Commands::JDClient (role_args)) => {
+            match &role_args.command {
+                Some(RoleCommands::StartLocal) => {
+                    let path = "../roles/jd-client/config-examples/jdc-config-local-example.toml".to_string();
+                    let config: jd_client::proxy_config::ProxyConfig  = parse_file(&path).unwrap();
+                    jd_client::JobDeclaratorClient::new(config).start().await;
+                }
+                Some(RoleCommands::StartHosted) => {
+                    let path = "../roles/jd-client/config-examples/jdc-config-hosted-example.toml".to_string();
+                    let config: jd_client::proxy_config::ProxyConfig  = parse_file(&path).unwrap();
+                    jd_client::JobDeclaratorClient::new(config).start().await;
+                }
+                Some(RoleCommands::StartConfig { path }) => {
+                    let config: jd_client::proxy_config::ProxyConfig = parse_file(path).unwrap();
+                    jd_client::JobDeclaratorClient::new(config).start().await;
+                }
+                None => {
+                    println!("JDClient No command");
+                }
+            }
+        }
         None => {
             println!("No command");
         }