cf-modkit-node-info 0.6.2

ModKit node info library
Documentation 
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

use crate::error::NodeInfoError;
use crate::model::{Node, NodeSysCap, NodeSysInfo};
use crate::syscap_collector::SysCapCollector;
use crate::sysinfo_collector::SysInfoCollector;
use std::sync::Arc;

/// Main collector for node information
pub struct NodeInfoCollector {
    sysinfo_collector: Arc<SysInfoCollector>,
    syscap_collector: Arc<SysCapCollector>,
}

impl NodeInfoCollector {
    #[must_use]
    pub fn new() -> Self {
        let sysinfo_collector = Arc::new(SysInfoCollector::new());
        Self {
            syscap_collector: Arc::new(SysCapCollector::new(Arc::clone(&sysinfo_collector))),
            sysinfo_collector,
        }
    }

    /// Create a Node instance for the current machine
    /// Uses hardware UUID for node ID and collects hostname and local IP
    #[must_use]
    pub fn create_current_node() -> Node {
        let id = crate::get_hardware_uuid();
        let hostname = sysinfo::System::host_name().unwrap_or_else(|| "unknown".to_owned());
        let ip_address = Self::detect_local_ip();
        let now = chrono::Utc::now();

        Node {
            id,
            hostname,
            ip_address,
            created_at: now,
            updated_at: now,
        }
    }

    /// Detect the local IP address used for the default route to the internet
    /// This returns the IP address of the network interface that would be used
    /// for outbound internet traffic, not the public external IP.
    fn detect_local_ip() -> Option<String> {
        match local_ip_address::local_ip() {
            Ok(ip) => {
                let ip_str = ip.to_string();
                tracing::debug!(ip = %ip_str, "Detected local IP address");
                Some(ip_str)
            }
            Err(e) => {
                tracing::warn!(error = %e, "Failed to detect local IP address");
                None
            }
        }
    }

    /// Collect system information for the current node.
    ///
    /// # Errors
    /// Returns `NodeInfoError::SysInfoCollectionFailed` if system info collection fails.
    pub fn collect_sysinfo(&self, node_id: uuid::Uuid) -> Result<NodeSysInfo, NodeInfoError> {
        self.sysinfo_collector
            .collect(node_id)
            .map_err(|e| NodeInfoError::SysInfoCollectionFailed(e.to_string()))
    }

    /// Collect system capabilities for the current node.
    ///
    /// # Errors
    /// Returns `NodeInfoError::SysCapCollectionFailed` if capability collection fails.
    pub fn collect_syscap(&self, node_id: uuid::Uuid) -> Result<NodeSysCap, NodeInfoError> {
        self.syscap_collector
            .collect(node_id)
            .map_err(|e| NodeInfoError::SysCapCollectionFailed(e.to_string()))
    }

    /// Collect both sysinfo and syscap.
    ///
    /// # Errors
    /// Returns `NodeInfoError::SysInfoCollectionFailed` if system info collection fails.
    /// Returns `NodeInfoError::SysCapCollectionFailed` if capability collection fails.
    pub fn collect_all(
        &self,
        node_id: uuid::Uuid,
    ) -> Result<(NodeSysInfo, NodeSysCap), NodeInfoError> {
        let sysinfo = self.collect_sysinfo(node_id)?;
        let syscap = self.collect_syscap(node_id)?;
        Ok((sysinfo, syscap))
    }
}

impl Default for NodeInfoCollector {
    fn default() -> Self {
        Self::new()
    }
}