feat: export history

README.md

@@ -11,15 +11,17 @@
 
 **Latest:**
 
-- `Mac`: [ChatGPT_0.1.5_x64.dmg](https://github.com/lencx/ChatGPT/releases/download/v0.1.5/ChatGPT_0.1.5_x64.dmg)
-- `Linux`: [chat-gpt_0.1.5_amd64.deb](https://github.com/lencx/ChatGPT/releases/download/v0.1.5/chat-gpt_0.1.5_amd64.deb)
-- `Windows`: [ChatGPT_0.1.5_x64_en-US.msi](https://github.com/lencx/ChatGPT/releases/download/v0.1.5/ChatGPT_0.1.5_x64_en-US.msi)
+- `Mac`: [ChatGPT_0.1.6_x64.dmg](https://github.com/lencx/ChatGPT/releases/download/v0.1.5/ChatGPT_0.1.6_x64.dmg)
+- `Linux`: [chat-gpt_0.1.6_amd64.deb](https://github.com/lencx/ChatGPT/releases/download/v0.1.5/chat-gpt_0.1.6_amd64.deb)
+- `Windows`: [ChatGPT_0.1.6_x64_en-US.msi](https://github.com/lencx/ChatGPT/releases/download/v0.1.5/ChatGPT_0.1.6_x64_en-US.msi)
 
 [Other version...](https://github.com/lencx/ChatGPT/releases)
 
 ## Features
 
 - multi-platform: `macOS` `Linux` `Windows`
+- export ChatGPT history (PNG, PDF and Share Link)
+- always on top (whether the window should always be on top of other windows)
 - inject script
 - auto updater
 - app menu
@@ -32,10 +34,9 @@
 <img width="600" src="./assets/chat.png" alt="chat">
 <img width="600" src="./assets/auto-update.png" alt="auto update">
 
-## TODO
+## Related
 
-- [ ] export chat history
-- [ ] ...
+- [ChatGPT Export and Share](https://github.com/liady/ChatGPT-pdf) - A Chrome extension for downloading your ChatGPT history to PNG, PDF or creating a sharable link
 
 ## FAQ
 

src-tauri/src/app/cmd.rs

@@ -1,12 +1,14 @@
-use tauri::Manager;
+use crate::utils;
+use std::fs;
+use tauri::{api, command, AppHandle, Manager};
 
-#[tauri::command]
-pub fn drag_window(app: tauri::AppHandle) {
+#[command]
+pub fn drag_window(app: AppHandle) {
     app.get_window("core").unwrap().start_dragging().unwrap();
 }
 
-#[tauri::command]
-pub fn fullscreen(app: tauri::AppHandle) {
+#[command]
+pub fn fullscreen(app: AppHandle) {
     let win = app.get_window("core").unwrap();
     if win.is_fullscreen().unwrap() {
         win.set_fullscreen(false).unwrap();
@@ -14,3 +16,15 @@ pub fn fullscreen(app: tauri::AppHandle) {
         win.set_fullscreen(true).unwrap();
     }
 }
+
+#[command]
+pub fn download(_app: AppHandle, name: String, blob: Vec<u8>) {
+    let path = api::path::download_dir().unwrap().join(name);
+    fs::write(&path, blob).unwrap();
+    utils::open_file(path);
+}
+
+#[command]
+pub fn open_link(app: AppHandle, url: String) {
+    api::shell::open(&app.shell_scope(), url, None).unwrap();
+}

src-tauri/src/app/menu.rs

@@ -1,22 +1,17 @@
-use crate::utils;
+use crate::{conf, utils};
 use tauri::{
     utils::assets::EmbeddedAssets, AboutMetadata, AppHandle, Context, CustomMenuItem, Manager,
-    Menu, MenuItem, Submenu, SystemTray, SystemTrayEvent, SystemTrayMenu, SystemTrayMenuItem,
-    WindowMenuEvent,
+    Menu, MenuItem, Submenu, SystemTray, SystemTrayEvent, SystemTrayMenu, WindowMenuEvent,
 };
 
 // --- Menu
-pub fn init(context: &Context<EmbeddedAssets>) -> Menu {
+pub fn init(chat_conf: &conf::ChatConfJson, 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)
@@ -25,6 +20,28 @@ pub fn init(context: &Context<EmbeddedAssets>) -> Menu {
             .add_native_item(MenuItem::Quit),
     );
 
+    let always_on_top = CustomMenuItem::new("always_on_top".to_string(), "Always On Top")
+        .accelerator("CmdOrCtrl+T");
+
+    let preferences_menu = Submenu::new(
+        "Preferences",
+        Menu::new()
+            .add_item(
+                CustomMenuItem::new("inject_script".to_string(), "Inject Script")
+                    .accelerator("CmdOrCtrl+J"),
+            )
+            .add_item(if chat_conf.always_on_top {
+                always_on_top.selected()
+            } else {
+                always_on_top
+            })
+            .add_native_item(MenuItem::Separator)
+            .add_item(
+                CustomMenuItem::new("awesome".to_string(), "Awesome ChatGPT")
+                    .accelerator("CmdOrCtrl+Z"),
+            ),
+    );
+
     let edit_menu = Submenu::new(
         "Edit",
         Menu::new()
@@ -74,6 +91,7 @@ pub fn init(context: &Context<EmbeddedAssets>) -> Menu {
 
     Menu::new()
         .add_submenu(app_menu)
+        .add_submenu(preferences_menu)
         .add_submenu(edit_menu)
         .add_submenu(view_menu)
         .add_submenu(help_menu)
@@ -83,12 +101,27 @@ 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 state: tauri::State<conf::ChatState> = app.state();
     let script_path = utils::script_path().to_string_lossy().to_string();
-    let issues_url = "https://github.com/lencx/ChatGPT/issues".to_string();
+    let menu_id = event.menu_item_id();
 
-    match event.menu_item_id() {
-        // App
-        "inject_script" => inject_script(&app, script_path),
+    let core_window = app.get_window("core").unwrap();
+    let menu_handle = core_window.menu_handle();
+
+    match menu_id {
+        // Preferences
+        "inject_script" => open(&app, script_path),
+        "awesome" => open(&app, conf::AWESOME_URL.to_string()),
+        "always_on_top" => {
+            let mut always_on_top = state.always_on_top.lock().unwrap();
+            *always_on_top = !*always_on_top;
+            menu_handle
+                .get_item(menu_id)
+                .set_selected(*always_on_top)
+                .unwrap();
+            win.set_always_on_top(*always_on_top).unwrap();
+            conf::ChatConfJson::update_chat_conf(*always_on_top);
+        }
         // View
         "reload" => win.eval("window.location.reload()").unwrap(),
         "go_back" => win.eval("window.history.go(-1)").unwrap(),
@@ -111,7 +144,7 @@ pub fn menu_handler(event: WindowMenuEvent<tauri::Wry>) {
             )
             .unwrap(),
         // Help
-        "report_bug" => inject_script(&app, issues_url),
+        "report_bug" => open(&app, conf::ISSUES_URL.to_string()),
         "dev_tools" => {
             win.open_devtools();
             win.close_devtools();
@@ -122,35 +155,24 @@ 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::new().with_menu(SystemTrayMenu::new())
 }
 
 // --- SystemTray Event
 pub fn tray_handler(app: &AppHandle, event: SystemTrayEvent) {
-    let script_path = utils::script_path().to_string_lossy().to_string();
     let win = app.get_window("core").unwrap();
 
-    if let SystemTrayEvent::MenuItemClick { id, .. } = event {
-        match id.as_str() {
-            "quit" => std::process::exit(0),
-            "show" => win.show().unwrap(),
-            "hide" => win.hide().unwrap(),
-            "inject_script" => inject_script(app, script_path),
-            _ => (),
+    if let SystemTrayEvent::LeftClick { .. } = event {
+        // TODO: tray window
+        if win.is_visible().unwrap() {
+            win.hide().unwrap();
+        } else {
+            win.show().unwrap();
+            win.set_focus().unwrap();
         }
     }
 }
 
-pub fn inject_script(app: &AppHandle, path: String) {
+pub fn open(app: &AppHandle, path: String) {
     tauri::api::shell::open(&app.shell_scope(), path, None).unwrap();
 }

src-tauri/src/app/setup.rs

@@ -1,10 +1,13 @@
-use crate::utils;
+use crate::{conf, utils};
 use tauri::{utils::config::WindowUrl, window::WindowBuilder, App};
 
 #[cfg(target_os = "macos")]
 use tauri::TitleBarStyle;
 
-pub fn init(app: &mut App) -> std::result::Result<(), Box<dyn std::error::Error>> {
+pub fn init(
+    app: &mut App,
+    chat_conf: conf::ChatConfJson,
+) -> std::result::Result<(), Box<dyn std::error::Error>> {
     let conf = utils::get_tauri_conf().unwrap();
     let url = conf.build.dev_path.to_string();
 
@@ -15,12 +18,13 @@ pub fn init(app: &mut App) -> std::result::Result<(), Box<dyn std::error::Error>
         .inner_size(800.0, 600.0)
         .hidden_title(true)
         .title_bar_style(TitleBarStyle::Overlay)
+        .always_on_top(chat_conf.always_on_top)
         .initialization_script(&utils::user_script())
         .initialization_script(include_str!("../assets/html2canvas.js"))
         .initialization_script(include_str!("../assets/jspdf.js"))
         .initialization_script(include_str!("../assets/core.js"))
         .initialization_script(include_str!("../assets/import.js"))
-        .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")
+        .user_agent(conf::USER_AGENT)
         .build()?;
 
     #[cfg(not(target_os = "macos"))]
@@ -29,12 +33,13 @@ pub fn init(app: &mut App) -> std::result::Result<(), Box<dyn std::error::Error>
         .resizable(true)
         .fullscreen(false)
         .inner_size(800.0, 600.0)
+        .always_on_top(chat_conf.always_on_top)
         .initialization_script(&utils::user_script())
         .initialization_script(include_str!("../assets/html2canvas.js"))
         .initialization_script(include_str!("../assets/jspdf.js"))
         .initialization_script(include_str!("../assets/core.js"))
         .initialization_script(include_str!("../assets/import.js"))
-        .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")
+        .user_agent(conf::USER_AGENT)
         .build()?;
 
     Ok(())

src-tauri/src/assets/core.js

@@ -1,5 +1,4 @@
 // *** Core Script - IPC ***
-document.addEventListener('DOMContentLoaded', async () => {
 const uid = () => window.crypto.getRandomValues(new Uint32Array(1))[0];
 function transformCallback(callback = () => {}, once = false) {
   const identifier = uid();
@@ -36,6 +35,11 @@ document.addEventListener('DOMContentLoaded', async () => {
   });
 }
 
+window.uid = uid;
+window.invoke = invoke;
+window.transformCallback = transformCallback;
+
+async function init() {
   async function platform() {
     return invoke('platform', {
       __tauriModule: 'Os',
@@ -74,4 +78,13 @@ document.addEventListener('DOMContentLoaded', async () => {
       }
     }
   });
-})
+}
+
+if (
+  document.readyState === "complete" ||
+  document.readyState === "interactive"
+) {
+  init();
+} else {
+  document.addEventListener("DOMContentLoaded", init);
+}

src-tauri/src/assets/import.js

@@ -109,15 +109,7 @@ function handleImg(imgData) {
   for (let i = 0; i < binaryData.length; i++) {
     data.push(binaryData.charCodeAt(i));
   }
-  const blob = new Blob([new Uint8Array(data)], { type: "image/png" });
-  const url = URL.createObjectURL(blob);
-
-  // window.open(url, "_blank");
-
-  //   const a = document.createElement("a");
-  //   a.href = url;
-  //   a.download = "chat-gpt-image.png";
-  //   a.click();
+  invoke('download', { name: `chatgpt-${Date.now()}.png`, blob: Array.from(new Uint8Array(data)) });
 }
 
 function handlePdf(imgData, canvas, pixelRatio) {
@@ -130,7 +122,9 @@ function handlePdf(imgData, canvas, pixelRatio) {
   var pdfWidth = pdf.internal.pageSize.getWidth();
   var pdfHeight = pdf.internal.pageSize.getHeight();
   pdf.addImage(imgData, "PNG", 0, 0, pdfWidth, pdfHeight);
-  pdf.save("chat-gpt.pdf");
+
+  const data = pdf.__private__.getArrayBuffer(pdf.__private__.buildDocument());
+  invoke('download', { name: `chatgpt-${Date.now()}.pdf`, blob: Array.from(new Uint8Array(data)) });
 }
 
 class Elements {
@@ -206,7 +200,7 @@ async function sendRequest() {
   })
     .then((response) => response.json())
     .then((data) => {
-      window.open(data.url, "_blank");
+      invoke('open_link', { url: data.url });
     });
 }
 

src-tauri/src/conf.rs

@@ -0,0 +1,66 @@
+use crate::utils::{chat_root, create_file, exists};
+use std::fs;
+use std::path::PathBuf;
+use std::sync::Mutex;
+
+pub const USER_AGENT: &str = "5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36";
+pub const ISSUES_URL: &str = "https://github.com/lencx/ChatGPT/issues";
+pub const AWESOME_URL: &str = "https://github.com/lencx/ChatGPT/blob/main/AWESOME.md";
+
+pub struct ChatState {
+    pub always_on_top: Mutex<bool>,
+}
+
+impl ChatState {
+    pub fn default(chat_conf: &ChatConfJson) -> Self {
+        ChatState {
+            always_on_top: Mutex::new(chat_conf.always_on_top),
+        }
+    }
+}
+
+#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
+pub struct ChatConfJson {
+    pub always_on_top: bool,
+}
+
+impl ChatConfJson {
+    /// init chat.conf.json
+    /// path: ~/.chatgpt/chat.conf.json
+    pub fn init() -> PathBuf {
+        let conf_file = ChatConfJson::conf_path();
+        if !exists(&conf_file) {
+            create_file(&conf_file).unwrap();
+            fs::write(&conf_file, r#"{"always_on_top": false}"#).unwrap();
+        }
+        conf_file
+    }
+
+    pub fn conf_path() -> PathBuf {
+        chat_root().join("chat.conf.json")
+    }
+
+    pub fn get_chat_conf() -> Self {
+        let config_file = fs::read_to_string(ChatConfJson::conf_path()).unwrap();
+        let config: serde_json::Value =
+            serde_json::from_str(&config_file).expect("failed to parse chat.conf.json");
+        serde_json::from_value(config).unwrap_or_else(|_| ChatConfJson::chat_conf_default())
+    }
+
+    pub fn update_chat_conf(always_on_top: bool) {
+        let mut conf = ChatConfJson::get_chat_conf();
+        conf.always_on_top = always_on_top;
+        fs::write(
+            ChatConfJson::conf_path(),
+            serde_json::to_string(&conf).unwrap(),
+        )
+        .unwrap();
+    }
+
+    pub fn chat_conf_default() -> Self {
+        serde_json::from_value(serde_json::json!({
+            "always_on_top": false,
+        }))
+        .unwrap()
+    }
+}

src-tauri/src/main.rs

@@ -4,20 +4,41 @@
 )]
 
 mod app;
+mod conf;
 mod utils;
 
 use app::{cmd, menu, setup};
+use conf::ChatConfJson;
 
 fn main() {
+    ChatConfJson::init();
     let context = tauri::generate_context!();
+    let chat_conf = ChatConfJson::get_chat_conf();
+    let chat_conf2 = chat_conf.clone();
 
     tauri::Builder::default()
-        .invoke_handler(tauri::generate_handler![cmd::drag_window, cmd::fullscreen])
-        .setup(setup::init)
-        .menu(menu::init(&context))
+        .manage(conf::ChatState::default(&chat_conf))
+        .invoke_handler(tauri::generate_handler![
+            cmd::drag_window,
+            cmd::fullscreen,
+            cmd::download,
+            cmd::open_link
+        ])
+        .setup(|app| setup::init(app, chat_conf2))
+        .menu(menu::init(&chat_conf, &context))
         .system_tray(menu::tray_menu())
         .on_menu_event(menu::menu_handler)
         .on_system_tray_event(menu::tray_handler)
+        .on_window_event(|event| {
+            // https://github.com/tauri-apps/tauri/discussions/2684
+            if let tauri::WindowEvent::CloseRequested { api, .. } = event.event() {
+                // TODO: https://github.com/tauri-apps/tauri/issues/3084
+                // event.window().hide().unwrap();
+                // https://github.com/tauri-apps/tao/pull/517
+                event.window().minimize().unwrap();
+                api.prevent_close();
+            }
+        })
         .run(context)
         .expect("error while running ChatGPT application");
 }

src-tauri/src/utils.rs

@@ -1,8 +1,15 @@
 use anyhow::Result;
-use std::fs::{self, File};
-use std::path::{Path, PathBuf};
+use std::{
+    fs::{self, File},
+    path::{Path, PathBuf},
+    process::Command,
+};
 use tauri::utils::config::Config;
 
+pub fn chat_root() -> PathBuf {
+    tauri::api::path::home_dir().unwrap().join(".chatgpt")
+}
+
 pub fn get_tauri_conf() -> Option<Config> {
     let config_file = include_str!("../tauri.conf.json");
     let config: Config =
@@ -22,8 +29,7 @@ pub fn create_file(path: &Path) -> Result<File> {
 }
 
 pub fn script_path() -> PathBuf {
-    let root = tauri::api::path::home_dir().unwrap().join(".chatgpt");
-    let script_file = root.join("main.js");
+    let script_file = chat_root().join("main.js");
     if !exists(&script_file) {
         create_file(&script_file).unwrap();
         fs::write(&script_file, format!("// *** ChatGPT User Script ***\n// @github: https://github.com/lencx/ChatGPT \n// @path: {}\n\nconsole.log('🤩 Hello ChatGPT!!!');", &script_file.to_string_lossy())).unwrap();
@@ -39,3 +45,19 @@ pub fn user_script() -> String {
         user_script_content
     )
 }
+
+pub fn open_file(path: PathBuf) {
+    #[cfg(target_os = "macos")]
+    Command::new("open").arg("-R").arg(path).spawn().unwrap();
+
+    #[cfg(target_os = "windows")]
+    Command::new("explorer")
+        .arg("/select,")
+        .arg(path)
+        .spawn()
+        .unwrap();
+
+    // https://askubuntu.com/a/31071
+    #[cfg(target_os = "linux")]
+    Command::new("xdg-open").arg(path).spawn().unwrap();
+}

src-tauri/tauri.conf.json

@@ -44,6 +44,10 @@
       "shortDescription": "ChatGPT",
       "targets": "all",
       "windows": {
+        "webviewInstallMode": {
+          "silent": true,
+          "type": "downloadBootstrapper"
+        },
         "certificateThumbprint": null,
         "digestAlgorithm": "sha256",
         "timestampUrl": ""