Add command to generate JWT tokens

We should not search for 3rd party JWT generators, but have
kernelci-storage embedded features.

Signed-off-by: Denys Fedoryshchenko <denys.f@collabora.com>

Author: nuclearcat

src/main.rs

@@ -57,9 +57,17 @@ struct Args {
     #[clap(
         short,
         long,
+        default_value = "false",
         help = "Generate JWT secret"
     )]
     generate_jwt_secret: bool,
+
+    #[clap(
+        long,
+        default_value = "",
+        help = "Generate JWT token for email"
+    )]
+    generate_jwt_token: String,
 }
 
 struct ReceivedFile {
@@ -99,6 +107,19 @@ async fn initial_setup() -> Option<RustlsConfig> {
         std::process::exit(0);
     }
 
+    if args.generate_jwt_token != "" {
+        let token_r = storjwt::generate_jwt_token(&args.generate_jwt_token);
+        let token = match token_r {
+            Ok(token) => token,
+            Err(e) => {
+                eprintln!("Error generating JWT token: {}", e);
+                std::process::exit(1);
+            }
+        };
+        println!("JWT token: {}", token);
+        std::process::exit(0);
+    }
+
     if let Err(e) = std::env::set_current_dir(&args.files_directory) {
         eprintln!("Error changing directory: {}", e);
         std::process::exit(1);

src/storjwt.rs

@@ -1,11 +1,11 @@
 use crate::Args;
 use clap::Parser;
 use hmac::{Hmac, Mac};
-use jwt::{Header, Token, VerifyWithKey};
+use jwt::{Header, Token, VerifyWithKey, SignWithKey};
 use sha2::Sha256;
 use std::collections::BTreeMap;
 use toml::value::Table;
-
+use rand::{distributions::Alphanumeric, Rng};
 pub fn verify_jwt_token(token_str: &str) -> Result<BTreeMap<String, String>, jwt::Error> {
     // config.toml, jwt_secret parameter
     let args = Args::parse();
@@ -48,3 +48,16 @@ pub fn generate_jwt_secret() {
         .collect();
     println!("jwt_secret=\"{}\"", secret);
 }
+
+pub fn generate_jwt_token(email: &str) -> Result<String, jwt::Error> {
+    let args = Args::parse();
+    let cfg_file = args.config_file;
+    let toml_cfg = std::fs::read_to_string(&cfg_file).unwrap();
+    let parsed_toml = toml_cfg.parse::<Table>().unwrap();
+    let key_str = parsed_toml["jwt_secret"].as_str().unwrap();
+    let key: Hmac<Sha256> = Hmac::new_from_slice(key_str.as_bytes())?;
+    let mut claims = BTreeMap::new();
+    claims.insert("email".to_string(), email.to_string());
+    let token_str = claims.sign_with_key(&key)?;
+    Ok(token_str)
+}