cosmic-edit/src/menu.rs

// SPDX-License-Identifier: GPL-3.0-only

use cosmic::{
    //TODO: export in cosmic::widget
    iced::{
        widget::{column, horizontal_rule},
        Alignment, Background, Length,
    },
    theme,
    widget::{
        self, horizontal_space,
        menu::{ItemHeight, ItemWidth, MenuBar, MenuTree},
        segmented_button,
    },
    Element,
};
use std::collections::HashMap;

use crate::{fl, icon_cache_get, Action, Config, ContextPage, KeyBind, Message};

macro_rules! menu_button {
    ($($x:expr),+ $(,)?) => (
        widget::button(
            widget::Row::with_children(
                vec![$(Element::from($x)),+]
            )
            .align_items(Alignment::Center)
        )
        .height(Length::Fixed(32.0))
        .padding([4, 16])
        .width(Length::Fill)
        .style(theme::Button::MenuItem)
    );
}

pub fn context_menu<'a>(
    key_binds: &HashMap<KeyBind, Action>,
    entity: segmented_button::Entity,
) -> Element<'a, Message> {
    let menu_item = |menu_label, menu_action| {
        let mut key = String::new();
        for (key_bind, key_action) in key_binds.iter() {
            if key_action == &menu_action {
                key = key_bind.to_string();
                break;
            }
        }
        menu_button!(
            widget::text(menu_label),
            horizontal_space(Length::Fill),
            widget::text(key)
        )
        .on_press(Message::TabContextAction(entity, menu_action))
    };

    widget::container(column!(
        menu_item(fl!("undo"), Action::Undo),
        menu_item(fl!("redo"), Action::Redo),
        horizontal_rule(1),
        menu_item(fl!("cut"), Action::Cut),
        menu_item(fl!("copy"), Action::Copy),
        menu_item(fl!("paste"), Action::Paste),
        menu_item(fl!("select-all"), Action::SelectAll),
    ))
    .padding(1)
    //TODO: move style to libcosmic
    .style(theme::Container::custom(|theme| {
        let cosmic = theme.cosmic();
        let component = &cosmic.background.component;
        widget::container::Appearance {
            icon_color: Some(component.on.into()),
            text_color: Some(component.on.into()),
            background: Some(Background::Color(component.base.into())),
            border_radius: 8.0.into(),
            border_width: 1.0,
            border_color: component.divider.into(),
        }
    }))
    .width(Length::Fixed(240.0))
    .into()
}

pub fn menu_bar<'a>(config: &Config, key_binds: &HashMap<KeyBind, Action>) -> Element<'a, Message> {
    //TODO: port to libcosmic
    let menu_root = |label| {
        widget::button(widget::text(label))
            .padding([4, 12])
            .style(theme::Button::MenuRoot)
    };

    let menu_folder =
        |label| menu_button!(widget::text(label), horizontal_space(Length::Fill), ">");

    let find_key = |message: &Message| -> String {
        let mut key = String::new();
        for (key_bind, action) in key_binds.iter() {
            if &action.message() == message {
                key = key_bind.to_string();
                break;
            }
        }
        key
    };

    let menu_item = |label, message| {
        let key = find_key(&message);
        MenuTree::new(
            menu_button!(
                widget::text(label),
                horizontal_space(Length::Fill),
                widget::text(key)
            )
            .on_press(message),
        )
    };

    //TODO: support key lookup?
    let menu_checkbox = |label, value, message| {
        let check: Element<_> = if value {
            icon_cache_get("object-select-symbolic", 16).into()
        } else {
            widget::Space::with_width(Length::Fixed(16.0)).into()
        };
        let key = find_key(&message);
        MenuTree::new(
            menu_button!(
                check,
                widget::Space::with_width(Length::Fixed(8.0)),
                widget::text(label),
                horizontal_space(Length::Fill),
                widget::text(key)
            )
            .on_press(message),
        )
    };

    let menu_key = |label, key, message| {
        MenuTree::new(
            menu_button!(widget::text(label), horizontal_space(Length::Fill), key)
                .on_press(message),
        )
    };

    let menu_tab_width = |tab_width: u16| {
        menu_checkbox(
            fl!("tab-width", tab_width = tab_width),
            config.tab_width == tab_width,
            Message::TabWidth(tab_width),
        )
    };

    MenuBar::new(vec![
        MenuTree::with_children(
            menu_root(fl!("file")),
            vec![
                menu_item(fl!("new-file"), Message::NewFile),
                menu_item(fl!("new-window"), Message::NewWindow),
                MenuTree::new(horizontal_rule(1)),
                menu_item(fl!("open-file"), Message::OpenFileDialog),
                MenuTree::with_children(
                    menu_folder(fl!("open-recent-file")),
                    vec![menu_item(fl!("todo"), Message::Todo)],
                ),
                menu_item(fl!("close-file"), Message::CloseFile),
                MenuTree::new(horizontal_rule(1)),
                menu_item(fl!("menu-open-project"), Message::OpenProjectDialog),
                MenuTree::with_children(
                    menu_folder(fl!("open-recent-project")),
                    vec![menu_item(fl!("todo"), Message::Todo)],
                ),
                menu_item(fl!("close-project"), Message::CloseProject),
                MenuTree::new(horizontal_rule(1)),
                menu_item(fl!("save"), Message::Save),
                menu_key(fl!("save-as"), "Ctrl + Shift + S", Message::Todo),
                MenuTree::new(horizontal_rule(1)),
                menu_item(fl!("revert-all-changes"), Message::Todo),
                MenuTree::new(horizontal_rule(1)),
                menu_item(
                    fl!("menu-document-statistics"),
                    Message::ToggleContextPage(ContextPage::DocumentStatistics),
                ),
                menu_item(fl!("document-type"), Message::Todo),
                menu_item(fl!("encoding"), Message::Todo),
                menu_item(
                    fl!("menu-git-management"),
                    Message::ToggleContextPage(ContextPage::GitManagement),
                ),
                menu_item(fl!("print"), Message::Todo),
                MenuTree::new(horizontal_rule(1)),
                menu_item(fl!("quit"), Message::Quit),
            ],
        ),
        MenuTree::with_children(
            menu_root(fl!("edit")),
            vec![
                menu_item(fl!("undo"), Message::Undo),
                menu_item(fl!("redo"), Message::Redo),
                MenuTree::new(horizontal_rule(1)),
                menu_item(fl!("cut"), Message::Cut),
                menu_item(fl!("copy"), Message::Copy),
                menu_item(fl!("paste"), Message::Paste),
                menu_item(fl!("select-all"), Message::SelectAll),
                MenuTree::new(horizontal_rule(1)),
                menu_item(fl!("find"), Message::Find(Some(false))),
                menu_item(fl!("replace"), Message::Find(Some(true))),
                menu_item(
                    fl!("find-in-project"),
                    Message::ToggleContextPage(ContextPage::ProjectSearch),
                ),
                MenuTree::new(horizontal_rule(1)),
                menu_item(fl!("spell-check"), Message::Todo),
            ],
        ),
        MenuTree::with_children(
            menu_root(fl!("view")),
            vec![
                MenuTree::with_children(
                    menu_folder(fl!("indentation")),
                    vec![
                        menu_checkbox(
                            fl!("automatic-indentation"),
                            config.auto_indent,
                            Message::ToggleAutoIndent,
                        ),
                        MenuTree::new(horizontal_rule(1)),
                        menu_tab_width(1),
                        menu_tab_width(2),
                        menu_tab_width(3),
                        menu_tab_width(4),
                        menu_tab_width(5),
                        menu_tab_width(6),
                        menu_tab_width(7),
                        menu_tab_width(8),
                        MenuTree::new(horizontal_rule(1)),
                        menu_item(fl!("convert-indentation-to-spaces"), Message::Todo),
                        menu_item(fl!("convert-indentation-to-tabs"), Message::Todo),
                    ],
                ),
                MenuTree::new(horizontal_rule(1)),
                menu_checkbox(fl!("word-wrap"), config.word_wrap, Message::ToggleWordWrap),
                menu_checkbox(
                    fl!("show-line-numbers"),
                    config.line_numbers,
                    Message::ToggleLineNumbers,
                ),
                menu_checkbox(fl!("highlight-current-line"), false, Message::Todo),
                menu_item(fl!("syntax-highlighting"), Message::Todo),
                MenuTree::new(horizontal_rule(1)),
                menu_item(
                    fl!("menu-settings"),
                    Message::ToggleContextPage(ContextPage::Settings),
                ),
                MenuTree::new(horizontal_rule(1)),
                menu_item(fl!("menu-keyboard-shortcuts"), Message::Todo),
                MenuTree::new(horizontal_rule(1)),
                menu_item(fl!("about-cosmic-text-editor"), Message::Todo),
            ],
        ),
    ])
    .item_height(ItemHeight::Dynamic(40))
    .item_width(ItemWidth::Uniform(240))
    .spacing(4.0)
    .into()
}