fanctl 0.6.0

Fancontrol replacement in Rust with YAML configuration
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

use std::fs;
use std::io;
use std::path::{Path, PathBuf};

use super::Fan;
use super::util::{ReadFileError, read_file_value};

#[derive(Debug, Clone)]
pub struct AmdgpuFan {
    enable_path: PathBuf,
    target_path: PathBuf,
    min_path: PathBuf,
    max_path: PathBuf,
}

pub enum InitializationError {
    Serde(serde_yaml::Error),
    Io(io::Error),
}

impl AmdgpuFan {
    pub fn new<P: AsRef<Path>, S: AsRef<str>>(base_path: P, name: S) -> Self {
        let base_path = base_path.as_ref();
        let name = name.as_ref();

        let make_path = |filename: &str| {
            let mut path = base_path.to_owned();
            path.push(format!("{}_{}", name, filename));
            path
        };

        AmdgpuFan {
            enable_path: make_path("enable"),
            target_path: make_path("target"),
            min_path: make_path("min"),
            max_path: make_path("max")
        }
    }

    pub fn min(&self) -> io::Result<u64> {
        read_file_value(&self.min_path, 8).map_err(|e| match e {
            ReadFileError::Io(e) => e,
            ReadFileError::Parse(e) => panic!("{:?}", e),
        })
    }

    pub fn max(&self) -> io::Result<u64> {
        read_file_value(&self.max_path, 8).map_err(|e| match e {
            ReadFileError::Io(e) => e,
            ReadFileError::Parse(e) => panic!("{:?}", e),
        })
    }

    pub fn enabled(&self) -> io::Result<bool> {
        read_file_value::<u8, _>(&self.enable_path, 8)
            .map(|v| v > 0)
            .map_err(|e| match e {
                ReadFileError::Io(e) => e,
                ReadFileError::Parse(e) => panic!("{:?}", e),
            })
    }

    fn set_target(&mut self, target: u64) -> io::Result<()> {
        use io::Write;
        let mut file = fs::OpenOptions::new()
            .write(true)
            .open(&self.target_path)?;
        write!(file, "{}", target)
    }
}

impl Fan for AmdgpuFan {
    fn set_enabled(&mut self, enabled: bool) -> io::Result<()> {
        use io::Write;

        let value = if enabled {
            "1"
        } else {
            "0"
        };
        let mut file = fs::OpenOptions::new()
            .write(true)
            .open(&self.enable_path)?;
        write!(file, "{}", value)
    }

    fn set_value(&mut self, value: f64) -> io::Result<()> {
        let min = self.min()?;
        let max = self.max()?;
        let target = if value >= 1.0 {
            max
        } else if value <= 0.0 {
            min
        } else {
            let mut target = (value * ((max - min) as f64)).round() as u64;
            target += min;
            target
        };
        self.set_target(target)
    }

    fn close(&mut self) -> io::Result<()> {
        use io::Write;
        fs::OpenOptions::new()
            .write(true)
            .open(&self.enable_path)
            .and_then(|mut f| write!(f, "0"))
    }
}