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
use crate::{resources, squire};
use std::fs::File;
use std::io::{self, BufRead};

/// Function to get processor information.
///
/// # Arguments
///
/// * `lib_path` - The path to the library used to get processor information.
///
/// # Returns
///
/// A `Option` containing the processor information if successful, otherwise `None`.
fn get_processor_info_darwin(lib_path: &str) -> Result<String, &'static str> {
    let result = squire::util::run_command(lib_path, &["-n", "machdep.cpu.brand_string"]);
    if result.is_err() {
        return Err("Failed to get processor info");
    }
    Ok(result.unwrap())
}

/// Function to get processor information on Linux.
///
/// # Arguments
///
/// * `lib_path` - The path to the library used to get processor information.
///
/// # Returns
///
/// A `Option` containing the processor information if successful, otherwise `None`.
fn get_processor_info_linux(lib_path: &str) -> Result<String, &'static str> {
    let file = match File::open(lib_path) {
        Ok(file) => file,
        Err(_) => return Err("Failed to open file"),
    };
    for line in io::BufReader::new(file).lines() {
        match line {
            Ok(line) => {
                if line.contains("model name") {
                    let parts: Vec<&str> = line.split(':').collect();
                    if parts.len() == 2 {
                        return Ok(parts[1].trim().to_string());
                    }
                }
            }
            Err(_) => return Err("Error reading line"),
        }
    }
    Err("Model name not found")
}

/// Function to get processor information on Windows.
///
/// # Arguments
///
/// * `lib_path` - The path to the library used to get processor information.
///
/// # Returns
///
/// A `Option` containing the processor information if successful, otherwise `None`.
fn get_processor_info_windows(lib_path: &str) -> Result<String, &'static str> {
    let result = squire::util::run_command(lib_path, &["cpu", "get", "name"]);
    let output = match result {
        Ok(output) => output,
        Err(_) => return Err("Failed to get processor info"),
    };
    let lines: Vec<&str> = output.trim().split('\n').collect();
    if lines.len() > 1 {
        Ok(lines[1].trim().to_string())
    } else {
        Err("Invalid output from command")
    }
}

/// OS-agnostic function to get processor name.
///
/// # Returns
///
/// A `Option` containing the processor name if successful, otherwise `None`.
pub fn get_name() -> Option<String> {
    let os = resources::system::os_arch().name;
    let result = match os.as_str() {
        "darwin" => get_processor_info_darwin("/usr/sbin/sysctl"),
        "linux" => get_processor_info_linux("/proc/cpuinfo"),
        "windows" => get_processor_info_windows("C:\\Windows\\System32\\wbem\\wmic.exe"),
        _ => {
            log::error!("Unsupported operating system: {}", os);
            Err("Unsupported operating system")
        },
    };
    match result {
        Ok(info) => Some(info),
        Err(err) => {
            log::error!("{}", err);
            None
        }
    }
}