yazi/src/config/preset.rs

use std::fs;

use toml::Table;
use xdg::BaseDirectories;

pub(crate) struct Preset;

impl Preset {
	fn merge(a: &mut Table, b: &Table, max: u8) {
		for (k, v) in b {
			let a = if let Some(a) = a.get_mut(k) {
				a
			} else {
				a.insert(k.clone(), v.clone());
				continue;
			};

			if k == "icons" || max <= 1 {
				continue;
			}

			if let Some(a) = a.as_table_mut() {
				if let Some(b) = v.as_table() {
					Self::merge(a, b, max - 1);
					continue;
				}
			}
			*a = v.clone();
		}
	}

	fn merge_str(user: &str, base: &str) -> String {
		let path = BaseDirectories::new().unwrap().get_config_file(user);
		let mut user = fs::read_to_string(path).unwrap_or("".to_string()).parse::<Table>().unwrap();

		let base = base.parse::<Table>().unwrap();
		Self::merge(&mut user, &base, 2);
		user.to_string()
	}

	#[inline]
	pub(crate) fn keymap() -> String {
		Self::merge_str("yazi/keymap.toml", include_str!("../../config/keymap.toml"))
	}

	#[inline]
	pub(crate) fn theme() -> String {
		Self::merge_str("yazi/theme.toml", include_str!("../../config/theme.toml"))
	}

	#[inline]
	pub(crate) fn yazi() -> String {
		Self::merge_str("yazi/yazi.toml", include_str!("../../config/yazi.toml"))
	}
}