rskit 0.0.37

rust kit
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

#[cfg(target_os = "linux")]
pub fn is_running(process: &str) -> bool {
    use std::process::Command;
    let output = Command::new("pgrep")
        .arg(process)
        .output()
        .expect("failed to execute pgrep");

    !output.stdout.is_empty()
}

pub fn is_running_current() -> bool {
    let pid = std::process::id();
    let exe = std::env::current_exe().unwrap();
    let exe_name = exe.to_str().unwrap();
    let system = sysinfo::System::new_all();
    for ele in system.processes().values() {
        if ele.name().to_str().unwrap() == exe_name && ele.pid().as_u32() != pid {
            return true;
        }
    }
    return false;
}

#[cfg(target_os = "windows")]
pub fn set_windows_startup(name: &str) -> Result<(), anyhow::Error> {
    let exe = std::env::current_exe()?;
    let exe_name = exe.to_str().unwrap();
    let hkcu = winreg::RegKey::predef(winreg::enums::HKEY_CURRENT_USER);
    let (run, _) = hkcu.create_subkey(r"Software\Microsoft\Windows\CurrentVersion\Run")?;
    run.set_value(name, &exe_name)?;
    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_is_running_current() {
        assert_eq!(false, is_running_current());
    }

    #[test]
    #[cfg(target_os = "windows")]
    fn test_set_windows_startup() {
        set_windows_startup("test").unwrap();
    }
}