swayr/swayrbar/src/config.rs

// Copyright (C) 2022  Tassilo Horn <tsdh@gnu.org>
//
// This program is free software: you can redistribute it and/or modify it
// under the terms of the GNU General Public License as published by the Free
// Software Foundation, either version 3 of the License, or (at your option)
// any later version.
//
// This program is distributed in the hope that it will be useful, but WITHOUT
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
// FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
// more details.
//
// You should have received a copy of the GNU General Public License along with
// this program.  If not, see <https://www.gnu.org/licenses/>.

//! TOML configuration for swayrbar.

use crate::module::BarModuleFn;
use crate::shared::cfg;
use serde::{Deserialize, Serialize};
use std::collections::HashMap;

#[derive(Debug, Serialize, Deserialize)]
pub struct Config {
    /// The status is refreshed every `refresh_interval` milliseconds.
    pub refresh_interval: u64,
    /// The list of modules to display in the given order, each one specified
    /// as `"<module_type>/<instance>"`.
    pub modules: Vec<ModuleConfig>,
}

#[derive(Debug, Serialize, Deserialize)]
pub struct ModuleConfig {
    pub name: String,
    pub instance: String,
    pub format: String,
    pub html_escape: Option<bool>,
    pub on_click: Option<HashMap<String, Vec<String>>>,
}

impl ModuleConfig {
    pub fn is_html_escape(&self) -> bool {
        self.html_escape.unwrap_or(false)
    }
}

impl Default for Config {
    fn default() -> Self {
        Config {
            refresh_interval: 1000,
            modules: vec![
                crate::module::window::BarModuleWindow::default_config(
                    "0".to_owned(),
                ),
                crate::module::sysinfo::BarModuleSysInfo::default_config(
                    "0".to_owned(),
                ),
                crate::module::battery::BarModuleBattery::default_config(
                    "0".to_owned(),
                ),
                crate::module::date::BarModuleDate::default_config(
                    "0".to_owned(),
                ),
            ],
        }
    }
}

pub fn load_config() -> Config {
    cfg::load_config::<Config>("swayrbar")
}

#[test]
fn test_load_swayrbar_config() {
    let cfg = cfg::load_config::<Config>("swayrbar");
    println!("{:?}", cfg);
}