cosmic-freedesktop-icons/src/theme/mod.rs

use crate::theme::error::ThemeError;
use crate::theme::paths::ThemePath;
use ini::Ini;
use once_cell::sync::Lazy;
use paths::BASE_PATHS;
use std::collections::BTreeMap;
use std::fmt::{Debug, Formatter};
use std::path::{Path, PathBuf};

mod directories;
pub mod error;
mod parse;
mod paths;

type Result<T> = std::result::Result<T, ThemeError>;

pub static THEMES: Lazy<BTreeMap<String, Theme>> =
    Lazy::new(|| get_all_themes().expect("Failed to get theme paths"));

pub struct Theme {
    path: ThemePath,
    index: Ini,
}

impl Theme {
    pub fn try_get_icon(&self, name: &str, size: u16, scale: u16) -> Option<PathBuf> {
        self.try_get_icon_exact_size(name, size, scale)
            .or_else(|| self.try_get_icon_closest_size(name, size, scale))
    }

    fn try_get_icon_exact_size(&self, name: &str, size: u16, scale: u16) -> Option<PathBuf> {
        self.match_size(size, scale)
            .iter()
            .find_map(|path| try_build_icon_path(name, path))
    }

    fn match_size(&self, size: u16, scale: u16) -> Vec<PathBuf> {
        let dirs = self.get_all_directories();

        dirs.iter()
            .filter(|directory| directory.match_size(size, scale))
            .map(|dir| dir.name)
            .map(|dir| self.path().join(dir))
            .collect()
    }

    fn try_get_icon_closest_size(&self, name: &str, size: u16, scale: u16) -> Option<PathBuf> {
        self.closest_match_size(size, scale)
            .iter()
            .find_map(|path| try_build_icon_path(name, path))
    }

    fn closest_match_size(&self, size: u16, scale: u16) -> Vec<PathBuf> {
        let dirs = self.get_all_directories();

        dirs.iter()
            .filter(|directory| directory.directory_size_distance(size, scale) < i16::MAX)
            .map(|dir| dir.name)
            .map(|dir| self.path().join(dir))
            .collect()
    }

    fn path(&self) -> &PathBuf {
        &self.path.0
    }
}

pub(super) fn try_build_icon_path<P: AsRef<Path>>(name: &str, path: P) -> Option<PathBuf> {
    let path = path.as_ref();
    let png = path.join(format!("{name}.png"));
    if png.exists() {
        return Some(png);
    }

    let svg = path.join(format!("{name}.svg"));
    if svg.exists() {
        return Some(svg);
    }
    let xmp = path.join(format!("{name}.xmp"));

    if xmp.exists() {
        return Some(xmp);
    }

    None
}

// Iter through the base paths and get all theme directories
pub fn get_all_themes() -> Result<BTreeMap<String, Theme>> {
    let mut icon_themes = BTreeMap::new();
    for theme_base_dir in BASE_PATHS.iter() {
        for entry in theme_base_dir.read_dir()? {
            let entry = entry?;
            if let Some(theme) = Theme::from_path(entry.path()) {
                let name = entry.file_name().to_string_lossy().to_string();
                icon_themes.insert(name, theme);
            }
        }
    }
    Ok(icon_themes)
}

pub fn theme_names() -> Vec<&'static str> {
    THEMES
        .values()
        .map(|path| &path.index)
        .filter_map(|index| {
            index
                .section(Some("Icon Theme"))
                .and_then(|section| section.get("Name"))
        })
        .collect()
}

impl Theme {
    fn from_path<P: AsRef<Path>>(path: P) -> Option<Self> {
        let path = path.as_ref();

        let has_index = path.join("index.theme").exists();

        if !has_index || !path.is_dir() {
            return None;
        }

        let path = ThemePath(path.into());

        match path.index() {
            Ok(index) => Some(Theme { path, index }),
            Err(_) => None,
        }
    }
}

impl Debug for Theme {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        let mut content = vec![];
        self.index.write_to(&mut content).expect("Write error");
        let content = String::from_utf8_lossy(&content);
        writeln!(f, "ThemeIndex{{path: {:?}, index: {content:?}}}", self.path)
    }
}

#[cfg(test)]
mod test {
    use crate::THEMES;

    #[test]
    fn get_one_icon() {
        let theme = THEMES.get("Adwaita").unwrap();
        println!(
            "{:?}",
            theme.try_get_icon_exact_size("edit-delete-symbolic", 24, 1)
        );
    }
}