coop/crates/ui/src/theme/mod.rs

use crate::scroll::ScrollbarShow;
use colors::{default_color_scales, hsl};
use gpui::{
    AppContext, Global, Hsla, ModelContext, SharedString, ViewContext, WindowAppearance,
    WindowContext,
};
use scale::ColorScaleSet;
use std::ops::{Deref, DerefMut};

pub mod colors;
pub mod scale;

#[derive(Debug, Clone, Copy, Default)]
pub struct ThemeColors {
    pub background: Hsla,
    pub transparent: Hsla,
    pub scrollbar: Hsla,
    pub scrollbar_thumb: Hsla,
    pub scrollbar_thumb_hover: Hsla,
    pub window_border: Hsla,
}

impl ThemeColors {
    pub fn light() -> Self {
        Self {
            background: hsl(0.0, 0.0, 100.),
            transparent: Hsla::transparent_black(),
            window_border: hsl(240.0, 5.9, 78.0),
            scrollbar: hsl(0., 0., 97.).opacity(0.75),
            scrollbar_thumb: hsl(0., 0., 69.).opacity(0.9),
            scrollbar_thumb_hover: hsl(0., 0., 59.),
        }
    }

    pub fn dark() -> Self {
        Self {
            background: hsl(0.0, 0.0, 8.0),
            transparent: Hsla::transparent_black(),
            window_border: hsl(240.0, 3.7, 28.0),
            scrollbar: hsl(240., 1., 15.).opacity(0.75),
            scrollbar_thumb: hsl(0., 0., 48.).opacity(0.9),
            scrollbar_thumb_hover: hsl(0., 0., 68.),
        }
    }
}

pub trait ActiveTheme {
    fn theme(&self) -> &Theme;
}

impl ActiveTheme for AppContext {
    fn theme(&self) -> &Theme {
        Theme::global(self)
    }
}

impl<V> ActiveTheme for ViewContext<'_, V> {
    fn theme(&self) -> &Theme {
        self.deref().theme()
    }
}

impl<V> ActiveTheme for ModelContext<'_, V> {
    fn theme(&self) -> &Theme {
        self.deref().theme()
    }
}

impl ActiveTheme for WindowContext<'_> {
    fn theme(&self) -> &Theme {
        self.deref().theme()
    }
}

pub fn init(cx: &mut AppContext) {
    Theme::sync_system_appearance(cx)
}

pub struct Theme {
    colors: ThemeColors,
    /// Base colors.
    pub base: ColorScaleSet,
    /// Accent colors.
    pub accent: ColorScaleSet,
    /// Window appearances.
    pub appearance: Appearance,
    pub font_family: SharedString,
    pub font_size: f32,
    pub radius: f32,
    pub shadow: bool,
    /// Show the scrollbar mode, default: Scrolling
    pub scrollbar_show: ScrollbarShow,
}

impl Deref for Theme {
    type Target = ThemeColors;

    fn deref(&self) -> &Self::Target {
        &self.colors
    }
}

impl DerefMut for Theme {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.colors
    }
}

impl Global for Theme {}

impl Theme {
    /// Returns the global theme reference
    pub fn global(cx: &AppContext) -> &Theme {
        cx.global::<Theme>()
    }

    /// Returns the global theme mutable reference
    pub fn global_mut(cx: &mut AppContext) -> &mut Theme {
        cx.global_mut::<Theme>()
    }

    /// Sync the theme with the system appearance
    pub fn sync_system_appearance(cx: &mut AppContext) {
        match cx.window_appearance() {
            WindowAppearance::Dark | WindowAppearance::VibrantDark => {
                Self::change(Appearance::Dark, cx)
            }
            WindowAppearance::Light | WindowAppearance::VibrantLight => {
                Self::change(Appearance::Light, cx)
            }
        }
    }

    pub fn change(mode: Appearance, cx: &mut AppContext) {
        let colors = match mode {
            Appearance::Light => ThemeColors::light(),
            Appearance::Dark => ThemeColors::dark(),
        };

        let mut theme = Theme::from(colors);
        theme.appearance = mode;

        cx.set_global(theme);
        cx.refresh();
    }
}

impl From<ThemeColors> for Theme {
    fn from(colors: ThemeColors) -> Self {
        let color_scales = default_color_scales();

        Theme {
            base: color_scales.gray,
            accent: color_scales.yellow,
            appearance: Appearance::default(),
            font_size: 16.0,
            font_family: if cfg!(target_os = "macos") {
                ".SystemUIFont".into()
            } else if cfg!(target_os = "windows") {
                "Segoe UI".into()
            } else {
                "FreeMono".into()
            },
            radius: 6.0,
            shadow: false,
            scrollbar_show: ScrollbarShow::default(),
            colors,
        }
    }
}

#[derive(Debug, Clone, Copy, Default, PartialEq, PartialOrd, Eq)]
pub enum Appearance {
    #[default]
    Light,
    Dark,
}

impl Appearance {
    pub fn is_dark(&self) -> bool {
        matches!(self, Self::Dark)
    }
}