Dave Hrycyszyn
127 lines
4.2 KiB
Rust
127 lines
4.2 KiB
Rust
use clap::{App, Arg, ArgMatches, SubCommand};
|
|
use std::process;
|
|
|
|
pub mod clients;
|
|
mod commands;
|
|
mod identity;
|
|
mod persistence;
|
|
mod sockets;
|
|
mod utils;
|
|
|
|
fn main() {
|
|
let arg_matches = App::new("Nym Client")
|
|
.version("0.1.0")
|
|
.author("Nymtech")
|
|
.about("Implementation of the Nym Client")
|
|
.subcommand(
|
|
SubCommand::with_name("init")
|
|
.about("Initialise a Nym client. Do this first!")
|
|
.arg(Arg::with_name("id")
|
|
.long("id")
|
|
.help("Id of the nym-mixnet-client we want to create config for.")
|
|
.takes_value(true)
|
|
.required(true)
|
|
)
|
|
.arg(Arg::with_name("provider")
|
|
.long("provider")
|
|
.help("Id of the provider we have preference to connect to. If left empty, a random provider will be chosen.")
|
|
.takes_value(true)
|
|
)
|
|
)
|
|
.subcommand(
|
|
SubCommand::with_name("run")
|
|
.about("Run a persistent Nym client process")
|
|
.arg(Arg::with_name("id")
|
|
.long("id")
|
|
.help("Id of the nym-mixnet-client we want to run.")
|
|
.takes_value(true)
|
|
.required(true)
|
|
)
|
|
.arg(Arg::with_name("local")
|
|
.long("local")
|
|
.help("Flag to indicate whether the client is expected to run on the local deployment.")
|
|
.takes_value(false)
|
|
)
|
|
)
|
|
.subcommand(
|
|
SubCommand::with_name("tcpsocket")
|
|
.about("Run Nym client that listens for bytes on a TCP socket")
|
|
.arg(
|
|
Arg::with_name("port")
|
|
.short("p")
|
|
.long("port")
|
|
.help("Port for TCP socket to listen on")
|
|
.takes_value(true)
|
|
.required(true),
|
|
)
|
|
.arg(Arg::with_name("local")
|
|
.long("local")
|
|
.help("Flag to indicate whether the client is expected to run on the local deployment.")
|
|
.takes_value(false)
|
|
)
|
|
.arg(Arg::with_name("id")
|
|
.long("id")
|
|
.help("Id of the nym-mixnet-client we want to run.")
|
|
.takes_value(true)
|
|
.required(true)
|
|
)
|
|
)
|
|
.subcommand(
|
|
SubCommand::with_name("websocket")
|
|
.about("Run Nym client that listens on a websocket")
|
|
.arg(
|
|
Arg::with_name("port")
|
|
.short("p")
|
|
.long("port")
|
|
.help("Port for websocket to listen on")
|
|
.takes_value(true)
|
|
)
|
|
.arg(Arg::with_name("local")
|
|
.long("local")
|
|
.help("Flag to indicate whether the client is expected to run on the local deployment.")
|
|
.takes_value(false)
|
|
)
|
|
.arg(Arg::with_name("id")
|
|
.long("id")
|
|
.help("Id of the nym-mixnet-client we want to run.")
|
|
.takes_value(true)
|
|
.required(true)
|
|
)
|
|
)
|
|
.get_matches();
|
|
|
|
if let Err(e) = execute(arg_matches) {
|
|
println!("{}", e);
|
|
process::exit(1);
|
|
}
|
|
}
|
|
|
|
fn execute(matches: ArgMatches) -> Result<(), String> {
|
|
match matches.subcommand() {
|
|
("init", Some(m)) => Ok(commands::init::execute(m)),
|
|
("run", Some(m)) => Ok(commands::run::execute(m)),
|
|
("tcpsocket", Some(m)) => Ok(commands::tcpsocket::execute(m)),
|
|
("websocket", Some(m)) => Ok(commands::websocket::execute(m)),
|
|
_ => Err(usage()),
|
|
}
|
|
}
|
|
|
|
fn usage() -> String {
|
|
banner() + "usage: --help to see available options.\n\n"
|
|
}
|
|
|
|
fn banner() -> String {
|
|
return r#"
|
|
|
|
_ __ _ _ _ __ ___
|
|
| '_ \| | | | '_ \ _ \
|
|
| | | | |_| | | | | | |
|
|
|_| |_|\__, |_| |_| |_|
|
|
|___/
|
|
|
|
(client)
|
|
|
|
"#
|
|
.to_string();
|
|
}
|