ChatGPT/src-tauri/src/wa/menu.rs

use tauri::{
    utils::assets::EmbeddedAssets, AboutMetadata, AppHandle, Context, CustomMenuItem, Manager,
    Menu, MenuItem, Submenu, SystemTrayEvent, WindowMenuEvent,
};

use crate::utils;

// --- Menu
pub fn init(context: &Context<EmbeddedAssets>) -> Menu {
    let name = &context.package_info().name;
    let app_menu = Submenu::new(
        name,
        Menu::new()
            .add_native_item(MenuItem::About(name.into(), AboutMetadata::default()))
            .add_native_item(MenuItem::Separator)
            .add_item(
                CustomMenuItem::new("inject_script".to_string(), "Inject Script")
                    .accelerator("CmdOrCtrl+J"),
            )
            .add_native_item(MenuItem::Separator)
            .add_native_item(MenuItem::Hide)
            .add_native_item(MenuItem::HideOthers)
            .add_native_item(MenuItem::ShowAll)
            .add_native_item(MenuItem::Separator)
            .add_native_item(MenuItem::Quit),
    );

    let edit_menu = Submenu::new(
        "Edit",
        Menu::new()
            .add_native_item(MenuItem::Undo)
            .add_native_item(MenuItem::Redo)
            .add_native_item(MenuItem::Separator)
            .add_native_item(MenuItem::Cut)
            .add_native_item(MenuItem::Copy)
            .add_native_item(MenuItem::Paste)
            .add_native_item(MenuItem::SelectAll),
    );

    let view_menu = Submenu::new(
        "View",
        Menu::new()
            .add_item(
                CustomMenuItem::new("go_back".to_string(), "Go Back").accelerator("CmdOrCtrl+Left"),
            )
            .add_item(
                CustomMenuItem::new("go_forward".to_string(), "Go Forward")
                    .accelerator("CmdOrCtrl+Right"),
            )
            .add_item(
                CustomMenuItem::new("scroll_top".to_string(), "Scroll to Top of Screen")
                    .accelerator("CmdOrCtrl+Up"),
            )
            .add_item(
                CustomMenuItem::new("scroll_bottom".to_string(), "Scroll to Bottom of Screen")
                    .accelerator("CmdOrCtrl+Down"),
            )
            .add_native_item(MenuItem::Separator)
            .add_item(
                CustomMenuItem::new("reload".to_string(), "Refresh the Screen")
                    .accelerator("CmdOrCtrl+R"),
            ),
    );

    let help_menu = Submenu::new(
        "Help",
        Menu::new()
            .add_item(CustomMenuItem::new("report_bug".to_string(), "Report Bug"))
            .add_item(
                CustomMenuItem::new("dev_tools".to_string(), "Toggle Developer Tools")
                    .accelerator("CmdOrCtrl+Shift+I"),
            ),
    );

    Menu::new()
        .add_submenu(app_menu)
        .add_submenu(edit_menu)
        .add_submenu(view_menu)
        .add_submenu(help_menu)
}

// --- Menu Event
pub fn menu_handler(event: WindowMenuEvent<tauri::Wry>) {
    let win = Some(event.window()).unwrap();
    let app = win.app_handle();

    match event.menu_item_id() {
        // App
        "inject_script" => {
            tauri::api::shell::open(
                &app.shell_scope(),
                utils::script_path().to_string_lossy(),
                None,
            )
            .unwrap();
        }
        // View
        "go_back" => {
            win.eval("window.history.go(-1)").unwrap();
        }
        "go_forward" => {
            win.eval("window.history.go(1)").unwrap();
        }
        "scroll_top" => {
            win.eval(
                r#"window.scroll({
                top: 0,
                left: 0,
                behavior: "smooth"
            })"#,
            )
            .unwrap();
        }
        "scroll_bottom" => {
            win.eval(
                r#"window.scroll({
                top: document.body.scrollHeight,
                left: 0,
                behavior: "smooth",
            })"#,
            )
            .unwrap();
        }
        "reload" => {
            win.eval("window.location.reload()").unwrap();
        }
        // Help
        "report_bug" => {
            tauri::api::shell::open(
                &app.shell_scope(),
                "https://github.com/lencx/ChatGPT/issues",
                None,
            )
            .unwrap();
        }
        "dev_tools" => {
            win.open_devtools();
            win.close_devtools();
        }
        _ => (),
    }
}

// --- SystemTray Event
pub fn tray_handler(app: &AppHandle, event: SystemTrayEvent) {
    if let SystemTrayEvent::LeftClick {
        position: _,
        size: _,
        ..
    } = event
    {
        let win = app.get_window("core").unwrap();
        if win.is_visible().unwrap() {
            win.hide().unwrap();
        } else {
            win.show().unwrap();
            win.set_focus().unwrap();
        }
    }
}