chore: fmt

src-tauri/src/app/cmd.rs

@@ -0,0 +1,16 @@
+use tauri::Manager;
+
+#[tauri::command]
+pub fn drag_window(app: tauri::AppHandle) {
+    app.get_window("core").unwrap().start_dragging().unwrap();
+}
+
+#[tauri::command]
+pub fn fullscreen(app: tauri::AppHandle) {
+    let win = app.get_window("core").unwrap();
+    if win.is_fullscreen().unwrap() {
+        win.set_fullscreen(false).unwrap();
+    } else {
+        win.set_fullscreen(true).unwrap();
+    }
+}

src-tauri/src/app/menu.rs

@@ -0,0 +1,159 @@
+use crate::utils;
+use tauri::{
+    utils::assets::EmbeddedAssets, AboutMetadata, AppHandle, Context, CustomMenuItem, Manager,
+    Menu, MenuItem, Submenu, SystemTrayEvent, WindowMenuEvent,
+};
+
+// --- 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();
+        }
+    }
+}

src-tauri/src/app/mod.rs

@@ -0,0 +1,3 @@
+pub mod cmd;
+pub mod menu;
+pub mod setup;

src-tauri/src/app/setup.rs

@@ -0,0 +1,20 @@
+use crate::utils;
+use tauri::{utils::config::WindowUrl, window::WindowBuilder, App, TitleBarStyle};
+
+pub fn init(app: &mut App) -> std::result::Result<(), Box<dyn std::error::Error>> {
+    let conf = utils::get_tauri_conf().unwrap();
+    let url = conf.build.dev_path.to_string();
+
+    WindowBuilder::new(app, "core", WindowUrl::App(url.into()))
+        .resizable(true)
+        .fullscreen(false)
+        .initialization_script(include_str!("../core.js"))
+        .initialization_script(&utils::user_script())
+        .title_bar_style(TitleBarStyle::Overlay)
+        .inner_size(800.0, 600.0)
+        .hidden_title(true)
+        .user_agent("5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36")
+        .build()?;
+
+    Ok(())
+}