From 763ba1de91727cd8132deb8c2f8e60709ba9ec9a Mon Sep 17 00:00:00 2001
From: Razvan Dimescu <ssaricu@gmail.com>
Date: Mon, 23 Mar 2026 10:30:22 +0200
Subject: [PATCH] add numa lan on/off CLI command, update README

- numa lan on/off toggles LAN discovery in numa.toml
- Writes [lan] section if missing, updates enabled if present
- Colored output with restart hint
- README: add lan on/off to help text

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
---
 src/main.rs | 71 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 71 insertions(+)

diff --git a/src/main.rs b/src/main.rs
index ed661a9..678a8bf 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -50,6 +50,17 @@ async fn main() -> numa::Result<()> {
                 }
             };
         }
+        "lan" => {
+            let sub = std::env::args().nth(2).unwrap_or_default();
+            return match sub.as_str() {
+                "on" => set_lan_enabled(true),
+                "off" => set_lan_enabled(false),
+                _ => {
+                    eprintln!("Usage: numa lan <on|off>");
+                    Ok(())
+                }
+            };
+        }
         "version" | "--version" | "-V" => {
             eprintln!("numa {}", env!("CARGO_PKG_VERSION"));
             return Ok(());
@@ -65,6 +76,8 @@ async fn main() -> numa::Result<()> {
             eprintln!("  service stop    Uninstall the system service");
             eprintln!("  service restart Restart the service with updated binary");
             eprintln!("  service status  Check if the service is running");
+            eprintln!("  lan on          Enable LAN service discovery (mDNS)");
+            eprintln!("  lan off         Disable LAN service discovery");
             eprintln!("  help            Show this help");
             eprintln!();
             eprintln!("Config path defaults to numa.toml");
@@ -338,6 +351,64 @@ async fn network_watch_loop(ctx: Arc<numa::ctx::ServerCtx>) {
     }
 }
 
+fn set_lan_enabled(enabled: bool) -> numa::Result<()> {
+    let path = "numa.toml";
+
+    if std::path::Path::new(path).exists() {
+        let contents = std::fs::read_to_string(path)?;
+
+        // Track current TOML section while scanning lines
+        let mut in_lan = false;
+        let mut found = false;
+        let mut lines: Vec<String> = contents
+            .lines()
+            .map(|line| {
+                let trimmed = line.trim();
+                if trimmed.starts_with('[') {
+                    in_lan = trimmed == "[lan]";
+                }
+                if in_lan && !found && trimmed.starts_with("enabled") && trimmed.contains('=') {
+                    found = true;
+                    return format!("enabled = {}", enabled);
+                }
+                line.to_string()
+            })
+            .collect();
+
+        let has_lan_section = lines.iter().any(|l| l.trim() == "[lan]");
+        if !found && has_lan_section {
+            // [lan] exists but no enabled line — insert after it
+            if let Some(i) = lines.iter().position(|l| l.trim() == "[lan]") {
+                lines.insert(i + 1, format!("enabled = {}", enabled));
+            }
+        } else if !found {
+            // No [lan] section — append
+            lines.push(String::new());
+            lines.push("[lan]".to_string());
+            lines.push(format!("enabled = {}", enabled));
+        }
+
+        let mut result = lines.join("\n");
+        if contents.ends_with('\n') && !result.ends_with('\n') {
+            result.push('\n');
+        }
+        std::fs::write(path, result)?;
+    } else {
+        std::fs::write(path, format!("[lan]\nenabled = {}\n", enabled))?;
+    }
+
+    let label = if enabled { "enabled" } else { "disabled" };
+    let color = if enabled { "32" } else { "33" };
+    eprintln!(
+        "\x1b[1;38;2;192;98;58mNuma\x1b[0m — LAN discovery \x1b[{}m{}\x1b[0m",
+        color, label
+    );
+    if enabled {
+        eprintln!("  Restart Numa to start mDNS discovery");
+    }
+    Ok(())
+}
+
 async fn load_blocklists(ctx: &ServerCtx, lists: &[String]) {
     let downloaded = download_blocklists(lists).await;