Env config cli tool

Cargo.toml

@@ -1,11 +1,11 @@
 [workspace]
 resolver = "2"
-members = ["compute", "p2p", "workflows", "utils", "monitor"]
+members = ["compute", "p2p", "workflows", "utils", "monitor", "launcher"]
 
 # FIXME: removing this breaks the workflows
 # compute node is the default member, until Oracle comes in
 # then, a Launcher will be the default member
-default-members = ["compute"]
+default-members = ["launcher"]
 
 [workspace.package]
 edition = "2021"

compute/src/main.rs → compute/src/launch.rs

@@ -1,41 +1,11 @@
-use dkn_compute::*;
+use crate::*;
 use dkn_workflows::DriaWorkflowsConfig;
 use eyre::Result;
 use std::env;
 use tokio_util::{sync::CancellationToken, task::TaskTracker};
 use workers::workflow::WorkflowsWorker;
 
-#[tokio::main]
-async fn main() -> Result<()> {
-    let dotenv_result = dotenvy::dotenv();
-
-    env_logger::builder()
-        .format_timestamp(Some(env_logger::TimestampPrecision::Millis))
-        .filter(None, log::LevelFilter::Off)
-        .filter_module("dkn_compute", log::LevelFilter::Info)
-        .filter_module("dkn_p2p", log::LevelFilter::Info)
-        .filter_module("dkn_workflows", log::LevelFilter::Info)
-        .parse_default_env() // reads RUST_LOG variable
-        .init();
-
-    log::info!(
-        r#"
-
-██████╗ ██████╗ ██╗ █████╗ 
-██╔══██╗██╔══██╗██║██╔══██╗   Dria Compute Node 
-██║  ██║██████╔╝██║███████║   v{DRIA_COMPUTE_NODE_VERSION}
-██║  ██║██╔══██╗██║██╔══██║   https://dria.co
-██████╔╝██║  ██║██║██║  ██║
-╚═════╝ ╚═╝  ╚═╝╚═╝╚═╝  ╚═╝
-"#
-    );
-
-    // log about env usage
-    match dotenv_result {
-        Ok(path) => log::info!("Loaded .env file at: {}", path.display()),
-        Err(e) => log::warn!("Could not load .env file: {}", e),
-    }
-
+pub async fn launch() -> Result<()> {
     // task tracker for multiple threads
     let task_tracker = TaskTracker::new();
     let cancellation = CancellationToken::new();

compute/src/lib.rs

@@ -1,5 +1,6 @@
 pub mod config;
 pub mod handlers;
+pub mod launch;
 pub mod node;
 pub mod payloads;
 pub mod utils;
@@ -9,7 +10,7 @@ pub mod workers;
 /// This value is attached within the published messages.
 pub const DRIA_COMPUTE_NODE_VERSION: &str = env!("CARGO_PKG_VERSION");
 
-pub use utils::refresh_dria_nodes;
-
 pub use config::DriaComputeNodeConfig;
+pub use launch::launch;
 pub use node::DriaComputeNode;
+pub use utils::refresh_dria_nodes;

launcher/Cargo.toml

@@ -0,0 +1,38 @@
+[package]
+name = "dkn-launcher"
+version.workspace = true
+edition.workspace = true
+license.workspace = true
+readme = "README.md"
+authors = ["Erhan Tezcan <erhan@firstbatch.xyz>"]
+
+[dependencies]
+clap = { version = "4.5.20", features = ["derive"] }
+crossterm = "0.28.1"
+which = "7.0.0"
+inquire = "0.7.5"
+tempfile = "3.14.0"
+
+# async stuff
+tokio-util.workspace = true
+tokio.workspace = true
+async-trait.workspace = true
+
+# serialize & deserialize
+serde.workspace = true
+serde_json.workspace = true
+
+# utilities
+dotenvy.workspace = true
+hex = "0.4.3"
+hex-literal = "0.4.1"
+
+env_logger.workspace = true
+log.workspace = true
+eyre.workspace = true
+
+# dria subcrates
+dkn-compute = { path = "../compute" }
+dkn-workflows = { path = "../workflows" }
+self-replace = "1.5.0"
+self_update = "0.41.0"

launcher/src/lib.rs

@@ -0,0 +1,5 @@
+pub mod utils;
+
+pub use utils::gemini;
+pub use utils::openai;
+pub use utils::openrouter;