refactor: use iced workspaces applet

applets/cosmic-applet-workspaces/src/components/app.rs

@@ -0,0 +1,210 @@
+use std::{cmp::Ordering, env};
+
+use calloop::channel::SyncSender;
+use cosmic::iced::alignment::{Horizontal, Vertical};
+use cosmic::iced::mouse::{self, ScrollDelta};
+use cosmic::iced::widget::{column, container, row, text};
+use cosmic::iced::{
+    executor, subscription, widget::button, window, Application, Command, Event::Mouse, Length,
+    Settings, Subscription,
+};
+use cosmic::iced_style::application::{self, Appearance};
+use cosmic::theme::Button;
+use cosmic::{Element, Theme};
+use cosmic_panel_config::PanelAnchor;
+use cosmic_protocols::workspace::v1::client::zcosmic_workspace_handle_v1;
+use iced_sctk::application::SurfaceIdWrapper;
+use iced_sctk::command::platform_specific::wayland::window::SctkWindowSettings;
+use iced_sctk::settings::InitialSurface;
+use iced_sctk::{commands, Color};
+use wayland_backend::client::ObjectId;
+
+use crate::config;
+use crate::wayland::{WorkspaceEvent, WorkspaceList};
+use crate::wayland_subscription::{workspaces, WorkspacesUpdate};
+
+pub fn run() -> cosmic::iced::Result {
+    let mut settings = Settings::default();
+    settings.initial_surface = InitialSurface::XdgWindow(SctkWindowSettings {
+        iced_settings: cosmic::iced_native::window::Settings {
+            size: (32, 32),
+            ..Default::default()
+        },
+        ..Default::default()
+    });
+    IcedWorkspacesApplet::run(settings)
+}
+
+#[derive(Debug, Clone, Copy, PartialEq, Eq)]
+pub enum Layout {
+    Row,
+    Column,
+}
+
+struct IcedWorkspacesApplet {
+    theme: Theme,
+    workspaces: WorkspaceList,
+    workspace_tx: Option<SyncSender<WorkspaceEvent>>,
+    layout: Layout,
+}
+
+#[derive(Debug, Clone)]
+enum Message {
+    WorkspaceUpdate(WorkspacesUpdate),
+    WorkspacePressed(ObjectId),
+    WheelScrolled(ScrollDelta),
+    Errored,
+}
+
+impl Application for IcedWorkspacesApplet {
+    type Message = Message;
+    type Theme = Theme;
+    type Executor = executor::Default;
+    type Flags = ();
+
+    fn new(_flags: ()) -> (Self, Command<Message>) {
+        (
+            IcedWorkspacesApplet {
+                layout: match env::var("COSMIC_PANEL_ANCHOR")
+                    .ok()
+                    .and_then(|anchor| anchor.parse::<PanelAnchor>().ok())
+                    .unwrap_or_default()
+                {
+                    PanelAnchor::Left | PanelAnchor::Right => Layout::Column,
+                    PanelAnchor::Top | PanelAnchor::Bottom => Layout::Row,
+                },
+                theme: Default::default(),
+                workspaces: Vec::new(),
+                workspace_tx: Default::default(),
+            },
+            Command::none(),
+        )
+    }
+
+    fn title(&self) -> String {
+        config::APP_ID.to_string()
+    }
+
+    fn update(&mut self, message: Message) -> Command<Message> {
+        match message {
+            Message::WorkspaceUpdate(msg) => match msg {
+                WorkspacesUpdate::Workspaces(mut list) => {
+                    list.retain(|w| {
+                        !matches!(w.1, Some(zcosmic_workspace_handle_v1::State::Hidden))
+                    });
+                    list.sort_by(|a, b| match a.0.len().cmp(&b.0.len()) {
+                        Ordering::Equal => a.0.cmp(&b.0),
+                        Ordering::Less => Ordering::Less,
+                        Ordering::Greater => Ordering::Greater,
+                    });
+                    self.workspaces = list;
+                    let unit = 32;
+                    let (w, h) = match self.layout {
+                        Layout::Row => (unit * self.workspaces.len() as u32, unit),
+                        Layout::Column => (unit, unit * self.workspaces.len() as u32),
+                    };
+                    return commands::window::resize_window(window::Id::new(0), w, h);
+                }
+                WorkspacesUpdate::Started(tx) => {
+                    self.workspace_tx.replace(tx);
+                }
+                WorkspacesUpdate::Errored => {
+                    // TODO
+                }
+            },
+            Message::WorkspacePressed(id) => {
+                if let Some(tx) = self.workspace_tx.as_mut() {
+                    let _ = tx.try_send(WorkspaceEvent::Activate(id));
+                }
+            }
+            Message::WheelScrolled(delta) => {
+                let delta = match delta {
+                    ScrollDelta::Lines { x, y } => x + y,
+                    ScrollDelta::Pixels { x, y } => x + y,
+                } as f64;
+                if let Some(tx) = self.workspace_tx.as_mut() {
+                    let _ = tx.try_send(WorkspaceEvent::Scroll(delta));
+                }
+            }
+            Message::Errored => {}
+        }
+        Command::none()
+    }
+
+    fn view(&self, _id: SurfaceIdWrapper) -> Element<Message> {
+        let buttons = self
+            .workspaces
+            .iter()
+            .filter_map(|w| {
+                let btn = button(
+                    text(w.0.clone())
+                        .horizontal_alignment(Horizontal::Center)
+                        .vertical_alignment(Vertical::Center)
+                        .width(Length::Fill)
+                        .height(Length::Fill),
+                )
+                .width(Length::Fill)
+                .height(Length::Fill)
+                .on_press(Message::WorkspacePressed(w.2.clone()))
+                .padding(0);
+                Some(
+                    btn.style(match w.1 {
+                        Some(zcosmic_workspace_handle_v1::State::Active) => Button::Primary,
+                        Some(zcosmic_workspace_handle_v1::State::Urgent) => Button::Destructive,
+                        None => Button::Secondary,
+                        _ => return None,
+                    })
+                    .into(),
+                )
+            })
+            .collect();
+        let layout_section: Element<_> = match self.layout {
+            Layout::Row => row(buttons)
+                .width(Length::Fill)
+                .height(Length::Fill)
+                .padding(0)
+                .into(),
+            Layout::Column => column(buttons)
+                .width(Length::Fill)
+                .height(Length::Fill)
+                .padding(0)
+                .into(),
+        };
+
+        container(layout_section)
+            .width(Length::Fill)
+            .height(Length::Fill)
+            .padding(0)
+            .into()
+    }
+
+    fn subscription(&self) -> Subscription<Message> {
+        Subscription::batch(
+            vec![
+                workspaces(0).map(|(_, msg)| Message::WorkspaceUpdate(msg)),
+                subscription::events_with(|e, _| match e {
+                    Mouse(mouse::Event::WheelScrolled { delta }) => {
+                        Some(Message::WheelScrolled(delta))
+                    }
+                    _ => None,
+                }),
+            ]
+            .into_iter(),
+        )
+    }
+
+    fn theme(&self) -> Theme {
+        self.theme
+    }
+
+    fn close_requested(&self, _id: iced_sctk::application::SurfaceIdWrapper) -> Self::Message {
+        unimplemented!()
+    }
+
+    fn style(&self) -> <Self::Theme as application::StyleSheet>::Style {
+        <Self::Theme as application::StyleSheet>::Style::Custom(|theme| Appearance {
+            background_color: Color::from_rgba(0.0, 0.0, 0.0, 0.0),
+            text_color: theme.cosmic().on_bg_color().into(),
+        })
+    }
+}

applets/cosmic-applet-workspaces/src/components/mod.rs

@@ -0,0 +1 @@
+pub mod app;