hostab 0.0.3

Your dev tool to manage /etc/hosts like a pro — written in Rust
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

use clap::Parser;

mod cli;
mod core;
mod output;
mod store;
mod util;

use cli::Shell;
use cli::{Cli, Commands, EntryCommands};

fn main() {
    let cli = Cli::parse();

    if cli.hosts_file.to_string_lossy().contains("etc/hosts") && !util::validation::is_root() {
        match &cli.command {
            Commands::Entry(EntryCommands::List { .. })
            | Commands::Resolve { .. }
            | Commands::Verify { .. }
            | Commands::Merge {
                target: Some(_), ..
            }
            | Commands::Completion { .. }
            | Commands::Version
            | Commands::Cat
            | Commands::Serve { .. } => {}
            _ => eprintln!(
                "Warning: modifying {} may require root privileges.",
                cli.hosts_file.display()
            ),
        }
    }

    match &cli.command {
        Commands::Entry(cmd) => match cmd {
            EntryCommands::List {
                ipv4,
                ipv6,
                pattern,
                ignore_case,
            } => {
                cli::entry::handle_list(&cli, *ipv4, *ipv6, pattern.as_deref(), *ignore_case);
            }
            EntryCommands::Add {
                ip,
                hosts,
                canonical,
                alias,
                comment,
            } => cli::entry::handle_add(
                &cli,
                ip,
                hosts,
                canonical.as_deref(),
                alias,
                comment.as_deref(),
            ),
            EntryCommands::Rm { hosts, ip } => cli::entry::handle_rm(&cli, hosts, ip.as_deref()),
            EntryCommands::Disable { hosts, ip } => {
                cli::entry::handle_disable(&cli, hosts, ip.as_deref())
            }
            EntryCommands::Enable { hosts, ip } => {
                cli::entry::handle_enable(&cli, hosts, ip.as_deref())
            }
            EntryCommands::Toggle { host, ip } => {
                cli::entry::handle_toggle(&cli, host, ip.as_deref())
            }
            EntryCommands::Edit { host, ip } => cli::entry::handle_edit(&cli, host, ip),
        },
        Commands::Serve {
            port,
            bind,
            no_docs,
        } => cli::serve::handle(&cli, *port, bind, *no_docs),
        Commands::Verify { strict } => cli::verify::handle_verify(&cli, *strict),
        Commands::Merge { src, target } => {
            cli::merge::handle(&cli, src, target.as_ref());
        }
        Commands::Resolve { hosts, local } => cli::resolve::handle(&cli, hosts, *local),
        Commands::Cat => {
            if let Err(e) = util::validation::validate_secure_path(&cli.hosts_file) {
                eprintln!("Error: {}", e);
                std::process::exit(1);
            }
            match std::fs::read_to_string(&cli.hosts_file) {
                Ok(c) => print!("{}", c),
                Err(e) => {
                    eprintln!("Error: {}", e);
                    std::process::exit(1);
                }
            }
        }
        Commands::Completion { shell } => {
            use clap::CommandFactory;
            use clap_complete::{generate, shells};
            let mut cmd = Cli::command();
            match shell {
                Shell::Bash => generate(shells::Bash, &mut cmd, "hostab", &mut std::io::stdout()),
                Shell::Zsh => generate(shells::Zsh, &mut cmd, "hostab", &mut std::io::stdout()),
                Shell::Fish => generate(shells::Fish, &mut cmd, "hostab", &mut std::io::stdout()),
            }
        }
        Commands::Version => {
            println!("hostab {}", env!("CARGO_PKG_VERSION"));
            println!("  commit:  {}", env!("GIT_HASH"));
            println!("  built:   {}", env!("BUILD_DATE"));
            println!("  target:  {}", env!("TARGET_TRIPLE"));
        }
    }
}