From 266f30af09c76a10e18214557e323455fa043be3 Mon Sep 17 00:00:00 2001
From: Alexander Rosenberg <zanderpkg@pm.me>
Date: Fri, 7 Oct 2022 04:57:59 -0700
Subject: [PATCH] Added user license file and license bad name check

---
 .cargo/config.toml |  2 +-
 install.sh         |  8 ++--
 src/main.rs        | 92 +++++++++++++++++++++++++++++++++-------------
 3 files changed, 71 insertions(+), 31 deletions(-)

diff --git a/.cargo/config.toml b/.cargo/config.toml
index d387223..688ef2b 100644
--- a/.cargo/config.toml
+++ b/.cargo/config.toml
@@ -1,2 +1,2 @@
 [env]
-DATA_PATH = { value = "licenses", relative = true }
+GLOBAL_DATA_PATH = { value = "licenses", relative = true }
diff --git a/install.sh b/install.sh
index 58a5bbc..3d7b10f 100755
--- a/install.sh
+++ b/install.sh
@@ -6,9 +6,9 @@ if [ "$#" -ge 1 ]; then
     PREFIX="$1"
 fi
 
-export DATA_PATH="$PREFIX/share/license-tool"
+export GLOBAL_DATA_PATH="$PREFIX/share/license-tool"
 
 cargo install --path . --root "$PREFIX"
-printf "Installing licenses to '%s'\n" "$DATA_PATH"
-rm -rf "$DATA_PATH"
-cp -R licenses "$DATA_PATH"
+printf "Installing licenses to '%s'\n" "$GLOBAL_DATA_PATH"
+rm -rf "$GLOBAL_DATA_PATH"
+cp -R licenses "$GLOBAL_DATA_PATH"
diff --git a/src/main.rs b/src/main.rs
index ad1453e..d098db7 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -2,16 +2,18 @@ use std::env;
 use std::fs;
 use std::io;
 use std::process::exit;
+use std::collections::HashSet;
 
-const DATA_PATH: &str = env!("DATA_PATH");
+const GLOBAL_DATA_PATH: &str = env!("GLOBAL_DATA_PATH");
 
 fn main() {
+    let user_data_file = get_user_data_file();
     let args: Vec<String> = env::args().collect();
     let license_index;
     let outfile: &str;
     if args.len() < 2 || args[1] == "-h" {
-        print_help();
-        return;
+        print_help(&user_data_file);
+        return
     } else if args[1] == "-o" {
         outfile = get_outfile(&args).unwrap_or_else(|| exit(1));
         license_index = 3;
@@ -20,45 +22,83 @@ fn main() {
         license_index = 1;
     }
     if args.len() <= license_index {
-        print_help();
-        return;
+        print_help(&user_data_file);
+        return
     }
-    copy_license(&args[license_index], outfile);
+    let license_name = &args[license_index];
+    if license_name.contains("/") {
+        eprintln!("error: malformatted license name: '{}'", license_name);
+        exit(1)
+    }
+    copy_license(&license_name, outfile, &user_data_file);
 }
 
-fn copy_license(license: &str, outfile: &str) {
-    let license_path = DATA_PATH.to_owned() + "/" + license;
-    let license_content = fs::read_to_string(license_path).unwrap_or_else(|_| {
-        eprintln!("error: could not open license for reading: '{}'", license);
-        exit(1);
+fn copy_license(license: &str, outfile: &str, user_data_file: &String) {
+    let user_path = user_data_file.to_owned() + "/" + license;
+    let license_content = fs::read_to_string(user_path).unwrap_or_else(|_| {
+        let global_path = GLOBAL_DATA_PATH.to_owned() + "/" + license;
+        fs::read_to_string(global_path).unwrap_or_else(|_| {
+            eprintln!("error: could not open license for reading: '{}'", license);
+            exit(1)
+        })
     });
     let _ = fs::write(outfile, license_content).or_else(|_| -> io::Result<()> {
         eprintln!("error: could not write license: '{}'", outfile);
-        exit(1);
+        exit(1)
     });
 }
 
 fn get_outfile(args: &Vec<String>) -> Option<&str> {
     if args.len() < 3 {
         eprintln!("error: -o requires an option");
-        return None;
+        return None
     }
-    return Some(&args[2]);
+    Some(&args[2])
 }
 
-fn print_help() {
+fn print_help(user_data_file: &String) {
     println!("license-tool [-h] [-o OUTFILE] <license>");
-    println!("Licenses:");
-    let dirs = fs::read_dir(DATA_PATH);
-    if dirs.is_err() {
-        eprintln!("error: could not open data directory");
-        exit(1);
-    }
-    for file in dirs.unwrap() {
-        if file.is_err() {
-            eprintln!("error: error reading data directory");
-            exit(1);
+    let mut user_dirs = HashSet::<String>::new();
+    let user_dir = fs::read_dir(user_data_file);
+    if user_dir.is_ok() {
+        println!("User Licenses:");
+        let user_error_fun = |_| {
+            eprintln!("error: could not read user data directory");
+            exit(1)
+        };
+        for file in user_dir.unwrap() {
+            let file_name = file.unwrap_or_else(user_error_fun).file_name()
+                .to_string_lossy().to_string();
+            println!("  {}", file_name);
+            user_dirs.insert(file_name);
+        }
+    }
+    let global_dir = fs::read_dir(GLOBAL_DATA_PATH);
+    if global_dir.is_ok() {
+        println!("Global Licenses:");
+        let global_error_fun = |_| {
+            eprintln!("error: error reading data directory");
+            exit(1)
+        };
+        for file in global_dir.unwrap() {
+            let file_name = file.unwrap_or_else(global_error_fun).file_name()
+                .to_string_lossy().to_string();
+            if user_dirs.contains(&file_name) {
+                println!("  {} [Overridden]", file_name);
+            } else {
+                println!("  {}", file_name);
+            }
         }
-        println!("  {}", file.as_ref().unwrap().file_name().to_string_lossy());
     }
 }
+
+fn get_user_data_file() -> String {
+    let xdg_config = env::var("XDG_CONFIG_HOME").unwrap_or_else(|_| {
+        let home_dir = env::var("HOME").unwrap_or_else(|_| {
+            eprintln!("error: '$HOME' not set");
+            exit(1)
+        });
+        home_dir + "/.config"
+    });
+    xdg_config + "/license-tool"
+}