rio-window 0.3.8

use objc2::rc::Retained;
use objc2::runtime::Sel;
use objc2::sel;
use objc2_app_kit::{NSApplication, NSEventModifierFlags, NSMenu, NSMenuItem};
use objc2_foundation::{ns_string, MainThreadMarker, NSProcessInfo, NSString};

pub struct KeyEquivalent<'a> {
    key: &'a NSString,
    masks: Option<NSEventModifierFlags>,
}

pub fn initialize(app: &NSApplication) {
    let mtm = MainThreadMarker::from(app);
    let menubar = NSMenu::new(mtm);

    let app_menu_item = NSMenuItem::new(mtm);
    let shell_menu_item = NSMenuItem::new(mtm);
    let edit_menu_item = NSMenuItem::new(mtm);
    let view_menu_item = NSMenuItem::new(mtm);
    let help_menu_item = NSMenuItem::new(mtm);
    let window_menu_item = NSMenuItem::new(mtm);

    menubar.addItem(&app_menu_item);
    menubar.addItem(&shell_menu_item);
    menubar.addItem(&edit_menu_item);
    menubar.addItem(&view_menu_item);
    menubar.addItem(&window_menu_item);
    menubar.addItem(&help_menu_item);

    let app_menu = NSMenu::new(mtm);
    let process_name = NSProcessInfo::processInfo().processName();

    // About menu item
    let about_item_title = ns_string!("About ").stringByAppendingString(&process_name);
    let about_item = menu_item(
        mtm,
        &about_item_title,
        Some(sel!(orderFrontStandardAboutPanel:)),
        None,
    );

    // Services menu item
    let services_menu = NSMenu::new(mtm);
    let services_item = menu_item(mtm, ns_string!("Services"), None, None);
    services_item.setSubmenu(Some(&services_menu));

    // Separator menu item
    let sep_first = NSMenuItem::separatorItem(mtm);

    let open_config_title = ns_string!("Edit Configuration");
    let open_config = menu_item(
        mtm,
        open_config_title,
        Some(sel!(openConfig:)),
        Some(KeyEquivalent {
            key: ns_string!(","),
            masks: Some(NSEventModifierFlags::NSEventModifierFlagCommand),
        }),
    );

    // Hide application menu item
    let hide_item_title = ns_string!("Hide ").stringByAppendingString(&process_name);
    let hide_item = menu_item(
        mtm,
        &hide_item_title,
        Some(sel!(hide:)),
        Some(KeyEquivalent {
            key: ns_string!("h"),
            masks: None,
        }),
    );

    // Hide other applications menu item
    let hide_others_item_title = ns_string!("Hide Others");
    let hide_others_item = menu_item(
        mtm,
        hide_others_item_title,
        Some(sel!(hideOtherApplications:)),
        Some(KeyEquivalent {
            key: ns_string!("h"),
            masks: Some(
                NSEventModifierFlags::NSEventModifierFlagOption
                    | NSEventModifierFlags::NSEventModifierFlagCommand,
            ),
        }),
    );

    // Show applications menu item
    let show_all_item_title = ns_string!("Show All");
    let show_all_item = menu_item(
        mtm,
        show_all_item_title,
        Some(sel!(unhideAllApplications:)),
        None,
    );

    // Separator menu item
    let sep = NSMenuItem::separatorItem(mtm);

    // Quit application menu item
    let quit_item_title = ns_string!("Quit ").stringByAppendingString(&process_name);
    let quit_item = menu_item(
        mtm,
        &quit_item_title,
        Some(sel!(terminate:)),
        Some(KeyEquivalent {
            key: ns_string!("q"),
            masks: Some(NSEventModifierFlags::NSEventModifierFlagCommand),
        }),
    );

    // New window menu item
    let create_window_item_title = ns_string!("New Window");
    let create_window_item = menu_item(
        mtm,
        create_window_item_title,
        Some(sel!(rioCreateWindow:)),
        Some(KeyEquivalent {
            key: ns_string!("n"),
            masks: Some(NSEventModifierFlags::NSEventModifierFlagCommand),
        }),
    );

    let create_tab_item_title = ns_string!("New Tab");
    let create_tab_item = menu_item(
        mtm,
        create_tab_item_title,
        Some(sel!(rioCreateTab:)),
        Some(KeyEquivalent {
            key: ns_string!("t"),
            masks: Some(NSEventModifierFlags::NSEventModifierFlagCommand),
        }),
    );

    let close_item_title = ns_string!("Close");
    let close_item = menu_item(
        mtm,
        close_item_title,
        Some(sel!(rioClose:)),
        Some(KeyEquivalent {
            key: ns_string!("w"),
            masks: Some(NSEventModifierFlags::NSEventModifierFlagCommand),
        }),
    );

    let create_split_horizontally_item_title = ns_string!("Split Right");
    let create_split_horizontally_item = menu_item(
        mtm,
        create_split_horizontally_item_title,
        Some(sel!(rioSplitRight:)),
        Some(KeyEquivalent {
            key: ns_string!("d"),
            masks: Some(NSEventModifierFlags::NSEventModifierFlagCommand),
        }),
    );

    let create_split_vertical_item_title = ns_string!("Split Down");
    let create_split_vertical_item = menu_item(
        mtm,
        create_split_vertical_item_title,
        Some(sel!(rioSplitDown:)),
        Some(KeyEquivalent {
            key: ns_string!("d"),
            masks: Some(
                NSEventModifierFlags::NSEventModifierFlagCommand
                    | NSEventModifierFlags::NSEventModifierFlagShift,
            ),
        }),
    );

    let copy_title = ns_string!("Copy");
    let copy_item = menu_item(
        mtm,
        copy_title,
        Some(sel!(copy:)),
        Some(KeyEquivalent {
            key: ns_string!("c"),
            masks: Some(NSEventModifierFlags::NSEventModifierFlagCommand),
        }),
    );
    let paste_title = ns_string!("Paste");
    let paste_item = menu_item(
        mtm,
        paste_title,
        Some(sel!(paste:)),
        Some(KeyEquivalent {
            key: ns_string!("v"),
            masks: Some(NSEventModifierFlags::NSEventModifierFlagCommand),
        }),
    );

    let shell_menu = unsafe { NSMenu::initWithTitle(mtm.alloc(), ns_string!("Shell")) };
    let edit_menu = unsafe { NSMenu::initWithTitle(mtm.alloc(), ns_string!("Edit")) };
    let view_menu = unsafe { NSMenu::initWithTitle(mtm.alloc(), ns_string!("View")) };
    let window_menu = unsafe { NSMenu::initWithTitle(mtm.alloc(), ns_string!("Window")) };
    let help_menu = unsafe { NSMenu::initWithTitle(mtm.alloc(), ns_string!("Help")) };

    app_menu.addItem(&about_item);
    app_menu.addItem(&sep_first);
    app_menu.addItem(&services_item);
    app_menu.addItem(&open_config);
    app_menu.addItem(&hide_item);
    app_menu.addItem(&hide_others_item);
    app_menu.addItem(&show_all_item);
    app_menu.addItem(&sep);
    app_menu.addItem(&quit_item);
    app_menu_item.setSubmenu(Some(&app_menu));
    shell_menu.addItem(&create_window_item);
    shell_menu.addItem(&create_tab_item);
    shell_menu.addItem(&close_item);
    shell_menu.addItem(&create_split_horizontally_item);
    shell_menu.addItem(&create_split_vertical_item);
    shell_menu_item.setSubmenu(Some(&shell_menu));
    edit_menu.addItem(&copy_item);
    edit_menu.addItem(&paste_item);
    edit_menu_item.setSubmenu(Some(&edit_menu));
    view_menu_item.setSubmenu(Some(&view_menu));
    window_menu_item.setSubmenu(Some(&window_menu));
    help_menu_item.setSubmenu(Some(&help_menu));

    unsafe {
        app.setServicesMenu(Some(&services_menu));
        app.setWindowsMenu(Some(&window_menu));
        app.setHelpMenu(Some(&help_menu));
    };
    app.setMainMenu(Some(&menubar));
}

pub fn menu_item(
    mtm: MainThreadMarker,
    title: &NSString,
    selector: Option<Sel>,
    key_equivalent: Option<KeyEquivalent<'_>>,
) -> Retained<NSMenuItem> {
    let (key, masks) = match key_equivalent {
        Some(ke) => (ke.key, ke.masks),
        None => (ns_string!(""), None),
    };
    let item = unsafe {
        NSMenuItem::initWithTitle_action_keyEquivalent(mtm.alloc(), title, selector, key)
    };
    if let Some(masks) = masks {
        item.setKeyEquivalentModifierMask(masks)
    }

    item
}