added install command

Author: cqb13

src/main.rs

@@ -6,7 +6,7 @@ use crate::cli::{Arg, Cli, Command};
 use crate::commands::all::all_command;
 use crate::commands::releases::releases_command;
 use crate::commands::user::user_command;
-use crate::utils::validate_and_convert_path;
+use crate::utils::{install, validate_and_convert_path, OS};
 
 fn main() {
     let cli = Cli::new().with_default_command("help").with_commands(vec![
@@ -128,6 +128,15 @@ fn main() {
 
     match command.name {
         "version" => cli.version(),
+        "install" => {
+            let os = match std::env::consts::OS {
+                "windows" => OS::Windows,
+                "macos" => OS::Mac,
+                _ => panic!("Unsupported OS"),
+            };
+
+            install(&os);
+        }
         "all" => {
             let user = command.get_value_of("user").throw_if_none();
             let repo = command.get_value_of("repository").throw_if_none();

src/utils.rs

@@ -1,9 +1,26 @@
 use reqwest::header::{HeaderMap, ACCEPT, USER_AGENT};
 use serde_json::Value;
+use std::env;
+use std::fs;
 use std::fs::{create_dir_all, remove_file, File};
 use std::io::{BufWriter, Write};
 use std::path::{Path, PathBuf};
 
+#[derive(Debug)]
+pub enum OS {
+    Windows,
+    Mac,
+}
+
+impl OS {
+    fn get_name(&self) -> &str {
+        match self {
+            OS::Windows => "Windows",
+            OS::Mac => "Mac",
+        }
+    }
+}
+
 pub fn request(url: String) -> Result<Value, Box<dyn std::error::Error>> {
     let headers = construct_header();
     let client = reqwest::blocking::Client::new();
@@ -96,3 +113,110 @@ pub fn bytes_to_best_size(bytes: i64) -> String {
 
     format!("{:.2} {}", size, unit)
 }
+
+pub fn install(os: &OS) {
+    println!("starting install on {}", os.get_name());
+
+    let home_dir = get_user_home_dir(os);
+    let bin_path = match os {
+        OS::Windows => "AppData/Roaming/gstats",
+        OS::Mac => ".local/bin",
+    };
+
+    let local_bin_path = format!("{}/{}", home_dir, bin_path);
+
+    // Create directory if it doesn't exist
+    if !Path::new(&local_bin_path).exists() {
+        println!("Creating {} directory", local_bin_path);
+        fs::create_dir_all(&local_bin_path).unwrap();
+    }
+
+    let binary_name = match os {
+        OS::Windows => "gstats.exe",
+        OS::Mac => "gstats",
+    };
+
+    let new_binary_path = format!("{}/{}", local_bin_path, binary_name);
+
+    if Path::new(&new_binary_path).exists() {
+        println!("Replacing binary in {}", &local_bin_path);
+        fs::remove_file(&new_binary_path).unwrap();
+    }
+
+    println!("Moving binary to {}", local_bin_path);
+    fs::copy(
+        format!("{}/{}", get_current_directory_path(), binary_name),
+        &new_binary_path,
+    )
+    .unwrap();
+
+    match os {
+        OS::Windows => {
+            if let Err(e) = add_registry_path(&local_bin_path) {
+                eprintln!("Failed to modify system PATH: {}", e);
+                eprintln!("This action may require administrator permissions.");
+                return;
+            }
+        }
+        OS::Mac => {
+            let zprofile_path = format!("{}/.zprofile", home_dir);
+            if let Ok(zprofile_content) = fs::read_to_string(&zprofile_path) {
+                if !zprofile_content.contains("export PATH=\"$PATH:$HOME/.local/bin\"") {
+                    println!("Adding .local/bin to path in .zprofile");
+                    let mut zprofile_file = File::create(&zprofile_path).unwrap();
+                    writeln!(zprofile_file, "export PATH=\"$PATH:$HOME/.local/bin\"").unwrap();
+                }
+            }
+        }
+    }
+
+    println!("install complete");
+}
+
+fn add_registry_path(new_path: &str) -> std::io::Result<()> {
+    use std::process::Command;
+
+    // Escape percent signs by doubling them
+    let escaped_path = new_path.replace("%", "%%");
+
+    // Prepare the command to modify the registry
+    let status = Command::new("reg")
+        .args(&[
+            "ADD",
+            "HKLM\\SYSTEM\\CurrentControlSet\\Control\\Session Manager\\Environment",
+            "/v",
+            "Path",
+            "/t",
+            "REG_EXPAND_SZ",
+            "/d",
+            &escaped_path,
+            "/f",
+        ])
+        .status()?;
+
+    if !status.success() {
+        return Err(std::io::Error::new(
+            std::io::ErrorKind::Other,
+            "Failed to modify registry",
+        ));
+    }
+
+    Ok(())
+}
+
+fn get_current_directory_path() -> String {
+    let current_dir_path = match env::current_dir() {
+        Ok(path) => path,
+        Err(_) => panic!("Could not get current directory path"),
+    };
+
+    current_dir_path.to_str().unwrap().to_string()
+}
+
+fn get_user_home_dir(os: &OS) -> String {
+    let user_dir = match os {
+        OS::Windows => "USERPROFILE",
+        OS::Mac => "HOME",
+    };
+    std::env::var(format!("{}", user_dir)).unwrap()
+}