mirror of
https://github.com/mii443/encrypt.git
synced 2025-08-22 15:05:33 +00:00
make args simply
This commit is contained in:
15
src/args.rs
15
src/args.rs
@ -1,17 +1,14 @@
|
|||||||
|
use std::path::PathBuf;
|
||||||
|
|
||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
|
use clap::ValueHint;
|
||||||
|
|
||||||
#[derive(Parser, Debug)]
|
#[derive(Parser, Debug)]
|
||||||
#[clap(author, version, about, long_about = None)]
|
#[clap(author, version, about, long_about = None)]
|
||||||
pub struct Args {
|
pub struct Args {
|
||||||
#[clap(short, long, value_parser)]
|
|
||||||
pub mode: String,
|
|
||||||
|
|
||||||
#[clap(short, long, value_parser)]
|
|
||||||
pub file: Option<String>,
|
|
||||||
|
|
||||||
#[clap(short, long, value_parser)]
|
|
||||||
pub ip: Option<String>,
|
|
||||||
|
|
||||||
#[clap(short, long, value_parser)]
|
#[clap(short, long, value_parser)]
|
||||||
pub port: Option<u16>,
|
pub port: Option<u16>,
|
||||||
|
|
||||||
|
#[clap(name = "FILE", value_hint = ValueHint::AnyPath)]
|
||||||
|
pub file: Option<PathBuf>,
|
||||||
}
|
}
|
||||||
|
14
src/main.rs
14
src/main.rs
@ -16,15 +16,9 @@ fn main() {
|
|||||||
env_logger::init();
|
env_logger::init();
|
||||||
let args = Args::parse();
|
let args = Args::parse();
|
||||||
|
|
||||||
match &*args.mode {
|
if let Some(_) = args.port {
|
||||||
"server" => {
|
start_server(args);
|
||||||
start_server(args);
|
} else {
|
||||||
}
|
start_client(args);
|
||||||
"client" => {
|
|
||||||
start_client(args);
|
|
||||||
}
|
|
||||||
_ => {
|
|
||||||
println!("Unknown mode");
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user