chore: flesh out init command; create data and config directories

Author: sutt0n

.gitignore

@@ -131,4 +131,8 @@ rust-project.json
 .cache
 
 # Flamegraph
-*.svg
+*.svg
+
+# Data and config directories
+/data
+/config

Cargo.toml

@@ -1,2 +1,25 @@
 [workspace]
 members = ["horizon_cli", "horizon_common", "horizon_registry"]
+
+[workspace.dependencies]
+clap = { version = "4.5", features = [
+    "derive",
+    "cargo",
+    "wrap_help",
+    "unicode",
+    "string",
+    "unstable-styles",
+] }
+
+# tracing and error reporting dependencies
+color-eyre = { version = "0.6.3" }
+thiserror = { version = "2.0" }
+
+# async dependencies
+tokio = { version = "1.43", features = ["rt-multi-thread", "fs"] }
+tokio-util = "0.7.12"
+
+# app initialization dependencies
+lazy_static = "1.5"
+config = "0.15"
+directories = "6.0"

horizon_cli/Cargo.toml

@@ -4,4 +4,11 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
-clap = { version = "4.5.32", features = ["derive", "env"] }
+clap.workspace = true
+tokio.workspace = true
+config.workspace = true
+color-eyre.workspace = true
+lazy_static.workspace = true
+directories.workspace = true
+
+horizon_common = { path = "../horizon_common" }

horizon_cli/src/commands.rs

@@ -1,5 +1,39 @@
 use clap::{Parser, Subcommand};
 
+/**
+horizon init # Initializes the Horizon environment locally
+
+horizon list-gpus # Lists available GPUs in the Horizon marketplace
+horizon rent <gpu_id> --duration <hours> [--price <price>] # Rent a GPU for a specified duration
+horizon submit-job <gpu_id> --job <job_file> # Submit a job for execution on a specified GPU
+horizon bid <gpu_id> --price <price> # Place a bid for a GPU rental
+horizon auction start <gpu_id> --duration <duration> # Start an auction for a GPU
+horizon auction end <gpu_id> # End an auction for a GPU
+horizon status <job_id> # Retrieve the status of a job submitted to the marketplace
+horizon cancel-job <job_id> # Cancel a job in progress
+
+horizon config --set <key> <value> # Set a configuration option (e.g., price limits, security keys)
+horizon config --get <key> # Retrieve a configuration value
+
+horizon network status # Check the status of the Horizon network and nodes
+horizon network connect <node_id> # Connect to a specific node in the network
+horizon network disconnect <node_id> # Disconnect from a node in the network
+
+horizon marketplace sync # Synchronize local data with the decentralized GPU marketplace
+horizon marketplace update # Update local GPU listings from the marketplace
+
+horizon register-gpu --gpu-id <gpu_id> --gpu-type <type> --price <price> # Register a GPU for rental
+horizon deregister-gpu <gpu_id> # Deregister a GPU from the marketplace
+
+horizon explore <gpu_id> [--filter <json_path>] # Interactively explore the details of a GPU
+
+horizon logs # Display logs for Horizon operations and node status
+horizon logs clear # Clear the local logs
+
+horizon version # Displays the Horizon CLI version
+horizon <command> --help # Shows help for a specific command
+*/
+
 #[derive(Parser)]
 #[clap(version, name = "horizon", long_about = None)]
 pub struct Cli {

horizon_cli/src/job_cmds.rs

@@ -0,0 +1,39 @@
+use std::env;
+use std::path::PathBuf;
+
+use lazy_static::lazy_static;
+
+use horizon_common::HorizonCliState;
+
+// get directories; default to ./data and ./config if env vars don't exist or aren't set
+// otherwise get HORIZON_DATA_DIRECTORY and HORIZON_CONFIG_DIRECTORY from env vars
+
+lazy_static! {
+    static ref HORIZON_DATA_DIRECTORY: PathBuf = {
+        match env::var("HORIZON_DATA_DIRECTORY") {
+            Ok(val) => PathBuf::from(val),
+            Err(_) => PathBuf::from("./data"),
+        }
+    };
+    static ref HORIZON_CONFIG_DIRECTORY: PathBuf = {
+        match env::var("HORIZON_CONFIG_DIRECTORY") {
+            Ok(val) => PathBuf::from(val),
+            Err(_) => PathBuf::from("./config"),
+        }
+    };
+}
+
+#[derive(Debug)]
+pub struct JobCommand {}
+
+impl JobCommand {
+    pub async fn init_command() {
+        let state = HorizonCliState::new(
+            HORIZON_DATA_DIRECTORY.to_path_buf(),
+            HORIZON_CONFIG_DIRECTORY.to_path_buf(),
+        );
+
+        state.init_data_folder().await;
+        state.init_config_folder().await;
+    }
+}

horizon_cli/src/main.rs

@@ -1,16 +1,20 @@
 pub mod commands;
+pub mod job_cmds;
 
 use clap::Parser;
 use commands::{Cli, Command};
 
-fn main() {
+use color_eyre::eyre::Report;
+use job_cmds::JobCommand;
+
+#[tokio::main]
+async fn main() -> Result<(), Report> {
     let cli = Cli::parse();
 
     match cli.cmd {
         Command::Init => {
-            // TODO: How do we want to manage local configuration? Initialization of a toml with
-            // mappings? Confy crate? Other ideas/suggestions?
-            todo!()
+            JobCommand::init_command().await;
+            Ok(())
         }
     }
 }

horizon_common/Cargo.toml

@@ -4,3 +4,4 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
+tokio.workspace = true

horizon_common/src/api/mod.rs

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

horizon_common/src/api/state.rs

@@ -0,0 +1,30 @@
+use std::path::PathBuf;
+use tokio::fs::create_dir_all;
+
+#[derive(Debug)]
+pub struct HorizonState {
+    data_folder: PathBuf,
+
+    config_folder: PathBuf,
+}
+
+impl HorizonState {
+    pub fn new(data_folder: PathBuf, config_folder: PathBuf) -> Self {
+        Self {
+            data_folder,
+            config_folder,
+        }
+    }
+
+    pub async fn init_data_folder(&self) {
+        if !self.data_folder.exists() {
+            create_dir_all(&self.data_folder).await.unwrap();
+        }
+    }
+
+    pub async fn init_config_folder(&self) {
+        if !self.config_folder.exists() {
+            create_dir_all(&self.config_folder).await.unwrap();
+        }
+    }
+}

horizon_common/src/lib.rs

@@ -1,7 +1,11 @@
+pub mod api;
+
 pub fn add(left: u64, right: u64) -> u64 {
     left + right
 }
 
+pub use api::state::HorizonState as HorizonCliState;
+
 #[cfg(test)]
 mod tests {
     use super::*;