libcosmic-yoda/examples/cosmic-applet-graphics/src/window.rs

use crate::dbus::{self, PowerDaemonProxy};
use crate::graphics::{get_current_graphics, Graphics};
use cosmic::widget::{expander, icon, nav_bar, nav_bar_page, nav_bar_section};
use cosmic::{
    iced::widget::{
        checkbox, column, container, horizontal_space, pick_list, progress_bar, radio, row, slider,
        text,
    },
    iced::{self, Alignment, Application, Color, Command, Length},
    iced_lazy::responsive,
    iced_native::window,
    iced_winit::window::{drag, maximize, minimize},
    list_view, list_view_item, list_view_row, list_view_section, scrollable,
    theme::{self, Theme},
    widget::{button, header_bar, list_box, list_row, list_view::*, toggler},
    Element,
};
use cosmic::{iced_native, separator};
use cosmic_panel_config::PanelSize;
use iced_sctk::alignment::Horizontal;
use iced_sctk::command::platform_specific::wayland::popup::{SctkPopupSettings, SctkPositioner};
use iced_sctk::commands::popup::{destroy_popup, get_popup};
use iced_sctk::{Point, Rectangle, Size};
use sctk::reexports::protocols::xdg::shell::client::xdg_positioner::{Anchor, Gravity};
use zbus::Connection;

#[derive(Default)]
pub struct Window {
    popup: Option<window::Id>,
    graphics_mode: Option<Graphics>,
    id_ctr: u32,
    icon_size: u16,
    theme: Theme,
    dbus: Option<(Connection, PowerDaemonProxy<'static>)>,
}

#[allow(dead_code)]
#[derive(Clone, Debug)]
pub enum Message {
    CurrentGraphics(Option<Graphics>),
    DBusInit(Option<(Connection, PowerDaemonProxy<'static>)>),
    SelectGraphicsMode(Graphics),
    TogglePopup,
}

impl Application for Window {
    type Executor = iced::executor::Default;
    type Flags = ();
    type Message = Message;
    type Theme = Theme;

    fn new(_flags: ()) -> (Self, Command<Self::Message>) {
        let mut window = Window::default();
        let pixels = std::env::var("COSMIC_PANEL_SIZE")
            .ok()
            .and_then(|size| match size.parse::<PanelSize>() {
                Ok(PanelSize::XL) => Some(64),
                Ok(PanelSize::L) => Some(36),
                Ok(PanelSize::M) => Some(24),
                Ok(PanelSize::S) => Some(16),
                Ok(PanelSize::XS) => Some(12),
                Err(_) => Some(12),
            })
            .unwrap_or(16);
        window.icon_size = pixels;
        (
            window,
            Command::perform(dbus::init(), |dbus_init| Message::DBusInit(dbus_init)),
        )
    }

    fn title(&self) -> String {
        String::from("Cosmic Graphics Applet")
    }

    fn update(&mut self, message: Message) -> iced::Command<Self::Message> {
        match message {
            Message::SelectGraphicsMode(_) => {}
            Message::TogglePopup => {
                if let Some(p) = self.popup.take() {
                    return destroy_popup(p);
                } else {
                    self.id_ctr += 1;
                    let new_id = window::Id::new(self.id_ctr);
                    self.popup.replace(new_id);
                    let mut commands = Vec::new();
                    if let Some((_, proxy)) = self.dbus.as_ref() {
                        commands.push(Command::perform(
                            get_current_graphics(proxy.clone()),
                            |cur_graphics| Message::CurrentGraphics(cur_graphics.ok()),
                        ));
                    }
                    commands.push(Command::batch(vec![get_popup(SctkPopupSettings {
                        parent: window::Id::new(0),
                        id: new_id,
                        positioner: SctkPositioner {
                            size: (200, 200),
                            anchor_rect: Rectangle {
                                x: 0,
                                y: 0,
                                width: 32 + self.icon_size as i32,
                                height: 16 + self.icon_size as i32,
                            },
                            anchor: Anchor::Bottom,
                            gravity: Gravity::Bottom,
                            reactive: true,
                            ..Default::default()
                        },
                        parent_size: None,
                        grab: true,
                    })]));
                    return Command::batch(commands);
                }
            }
            Message::DBusInit(dbus) => {
                self.dbus = dbus;
                return Command::perform(
                    get_current_graphics(self.dbus.as_ref().unwrap().1.clone()),
                    |cur_graphics| {
                        Message::CurrentGraphics(match cur_graphics {
                            Ok(g) => Some(g),
                            Err(err) => {
                                dbg!(err);
                                None
                            },
                        })
                    },
                );
            }
            Message::CurrentGraphics(g) => {
                if let Some(g) = g {
                    self.graphics_mode.replace(g);
                }
            }
        }
        Command::none()
    }

    fn view_popup(&self, _: window::Id) -> Element<Message> {
        let content = column(vec![
            text("Graphics Mode")
                .width(Length::Fill)
                .horizontal_alignment(Horizontal::Center)
                .size(24)
                .into(),
            separator!(1).into(),
            column(vec![
                radio(
                    "Integrated Graphics",
                    Graphics::Integrated,
                    self.graphics_mode,
                    |g| Message::SelectGraphicsMode(g),
                )
                .into(),
                radio(
                    "Nvidia Graphics",
                    Graphics::Nvidia,
                    self.graphics_mode,
                    |g| Message::SelectGraphicsMode(g),
                )
                .into(),
                radio(
                    "Hybrid Graphics",
                    Graphics::Hybrid,
                    self.graphics_mode,
                    |g| Message::SelectGraphicsMode(g),
                )
                .into(),
                radio(
                    "Compute Graphics",
                    Graphics::Compute,
                    self.graphics_mode,
                    |g| Message::SelectGraphicsMode(g),
                )
                .into(),
            ])
            .padding([8, 0])
            .spacing(8)
            .into(),
        ])
        .padding(4)
        .spacing(4)
        .into();
        content
    }

    fn view_layer_surface(
        &self,
        _: cosmic::iced_native::window::Id,
    ) -> iced::Element<'_, Self::Message, iced::Renderer<Self::Theme>> {
        unimplemented!()
    }
    fn close_window_requested(&self, _: cosmic::iced_native::window::Id) -> Self::Message {
        unimplemented!()
    }
    fn popup_done(&self, _: cosmic::iced_native::window::Id) -> Self::Message {
        unimplemented!()
    }
    fn layer_surface_done(&self, _: cosmic::iced_native::window::Id) -> Self::Message {
        unimplemented!()
    }

    fn view_window(&self, id: window::Id) -> Element<Message> {
        // TODO use panel config crate after resolving version mismatch

        button!(icon("input-gaming-symbolic", self.icon_size).style(theme::Svg::Accent))
            .on_press(Message::TogglePopup)
            .into()
    }

    fn should_exit(&self) -> bool {
        false
    }

    fn theme(&self) -> Theme {
        self.theme
    }
}