v0.1.4

UPDATE_LOG.md

@@ -1,12 +1,22 @@
 # UPDATE LOG
 
+## v0.1.4
+
+feat:
+- beautify icons
+- add system tray menu
+
+## v0.1.3
+
+fix: only mac supports `TitleBarStyle`
+
 ## v0.1.2
 
-- initialization
+initialization
 
 ## v0.1.1
 
-- initialization
+initialization
 
 ## v0.1.0
 

src-tauri/src/app/menu.rs

@@ -1,7 +1,8 @@
 use crate::utils;
 use tauri::{
     utils::assets::EmbeddedAssets, AboutMetadata, AppHandle, Context, CustomMenuItem, Manager,
-    Menu, MenuItem, Submenu, SystemTrayEvent, WindowMenuEvent,
+    Menu, MenuItem, Submenu, SystemTray, SystemTrayEvent, SystemTrayMenu, SystemTrayMenuItem,
+    WindowMenuEvent,
 };
 
 // --- Menu
@@ -82,56 +83,35 @@ pub fn init(context: &Context<EmbeddedAssets>) -> Menu {
 pub fn menu_handler(event: WindowMenuEvent<tauri::Wry>) {
     let win = Some(event.window()).unwrap();
     let app = win.app_handle();
+    let script_path = utils::script_path().to_string_lossy().to_string();
+    let issues_url = "https://github.com/lencx/ChatGPT/issues".to_string();
 
     match event.menu_item_id() {
         // App
-        "inject_script" => {
+        "inject_script" => inject_script(&app, script_path),
-            tauri::api::shell::open(
-                &app.shell_scope(),
-                utils::script_path().to_string_lossy(),
-                None,
-            )
-            .unwrap();
-        }
         // View
-        "go_back" => {
+        "reload" => win.eval("window.location.reload()").unwrap(),
-            win.eval("window.history.go(-1)").unwrap();
+        "go_back" => win.eval("window.history.go(-1)").unwrap(),
-        }
+        "go_forward" => win.eval("window.history.go(1)").unwrap(),
-        "go_forward" => {
+        "scroll_top" => win
-            win.eval("window.history.go(1)").unwrap();
+            .eval(
-        }
-        "scroll_top" => {
-            win.eval(
                 r#"window.scroll({
                 top: 0,
                 left: 0,
                 behavior: "smooth"
-            })"#,
+                })"#,
             )
-            .unwrap();
+            .unwrap(),
-        }
+        "scroll_bottom" => win
-        "scroll_bottom" => {
+            .eval(
-            win.eval(
                 r#"window.scroll({
                 top: document.body.scrollHeight,
                 left: 0,
-                behavior: "smooth",
+                behavior: "smooth"})"#,
-            })"#,
             )
-            .unwrap();
+            .unwrap(),
-        }
-        "reload" => {
-            win.eval("window.location.reload()").unwrap();
-        }
         // Help
-        "report_bug" => {
+        "report_bug" => inject_script(&app, issues_url),
-            tauri::api::shell::open(
-                &app.shell_scope(),
-                "https://github.com/lencx/ChatGPT/issues",
-                None,
-            )
-            .unwrap();
-        }
         "dev_tools" => {
             win.open_devtools();
             win.close_devtools();
@@ -140,20 +120,37 @@ pub fn menu_handler(event: WindowMenuEvent<tauri::Wry>) {
     }
 }
 
+// --- SystemTray Menu
+pub fn tray_menu() -> SystemTray {
+    SystemTray::new().with_menu(
+        SystemTrayMenu::new()
+            .add_item(CustomMenuItem::new("show".to_string(), "Show ChatGPT"))
+            .add_item(CustomMenuItem::new("hide".to_string(), "Hide ChatGPT"))
+            .add_item(CustomMenuItem::new(
+                "inject_script".to_string(),
+                "Inject Script",
+            ))
+            .add_native_item(SystemTrayMenuItem::Separator)
+            .add_item(CustomMenuItem::new("quit".to_string(), "Quit ChatGPT")),
+    )
+}
+
 // --- SystemTray Event
 pub fn tray_handler(app: &AppHandle, event: SystemTrayEvent) {
-    if let SystemTrayEvent::LeftClick {
+    let script_path = utils::script_path().to_string_lossy().to_string();
-        position: _,
+    let win = app.get_window("core").unwrap();
-        size: _,
+
-        ..
+    if let SystemTrayEvent::MenuItemClick { id, .. } = event {
-    } = event
+        match id.as_str() {
-    {
+            "quit" => std::process::exit(0),
-        let win = app.get_window("core").unwrap();
+            "show" => win.show().unwrap(),
-        if win.is_visible().unwrap() {
+            "hide" => win.hide().unwrap(),
-            win.hide().unwrap();
+            "inject_script" => inject_script(app, script_path),
-        } else {
+            _ => (),
-            win.show().unwrap();
-            win.set_focus().unwrap();
         }
     }
 }
+
+pub fn inject_script(app: &AppHandle, path: String) {
+    tauri::api::shell::open(&app.shell_scope(), path, None).unwrap();
+}

src-tauri/src/app/setup.rs

@@ -27,7 +27,7 @@ pub fn init(app: &mut App) -> std::result::Result<(), Box<dyn std::error::Error>
         .initialization_script(include_str!("../core.js"))
         .initialization_script(&utils::user_script())
         .inner_size(800.0, 600.0)
-        .hidden_title(true)
+        .title("ChatGPT")
         .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()?;
 

src-tauri/src/core.js

@@ -36,16 +36,26 @@ document.addEventListener('DOMContentLoaded', async () => {
     });
   }
 
-  const topStyleDom = document.createElement("style");
+  async function platform() {
-  topStyleDom.innerHTML = `#chatgpt-app-window-top{position:fixed;top:0;z-index:999999999;width:100%;height:24px;background:transparent;cursor:grab;cursor:-webkit-grab;user-select:none;-webkit-user-select:none;}#chatgpt-app-window-top:active {cursor:grabbing;cursor:-webkit-grabbing;}`;
+    return invoke('platform', {
-  document.head.appendChild(topStyleDom);
+      __tauriModule: 'Os',
-  const topDom = document.createElement("div");
+      message: { cmd: 'platform' }
-  topDom.id = "chatgpt-app-window-top";
+    });
-  document.body.appendChild(topDom);
+  }
 
-  topDom.addEventListener("mousedown", () => invoke("drag_window"));
+  const _platform = await platform();
-  topDom.addEventListener("touchstart", () => invoke("drag_window"));
+  if (/darwin/.test(_platform)) {
-  topDom.addEventListener("dblclick", () => invoke("fullscreen"));
+    const topStyleDom = document.createElement("style");
+    topStyleDom.innerHTML = `#chatgpt-app-window-top{position:fixed;top:0;z-index:999999999;width:100%;height:24px;background:transparent;cursor:grab;cursor:-webkit-grab;user-select:none;-webkit-user-select:none;}#chatgpt-app-window-top:active {cursor:grabbing;cursor:-webkit-grabbing;}`;
+    document.head.appendChild(topStyleDom);
+    const topDom = document.createElement("div");
+    topDom.id = "chatgpt-app-window-top";
+    document.body.appendChild(topDom);
+
+    topDom.addEventListener("mousedown", () => invoke("drag_window"));
+    topDom.addEventListener("touchstart", () => invoke("drag_window"));
+    topDom.addEventListener("dblclick", () => invoke("fullscreen"));
+  }
 
   document.addEventListener("click", (e) => {
     const origin = e.target.closest("a");

src-tauri/src/main.rs

@@ -7,7 +7,6 @@ mod app;
 mod utils;
 
 use app::{cmd, menu, setup};
-use tauri::SystemTray;
 
 fn main() {
     let context = tauri::generate_context!();
@@ -16,7 +15,7 @@ fn main() {
         .invoke_handler(tauri::generate_handler![cmd::drag_window, cmd::fullscreen])
         .setup(setup::init)
         .menu(menu::init(&context))
-        .system_tray(SystemTray::new())
+        .system_tray(menu::tray_menu())
         .on_menu_event(menu::menu_handler)
         .on_system_tray_event(menu::tray_handler)
         .run(context)

src-tauri/tauri.conf.json

@@ -7,7 +7,7 @@
   },
   "package": {
     "productName": "ChatGPT",
-    "version": "0.1.2"
+    "version": "0.1.4"
   },
   "tauri": {
     "allowlist": {