// Copyright 2023 System76 // SPDX-License-Identifier: MPL-2.0 use std::borrow::Borrow; use std::collections::HashMap; use std::sync::Arc; use super::{Action, Application, ApplicationExt, Subscription}; use crate::theme::{Theme, ThemeType, THEME}; use crate::{keyboard_nav, Core, Element}; #[cfg(feature = "wayland")] use cctk::sctk::reexports::csd_frame::{WindowManagerCapabilities, WindowState}; use cosmic_theme::ThemeMode; #[cfg(feature = "wayland")] use iced::event::wayland; #[cfg(not(any(feature = "multi-window", feature = "wayland")))] use iced::Application as IcedApplication; use iced::{window, Task}; use iced_futures::event::listen_with; use palette::color_difference::EuclideanDistance; #[derive(Default)] pub struct Cosmic { pub app: App, #[cfg(feature = "wayland")] pub surface_views: HashMap< window::Id, Box Fn(&'a App) -> Element<'a, crate::Action>>, >, } impl Cosmic where T::Message: Send + 'static, { pub fn init( (mut core, flags): (Core, T::Flags), ) -> (Self, iced::Task>) { #[cfg(feature = "dbus-config")] { use iced_futures::futures::executor::block_on; core.settings_daemon = block_on(cosmic_config::dbus::settings_daemon_proxy()).ok(); } let (model, command) = T::init(core, flags); (Self::new(model), command) } #[cfg(not(feature = "multi-window"))] pub fn title(&self) -> String { self.app.title().to_string() } #[cfg(feature = "multi-window")] pub fn title(&self, id: window::Id) -> String { self.app.title(id).to_string() } pub fn surface_update( &mut self, _surface_message: crate::surface::Action, ) -> iced::Task> { #[cfg(feature = "wayland")] match _surface_message { crate::surface::Action::AppSubsurface(settings, view) => { let Some(settings) = std::sync::Arc::try_unwrap(settings) .ok() .and_then(|s| s.downcast:: iced_runtime::platform_specific::wayland::subsurface::SctkSubsurfaceSettings + Send + Sync>>().ok()) else { tracing::error!("Invalid settings for subsurface"); return Task::none(); }; if let Some(view) = view.and_then(|view| { match std::sync::Arc::try_unwrap(view).ok()?.downcast:: Fn(&'a T) -> Element<'a, crate::Action> + Send + Sync, >>() { Ok(v) => Some(v), Err(err) => { tracing::error!("Invalid view for subsurface view: {err:?}"); None } } }) { let settings = settings(&mut self.app); self.get_subsurface(settings, *view) } else { iced_winit::commands::subsurface::get_subsurface(settings(&mut self.app)) } } crate::surface::Action::Subsurface(settings, view) => { let Some(settings) = std::sync::Arc::try_unwrap(settings) .ok() .and_then(|s| s.downcast:: iced_runtime::platform_specific::wayland::subsurface::SctkSubsurfaceSettings + Send + Sync>>().ok()) else { tracing::error!("Invalid settings for subsurface"); return Task::none(); }; if let Some(view) = view.and_then(|view| { match std::sync::Arc::try_unwrap(view).ok()?.downcast:: Element<'static, crate::Action> + Send + Sync, >>() { Ok(v) => Some(v), Err(err) => { tracing::error!("Invalid view for subsurface view: {err:?}"); None } } }) { let settings = settings(); self.get_subsurface(settings, Box::new(move |_| view())) } else { iced_winit::commands::subsurface::get_subsurface(settings()) } } crate::surface::Action::AppPopup(settings, view) => { let Some(settings) = std::sync::Arc::try_unwrap(settings) .ok() .and_then(|s| s.downcast:: iced_runtime::platform_specific::wayland::popup::SctkPopupSettings + Send + Sync>>().ok()) else { tracing::error!("Invalid settings for popup"); return Task::none(); }; if let Some(view) = view.and_then(|view| { match std::sync::Arc::try_unwrap(view).ok()?.downcast:: Fn(&'a T) -> Element<'a, crate::Action> + Send + Sync, >>() { Ok(v) => Some(v), Err(err) => { tracing::error!("Invalid view for subsurface view: {err:?}"); None } } }) { let settings = settings(&mut self.app); self.get_popup(settings, *view) } else { iced_winit::commands::popup::get_popup(settings(&mut self.app)) } } #[cfg(feature = "wayland")] crate::surface::Action::DestroyPopup(id) => { iced_winit::commands::popup::destroy_popup(id) } #[cfg(feature = "wayland")] crate::surface::Action::DestroySubsurface(id) => { iced_winit::commands::subsurface::destroy_subsurface(id) } crate::surface::Action::ResponsiveMenuBar { menu_bar, limits, size, } => { let core = self.app.core_mut(); core.menu_bars.insert(menu_bar, (limits, size)); iced::Task::none() } crate::surface::Action::Popup(settings, view) => { let Some(settings) = std::sync::Arc::try_unwrap(settings) .ok() .and_then(|s| s.downcast:: iced_runtime::platform_specific::wayland::popup::SctkPopupSettings + Send + Sync>>().ok()) else { tracing::error!("Invalid settings for popup"); return Task::none(); }; if let Some(view) = view.and_then(|view| { match std::sync::Arc::try_unwrap(view).ok()?.downcast:: Element<'static, crate::Action> + Send + Sync, >>() { Ok(v) => Some(v), Err(err) => { tracing::error!("Invalid view for subsurface view: {err:?}"); None } } }) { let settings = settings(); self.get_popup(settings, Box::new(move |_| view())) } else { iced_winit::commands::popup::get_popup(settings()) } } crate::surface::Action::Ignore => iced::Task::none(), crate::surface::Action::Task(f) => { f().map(|sm| crate::Action::Cosmic(Action::Surface(sm))) } } #[cfg(not(feature = "wayland"))] iced::Task::none() } pub fn update( &mut self, message: crate::Action, ) -> iced::Task> { let message = match message { crate::Action::App(message) => self.app.update(message), crate::Action::Cosmic(message) => self.cosmic_update(message), crate::Action::None => iced::Task::none(), #[cfg(feature = "single-instance")] crate::Action::DbusActivation(message) => self.app.dbus_activation(message), }; #[cfg(target_env = "gnu")] crate::malloc::trim(0); message } #[cfg(not(feature = "multi-window"))] pub fn scale_factor(&self) -> f64 { f64::from(self.app.core().scale_factor()) } #[cfg(feature = "multi-window")] pub fn scale_factor(&self, _id: window::Id) -> f64 { f64::from(self.app.core().scale_factor()) } pub fn style(&self, theme: &Theme) -> iced_runtime::Appearance { if let Some(style) = self.app.style() { style } else if self.app.core().window.sharp_corners { let theme = THEME.lock().unwrap(); crate::style::iced::application::appearance(theme.borrow()) } else { let theme = THEME.lock().unwrap(); iced_runtime::Appearance { background_color: iced_core::Color::TRANSPARENT, icon_color: theme.cosmic().on_bg_color().into(), text_color: theme.cosmic().on_bg_color().into(), } } } #[allow(clippy::too_many_lines)] pub fn subscription(&self) -> Subscription> { let window_events = listen_with(|event, _, id| { match event { iced::Event::Window(window::Event::Resized(iced::Size { width, height })) => { return Some(Action::WindowResize(id, width, height)); } iced::Event::Window(window::Event::Closed) => { return Some(Action::SurfaceClosed(id)); } iced::Event::Window(window::Event::Focused) => return Some(Action::Focus(id)), iced::Event::Window(window::Event::Unfocused) => return Some(Action::Unfocus(id)), #[cfg(feature = "wayland")] iced::Event::PlatformSpecific(iced::event::PlatformSpecific::Wayland(event)) => { match event { wayland::Event::Popup(wayland::PopupEvent::Done, _, id) | wayland::Event::Layer(wayland::LayerEvent::Done, _, id) => { return Some(Action::SurfaceClosed(id)); } #[cfg(feature = "applet")] wayland::Event::Window( iced::event::wayland::WindowEvent::SuggestedBounds(b), ) => { return Some(Action::SuggestedBounds(b)); } _ => (), } } _ => (), } None }); let mut subscriptions = vec![ self.app.subscription().map(crate::Action::App), self.app .core() .watch_config::(crate::config::ID) .map(|update| { for why in update .errors .into_iter() .filter(cosmic_config::Error::is_err) { tracing::error!(?why, "cosmic toolkit config update error"); } crate::Action::Cosmic(Action::ToolkitConfig(update.config)) }), self.app .core() .watch_config::( if if let ThemeType::System { prefer_dark, .. } = THEME.lock().unwrap().theme_type { prefer_dark } else { None } .unwrap_or_else(|| self.app.core().system_theme_mode.is_dark) { cosmic_theme::DARK_THEME_ID } else { cosmic_theme::LIGHT_THEME_ID }, ) .map(|update| { for why in update .errors .into_iter() .filter(cosmic_config::Error::is_err) { tracing::error!(?why, "cosmic theme config update error"); } Action::SystemThemeChange( update.keys, crate::theme::Theme::system(Arc::new(update.config)), ) }) .map(crate::Action::Cosmic), self.app .core() .watch_config::(cosmic_theme::THEME_MODE_ID) .map(|update| { for error in update .errors .into_iter() .filter(cosmic_config::Error::is_err) { tracing::error!(?error, "error reading system theme mode update"); } Action::SystemThemeModeChange(update.keys, update.config) }) .map(crate::Action::Cosmic), window_events.map(crate::Action::Cosmic), #[cfg(feature = "xdg-portal")] crate::theme::portal::desktop_settings() .map(Action::DesktopSettings) .map(crate::Action::Cosmic), ]; if self.app.core().keyboard_nav { subscriptions.push( keyboard_nav::subscription() .map(Action::KeyboardNav) .map(crate::Action::Cosmic), ); } #[cfg(feature = "single-instance")] if self.app.core().single_instance { subscriptions.push(crate::dbus_activation::subscription::()); } Subscription::batch(subscriptions) } #[cfg(not(feature = "multi-window"))] pub fn theme(&self) -> Theme { crate::theme::active() } #[cfg(feature = "multi-window")] pub fn theme(&self, _id: window::Id) -> Theme { crate::theme::active() } #[cfg(feature = "multi-window")] pub fn view(&self, id: window::Id) -> Element> { #[cfg(feature = "wayland")] if let Some(v) = self.surface_views.get(&id) { return v(&self.app); } if !self .app .core() .main_window_id() .is_some_and(|main_id| main_id == id) { return self.app.view_window(id).map(crate::Action::App); } let view = if self.app.core().window.use_template { self.app.view_main() } else { self.app.view().map(crate::Action::App) }; #[cfg(target_env = "gnu")] crate::malloc::trim(0); view } #[cfg(not(feature = "multi-window"))] pub fn view(&self) -> Element> { let view = self.app.view_main(); #[cfg(target_env = "gnu")] crate::malloc::trim(0); view } } impl Cosmic { #[allow(clippy::unused_self)] pub fn close(&mut self) -> iced::Task> { if let Some(id) = self.app.core().main_window_id() { iced::window::close(id) } else { iced::Task::none() } } #[allow(clippy::too_many_lines)] fn cosmic_update(&mut self, message: Action) -> iced::Task> { match message { Action::WindowMaximized(id, maximized) => { if self .app .core() .main_window_id() .is_some_and(|main_id| main_id == id) { self.app.core_mut().window.sharp_corners = maximized; } } Action::WindowResize(id, width, height) => { if self .app .core() .main_window_id() .is_some_and(|main_id| main_id == id) { self.app.core_mut().set_window_width(width); self.app.core_mut().set_window_height(height); } self.app.on_window_resize(id, width, height); //TODO: more efficient test of maximized (winit has no event for maximize if set by the OS) return iced::window::get_maximized(id).map(move |maximized| { crate::Action::Cosmic(Action::WindowMaximized(id, maximized)) }); } #[cfg(feature = "wayland")] Action::WindowState(id, state) => { if self .app .core() .main_window_id() .is_some_and(|main_id| main_id == id) { self.app.core_mut().window.sharp_corners = state.intersects( WindowState::MAXIMIZED | WindowState::FULLSCREEN | WindowState::TILED | WindowState::TILED_RIGHT | WindowState::TILED_LEFT | WindowState::TILED_TOP | WindowState::TILED_BOTTOM, ); } } #[cfg(feature = "wayland")] Action::WmCapabilities(id, capabilities) => { if self .app .core() .main_window_id() .is_some_and(|main_id| main_id == id) { self.app.core_mut().window.show_maximize = capabilities.contains(WindowManagerCapabilities::MAXIMIZE); self.app.core_mut().window.show_minimize = capabilities.contains(WindowManagerCapabilities::MINIMIZE); self.app.core_mut().window.show_window_menu = capabilities.contains(WindowManagerCapabilities::WINDOW_MENU); } } Action::KeyboardNav(message) => match message { keyboard_nav::Action::FocusNext => { return iced::widget::focus_next().map(crate::Action::Cosmic) } keyboard_nav::Action::FocusPrevious => { return iced::widget::focus_previous().map(crate::Action::Cosmic) } keyboard_nav::Action::Escape => return self.app.on_escape(), keyboard_nav::Action::Search => return self.app.on_search(), keyboard_nav::Action::Fullscreen => return self.app.core().toggle_maximize(None), }, Action::ContextDrawer(show) => { self.app.core_mut().set_show_context(show); return self.app.on_context_drawer(); } Action::Drag => return self.app.core().drag(None), Action::Minimize => return self.app.core().minimize(None), Action::Maximize => return self.app.core().toggle_maximize(None), Action::NavBar(key) => { self.app.core_mut().nav_bar_set_toggled_condensed(false); return self.app.on_nav_select(key); } Action::NavBarContext(key) => { self.app.core_mut().nav_bar_set_context(key); return self.app.on_nav_context(key); } Action::ToggleNavBar => { self.app.core_mut().nav_bar_toggle(); } Action::ToggleNavBarCondensed => { self.app.core_mut().nav_bar_toggle_condensed(); } Action::AppThemeChange(mut theme) => { if let ThemeType::System { theme: _, .. } = theme.theme_type { self.app.core_mut().theme_sub_counter += 1; let portal_accent = self.app.core().portal_accent; if let Some(a) = portal_accent { let t_inner = theme.cosmic(); if a.distance_squared(*t_inner.accent_color()) > 0.00001 { theme = Theme::system(Arc::new(t_inner.with_accent(a))); } }; } THEME.lock().unwrap().set_theme(theme.theme_type); } Action::SystemThemeChange(keys, theme) => { let cur_is_dark = THEME.lock().unwrap().theme_type.is_dark(); // Ignore updates if the current theme mode does not match. if cur_is_dark != theme.cosmic().is_dark { return iced::Task::none(); } let cmd = self.app.system_theme_update(&keys, theme.cosmic()); // Record the last-known system theme in event that the current theme is custom. self.app.core_mut().system_theme = theme.clone(); let portal_accent = self.app.core().portal_accent; { let mut cosmic_theme = THEME.lock().unwrap(); // Only apply update if the theme is set to load a system theme if let ThemeType::System { theme: _, prefer_dark, } = cosmic_theme.theme_type { let mut new_theme = if let Some(a) = portal_accent { let t_inner = theme.cosmic(); if a.distance_squared(*t_inner.accent_color()) > 0.00001 { Theme::system(Arc::new(t_inner.with_accent(a))) } else { theme } } else { theme }; new_theme.theme_type.prefer_dark(prefer_dark); cosmic_theme.set_theme(new_theme.theme_type); } } return cmd; } Action::ScaleFactor(factor) => { self.app.core_mut().set_scale_factor(factor); } Action::Close => { return match self.app.on_app_exit() { Some(message) => self.app.update(message), None => self.close(), }; } Action::SystemThemeModeChange(keys, mode) => { if !keys.contains(&"is_dark") { return iced::Task::none(); } if match THEME.lock().unwrap().theme_type { ThemeType::System { theme: _, prefer_dark, } => prefer_dark.is_some(), _ => false, } { return iced::Task::none(); } let mut cmds = vec![self.app.system_theme_mode_update(&keys, &mode)]; let core = self.app.core_mut(); let prev_is_dark = core.system_is_dark(); core.system_theme_mode = mode; let is_dark = core.system_is_dark(); let changed = prev_is_dark != is_dark; if changed { core.theme_sub_counter += 1; let mut new_theme = if is_dark { crate::theme::system_dark() } else { crate::theme::system_light() }; cmds.push(self.app.system_theme_update(&[], new_theme.cosmic())); let core = self.app.core_mut(); new_theme = if let Some(a) = core.portal_accent { let t_inner = new_theme.cosmic(); if a.distance_squared(*t_inner.accent_color()) > 0.00001 { Theme::system(Arc::new(t_inner.with_accent(a))) } else { new_theme } } else { new_theme }; core.system_theme = new_theme.clone(); { let mut cosmic_theme = THEME.lock().unwrap(); // Only apply update if the theme is set to load a system theme if let ThemeType::System { theme: _, .. } = cosmic_theme.theme_type { cosmic_theme.set_theme(new_theme.theme_type); } } } return Task::batch(cmds); } Action::Activate(_token) => { #[cfg(feature = "wayland")] if let Some(id) = self.app.core().main_window_id() { return iced_winit::platform_specific::commands::activation::activate( id, #[allow(clippy::used_underscore_binding)] _token, ); } } Action::Surface(action) => return self.surface_update(action), Action::SurfaceClosed(id) => { let mut ret = if let Some(msg) = self.app.on_close_requested(id) { self.app.update(msg) } else { Task::none() }; let core = self.app.core(); if core.exit_on_main_window_closed && core.main_window_id().is_some_and(|m_id| id == m_id) { ret = Task::batch(vec![iced::exit::>()]); } return ret; } Action::ShowWindowMenu => { if let Some(id) = self.app.core().main_window_id() { return iced::window::show_system_menu(id); } } #[cfg(feature = "xdg-portal")] Action::DesktopSettings(crate::theme::portal::Desktop::ColorScheme(s)) => { use ashpd::desktop::settings::ColorScheme; if match THEME.lock().unwrap().theme_type { ThemeType::System { theme: _, prefer_dark, } => prefer_dark.is_some(), _ => false, } { return iced::Task::none(); } let is_dark = match s { ColorScheme::NoPreference => None, ColorScheme::PreferDark => Some(true), ColorScheme::PreferLight => Some(false), }; let core = self.app.core_mut(); let prev_is_dark = core.system_is_dark(); core.portal_is_dark = is_dark; let is_dark = core.system_is_dark(); let changed = prev_is_dark != is_dark; if changed { core.theme_sub_counter += 1; let new_theme = if is_dark { crate::theme::system_dark() } else { crate::theme::system_light() }; core.system_theme = new_theme.clone(); { let mut cosmic_theme = THEME.lock().unwrap(); // Only apply update if the theme is set to load a system theme if let ThemeType::System { theme: _, .. } = cosmic_theme.theme_type { cosmic_theme.set_theme(new_theme.theme_type); } } } } #[cfg(feature = "xdg-portal")] Action::DesktopSettings(crate::theme::portal::Desktop::Accent(c)) => { use palette::Srgba; let c = Srgba::new(c.red() as f32, c.green() as f32, c.blue() as f32, 1.0); let core = self.app.core_mut(); core.portal_accent = Some(c); let cur_accent = core.system_theme.cosmic().accent_color(); if cur_accent.distance_squared(*c) < 0.00001 { // skip calculations if we already have the same color return iced::Task::none(); } { let mut cosmic_theme = THEME.lock().unwrap(); // Only apply update if the theme is set to load a system theme if let ThemeType::System { theme: t, prefer_dark, } = cosmic_theme.theme_type.clone() { cosmic_theme.set_theme(ThemeType::System { theme: Arc::new(t.with_accent(c)), prefer_dark, }); } } } #[cfg(feature = "xdg-portal")] Action::DesktopSettings(crate::theme::portal::Desktop::Contrast(_)) => { // TODO when high contrast is integrated in settings and all custom themes } Action::ToolkitConfig(config) => { // Change the icon theme if not defined by the application. if !self.app.core().icon_theme_override && crate::icon_theme::default() != config.icon_theme { crate::icon_theme::set_default(config.icon_theme.clone()); } *crate::config::COSMIC_TK.write().unwrap() = config; } Action::Focus(f) => { self.app.core_mut().focused_window = Some(f); } Action::Unfocus(id) => { let core = self.app.core_mut(); if core.focused_window.as_ref().is_some_and(|cur| *cur == id) { core.focused_window = None; } } #[cfg(feature = "applet")] Action::SuggestedBounds(b) => { tracing::info!("Suggested bounds: {b:?}"); let core = self.app.core_mut(); core.applet.suggested_bounds = b; } _ => {} } iced::Task::none() } } impl Cosmic { pub fn new(app: App) -> Self { Self { app, #[cfg(feature = "wayland")] surface_views: HashMap::new(), } } #[cfg(feature = "wayland")] /// Create a subsurface pub fn get_subsurface( &mut self, settings: iced_runtime::platform_specific::wayland::subsurface::SctkSubsurfaceSettings, view: Box< dyn for<'a> Fn(&'a App) -> Element<'a, crate::Action> + Send + Sync, >, ) -> Task> { use iced_winit::commands::subsurface::get_subsurface; self.surface_views.insert(settings.id, view); get_subsurface(settings) } #[cfg(feature = "wayland")] /// Create a subsurface pub fn get_popup( &mut self, settings: iced_runtime::platform_specific::wayland::popup::SctkPopupSettings, view: Box< dyn for<'a> Fn(&'a App) -> Element<'a, crate::Action> + Send + Sync, >, ) -> Task> { use iced_winit::commands::popup::get_popup; self.surface_views.insert(settings.id, view); get_popup(settings) } }