Make size variable

nc-apps · Feb 9, 2025 · 3f2b2fb · 3f2b2fb
1 parent cd60168
commit 3f2b2fb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/bin/generate_secret.rs b/src/bin/generate_secret.rs
@@ -1,7 +1,10 @@
 use base64::{prelude::BASE64_URL_SAFE_NO_PAD, Engine};
 
 fn main() {
-    let mut key = [0u8; 32];
+    let mut arguments = std::env::args();
+    _ = arguments.next();
+    let argument = arguments.next().unwrap();
+    let mut key = vec![0u8; argument.parse().unwrap()];
     getrandom::getrandom(&mut key).unwrap();
     let encoded = BASE64_URL_SAFE_NO_PAD.encode(&key);
     println!("{}", encoded);