From b706f78a8544fd2406e4a3ddcbe37680b4d25f2f Mon Sep 17 00:00:00 2001
From: Artur Meski <artur@meski.io>
Date: Sun, 22 Dec 2024 22:47:15 +0000
Subject: [PATCH] Daemonize

---
 Cargo.lock  | 16 ++++++++++++++++
 Cargo.toml  |  1 +
 src/main.rs | 25 ++++++++++++++-----------
 3 files changed, 31 insertions(+), 11 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index 6ba24cd..3c9c213 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -97,11 +97,21 @@ version = "1.0.3"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "5b63caa9aa9397e2d9480a9b13673856c78d8ac123288526c37d7839f2a86990"
 
+[[package]]
+name = "daemonize"
+version = "0.5.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ab8bfdaacb3c887a54d41bdf48d3af8873b3f5566469f8ba21b92057509f116e"
+dependencies = [
+ "libc",
+]
+
 [[package]]
 name = "dhcpleasemon"
 version = "0.1.0"
 dependencies = [
  "clap",
+ "daemonize",
 ]
 
 [[package]]
@@ -116,6 +126,12 @@ version = "1.70.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "7943c866cc5cd64cbc25b2e01621d07fa8eb2a1a23160ee81ce38704e97b8ecf"
 
+[[package]]
+name = "libc"
+version = "0.2.169"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b5aba8db14291edd000dfcc4d620c7ebfb122c613afb886ca8803fa4e128a20a"
+
 [[package]]
 name = "proc-macro2"
 version = "1.0.89"
diff --git a/Cargo.toml b/Cargo.toml
index cea3294..034f9e6 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -5,3 +5,4 @@ edition = "2021"
 
 [dependencies]
 clap = { version = "4.5.22", features = ["derive"] }
+daemonize = "0.5.0"
diff --git a/src/main.rs b/src/main.rs
index 25868a7..0da5bc3 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -7,6 +7,7 @@ use std::process::Command;
 use std::thread::sleep;
 use std::time::Duration;
 use std::time::{SystemTime, UNIX_EPOCH};
+use daemonize::Daemonize;
 
 #[derive(Parser, Debug, Clone)]
 #[command(version, about, long_about = None)]
@@ -55,8 +56,6 @@ impl Monitor {
 
     /// Was the file modified since the last check?
     fn check_file_modified(&mut self, lease_file_path: &str) -> bool {
-        println!("Checking: file {}", lease_file_path);
-
         let metadata = fs::metadata(&lease_file_path);
         let current_timestamp = metadata
             .expect("Unsupported platform")
@@ -80,12 +79,13 @@ impl Monitor {
         false
     }
 
-    /// Generates the lease file path
+    /// Generates the lease file path for a given interface
     fn get_lease_file_path(&self, iface_name: &str) -> String {
         let dhcp_lease_dir = &self.args.dhcp_lease_dir;
         format!("{dhcp_lease_dir}/{iface_name}")
     }
 
+    /// Generates the trigger script path for a given interface
     fn get_trigger_script_path(&self, iface_name: &str) -> String {
         let trigger_scripts_path = &self.args.scripts_dir;
         format!("{trigger_scripts_path}/lease_trigger_{iface_name}")
@@ -118,7 +118,6 @@ impl Monitor {
                 let route_dest = cols[0];
                 if route_iface == iface_name && route_dest == "default" {
                     let route_ip = cols[1];
-                    println!("{} {}", route_dest, route_ip);
                     return Some(route_ip.to_string());
                 }
             }
@@ -153,10 +152,6 @@ impl Monitor {
             .unwrap_or(String::from(""));
 
         let trigger_script_path = self.get_trigger_script_path(iface_name);
-        println!(
-            "EXEC: {} {} -> script |{}| IP: {:?} Route: {}",
-            iface_name, lease_file_path, trigger_script_path, lease_ip_addr, default_route
-        );
 
         let output = Command::new(trigger_script_path)
             .env("DHCP_IFACE", iface_name)
@@ -176,13 +171,10 @@ impl Monitor {
             for iface_name in self.args.interfaces.clone() {
                 let lease_file_path = self.get_lease_file_path(&iface_name);
                 if self.check_file_modified(&lease_file_path) {
-                    println!("Was modified!");
                     self.trigger_script(&iface_name, &lease_file_path);
                 }
             }
             sleep(Duration::new(self.args.interval.into(), 0));
-            println!("{:?}", self.args);
-            println!("hello");
         }
     }
 }
@@ -195,6 +187,17 @@ fn main() {
         panic!("No interfaces to monitor");
     }
 
+    let daemonize = Daemonize::new()
+        .pid_file(&args.pid_file);
+
+    match daemonize.start() {
+        Ok(_) => {},
+        Err(e) => {
+            eprintln!("Error: {}", e);
+            return;
+        },
+    }
+
     monitor.run();
 }