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
135
136
137
138
139
140
141
142
143
144
145
146

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

use std::collections::HashMap;
use std::ffi::OsStr;
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) {
        if f.read_to_string(&mut reader).is_ok() {
            Some(reader)
        } else {
            None
        }
    } 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_or_else(|| OsStr::new("/")).to_str()
                                           .unwrap_or("").starts_with("temp") {
                    continue;
                }
                if let Some(entry) = entry.file_name() {
                    if let Some(entry) = entry.to_str() {
                        if let Ok(id) = entry[4..5].parse::<u32>() {
                            matchings.entry(id)
                                     .or_insert_with(|| Vec::with_capacity(1))
                                     .push(entry[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));
                    if let Some(content) = get_file_line(p_label.as_path()) {
                        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_or(100_000f32) / 1000f32)
                        } else {
                            None
                        };
                        let crit = if let Some(crit) = get_file_line(p_crit.as_path()) {
                            Some(crit.replace("\n", "")
                                     .parse::<f32>().unwrap_or(100_000f32) / 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) {
        if let Some(content) = get_file_line(self.input_file.as_path()) {
            self.temperature = content.replace("\n", "")
                                      .parse::<f32>().unwrap_or(100_000f32) / 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_or_else(|| OsStr::new("/")).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
}