config.rs

use crate::{lapis::Lapis, objects::AttractionFactor, ui::ScaleFactor};
use avian2d::prelude::Gravity;
use bevy::{prelude::*, window::WindowMode};
use clap::Parser;
use figment::{
    providers::{Format, Serialized, Toml},
    Figment,
};
use serde::{Deserialize, Serialize};
use xdg::BaseDirectories;

pub struct ConfigPlugin;

#[derive(Parser, Debug, Resource, Serialize, Deserialize)]
#[command(version, about, long_about = None)]
pub struct Config {
    /// start with paused time
    #[arg(long, default_value_t = false)]
    pub pause: bool,

    /// start in full screen mode
    #[arg(long, default_value_t = false)]
    pub fullscreen: bool,

    /// enable lapis quiet evaluation
    #[arg(long, default_value_t = false)]
    pub lapis_quiet: bool,

    /// enable lapis keybindings
    #[arg(long, default_value_t = false)]
    pub lapis_keys: bool,

    #[arg(long, default_value_t = 0.0)]
    pub gravity_x: f32,

    #[arg(long, default_value_t = 0.0)]
    pub gravity_y: f32,

    #[arg(long, default_value_t = 0.01)]
    pub attraction: f32,

    #[arg(long, default_value_t = 1.0)]
    pub scale_factor: f32,

    #[arg(long, default_value_t = 1280.0)]
    pub win_width: f32,

    #[arg(long, default_value_t = 720.0)]
    pub win_height: f32,

    /// hex code
    #[arg(long, default_value_t = String::from("000000"))]
    pub clear_color: String,
}

impl Plugin for ConfigPlugin {
    fn build(&self, app: &mut App) {
        let mut config: Config = Figment::new()
            .merge(Serialized::defaults(Config::parse()))
            .extract()
            .unwrap();

        if let Ok(xdg_dirs) = BaseDirectories::with_prefix("bgawk") {
            if let Ok(config_path) = xdg_dirs.place_config_file("config.toml") {
                if let Ok(figment) = Figment::new()
                    .merge(Serialized::defaults(Config::parse()))
                    .merge(Toml::file(config_path))
                    .extract()
                {
                    config = figment;
                }
            }
        }

        app.insert_resource(config)
            .add_systems(PostStartup, configure);
    }
}

fn configure(
    config: Res<Config>,
    mut gravity: ResMut<Gravity>,
    mut attraction_factor: ResMut<AttractionFactor>,
    mut scale_factor: ResMut<ScaleFactor>,
    mut win: Query<&mut Window>,
    mut clear_color: ResMut<ClearColor>,
    mut lapis: Lapis,
) {
    if config.pause {
        lapis.time.pause();
    }

    gravity.0.x = config.gravity_x;
    gravity.0.y = config.gravity_y;

    attraction_factor.0 = config.attraction;

    scale_factor.0 = config.scale_factor;
    let res = &mut win.single_mut().resolution;
    res.set_scale_factor(config.scale_factor);
    res.set(config.win_width, config.win_height);

    if config.fullscreen {
        win.single_mut().mode = WindowMode::Fullscreen(MonitorSelection::Current);
    }

    if let Ok(color) = Srgba::hex(config.clear_color.clone()) {
        clear_color.0 = color.into();
    }

    lapis.data.keys_active = config.lapis_keys;
    lapis.data.quiet = config.lapis_quiet;
}