v0.1.7

.gitattributes

@@ -0,0 +1 @@
+*.js linguist-vendored

AWESOME.md

@@ -0,0 +1,19 @@
+# Awesome ChatGPT
+
+- [Awesome ChatGPT](https://github.com/humanloop/awesome-chatgpt) - Curated list of awesome tools, demos, docs for ChatGPT and GPT-3
+
+## Extension
+
+`Browser`
+
+- [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
+- [ChatGPT for Google](https://github.com/wong2/chat-gpt-google-extension) - A browser extension to display ChatGPT response alongside Google Search results
+- [ChatGPT Extension](https://github.com/kazuki-sf/ChatGPT_Extension) - ChatGPT Extension is a really simple Chrome Extension (manifest v3) that you can access OpenAI's ChatGPT from anywhere on the web.
+
+`VSCode`
+
+- [ChatGPT Extension for VSCode](https://github.com/mpociot/chatgpt-vscode) - A VSCode extension that allows you to use ChatGPT
+
+`Bot`
+
+- [ChatGPT Telegram Bot](https://github.com/altryne/chatGPT-telegram-bot) - This is a very early attempt at having chatGPT work within a telegram bot

README.md

@@ -5,23 +5,81 @@
 
 > ChatGPT Desktop Application
 
-[🚀 Download ChatGPT](https://github.com/lencx/ChatGPT/releases)
+[Awesome ChatGPT](./AWESOME.md)
+
+## Downloads
+
+[![ChatGPT downloads](https://img.shields.io/github/downloads/lencx/ChatGPT/total.svg?style=flat-square)](https://github.com/lencx/ChatGPT/releases)
+
+<!-- download start -->
+
+**Latest:**
+
+- `Mac`: [ChatGPT_0.1.7_x64.dmg](https://github.com/lencx/ChatGPT/releases/download/v0.1.7/ChatGPT_0.1.7_x64.dmg)
+- `Linux`: [chat-gpt_0.1.7_amd64.deb](https://github.com/lencx/ChatGPT/releases/download/v0.1.7/chat-gpt_0.1.7_amd64.deb)
+- `Windows`: [ChatGPT_0.1.7_x64_en-US.msi](https://github.com/lencx/ChatGPT/releases/download/v0.1.7/ChatGPT_0.1.7_x64_en-US.msi)
+
+[Other version...](https://github.com/lencx/ChatGPT/releases)
+
+<!-- download end -->
 
 ## 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
-- system tray
+- tray window
 - shortcut
 
 ## Preview
 
-<img width="600" src="./assets/install.png" alt="install">
+<img width="360" src="./assets/install.png" alt="install"> <img width="360" src="./assets/chat.png" alt="chat">
-<img width="600" src="./assets/chat.png" alt="chat">
+<img width="360" src="./assets/export.png" alt="export"> <img width="360" src="./assets/tray.png" alt="tray">
+<img width="360" src="./assets/auto-update.png" alt="auto update">
+## FAQ
 
-## TODO
+### Is it safe?
 
-- [ ] export chat history
+It's safe, just a wrapper for [OpenAI ChatGPT](https://chat.openai.com) website, no other data transfer exists (you can check the source code).
-- [ ] ...
+
+### Developer cannot be verified?
+
+- [Open a Mac app from an unidentified developer](https://support.apple.com/en-sg/guide/mac-help/mh40616/mac)
+
+### How do i build it?
+
+#### PreInstall
+
+- [Rust](https://www.rust-lang.org/)
+- [VS Code](https://code.visualstudio.com/)
+  - [rust-analyzer](https://marketplace.visualstudio.com/items?itemName=rust-lang.rust-analyzer)
+  - [tauri](https://marketplace.visualstudio.com/items?itemName=tauri-apps.tauri-vscode)
+
+#### Start
+
+```bash
+# step1:
+git clone https://github.com/lencx/ChatGPT.git
+
+# step2:
+cd ChatGPT
+
+# step3: install deps
+yarn
+
+# step4:
+yarn dev
+
+# step5:
+# bundle path: src-tauri/target/release/bundle
+yarn build
+```
+
+## Related
+
+- [Tauri](https://tauri.app) - Build an optimized, secure, and frontend-independent application for multi-platform deployment.
+- [ChatGPT](https://openai.com/blog/chatgpt) - ChatGPT: Optimizing Language Models for Dialogue.
+- [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.

UPDATE_LOG.md

@@ -1,12 +1,36 @@
 # UPDATE LOG
 
+## v0.1.7
+
+feat: tray window
+
+## v0.1.6
+
+feat:
+- always on top
+- export ChatGPT history
+
+## v0.1.5
+
+fix: mac can't use shortcut keys
+
+## 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
 

package.json

@@ -6,6 +6,7 @@
     "build": "yarn tauri build",
     "updater": "tr updater",
     "release": "tr release --git",
+    "download": "node ./scripts/download.js",
     "tr": "tr",
     "tauri": "tauri"
   },

scripts/download.js

@@ -0,0 +1,26 @@
+const fs = require('fs');
+
+const argv = process.argv.slice(2);
+
+async function init() {
+  const content = fs.readFileSync('README.md', 'utf8').split('\n');
+  const startRe = /<!-- download start -->/;
+  const endRe = /<!-- download end -->/;
+
+  let flag = false;
+  for (let i = 0; i < content.length; i++) {
+    if (startRe.test(content[i])) {
+      flag = true;
+    }
+    if (flag) {
+      content[i] = content[i].replace(/(\d+).(\d+).(\d+)/g, argv[0]);
+    }
+    if (endRe.test(content[i])) {
+      break;
+    }
+  }
+
+  fs.writeFileSync('README.md', content.join('\n'), 'utf8');
+}
+
+init().catch(console.error);

src-tauri/Cargo.toml

@@ -17,6 +17,7 @@ tauri-build = {version = "1.2.1", features = [] }
 serde_json = "1.0"
 serde = { version = "1.0", features = ["derive"] }
 tauri = { version = "1.2.1", features = ["api-all", "devtools", "system-tray", "updater"] }
+tauri-plugin-positioner = { version = "1.0.4", features = ["system-tray"] }
 anyhow = "1.0.66"
 
 [features]
@@ -26,3 +27,10 @@ default = [ "custom-protocol" ]
 # this feature is used used for production builds where `devPath` points to the filesystem
 # DO NOT remove this
 custom-protocol = [ "tauri/custom-protocol" ]
+
+# fix: mac v1.2.0 can not copy/paste
+# https://github.com/tauri-apps/tauri/issues/5669
+[profile.release]
+strip = true
+lto = true
+opt-level = "z"

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]
+#[command]
-pub fn drag_window(app: tauri::AppHandle) {
+pub fn drag_window(app: AppHandle) {
     app.get_window("core").unwrap().start_dragging().unwrap();
 }
 
-#[tauri::command]
+#[command]
-pub fn fullscreen(app: tauri::AppHandle) {
+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,21 +1,18 @@
-use crate::utils;
+use crate::{conf, utils};
 use tauri::{
     utils::assets::EmbeddedAssets, AboutMetadata, AppHandle, Context, CustomMenuItem, Manager,
-    Menu, MenuItem, Submenu, SystemTrayEvent, WindowMenuEvent,
+    Menu, MenuItem, Submenu, SystemTray, SystemTrayEvent, SystemTrayMenu, WindowMenuEvent,
 };
+use tauri_plugin_positioner::{on_tray_event, Position, WindowExt};
 
 // --- 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)
@@ -24,6 +21,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()
@@ -73,6 +92,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)
@@ -82,56 +102,50 @@ 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 menu_id = event.menu_item_id();
 
-    match event.menu_item_id() {
+    let core_window = app.get_window("core").unwrap();
-        // App
+    let menu_handle = core_window.menu_handle();
-        "inject_script" => {
+
-            tauri::api::shell::open(
+    match menu_id {
-                &app.shell_scope(),
+        // Preferences
-                utils::script_path().to_string_lossy(),
+        "inject_script" => open(&app, script_path),
-                None,
+        "awesome" => open(&app, conf::AWESOME_URL.to_string()),
-            )
+        "always_on_top" => {
-            .unwrap();
+            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
-        "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" => open(&app, conf::ISSUES_URL.to_string()),
-            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 +154,29 @@ pub fn menu_handler(event: WindowMenuEvent<tauri::Wry>) {
     }
 }
 
+// --- SystemTray Menu
+pub fn tray_menu() -> SystemTray {
+    SystemTray::new().with_menu(SystemTrayMenu::new())
+}
+
 // --- SystemTray Event
-pub fn tray_handler(app: &AppHandle, event: SystemTrayEvent) {
+pub fn tray_handler(handle: &AppHandle, event: SystemTrayEvent) {
-    if let SystemTrayEvent::LeftClick {
+    let core_win = handle.get_window("core").unwrap();
-        position: _,
+    on_tray_event(handle, &event);
-        size: _,
+
-        ..
+    if let SystemTrayEvent::LeftClick { .. } = event {
-    } = event
+        core_win.minimize().unwrap();
-    {
+        let mini_win = handle.get_window("mini").unwrap();
-        let win = app.get_window("core").unwrap();
+        mini_win.move_window(Position::TrayCenter).unwrap();
-        if win.is_visible().unwrap() {
+
-            win.hide().unwrap();
+        if mini_win.is_visible().unwrap() {
+            mini_win.hide().unwrap();
         } else {
-            win.show().unwrap();
+            mini_win.show().unwrap();
-            win.set_focus().unwrap();
         }
     }
 }
+
+pub fn open(app: &AppHandle, path: String) {
+    tauri::api::shell::open(&app.shell_scope(), path, None).unwrap();
+}

src-tauri/src/app/mod.rs

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

src-tauri/src/app/setup.rs

@@ -1,34 +1,46 @@
-use crate::utils;
+use crate::{app::window, conf, utils};
-use tauri::{utils::config::WindowUrl, window::WindowBuilder, App};
+use tauri::{utils::config::WindowUrl, window::WindowBuilder, App, Manager};
 
 #[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();
+    window::mini_window(&app.app_handle());
 
     #[cfg(target_os = "macos")]
     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")
+        .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/export.js"))
+        .user_agent(conf::USER_AGENT)
         .build()?;
 
     #[cfg(not(target_os = "macos"))]
     WindowBuilder::new(app, "core", WindowUrl::App(url.into()))
+        .title("ChatGPT")
         .resizable(true)
         .fullscreen(false)
-        .initialization_script(include_str!("../core.js"))
-        .initialization_script(&utils::user_script())
         .inner_size(800.0, 600.0)
-        .hidden_title(true)
+        .always_on_top(chat_conf.always_on_top)
-        .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")
+        .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/export.js"))
+        .user_agent(conf::USER_AGENT)
         .build()?;
 
     Ok(())

src-tauri/src/app/window.rs

@@ -0,0 +1,25 @@
+use crate::{conf, utils};
+use tauri::{utils::config::WindowUrl, window::WindowBuilder};
+
+pub fn mini_window(handle: &tauri::AppHandle) {
+    let conf = utils::get_tauri_conf().unwrap();
+    let url = conf.build.dev_path.to_string();
+
+    WindowBuilder::new(handle, "mini", WindowUrl::App(url.into()))
+        .resizable(false)
+        .fullscreen(false)
+        .inner_size(360.0, 540.0)
+        .decorations(false)
+        .always_on_top(true)
+        .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/export.js"))
+        .user_agent(conf::PHONE_USER_AGENT)
+        .menu(tauri::Menu::new())
+        .build()
+        .unwrap()
+        .hide()
+        .unwrap();
+}

src-tauri/src/assets/core.js

@@ -0,0 +1,96 @@
+// *** Core Script - IPC ***
+const uid = () => window.crypto.getRandomValues(new Uint32Array(1))[0];
+function transformCallback(callback = () => {}, once = false) {
+  const identifier = uid();
+  const prop = `_${identifier}`;
+  Object.defineProperty(window, prop, {
+    value: (result) => {
+      if (once) {
+        Reflect.deleteProperty(window, prop);
+      }
+      return callback(result)
+    },
+    writable: false,
+    configurable: true,
+  })
+  return identifier;
+}
+async function invoke(cmd, args) {
+  return new Promise((resolve, reject) => {
+    if (!window.__TAURI_POST_MESSAGE__) reject('__TAURI_POST_MESSAGE__ does not exist!');
+    const callback = transformCallback((e) => {
+      resolve(e);
+      Reflect.deleteProperty(window, `_${error}`);
+    }, true)
+    const error = transformCallback((e) => {
+      reject(e);
+      Reflect.deleteProperty(window, `_${callback}`);
+    }, true)
+    window.__TAURI_POST_MESSAGE__({
+      cmd,
+      callback,
+      error,
+      ...args
+    });
+  });
+}
+
+window.uid = uid;
+window.invoke = invoke;
+window.transformCallback = transformCallback;
+
+async function init() {
+  if (__TAURI_METADATA__.__currentWindow.label === 'mini') {
+    document.getElementsByTagName('html')[0].style['font-size'] = '70%';
+  }
+
+  if (__TAURI_METADATA__.__currentWindow.label !== 'core') return;
+
+  async function platform() {
+    return invoke('platform', {
+      __tauriModule: 'Os',
+      message: { cmd: 'platform' }
+    });
+  }
+
+  const _platform = await platform();
+  if (/darwin/.test(_platform)) {
+    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");
+    if (origin && origin.href && origin.target !== '_self') {
+      origin.target = "_self";
+    }
+  });
+
+  document.addEventListener('wheel', function(event) {
+    const deltaX = event.wheelDeltaX;
+    if (Math.abs(deltaX) >= 50) {
+      if (deltaX > 0) {
+        window.history.go(-1);
+      } else {
+        window.history.go(1);
+      }
+    }
+  });
+}
+
+if (
+  document.readyState === "complete" ||
+  document.readyState === "interactive"
+) {
+  init();
+} else {
+  document.addEventListener("DOMContentLoaded", init);
+}

src-tauri/src/assets/export.js

@@ -0,0 +1,238 @@
+// *** Core Script - Export ***
+// @ref: https://github.com/liady/ChatGPT-pdf/blob/main/src/content_script.js
+
+async function init() {
+  if (window.buttonsInterval) {
+    clearInterval(window.buttonsInterval);
+  }
+  window.buttonsInterval = setInterval(() => {
+    const actionsArea = document.querySelector("form>div>div");
+    if (!actionsArea) {
+      return;
+    }
+    const buttons = actionsArea.querySelectorAll("button");
+    const hasTryAgainButton = Array.from(buttons).some((button) => {
+      return !button.id?.includes("download");
+    });
+    if (hasTryAgainButton && buttons.length === 1) {
+      const TryAgainButton = actionsArea.querySelector("button");
+      addActionsButtons(actionsArea, TryAgainButton);
+    } else if (!hasTryAgainButton) {
+      removeButtons();
+    }
+  }, 200);
+}
+
+const Format = {
+  PNG: "png",
+  PDF: "pdf",
+};
+
+function addActionsButtons(actionsArea, TryAgainButton) {
+  const downloadButton = TryAgainButton.cloneNode(true);
+  downloadButton.id = "download-png-button";
+  downloadButton.innerText = "Generate PNG";
+  downloadButton.onclick = () => {
+    downloadThread();
+  };
+  actionsArea.appendChild(downloadButton);
+  const downloadPdfButton = TryAgainButton.cloneNode(true);
+  downloadPdfButton.id = "download-pdf-button";
+  downloadPdfButton.innerText = "Download PDF";
+  downloadPdfButton.onclick = () => {
+    downloadThread({ as: Format.PDF });
+  };
+  actionsArea.appendChild(downloadPdfButton);
+  const exportHtml = TryAgainButton.cloneNode(true);
+  exportHtml.id = "download-html-button";
+  exportHtml.innerText = "Share Link";
+  exportHtml.onclick = () => {
+    sendRequest();
+  };
+  actionsArea.appendChild(exportHtml);
+}
+
+function removeButtons() {
+  const downloadButton = document.getElementById("download-png-button");
+  const downloadPdfButton = document.getElementById("download-pdf-button");
+  const downloadHtmlButton = document.getElementById("download-html-button");
+  if (downloadButton) {
+    downloadButton.remove();
+  }
+  if (downloadPdfButton) {
+    downloadPdfButton.remove();
+  }
+  if (downloadHtmlButton) {
+    downloadHtmlButton.remove();
+  }
+}
+
+function downloadThread({ as = Format.PNG } = {}) {
+  const elements = new Elements();
+  elements.fixLocation();
+  const pixelRatio = window.devicePixelRatio;
+  const minRatio = as === Format.PDF ? 2 : 2.5;
+  window.devicePixelRatio = Math.max(pixelRatio, minRatio);
+  html2canvas(elements.thread, {
+    letterRendering: true,
+    onclone: function (cloneDoc) {
+      //Make small fix of position to all the text containers
+      let listOfTexts = cloneDoc.getElementsByClassName("min-h-[20px]");
+      Array.from(listOfTexts).forEach((text) => {
+        text.style.position = "relative";
+        text.style.top = "-8px";
+      });
+
+      //Delete copy button from code blocks
+      let listOfCopyBtns = cloneDoc.querySelectorAll("button.flex");
+      Array.from(listOfCopyBtns).forEach(
+        (btn) => (btn.style.visibility = "hidden")
+      );
+    },
+  }).then(async function (canvas) {
+    elements.restoreLocation();
+    window.devicePixelRatio = pixelRatio;
+    const imgData = canvas.toDataURL("image/png");
+    requestAnimationFrame(() => {
+      if (as === Format.PDF) {
+        return handlePdf(imgData, canvas, pixelRatio);
+      } else {
+        handleImg(imgData);
+      }
+    });
+  });
+}
+
+function handleImg(imgData) {
+  const binaryData = atob(imgData.split("base64,")[1]);
+  const data = [];
+  for (let i = 0; i < binaryData.length; i++) {
+    data.push(binaryData.charCodeAt(i));
+  }
+  invoke('download', { name: `chatgpt-${Date.now()}.png`, blob: Array.from(new Uint8Array(data)) });
+}
+
+function handlePdf(imgData, canvas, pixelRatio) {
+  const { jsPDF } = window.jspdf;
+  const orientation = canvas.width > canvas.height ? "l" : "p";
+  var pdf = new jsPDF(orientation, "pt", [
+    canvas.width / pixelRatio,
+    canvas.height / pixelRatio,
+  ]);
+  var pdfWidth = pdf.internal.pageSize.getWidth();
+  var pdfHeight = pdf.internal.pageSize.getHeight();
+  pdf.addImage(imgData, "PNG", 0, 0, pdfWidth, pdfHeight);
+
+  const data = pdf.__private__.getArrayBuffer(pdf.__private__.buildDocument());
+  invoke('download', { name: `chatgpt-${Date.now()}.pdf`, blob: Array.from(new Uint8Array(data)) });
+}
+
+class Elements {
+  constructor() {
+    this.init();
+  }
+  init() {
+    // this.threadWrapper = document.querySelector(".cdfdFe");
+    this.spacer = document.querySelector(".w-full.h-48.flex-shrink-0");
+    this.thread = document.querySelector(
+      "[class*='react-scroll-to-bottom']>[class*='react-scroll-to-bottom']>div"
+    );
+    this.positionForm = document.querySelector("form").parentNode;
+    // this.styledThread = document.querySelector("main");
+    // this.threadContent = document.querySelector(".gAnhyd");
+    this.scroller = Array.from(
+      document.querySelectorAll('[class*="react-scroll-to"]')
+    ).filter((el) => el.classList.contains("h-full"))[0];
+    this.hiddens = Array.from(document.querySelectorAll(".overflow-hidden"));
+    this.images = Array.from(document.querySelectorAll("img[srcset]"));
+  }
+  fixLocation() {
+    this.hiddens.forEach((el) => {
+      el.classList.remove("overflow-hidden");
+    });
+    this.spacer.style.display = "none";
+    this.thread.style.maxWidth = "960px";
+    this.thread.style.marginInline = "auto";
+    this.positionForm.style.display = "none";
+    this.scroller.classList.remove("h-full");
+    this.scroller.style.minHeight = "100vh";
+    this.images.forEach((img) => {
+      const srcset = img.getAttribute("srcset");
+      img.setAttribute("srcset_old", srcset);
+      img.setAttribute("srcset", "");
+    });
+  }
+  restoreLocation() {
+    this.hiddens.forEach((el) => {
+      el.classList.add("overflow-hidden");
+    });
+    this.spacer.style.display = null;
+    this.thread.style.maxWidth = null;
+    this.thread.style.marginInline = null;
+    this.positionForm.style.display = null;
+    this.scroller.classList.add("h-full");
+    this.scroller.style.minHeight = null;
+    this.images.forEach((img) => {
+      const srcset = img.getAttribute("srcset_old");
+      img.setAttribute("srcset", srcset);
+      img.setAttribute("srcset_old", "");
+    });
+  }
+}
+
+function selectElementByClassPrefix(classPrefix) {
+  const element = document.querySelector(`[class^='${classPrefix}']`);
+  return element;
+}
+
+async function sendRequest() {
+  const data = getData();
+  const uploadUrlResponse = await fetch(
+    "https://chatgpt-static.s3.amazonaws.com/url.txt"
+  );
+  const uploadUrl = await uploadUrlResponse.text();
+  fetch(uploadUrl, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+    },
+    body: JSON.stringify(data),
+  })
+    .then((response) => response.json())
+    .then((data) => {
+      invoke('open_link', { url: data.url });
+    });
+}
+
+function getData() {
+  const globalCss = getCssFromSheet(
+    document.querySelector("link[rel=stylesheet]").sheet
+  );
+  const localCss =
+    getCssFromSheet(
+      document.querySelector(`style[data-styled][data-styled-version]`).sheet
+    ) || "body{}";
+  const data = {
+    main: document.querySelector("main").outerHTML,
+    // css: `${globalCss} /* GLOBAL-LOCAL */ ${localCss}`,
+    globalCss,
+    localCss,
+  };
+  return data;
+}
+
+function getCssFromSheet(sheet) {
+  return Array.from(sheet.cssRules)
+    .map((rule) => rule.cssText)
+    .join("");
+}
+
+// run init
+if (
+  document.readyState === "complete" ||
+  document.readyState === "interactive"
+) {
+  init();
+} else {
+  document.addEventListener("DOMContentLoaded", init);
+}

src-tauri/src/conf.rs

@@ -0,0 +1,67 @@
+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 PHONE_USER_AGENT: &str = "Mozilla/5.0 (iPhone; CPU iPhone OS 13_2_3 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/13.0.3 Mobile/15E148 Safari/604.1";
+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/core.js

@@ -1,67 +0,0 @@
-// *** Core Script ***
-document.addEventListener('DOMContentLoaded', async () => {
-  const uid = () => window.crypto.getRandomValues(new Uint32Array(1))[0];
-  function transformCallback(callback = () => {}, once = false) {
-    const identifier = uid();
-    const prop = `_${identifier}`;
-    Object.defineProperty(window, prop, {
-      value: (result) => {
-        if (once) {
-          Reflect.deleteProperty(window, prop);
-        }
-        return callback(result)
-      },
-      writable: false,
-      configurable: true,
-    })
-    return identifier;
-  }
-  async function invoke(cmd, args) {
-    return new Promise((resolve, reject) => {
-      if (!window.__TAURI_POST_MESSAGE__) reject('__TAURI_POST_MESSAGE__ does not exist!');
-      const callback = transformCallback((e) => {
-        resolve(e);
-        Reflect.deleteProperty(window, `_${error}`);
-      }, true)
-      const error = transformCallback((e) => {
-        reject(e);
-        Reflect.deleteProperty(window, `_${callback}`);
-      }, true)
-      window.__TAURI_POST_MESSAGE__({
-        cmd,
-        callback,
-        error,
-        ...args
-      });
-    });
-  }
-
-  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");
-    if (origin && origin.href && origin.target !== '_self') {
-      origin.target = "_self";
-    }
-  });
-
-  document.addEventListener('wheel', function(event) {
-    const deltaX = event.wheelDeltaX;
-    if (Math.abs(deltaX) >= 50) {
-      if (deltaX > 0) {
-        window.history.go(-1);
-      } else {
-        window.history.go(1);
-      }
-    }
-  });
-})

src-tauri/src/main.rs

@@ -4,21 +4,42 @@
 )]
 
 mod app;
+mod conf;
 mod utils;
 
 use app::{cmd, menu, setup};
-use tauri::SystemTray;
+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])
+        .manage(conf::ChatState::default(&chat_conf))
-        .setup(setup::init)
+        .invoke_handler(tauri::generate_handler![
-        .menu(menu::init(&context))
+            cmd::drag_window,
-        .system_tray(SystemTray::new())
+            cmd::fullscreen,
+            cmd::download,
+            cmd::open_link
+        ])
+        .setup(|app| setup::init(app, chat_conf2))
+        .plugin(tauri_plugin_positioner::init())
+        .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::{
-use std::path::{Path, PathBuf};
+    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 = chat_root().join("main.js");
-    let script_file = 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

@@ -2,12 +2,12 @@
   "build": {
     "beforeDevCommand": "",
     "beforeBuildCommand": "",
-    "devPath": "https://chat.openai.com",
+    "devPath": "https://chat.openai.com/",
     "distDir": "../dist"
   },
   "package": {
     "productName": "ChatGPT",
-    "version": "0.1.2"
+    "version": "0.1.7"
   },
   "tauri": {
     "allowlist": {
@@ -44,6 +44,10 @@
       "shortDescription": "ChatGPT",
       "targets": "all",
       "windows": {
+        "webviewInstallMode": {
+          "silent": true,
+          "type": "downloadBootstrapper"
+        },
         "certificateThumbprint": null,
         "digestAlgorithm": "sha256",
         "timestampUrl": ""