hostcraft-cli 1.0.1

CLI for managing your system hosts file
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

mod utils;

use crate::display::{print_entries, print_success};
use clap::{Parser, Subcommand};
use hostcraft_core::{HostError, file, host};
use std::{error::Error, net::IpAddr};
use utils::write_hosts;

const HOSTS_FILE: &str = if cfg!(target_os = "windows") {
    r"C:\Windows\System32\drivers\etc\hosts"
} else {
    "/etc/hosts"
};

#[derive(Parser)]
#[command(
    name = "hostcraft",
    about = "Manage your system hosts file from the terminal",
    version
)]
pub struct Cli {
    /// Path to the hosts file (override for testing or non-standard setups)
    #[arg(long, default_value = HOSTS_FILE)]
    pub file: String,

    #[command(subcommand)]
    pub command: Command,
}

#[derive(Subcommand)]
pub enum Command {
    /// List all host entries
    List,

    /// Add a new host entry
    Add {
        /// The hostname to add (e.g. myapp.local)
        name: String,

        /// The IP address to associate with the hostname (e.g. 127.0.0.1)
        ip: IpAddr,
    },

    /// Remove a host entry by name (partial match supported)
    Remove {
        /// The hostname to remove (partial match supported)
        name: String,
    },

    /// Toggle a host entry on or off by name (partial match supported)
    Toggle {
        /// The hostname to toggle (partial match supported)
        name: String,
    },
}

pub fn run(cli: Cli) -> Result<(), Box<dyn Error>> {
    let lines = file::read_file(&cli.file)
        .map_err(|e| format!("Failed to read hosts file '{}': {}", cli.file, e))?;

    let mut entries = host::parse_contents(lines);

    match cli.command {
        Command::List => {
            print_entries(&entries);
        }

        Command::Add { name, ip } => {
            host::add_entry(&mut entries, ip, name.as_str()).map_err(|e| match e {
                HostError::DuplicateEntry => format!("Entry already exists. {}", e),
                _ => format!("Failed to add entry: {}", e),
            })?;

            write_hosts(&cli.file, &entries)?;

            print_success(&format!("Added '{}'", name));
            print_entries(&entries);
        }

        Command::Remove { name } => {
            host::remove_entry(&mut entries, &name).map_err(|e| match e {
                HostError::EntryNotFound => format!("No entry found matching '{}'. {}", name, e),
                _ => format!("Failed to remove entry: {}", e),
            })?;

            write_hosts(&cli.file, &entries)?;

            print_success(&format!("Removed '{}'", name));
            print_entries(&entries);
        }

        Command::Toggle { name } => {
            host::toggle_entry(&mut entries, &name).map_err(|e| match e {
                HostError::EntryNotFound => format!("No entry found matching '{}'. {}", name, e),
                _ => format!("Failed to toggle entry: {}", e),
            })?;

            write_hosts(&cli.file, &entries)?;

            print_success(&format!("Toggled '{}'", name));
            print_entries(&entries);
        }
    }

    Ok(())
}