System Initiative

use std::path::PathBuf; use clap::{ ArgAction, Parser, }; use module_index_server::{ Config, ConfigError, ConfigFile, StandardConfigFile, }; use si_std::SensitiveString; const NAME: &str = "module_index"; /// Parse, validate, and return the CLI arguments as a typed struct. pub(crate) fn parse() -> Args { Args::parse() } /// The System Initiative Module Index API service /// /// This is the centralized store of "modules" - ie how users will back up and share /// their customizations of our system #[derive(Parser, Debug)] #[command(name = NAME, max_term_width = 100)] pub(crate) struct Args { /// Sets the verbosity mode. /// /// Multiple -v options increase verbosity. The maximum is 6. #[arg(short = 'v', long = "verbose", action = ArgAction::Count)] pub(crate) verbose: u8, /// Disables ANSI coloring in log output, even if standard output refers to a terminal/TTY. /// /// For more details, visit: <http://no-color.org/>. #[arg( long = "no-color", default_value = "false", env = "SI_NO_COLOR", hide_env_values = true, conflicts_with = "force_color" )] pub(crate) no_color: bool, /// Forces ANSI coloring, even if standard output refers to a terminal/TTY. /// /// For more details, visit: <http://no-color.org/>. #[arg( long = "force-color", default_value = "false", env = "SI_FORCE_COLOR", hide_env_values = true, conflicts_with = "no_color" )] pub(crate) force_color: bool, /// Prints telemetry logging as JSON lines. /// /// For more details, visit: <https://jsonlines.org/>. #[arg( long = "log-json", default_value = "false", env = "SI_LOG_JSON", hide_env_values = true )] pub(crate) log_json: bool, /// Additionally appends logging to rolling files under the given directory. #[arg( long = "log-file-directory", env = "SI_LOG_FILE_DIRECTORY", hide_env_values = true )] pub(crate) log_file_directory: Option<PathBuf>, /// Enables support for emitting async runtime data to `tokio-console`. /// /// For more details, visit: <https://github.com/tokio-rs/console>. #[arg( long = "tokio-console", default_value = "false", env = "SI_TOKIO_CONSOLE", hide_env_values = true )] pub(crate) tokio_console: bool, /// Override for the auth api url #[arg(long, env = "SI_AUTH_API_URL")] pub(crate) auth_api_url: Option<String>, /// PostgreSQL connection pool dbname [example: myapp] #[arg(long, env)] pub(crate) pg_dbname: Option<String>, /// PostgreSQL connection pool hostname [example: prod.db.example.com] #[arg(long, env)] pub(crate) pg_hostname: Option<String>, /// PostgreSQL connection pool max size [example: 8] #[arg(long, env)] pub(crate) pg_pool_max_size: Option<u32>, /// PostgreSQL connection pool port [example: 5432] #[arg(long, env)] pub(crate) pg_port: Option<u16>, /// PostgreSQL connection pool user [example: dbuser] #[arg(long, env)] pub(crate) pg_user: Option<String>, /// PostgreSQL connection certification path #[arg(long)] pub(crate) pg_cert_path: Option<PathBuf>, /// PostgreSQL connection certification base64 string #[arg(long)] pub(crate) pg_cert_base64: Option<SensitiveString>, /// PostgreSQL connection certificate url #[arg(long)] pub(crate) pg_cert_url: Option<String>, /// The address and port to bind the HTTP server to [example: 0.0.0.0:80] #[arg(long, env)] pub(crate) socket_addr: Option<String>, /// The s3 bucket access key id #[arg(long, env)] pub(crate) s3_access_key_id: Option<SensitiveString>, /// The s3 bucket #[arg(long, env)] pub(crate) s3_bucket: Option<String>, /// The s3 bucket region #[arg(long, env)] pub(crate) s3_region: Option<String>, /// The s3 bucket secret access key #[arg(long, env)] pub(crate) s3_secret_access_key: Option<SensitiveString>, /// The s3 bucket path prefix #[arg(long, env)] pub(crate) s3_path_prefix: Option<String>, /// The path to the JWT public signing key #[arg(long, env)] pub(crate) jwt_public_key: Option<String>, #[arg(long, env)] pub(crate) jwt_public_key_algo: Option<String>, #[arg(long, env)] pub(crate) jwt_secondary_public_key: Option<String>, #[arg(long, env)] pub(crate) jwt_secondary_public_key_algo: Option<String>, // /// Database migration mode on startup // #[arg(long, value_parser = PossibleValuesParser::new(MigrationMode::variants()))] } impl TryFrom<Args> for Config { type Error = ConfigError; fn try_from(args: Args) -> Result<Self, Self::Error> { ConfigFile::layered_load(NAME, |config_map| { if let Some(auth_api_url) = args.auth_api_url { config_map.set("auth_api_url", auth_api_url); } if let Some(dbname) = args.pg_dbname { config_map.set("pg.dbname", dbname); } if let Some(hostname) = args.pg_hostname { config_map.set("pg.hostname", hostname); } if let Some(pool_max_size) = args.pg_pool_max_size { config_map.set("pg.pool_max_size", i64::from(pool_max_size)); } if let Some(port) = args.pg_port { config_map.set("pg.port", i64::from(port)); } if let Some(user) = args.pg_user { config_map.set("pg.user", user); } if let Some(cert_path) = args.pg_cert_path { config_map.set("pg.certificate_path", cert_path.display().to_string()); } if let Some(cert) = args.pg_cert_base64 { config_map.set("pg.certificate_base64", cert.to_string()); } if let Some(cert) = args.pg_cert_url { config_map.set("pg.certificate_url", cert); } if let Some(socket_addr) = args.socket_addr { config_map.set("socket_addr", socket_addr); } if let Some(s3_access_key_id) = args.s3_access_key_id { config_map.set("s3.access_key_id", s3_access_key_id.to_string()); } if let Some(s3_secret_access_key) = args.s3_secret_access_key { config_map.set("s3.secret_access_key", s3_secret_access_key.to_string()); } if let Some(s3_bucket) = args.s3_bucket { config_map.set("s3.bucket", s3_bucket); } if let Some(s3_region) = args.s3_region { config_map.set("s3.region", s3_region); } if let Some(s3_path_prefix) = args.s3_path_prefix { config_map.set("s3.path_prefix", s3_path_prefix); } if let Some(jwt_public_key) = args.jwt_public_key { config_map.set("jwt_signing_public_key_path", jwt_public_key.to_string()); } // if let Some(migration_mode) = args.migration_mode { // config_map.set("migration_mode", migration_mode); // } config_map.set("pg.application_name", NAME); })? .try_into() } } #[cfg(test)] mod tests { use super::*; #[test] fn verify_command() { use clap::CommandFactory; Args::command().debug_assert() } }