modularize

src/release.rs

@@ -0,0 +1,54 @@
+use color_eyre::Result;
+use std::fs::{self, read_to_string};
+use std::io::{self, Read};
+
+// Try to detect OS type as accurately as possible and without depending on uname.
+// /etc/os-release should generally imply Linux, and /etc/bsd-release would imply BSD system.
+fn detect_os() -> Result<String, io::Error> {
+    if fs::metadata("/etc/os-release").is_ok() || fs::metadata("/usr/lib/os-release").is_ok() {
+        Ok("Linux".to_string())
+    } else if fs::metadata("/etc/rc.conf").is_ok() || fs::metadata("/etc/bsd-release").is_ok() {
+        Ok("BSD".to_string())
+    } else {
+        Ok("Unknown".to_string())
+    }
+}
+
+fn get_architecture() -> Result<String, io::Error> {
+    // Read architecture from /proc/sys/kernel/arch
+    let mut arch = String::new();
+    fs::File::open("/proc/sys/kernel/arch")?.read_to_string(&mut arch)?;
+    let arch = arch.trim().to_string();
+    Ok(arch)
+}
+
+pub fn get_system_info() -> Result<String, io::Error> {
+    let system = detect_os()?;
+
+    let mut kernel_release = String::new();
+    fs::File::open("/proc/sys/kernel/osrelease")?.read_to_string(&mut kernel_release)?;
+    let kernel_release = kernel_release.trim().to_string();
+
+    let mut cpuinfo = String::new();
+    fs::File::open("/proc/cpuinfo")?.read_to_string(&mut cpuinfo)?;
+
+    let architecture = get_architecture()?;
+
+    let result = format!("{} {} ({})", system, kernel_release, architecture);
+    Ok(result)
+}
+
+pub fn get_os_pretty_name() -> Option<String> {
+    let os_release_content = read_to_string("/etc/os-release").ok()?;
+    let os_release_lines: Vec<&str> = os_release_content.lines().collect();
+    let pretty_name = os_release_lines
+        .iter()
+        .find(|line| line.starts_with("PRETTY_NAME="))
+        .map(|line| {
+            line.trim_start_matches("PRETTY_NAME=")
+                .trim_matches('"')
+                .to_string()
+        });
+
+    pretty_name
+}