wallheaven 1.1.1

Random image fetcher for wallhaven.cc
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

use std::{collections::HashMap, env};

use anyhow::Result;
use serde::Deserialize;

#[derive(Debug, Deserialize)]
struct Command {
    command: String,
}

#[derive(Debug, Deserialize)]
struct Config {
    linux: HashMap<String, Command>,
    macos: Command,
}

fn load_commands() -> Config {
    let commands_toml = include_str!("setcommands.toml");
    toml::from_str(commands_toml).expect("Failed to parse commands")
}

pub fn set_wallpaper(filepath: &str) -> Result<()> {
    let config = load_commands();
    let os = env::consts::OS;

    if os == "linux" {
        let desktop_session = env::var("DESKTOP_SESSION").unwrap_or_default();
        if let Some(config_command) = config.linux.get(&desktop_session) {
            let cmd_string = config_command.command.replace("{IMG}", filepath);
            let status = std::process::Command::new("sh")
                .arg("-c")
                .arg(cmd_string)
                .status()
                .expect("Failed to set wallpaper");
            if !status.success() {
                println!("{}", filepath);
            }
        } else {
            println!("{}", filepath);
        }
        return Ok(());
    } else if os == "macos" {
        let cmd_string = config.macos.command.replace("{IMG}", filepath);
        let status = std::process::Command::new("sh")
            .arg("-c")
            .arg(cmd_string)
            .status()
            .expect("Failed to set wallpaper");
        if !status.success() {
            println!("{}", filepath);
        }
    } else {
        println!("{}", filepath);
    }
    return Ok(());
}