jwkserve-cli 0.1.0

CLI tool for jwkserve - a fake authentication service for local development
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

use jwkserve::{Router, RouterState, key::KeyStore};
use poem::{Server, listener::TcpListener};
use std::{env, io::Error};
use tokio::signal;

#[tokio::main]
async fn main() -> Result<(), Error> {
    tracing_subscriber::fmt()
        .with_target(false)
        .with_thread_ids(false)
        .with_thread_names(false)
        .init();

    let host: String = match env::var("APP_HOST") {
        Ok(host) => host,
        Err(_) => "0.0.0.0".to_string(),
    };

    let port: String = match env::var("APP_PORT") {
        Ok(port) => port,
        Err(_) => "3000".to_string(),
    };

    let key_store: KeyStore = match env::var("KEY_FILE") {
        Ok(key_path) => KeyStore::from_file(&key_path),
        Err(_) => KeyStore::new(),
    };

    let issuer: String = match env::var("WEB_ISSUER") {
        Ok(issuer) => issuer,
        Err(_) => format!("http://{host}:{port}"),
    };

    let router_state = RouterState { issuer, key_store };

    let router = Router::with_state(router_state).await?;

    Server::new(TcpListener::bind(format!("{host}:{port}")))
        .name("jwkserve".to_string())
        .run_with_graceful_shutdown(
            router.poem,
            async move {
                let _ = signal::ctrl_c().await;
            },
            None,
        )
        .await
}