add args

Cargo.toml

@@ -9,3 +9,8 @@ edition = "2021"
 serde_json = "1.0.105"
 serde = {version = "1.0.183", features = ["derive"]}
 curl = "0.4.44"
+clap = { version = "4.4.3", features = ["derive"] }
+
+[profile.release]
+strip=true
+lto=true

src/main.rs

@@ -1,17 +1,33 @@
-use std::env;
 use std::io::Error as IoError;
-use std::path::Path;
 use std::process::exit;
 
+use clap::Parser;
+
 mod docker;
 mod hosts;
 use hosts::Hosts;
 
-const DOCKER_SOCKET: &str = "/var/run/docker.sock";
-const HOSTS: &str = "/etc/hosts";
+/// Generates /etc/hosts entries for all docker containers
+#[derive(Parser, Debug)]
+#[command(
+ version,
+ about,
+ long_about = None
+)]
+struct Args {
+ /// Unix socket path to docker
+ #[arg(long, default_value_t = String::from("/var/run/docker.sock"))]
+ docker_socket: String,
+
+ /// Path to hosts
+ #[arg(long, default_value_t = String::from("/etc/hosts"))]
+ hosts: String,
+}
 
 fn main() -> Result<(), IoError> {
- let lister = docker::ContainerLister::new(DOCKER_SOCKET);
+ let args = Args::parse();
+
+ let lister = docker::ContainerLister::new(args.docker_socket.as_str());
  let container_entries: Vec<String> = match lister.fetch() {
  Ok(v) => v
  .iter()
@@ -27,7 +43,7 @@ fn main() -> Result<(), IoError> {
  exit(0);
  }
 
- let mut hosts = match Hosts::new(HOSTS) {
+ let mut hosts = match Hosts::new(args.hosts) {
  Ok(v) => v,
  Err(e) => {
  eprintln!("Error: could not open hosts: {:?}", e);