mirror of
https://github.com/house-of-vanity/khm.git
synced 2025-07-07 23:34:07 +00:00
84 lines
3.2 KiB
Rust
84 lines
3.2 KiB
Rust
|
mod server;
|
||
|
|
mod client;
|
||
|
|
|
||
|
|
use clap::Parser;
|
||
|
|
use env_logger;
|
||
|
|
|
||
|
|
/// This application manages SSH keys and flows, either as a server or client.
|
||
|
|
/// In server mode, it stores keys and flows in a PostgreSQL database.
|
||
|
|
/// In client mode, it sends keys to the server and can update the known_hosts file with keys from the server.
|
||
|
|
#[derive(Parser, Debug)]
|
||
|
|
#[command(
|
||
|
|
author = "Your Name",
|
||
|
|
version = "1.0",
|
||
|
|
about = "SSH Key Manager",
|
||
|
|
long_about = None,
|
||
|
|
after_help = "Examples:\n\
|
||
|
|
\n\
|
||
|
|
Running in server mode:\n\
|
||
|
|
khm --server --ip 0.0.0.0 --port 1337 --db-host psql.psql.svc --db-name khm --db-user admin --db-password <SECRET> --flows work,home\n\
|
||
|
|
\n\
|
||
|
|
Running in client mode to send diff and sync ~/.ssh/known_hosts with remote flow in place:\n\
|
||
|
|
khm --host http://kh.example.com:8080 --known_hosts ~/.ssh/known_hosts --in-place\n\
|
||
|
|
\n\
|
||
|
|
"
|
||
|
|
)]struct Args {
|
||
|
|
/// IP address to bind the server or client to (default: 127.0.0.1)
|
||
|
|
#[arg(short, long, default_value = "127.0.0.1", help = "Server mode: IP address to bind the server to")]
|
||
|
|
ip: String,
|
||
|
|
|
||
|
|
/// Port to bind the server or client to (default: 8080)
|
||
|
|
#[arg(short, long, default_value = "8080", help = "Server mode: Port to bind the server to")]
|
||
|
|
port: u16,
|
||
|
|
|
||
|
|
/// Hostname or IP address of the PostgreSQL database (default: 127.0.0.1)
|
||
|
|
#[arg(long, default_value = "127.0.0.1", help = "Server mode: Hostname or IP address of the PostgreSQL database")]
|
||
|
|
db_host: String,
|
||
|
|
|
||
|
|
/// Name of the PostgreSQL database (default: khm)
|
||
|
|
#[arg(long, default_value = "khm", help = "Server mode: Name of the PostgreSQL database")]
|
||
|
|
db_name: String,
|
||
|
|
|
||
|
|
/// Username for the PostgreSQL database (required in server mode)
|
||
|
|
#[arg(long, required_if_eq("server", "true"), help = "Server mode: Username for the PostgreSQL database")]
|
||
|
|
db_user: Option<String>,
|
||
|
|
|
||
|
|
/// Password for the PostgreSQL database (required in server mode)
|
||
|
|
#[arg(long, required_if_eq("server", "true"), help = "Server mode: Password for the PostgreSQL database")]
|
||
|
|
db_password: Option<String>,
|
||
|
|
|
||
|
|
/// Host address of the server to connect to in client mode (required in client mode)
|
||
|
|
#[arg(long, required_if_eq("server", "false"), help = "Client mode: Host address of the server to connect to")]
|
||
|
|
host: Option<String>,
|
||
|
|
|
||
|
|
/// Run in server mode (default: false)
|
||
|
|
#[arg(long, help = "Run in server mode")]
|
||
|
|
server: bool,
|
||
|
|
|
||
|
|
/// Path to the known_hosts file (default: ~/.ssh/known_hosts)
|
||
|
|
#[arg(long, default_value = "~/.ssh/known_hosts", help = "Client mode: Path to the known_hosts file")]
|
||
|
|
known_hosts: String,
|
||
|
|
|
||
|
|
/// Update the known_hosts file with keys from the server after sending keys (default: false)
|
||
|
|
#[arg(long, help = "Server mode: Sync the known_hosts file with keys from the server")]
|
||
|
|
in_place: bool,
|
||
|
|
|
||
|
|
/// Comma-separated list of flows to manage (default: default)
|
||
|
|
#[arg(long, default_value = "default", value_parser, num_args = 1.., value_delimiter = ',', help = "Comma-separated list of flows to manage")]
|
||
|
|
flows: Vec<String>,
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
#[actix_web::main]
|
||
|
|
async fn main() -> std::io::Result<()> {
|
||
|
|
env_logger::init();
|
||
|
|
|
||
|
|
let args = Args::parse();
|
||
|
|
|
||
|
|
if args.server {
|
||
|
|
server::run_server(args).await
|
||
|
|
} else {
|
||
|
|
client::run_client(args).await
|
||
|
|
}
|
||
|
|
}
|