Add Command-Line Arguments

Cargo.toml

@@ -18,6 +18,7 @@ actix-ws = { version = "0.3.0", default-features = false }
 async-trait = "0.1.88"
 bytestring = "1.4.0"
 chrono = { version = "0.4.41", features = ["serde"] }
+clap = { version = "4.5.50", features = ["derive"] }
 dashmap = { version = "6.1.0", features = ["serde"] }
 dotenvy = "0.15.7"
 futures-util = "0.3.31"

src/guards.rs

@@ -1,9 +1,13 @@
+use crate::get_args;
 use actix_web::guard::GuardContext;
-
 pub const INTERNAL_KEY_HEADER: &str = "Kromer-Key";
 
 pub fn internal_key_guard(ctx: &GuardContext) -> bool {
-    let kromer_key = std::env::var("INTERNAL_KEY").expect("No INTERNAL_KEY set in .env file");
+    let args = get_args();
+    let kromer_key = args.key.clone().unwrap_or_else(|| {
+        std::env::var("INTERNAL_KEY")
+            .expect("No INTERNAL_KEY set in .env file or command line argument (--key)")
+    });
     ctx.head()
         .headers()
         .get(INTERNAL_KEY_HEADER)