Jon Häggblad
145 lines
3.7 KiB
Rust
145 lines
3.7 KiB
Rust
// Copyright 2021 - Nym Technologies SA <contact@nymtech.net>
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use crate::client::config::{Config, SocketType};
|
|
use crate::error::ClientError;
|
|
use clap::CommandFactory;
|
|
use clap::{Parser, Subcommand};
|
|
use completions::{fig_generate, ArgShell};
|
|
|
|
pub(crate) mod init;
|
|
pub(crate) mod run;
|
|
pub(crate) mod upgrade;
|
|
|
|
fn long_version() -> String {
|
|
format!(
|
|
r#"
|
|
{:<20}{}
|
|
{:<20}{}
|
|
{:<20}{}
|
|
{:<20}{}
|
|
{:<20}{}
|
|
{:<20}{}
|
|
{:<20}{}
|
|
{:<20}{}
|
|
"#,
|
|
"Build Timestamp:",
|
|
env!("VERGEN_BUILD_TIMESTAMP"),
|
|
"Build Version:",
|
|
env!("VERGEN_BUILD_SEMVER"),
|
|
"Commit SHA:",
|
|
env!("VERGEN_GIT_SHA"),
|
|
"Commit Date:",
|
|
env!("VERGEN_GIT_COMMIT_TIMESTAMP"),
|
|
"Commit Branch:",
|
|
env!("VERGEN_GIT_BRANCH"),
|
|
"rustc Version:",
|
|
env!("VERGEN_RUSTC_SEMVER"),
|
|
"rustc Channel:",
|
|
env!("VERGEN_RUSTC_CHANNEL"),
|
|
"cargo Profile:",
|
|
env!("VERGEN_CARGO_PROFILE"),
|
|
)
|
|
}
|
|
|
|
fn long_version_static() -> &'static str {
|
|
Box::leak(long_version().into_boxed_str())
|
|
}
|
|
|
|
#[derive(Parser)]
|
|
#[clap(author = "Nymtech", version, long_version = long_version_static(), about)]
|
|
pub(crate) struct Cli {
|
|
/// Path pointing to an env file that configures the client.
|
|
#[clap(short, long)]
|
|
pub(crate) config_env_file: Option<std::path::PathBuf>,
|
|
|
|
#[clap(subcommand)]
|
|
command: Commands,
|
|
}
|
|
|
|
#[derive(Subcommand)]
|
|
pub(crate) enum Commands {
|
|
/// Initialise a Nym client. Do this first!
|
|
Init(init::Init),
|
|
/// Run the Nym client with provided configuration client optionally overriding set parameters
|
|
Run(run::Run),
|
|
/// Try to upgrade the client
|
|
Upgrade(upgrade::Upgrade),
|
|
|
|
/// Generate shell completions
|
|
Completions(ArgShell),
|
|
|
|
/// Generate Fig specification
|
|
GenerateFigSpec,
|
|
}
|
|
|
|
// Configuration that can be overridden.
|
|
pub(crate) struct OverrideConfig {
|
|
validators: Option<String>,
|
|
disable_socket: bool,
|
|
port: Option<u16>,
|
|
fastmode: bool,
|
|
|
|
#[cfg(feature = "coconut")]
|
|
enabled_credentials_mode: bool,
|
|
}
|
|
|
|
pub(crate) async fn execute(args: &Cli) -> Result<(), ClientError> {
|
|
let bin_name = "nym-native-client";
|
|
|
|
match &args.command {
|
|
Commands::Init(m) => init::execute(m).await,
|
|
Commands::Run(m) => run::execute(m).await?,
|
|
Commands::Upgrade(m) => upgrade::execute(m),
|
|
Commands::Completions(s) => s.generate(&mut Cli::into_app(), bin_name),
|
|
Commands::GenerateFigSpec => fig_generate(&mut Cli::into_app(), bin_name),
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
pub(crate) fn override_config(mut config: Config, args: OverrideConfig) -> Config {
|
|
if let Some(raw_validators) = args.validators {
|
|
config
|
|
.get_base_mut()
|
|
.set_custom_validator_apis(config::parse_validators(&raw_validators));
|
|
} else if std::env::var(network_defaults::var_names::CONFIGURED).is_ok() {
|
|
let raw_validators = std::env::var(network_defaults::var_names::API_VALIDATOR)
|
|
.expect("api validator not set");
|
|
config
|
|
.get_base_mut()
|
|
.set_custom_validator_apis(config::parse_validators(&raw_validators));
|
|
}
|
|
|
|
if args.disable_socket {
|
|
config = config.with_socket(SocketType::None);
|
|
}
|
|
|
|
if let Some(port) = args.port {
|
|
config = config.with_port(port);
|
|
}
|
|
|
|
#[cfg(feature = "coconut")]
|
|
{
|
|
if args.enabled_credentials_mode {
|
|
config.get_base_mut().with_disabled_credentials(false)
|
|
}
|
|
}
|
|
|
|
if args.fastmode {
|
|
config.get_base_mut().set_high_default_traffic_volume();
|
|
}
|
|
|
|
config
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use clap::CommandFactory;
|
|
|
|
#[test]
|
|
fn verify_cli() {
|
|
Cli::command().debug_assert();
|
|
}
|
|
}
|