Pierre HUBERT
a3b9c7cdb1
All checks were successful
continuous-integration/drone/push Build is passing
77 lines
2.1 KiB
Rust
77 lines
2.1 KiB
Rust
use clap::Parser;
|
|
use jwt_simple::algorithms::HS256Key;
|
|
use jwt_simple::prelude::{Clock, Duration, JWTClaims, MACLike};
|
|
use moneymgr_backend::constants;
|
|
use moneymgr_backend::extractors::auth_extractor::TokenClaims;
|
|
use moneymgr_backend::utils::rand_utils::rand_string;
|
|
use std::ops::Add;
|
|
use std::os::unix::prelude::CommandExt;
|
|
use std::process::Command;
|
|
|
|
/// cURL wrapper to query Money manager
|
|
#[derive(Parser, Debug)]
|
|
#[command(version, about, long_about = None)]
|
|
struct Args {
|
|
/// URL to Money manager API
|
|
#[arg(short('U'), long, env, default_value = "http://localhost:8000/api")]
|
|
moneymgr_url: String,
|
|
|
|
/// Token ID
|
|
#[arg(short('i'), long, env)]
|
|
token_id: u32,
|
|
|
|
/// Token secret
|
|
#[arg(short('t'), long, env)]
|
|
token_secret: String,
|
|
|
|
/// Request verb
|
|
#[arg(short('X'), long, default_value = "GET")]
|
|
method: String,
|
|
|
|
/// Request URI
|
|
uri: String,
|
|
|
|
/// Command line arguments to pass to cURL
|
|
#[clap(trailing_var_arg = true, allow_hyphen_values = true)]
|
|
run: Vec<String>,
|
|
}
|
|
|
|
fn main() {
|
|
let args: Args = Args::parse();
|
|
|
|
let full_url = format!("{}{}", args.moneymgr_url, args.uri);
|
|
|
|
log::debug!("Full URL: {full_url}");
|
|
|
|
let key = HS256Key::from_bytes(args.token_secret.as_bytes());
|
|
|
|
let claims = JWTClaims::<TokenClaims> {
|
|
issued_at: Some(Clock::now_since_epoch()),
|
|
expires_at: Some(Clock::now_since_epoch().add(Duration::from_mins(15))),
|
|
invalid_before: None,
|
|
issuer: None,
|
|
subject: None,
|
|
audiences: None,
|
|
jwt_id: None,
|
|
nonce: Some(rand_string(10)),
|
|
custom: TokenClaims {
|
|
method: args.method.to_string(),
|
|
uri: args.uri,
|
|
},
|
|
};
|
|
|
|
let jwt = key
|
|
.with_key_id(&args.token_id.to_string())
|
|
.authenticate(claims)
|
|
.expect("Failed to sign JWT!");
|
|
|
|
let _ = Command::new("curl")
|
|
.args(["-X", &args.method])
|
|
.args(["-H", &format!("{}: {jwt}", constants::API_TOKEN_HEADER)])
|
|
.args(args.run)
|
|
.arg(full_url)
|
|
.exec();
|
|
|
|
panic!("Failed to run cURL!")
|
|
}
|