[go: up one dir, main page]

sysinfo 0.3.10

Library to handle processes
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134

// 
// Sysinfo
// 
// Copyright (c) 2015 Guillaume Gomez
//

use std::collections::HashMap;
use std::fs::{File, read_dir};
use std::io::Read;
use std::path::{Path, PathBuf};

/// More information can be found at
/// http://lxr.free-electrons.com/source/Documentation/hwmon/sysfs-interface
pub struct Component {
    /// Temperature is in celsius.
    pub temperature: f32,
    /// Temperature max value.
    pub max: f32,
    /// The highest temperature before the computer halts.
    pub critical: Option<f32>,
    /// Component's label.
    pub label: String,
    input_file: PathBuf,
}

fn get_file_line(file: &Path) -> Option<String> {
    let mut reader = String::new();
    if let Ok(mut f) = File::open(file) {
        f.read_to_string(&mut reader).unwrap();
        Some(reader)
    } else {
        None
    }
}

fn append_files(components: &mut Vec<Component>, folder: &Path) {
    let mut matchings = HashMap::new();
    if let Ok(dir) = read_dir(folder) {
        for entry in dir {
            if let Ok(entry) = entry {
                let entry = entry.path();
                if entry.is_dir() || !entry.file_name().unwrap().to_str()
                                           .unwrap_or("").starts_with("temp") {
                    continue;
                }
                if let Some(entry) = entry.file_name() {
                    let id = entry.to_str().unwrap()[4..5].parse::<u32>().unwrap();
                    matchings.entry(id)
                             .or_insert_with(|| Vec::with_capacity(1))
                             .push(entry.to_str().unwrap()[6..].to_owned());
                }
            }
        }
        for (key, val) in &matchings {
            let mut found_input = None;
            let mut found_label = None;
            for (pos, v) in val.iter().enumerate() {
                match v.as_str() {
                    "input" => { found_input = Some(pos); }
                    "label" => { found_label = Some(pos); }
                    _ => {}
                }
                if found_label.is_some() && found_input.is_some() {
                    let mut p_label = folder.to_path_buf();
                    let mut p_input = folder.to_path_buf();
                    let mut p_crit = folder.to_path_buf();
                    let mut p_max = folder.to_path_buf();

                    p_label.push(&format!("temp{}_label", key));
                    p_input.push(&format!("temp{}_input", key));
                    p_max.push(&format!("temp{}_max", key));
                    p_crit.push(&format!("temp{}_crit", key));
                    let content = get_file_line(p_label.as_path()).unwrap();
                    let label = content.replace("\n", "");
                    let max = if let Some(max) = get_file_line(p_max.as_path()) {
                        Some(max.replace("\n", "").parse::<f32>().unwrap() / 1000f32)
                    } else {
                        None
                    };
                    let crit = if let Some(crit) = get_file_line(p_crit.as_path()) {
                        Some(crit.replace("\n", "").parse::<f32>().unwrap() / 1000f32)
                    } else {
                        None
                    };
                    components.push(Component::new(label, p_input.as_path(), max, crit));
                    break
                }
            }
        }
    }
}

impl Component {
    /// Creates a new component with the given information.
    pub fn new(label: String, input_path: &Path, max: Option<f32>,
               critical: Option<f32>) -> Component {
        let mut c = Component {
            temperature: 0f32,
            label: label,
            input_file: input_path.to_path_buf(),
            max: max.unwrap_or(0.0),
            critical: critical,
        };
        c.update();
        c
    }

    /// Updates the component.
    pub fn update(&mut self) {
        let content = get_file_line(self.input_file.as_path()).unwrap();
        self.temperature = content.replace("\n", "").parse::<f32>().unwrap() / 1000f32;
        if self.temperature > self.max {
            self.max = self.temperature;
        }
    }
}

pub fn get_components() -> Vec<Component> {
    let mut ret = Vec::new();
    if let Ok(dir) = read_dir(&Path::new("/sys/class/hwmon/")) {
        for entry in dir {
            if let Ok(entry) = entry {
                let entry = entry.path();
                if !entry.is_dir() || !entry.file_name().unwrap().to_str()
                                            .unwrap_or("").starts_with("hwmon") {
                    continue;
                }
                append_files(&mut ret, &entry);
            }
        }
    }
    ret.sort_by(|c1, c2| c1.label.to_lowercase().cmp(&c2.label.to_lowercase()));
    ret
}