diff --git a/.github/ISSUE_TEMPLATE/bug_report.yml b/.github/ISSUE_TEMPLATE/bug_report.yml index 21688e9..cebefb1 100644 --- a/.github/ISSUE_TEMPLATE/bug_report.yml +++ b/.github/ISSUE_TEMPLATE/bug_report.yml @@ -1,43 +1,43 @@ -name: "🕷️ Bug report" -description: "report bugs" -title: "[Bug]" +name: '🕷️ Bug report' +description: 'report bugs' +title: '[Bug]' labels: - - "bug" + - 'bug' body: -- type: markdown - attributes: - value: "Please make sure to [search for existing issues](https://github.com/lencx/ChatGPT/issues) before filing a new one!" -- type: markdown - attributes: - value: | - ## Bug report - Please fill in the following information to help us reproduce the bug: -- type: input - id: version - attributes: - label: Version - description: "Please specify the version of ChatGPT you are using, a newer version may have fixed the bug you encountered.Check the [UPDATE_LOG](https://github.com/lencx/ChatGPT/blob/main/UPDATE_LOG.md) for more information." - placeholder: "e.g. v0.1.0" - validations: - required: true -- type: textarea - id: bug - attributes: - label: Bug description - description: | - Please describe the bug here,if possible, please provide a minimal example to reproduce the bug.The content of `~/.chatgpt/chatgpt.log` may be helpful if you encounter a crash. - validations: - required: true -- type: input - id: OS - attributes: - label: OS - description: "Please specify the OS you are using." - placeholder: "e.g. Ubuntu 22.04" - validations: - required: true -- type: textarea - id: environment - attributes: - label: Environment - description: "If you think your environment may be related to the problem, please describe it here." + - type: markdown + attributes: + value: 'Please make sure to [search for existing issues](https://github.com/lencx/ChatGPT/issues) before filing a new one!' + - type: markdown + attributes: + value: | + ## Bug report + Please fill in the following information to help us reproduce the bug: + - type: input + id: version + attributes: + label: Version + description: 'Please specify the version of ChatGPT you are using, a newer version may have fixed the bug you encountered.Check the [UPDATE_LOG](https://github.com/lencx/ChatGPT/blob/main/UPDATE_LOG.md) for more information.' + placeholder: 'e.g. v0.1.0' + validations: + required: true + - type: textarea + id: bug + attributes: + label: Bug description + description: | + Please describe the bug here,if possible, please provide a minimal example to reproduce the bug.The content of `~/.chatgpt/chatgpt.log` may be helpful if you encounter a crash. + validations: + required: true + - type: input + id: OS + attributes: + label: OS + description: 'Please specify the OS you are using.' + placeholder: 'e.g. Ubuntu 22.04' + validations: + required: true + - type: textarea + id: environment + attributes: + label: Environment + description: 'If you think your environment may be related to the problem, please describe it here.' diff --git a/.github/ISSUE_TEMPLATE/build_error_report.yml b/.github/ISSUE_TEMPLATE/build_error_report.yml index 7223b20..a2db2fc 100644 --- a/.github/ISSUE_TEMPLATE/build_error_report.yml +++ b/.github/ISSUE_TEMPLATE/build_error_report.yml @@ -1,37 +1,37 @@ -name: "❌ Build error report" -description: "report errors when building by yourself" -title: "[Build Error]" +name: '❌ Build error report' +description: 'report errors when building by yourself' +title: '[Build Error]' labels: - - "build error" + - 'build error' body: -- type: markdown - attributes: - value: "Please make sure to [search for existing issues](https://github.com/lencx/ChatGPT/issues) before filing a new one!" -- type: markdown - attributes: - value: "Please make sure to build from the source code with the latest version of ChatGPT." -- type: markdown - attributes: - value: | - ## Build error report - Please fill in the following information to help us reproduce the bug: -- type: textarea - id: error - attributes: - label: Error message - description: "Please paste the error message here." - validations: - required: true -- type: input - id: OS - attributes: - label: OS - description: "Please specify the OS you are using." - placeholder: "e.g. Ubuntu 22.04" - validations: - required: true -- type: textarea - id: environment - attributes: - label: Environment - description: "If you think your environment may be related to the problem, please describe it here." \ No newline at end of file + - type: markdown + attributes: + value: 'Please make sure to [search for existing issues](https://github.com/lencx/ChatGPT/issues) before filing a new one!' + - type: markdown + attributes: + value: 'Please make sure to build from the source code with the latest version of ChatGPT.' + - type: markdown + attributes: + value: | + ## Build error report + Please fill in the following information to help us reproduce the bug: + - type: textarea + id: error + attributes: + label: Error message + description: 'Please paste the error message here.' + validations: + required: true + - type: input + id: OS + attributes: + label: OS + description: 'Please specify the OS you are using.' + placeholder: 'e.g. Ubuntu 22.04' + validations: + required: true + - type: textarea + id: environment + attributes: + label: Environment + description: 'If you think your environment may be related to the problem, please describe it here.' diff --git a/.github/ISSUE_TEMPLATE/docmentation_issue.yml b/.github/ISSUE_TEMPLATE/docmentation_issue.yml index 6aff586..fc64dbd 100644 --- a/.github/ISSUE_TEMPLATE/docmentation_issue.yml +++ b/.github/ISSUE_TEMPLATE/docmentation_issue.yml @@ -1,19 +1,19 @@ -name: "📚 Documentation Issue" -description: "report documentation issues, typos welcome!" -title: "[Doc]" +name: '📚 Documentation Issue' +description: 'report documentation issues, typos welcome!' +title: '[Doc]' labels: - - "documentation" + - 'documentation' body: -- type: markdown - attributes: - value: "Please make sure to [search for existing issues](https://github.com/lencx/ChatGPT/issues) before creating a new one." -- type: textarea - id: doc-description - attributes: - label: "Provide a description of requested docs changes" - description: "Briefly describe the requested docs changes." - validations: - required: true -- type: markdown - attributes: - value: Please limit one request per issue. \ No newline at end of file + - type: markdown + attributes: + value: 'Please make sure to [search for existing issues](https://github.com/lencx/ChatGPT/issues) before creating a new one.' + - type: textarea + id: doc-description + attributes: + label: 'Provide a description of requested docs changes' + description: 'Briefly describe the requested docs changes.' + validations: + required: true + - type: markdown + attributes: + value: Please limit one request per issue. diff --git a/.github/ISSUE_TEMPLATE/feature_request.yml b/.github/ISSUE_TEMPLATE/feature_request.yml index 75eca7e..c1f82ed 100644 --- a/.github/ISSUE_TEMPLATE/feature_request.yml +++ b/.github/ISSUE_TEMPLATE/feature_request.yml @@ -1,34 +1,34 @@ -name: "⭐ Feature or enhancement request" -description: "suggest new features or enhancements" -title: "[Feature]" +name: '⭐ Feature or enhancement request' +description: 'suggest new features or enhancements' +title: '[Feature]' labels: - - "enhancement" + - 'enhancement' body: -- type: markdown - attributes: - value: "Please make sure to [search for existing issues](https://github.com/lencx/ChatGPT/issues) before creating a new one." -- type: textarea - id: feature-description - attributes: - label: "Feature description" - description: "Describe the feature or enhancements you'd like to see." - validations: - required: true -- type: textarea - id: motivation - attributes: - label: "Motivation" - description: "Describe the motivation for this feature or enhancement." -- type: textarea - id: alternatives - attributes: - label: "Alternatives" - description: "Describe any alternatives you've considered." -- type: textarea - id: additional-context - attributes: - label: "Additional context" - description: "Add any other context or screenshots about the feature request here." -- type: markdown - attributes: - value: Please limit one request per issue. + - type: markdown + attributes: + value: 'Please make sure to [search for existing issues](https://github.com/lencx/ChatGPT/issues) before creating a new one.' + - type: textarea + id: feature-description + attributes: + label: 'Feature description' + description: "Describe the feature or enhancements you'd like to see." + validations: + required: true + - type: textarea + id: motivation + attributes: + label: 'Motivation' + description: 'Describe the motivation for this feature or enhancement.' + - type: textarea + id: alternatives + attributes: + label: 'Alternatives' + description: "Describe any alternatives you've considered." + - type: textarea + id: additional-context + attributes: + label: 'Additional context' + description: 'Add any other context or screenshots about the feature request here.' + - type: markdown + attributes: + value: Please limit one request per issue. diff --git a/.github/ISSUE_TEMPLATE/security.yml b/.github/ISSUE_TEMPLATE/security.yml index 7c0653b..0bb5b56 100644 --- a/.github/ISSUE_TEMPLATE/security.yml +++ b/.github/ISSUE_TEMPLATE/security.yml @@ -1,34 +1,34 @@ -name: "⚠️ Security&Privacy issue" -description: "Report security or privacy issues" -title: "[Security]" +name: '⚠️ Security&Privacy issue' +description: 'Report security or privacy issues' +title: '[Security]' labels: - - "security" + - 'security' body: -- type: markdown - attributes: - value: "Please make sure to [search for existing issues](https://github.com/lencx/ChatGPT/issues) before creating a new one." -- type: textarea - id: security-description - attributes: - label: "Description" - description: "Describe the security or privacy issue." - validations: - required: true -- type: textarea - id: motivation - attributes: - label: "Motivation" - description: "Describe the motivation for this security or privacy issue." -- type: textarea - id: alternatives - attributes: - label: "Alternatives" - description: "Describe any alternatives you've considered." -- type: textarea - id: additional-context - attributes: - label: "Additional context" - description: "Add any other context or screenshots about the security or privacy issue here." -- type: markdown - attributes: - value: Please limit one request per issue. \ No newline at end of file + - type: markdown + attributes: + value: 'Please make sure to [search for existing issues](https://github.com/lencx/ChatGPT/issues) before creating a new one.' + - type: textarea + id: security-description + attributes: + label: 'Description' + description: 'Describe the security or privacy issue.' + validations: + required: true + - type: textarea + id: motivation + attributes: + label: 'Motivation' + description: 'Describe the motivation for this security or privacy issue.' + - type: textarea + id: alternatives + attributes: + label: 'Alternatives' + description: "Describe any alternatives you've considered." + - type: textarea + id: additional-context + attributes: + label: 'Additional context' + description: 'Add any other context or screenshots about the security or privacy issue here.' + - type: markdown + attributes: + value: Please limit one request per issue. diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml index 3913b95..c6f2d4e 100644 --- a/.github/workflows/release.yml +++ b/.github/workflows/release.yml @@ -90,13 +90,13 @@ jobs: publish_dir: ./updater force_orphan: true - # publish-winget: - # # Action can only be run on windows - # runs-on: windows-latest - # needs: [create-release, build-tauri] - # steps: - # - uses: vedantmgoyal2009/winget-releaser@v1 - # with: - # identifier: lencx.ChatGPT - # token: ${{ secrets.WINGET_TOKEN }} - # version: ${{ env.version }} + publish-winget: + # Action can only be run on windows + runs-on: windows-latest + needs: [create-release, build-tauri] + steps: + - uses: vedantmgoyal2009/winget-releaser@v1 + with: + identifier: lencx.ChatGPT + token: ${{ secrets.WINGET_TOKEN }} + version: ${{ env.version }} diff --git a/.husky/pre-commit b/.husky/pre-commit new file mode 100755 index 0000000..aef59bf --- /dev/null +++ b/.husky/pre-commit @@ -0,0 +1,6 @@ +#!/usr/bin/env sh +. "$(dirname -- "$0")/_/husky.sh" + +npm run pretty-quick +cargo fmt +git add . \ No newline at end of file diff --git a/.prettierignore b/.prettierignore new file mode 100644 index 0000000..c61b10f --- /dev/null +++ b/.prettierignore @@ -0,0 +1,45 @@ +package-lock.json +node_modules/ +yarn.lock +*.lock + +casks/ + +# rust +src-tauri/ +target/ +Cargo.lock +*.toml + +# Logs +logs +*.log +npm-debug.log* +yarn-debug.log* +yarn-error.log* +pnpm-debug.log* +lerna-debug.log* + +dist +dist-ssr +*.local + +# Editor directories and files +.vscode/* +!.vscode/extensions.json +.idea +.DS_Store +*.suo +*.ntvs* +*.njsproj +*.sln +*.sw? + +assets/** +public/** + +.gitattributes +.gitignore +.prettierignore + +LICENSE \ No newline at end of file diff --git a/.prettierrc b/.prettierrc new file mode 100644 index 0000000..c36a279 --- /dev/null +++ b/.prettierrc @@ -0,0 +1,7 @@ +{ + "trailingComma": "all", + "singleQuote": true, + "semi": true, + "tabWidth": 2, + "printWidth": 100 +} diff --git a/Cargo.toml b/Cargo.toml new file mode 100644 index 0000000..47083f7 --- /dev/null +++ b/Cargo.toml @@ -0,0 +1,9 @@ +[workspace] +members = ["src-tauri"] + +# 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 = "s" diff --git a/README-ZH_CN.md b/README-ZH_CN.md index f58c2c2..92d31a6 100644 --- a/README-ZH_CN.md +++ b/README-ZH_CN.md @@ -10,6 +10,7 @@ [![ChatGPT downloads](https://img.shields.io/github/downloads/lencx/ChatGPT/total.svg?style=flat-square)](https://github.com/lencx/ChatGPT/releases) [![chat](https://img.shields.io/badge/chat-discord-blue?style=flat&logo=discord)](https://discord.gg/aPhCRf4zZr) [![lencx](https://img.shields.io/badge/follow-lencx__-blue?style=flat&logo=Twitter)](https://twitter.com/lencx_) + Buy Me A Coffee @@ -23,8 +24,9 @@ ### Windows -- [ChatGPT_0.9.2_x64_en-US.msi](https://github.com/lencx/ChatGPT/releases/download/v0.9.2/ChatGPT_0.9.2_x64_en-US.msi): +- [ChatGPT_0.10.0_x64_en-US.msi](https://github.com/lencx/ChatGPT/releases/download/v0.10.0/ChatGPT_0.10.0_x64_en-US.msi): - 使用 [winget](https://winstall.app/apps/lencx.ChatGPT): + ```bash # install the latest version winget install --id=lencx.ChatGPT -e @@ -33,12 +35,12 @@ winget install --id=lencx.ChatGPT -e --version 0.9.0 ``` -**注意：如果安装路径和应用名称相同，会导致冲突 ([#142](https://github.com/lencx/ChatGPT/issues/142#issuecomment-0.9.2))** +**注意：如果安装路径和应用名称相同，会导致冲突 ([#142](https://github.com/lencx/ChatGPT/issues/142#issuecomment-0.10.0))** ### Mac -- [ChatGPT_0.9.2_x64.dmg](https://github.com/lencx/ChatGPT/releases/download/v0.9.2/ChatGPT_0.9.2_x64.dmg) -- [ChatGPT.app.tar.gz](https://github.com/lencx/ChatGPT/releases/download/v0.9.2/ChatGPT.app.tar.gz) +- [ChatGPT_0.10.0_x64.dmg](https://github.com/lencx/ChatGPT/releases/download/v0.10.0/ChatGPT_0.10.0_x64.dmg) +- [ChatGPT.app.tar.gz](https://github.com/lencx/ChatGPT/releases/download/v0.10.0/ChatGPT.app.tar.gz) - Homebrew \ _[Homebrew 快捷安装](https://brew.sh) ([Cask](https://docs.brew.sh/Cask-Cookbook)):_ ```sh @@ -54,8 +56,8 @@ ### Linux -- [chat-gpt_0.9.2_amd64.deb](https://github.com/lencx/ChatGPT/releases/download/v0.9.2/chat-gpt_0.9.2_amd64.deb) -- [chat-gpt_0.9.2_amd64.AppImage](https://github.com/lencx/ChatGPT/releases/download/v0.9.2/chat-gpt_0.9.2_amd64.AppImage): **工作可靠，`.deb` 运行失败时可以尝试它** +- [chat-gpt_0.10.0_amd64.deb](https://github.com/lencx/ChatGPT/releases/download/v0.10.0/chat-gpt_0.10.0_amd64.deb) +- [chat-gpt_0.10.0_amd64.AppImage](https://github.com/lencx/ChatGPT/releases/download/v0.10.0/chat-gpt_0.10.0_amd64.AppImage): **工作可靠，`.deb` 运行失败时可以尝试它** - 使用 [AUR](https://aur.archlinux.org/packages/chatgpt-desktop-bin): ```bash yay -S chatgpt-desktop-bin diff --git a/README.md b/README.md index 7bf5fb1..58af401 100644 --- a/README.md +++ b/README.md @@ -10,6 +10,7 @@ [![ChatGPT downloads](https://img.shields.io/github/downloads/lencx/ChatGPT/total.svg?style=flat-square)](https://github.com/lencx/ChatGPT/releases) [![chat](https://img.shields.io/badge/chat-discord-blue?style=flat&logo=discord)](https://discord.gg/aPhCRf4zZr) [![lencx](https://img.shields.io/badge/follow-lencx__-blue?style=flat&logo=Twitter)](https://twitter.com/lencx_) + @@ -25,8 +26,9 @@ ### Windows -- [ChatGPT_0.9.2_x64_en-US.msi](https://github.com/lencx/ChatGPT/releases/download/v0.9.2/ChatGPT_0.9.2_x64_en-US.msi): Direct download installer +- [ChatGPT_0.10.0_x64_en-US.msi](https://github.com/lencx/ChatGPT/releases/download/v0.10.0/ChatGPT_0.10.0_x64_en-US.msi): Direct download installer - Use [winget](https://winstall.app/apps/lencx.ChatGPT): + ```bash # install the latest version winget install --id=lencx.ChatGPT -e @@ -35,12 +37,12 @@ winget install --id=lencx.ChatGPT -e --version 0.9.0 ``` -**Note: If the installation path and application name are the same, it will lead to conflict ([#142](https://github.com/lencx/ChatGPT/issues/142#issuecomment-0.9.2))** +**Note: If the installation path and application name are the same, it will lead to conflict ([#142](https://github.com/lencx/ChatGPT/issues/142#issuecomment-0.10.0))** ### Mac -- [ChatGPT_0.9.2_x64.dmg](https://github.com/lencx/ChatGPT/releases/download/v0.9.2/ChatGPT_0.9.2_x64.dmg): Direct download installer -- [ChatGPT.app.tar.gz](https://github.com/lencx/ChatGPT/releases/download/v0.9.2/ChatGPT.app.tar.gz): Download the `.app` installer +- [ChatGPT_0.10.0_x64.dmg](https://github.com/lencx/ChatGPT/releases/download/v0.10.0/ChatGPT_0.10.0_x64.dmg): Direct download installer +- [ChatGPT.app.tar.gz](https://github.com/lencx/ChatGPT/releases/download/v0.10.0/ChatGPT.app.tar.gz): Download the `.app` installer - Homebrew \ Or you can install with _[Homebrew](https://brew.sh) ([Cask](https://docs.brew.sh/Cask-Cookbook)):_ ```sh @@ -56,8 +58,8 @@ ### Linux -- [chat-gpt_0.9.2_amd64.deb](https://github.com/lencx/ChatGPT/releases/download/v0.9.2/chat-gpt_0.9.2_amd64.deb): Download `.deb` installer, advantage small size, disadvantage poor compatibility -- [chat-gpt_0.9.2_amd64.AppImage](https://github.com/lencx/ChatGPT/releases/download/v0.9.2/chat-gpt_0.9.2_amd64.AppImage): Works reliably, you can try it if `.deb` fails to run +- [chat-gpt_0.10.0_amd64.deb](https://github.com/lencx/ChatGPT/releases/download/v0.10.0/chat-gpt_0.10.0_amd64.deb): Download `.deb` installer, advantage small size, disadvantage poor compatibility +- [chat-gpt_0.10.0_amd64.AppImage](https://github.com/lencx/ChatGPT/releases/download/v0.10.0/chat-gpt_0.10.0_amd64.AppImage): Works reliably, you can try it if `.deb` fails to run - Available on [AUR](https://aur.archlinux.org/packages/chatgpt-desktop-bin) with the package name `chatgpt-desktop-bin`, and you can use your favourite AUR package manager to install it. @@ -174,6 +176,7 @@ Currently, only json and csv are supported for synchronizing custom files, and t ## 📌 TODO + - `Control Center` enhancement - `Pop-up Search` enhancement - ... diff --git a/UPDATE_LOG.md b/UPDATE_LOG.md index fd708c1..e4322b6 100644 --- a/UPDATE_LOG.md +++ b/UPDATE_LOG.md @@ -1,177 +1,208 @@ # UPDATE LOG +## v0.10.0 + +Fix: + +- After exporting a file in Windows, open an empty file explorer (https://github.com/lencx/ChatGPT/issues/242) + +Feat: + +- Markdown files support editing and live preview +- Add `Awesome` menu to the `Control Center` (similar to bookmarks, but it's just a start, more possibilities in the future), custom URL support for the home and tray windows (if you're tired of ChatGPT as your home screen). + ## v0.9.2 -fix: slash command does not work +Fix: Slash command does not work ## v0.9.1 -fix: slash command does not work +Fix: Slash command does not work ## v0.9.0 -fix: -- export button does not work +Fix: -feat: -- add an export markdown button +- Export button does not work + +Feat: + +- Add an export markdown button - `Control Center` adds `Notes` and `Download` menus for managing exported chat files (Markdown, PNG, PDF). `Notes` supports markdown previews. ## v0.8.1 -fix: -- export button keeps blinking -- export button in the old chat does not work -- disable export sharing links because it is a security risk +Fix: + +- Export button keeps blinking +- Export button in the old chat does not work +- Disable export sharing links because it is a security risk ## v0.8.0 -feat: -- theme enhancement (Light, Dark, System) -- automatic updates support `silent` settings -- pop-up search: select the ChatGPT content with the mouse, the `DALL·E 2` button appears, and click to jump (note: because the search content filled by the script cannot trigger the event directly, you need to enter a space in the input box to make the button clickable). +Feat: -fix: -- close the main window and hide it in the tray (windows systems) +- Theme enhancement (Light, Dark, System) +- Automatic updates support `silent` settings +- Pop-up search: select the ChatGPT content with the mouse, the `DALL·E 2` button appears, and click to jump (note: because the search content filled by the script cannot trigger the event directly, you need to enter a space in the input box to make the button clickable). + +Fix: + +- Close the main window and hide it in the tray (windows systems) ## v0.7.4 -fix: -- trying to resolve linux errors: `error while loading shared libraries` -- customize global shortcuts (`Menu -> Preferences -> Control Center -> General -> Global Shortcut`) +Fix: + +- Trying to resolve linux errors: `error while loading shared libraries` +- Customize global shortcuts (`Menu -> Preferences -> Control Center -> General -> Global Shortcut`) ## v0.7.3 -chore: -- optimize slash command style -- optimize tray menu icon and button icons -- global shortcuts to the chatgpt app (mac: `Command + Shift + O`, windows: `Ctrl + Shift + O`) +Chore: + +- Optimize slash command style +- Optimize tray menu icon and button icons +- Global shortcuts to the chatgpt app (mac: `Command + Shift + O`, windows: `Ctrl + Shift + O`) ## v0.7.2 -fix: some windows systems cannot start the application +Fix: Some windows systems cannot start the application ## v0.7.1 -fix: -- some windows systems cannot start the application -- windows and linux add about menu (show version information) -- the tray icon is indistinguishable from the background in dark mode on window and linux +Fix: + +- Some windows systems cannot start the application +- Windows and linux add about menu (show version information) +- The tray icon is indistinguishable from the background in dark mode on window and linux ## v0.7.0 -fix: -- mac m1 copy/paste does not work on some system versions -- optimize the save chat log button to a small icon, the tray window no longer provides a save chat log button (the buttons causes the input area to become larger and the content area to become smaller) +Fix: -feat: -- use the keyboard `⇧` (arrow up) and `⇩` (arrow down) keys to select the slash command +- Mac m1 copy/paste does not work on some system versions +- Optimize the save chat log button to a small icon, the tray window no longer provides a save chat log button (the buttons causes the input area to become larger and the content area to become smaller) + +Feat: + +- Use the keyboard `⇧` (arrow up) and `⇩` (arrow down) keys to select the slash command ## v0.6.10 -fix: sync failure on windows +Fix: Sync failure on windows ## v0.6.4 -fix: path not allowed on the configured scope +Fix: Path not allowed on the configured scope -feat: -- optimize the generated pdf file size -- menu added `Sync Prompts` +Feat: + +- Optimize the generated pdf file size +- Menu added `Sync Prompts` - `Control Center` added `Sync Custom` -- the slash command is triggered by the enter key -- under the slash command, use the tab key to modify the contents of the `{q}` tag (only single changes are supported (https://github.com/lencx/ChatGPT/issues/54) +- The slash command is triggered by the enter key +- Under the slash command, use the tab key to modify the contents of the `{q}` tag (only single changes are supported (https://github.com/lencx/ChatGPT/issues/54) ## v0.6.0 -fix: -- windows show Chinese when upgrading +Fix: + +- Windows show Chinese when upgrading ## v0.5.1 -some optimization +Some optimization ## v0.5.0 -feat: `Control Center` added `chatgpt-prompts` synchronization +Feat: `Control Center` added `chatgpt-prompts` synchronization ## v0.4.2 -add chatgpt log (path: `~/.chatgpt/chatgpt.log`) +Add chatgpt log (path: `~/.chatgpt/chatgpt.log`) ## v0.4.1 -fix: -- tray window style optimization +Fix: + +- Tray window style optimization ## v0.4.0 -feat: -- customize the ChatGPT prompts command (https://github.com/lencx/ChatGPT#-announcement) -- menu enhancement: hide application icons from the Dock (support macOS only) +Feat: + +- Customize the ChatGPT prompts command (https://github.com/lencx/ChatGPT#-announcement) +- Menu enhancement: hide application icons from the Dock (support macOS only) ## v0.3.0 -fix: can't open ChatGPT +Fix: Can't open ChatGPT -feat: menu enhancement -- the control center of ChatGPT application -- open the configuration file directory +Feat: Menu enhancement + +- The control center of ChatGPT application +- Open the configuration file directory ## v0.2.2 -feat: -- menu: go to config +Feat: + +- Menu: go to config ## v0.2.1 -feat: menu optimization +Feat: Menu optimization ## v0.2.0 -feat: menu enhancement -- customize user-agent to prevent security detection interception -- clear all chatgpt configuration files +Feat: Menu enhancement + +- Customize user-agent to prevent security detection interception +- Clear all chatgpt configuration files ## v0.1.8 -feat: -- menu enhancement: theme, titlebar -- modify website address +Feat: + +- Menu enhancement: theme, titlebar +- Modify website address ## v0.1.7 -feat: tray window +Feat: Tray window ## v0.1.6 -feat: -- stay on top -- export ChatGPT history +Feat: + +- Stay on top +- Export ChatGPT history ## v0.1.5 -fix: mac can't use shortcut keys +Fix: Mac can't use shortcut keys ## v0.1.4 -feat: -- beautify icons -- add system tray menu +Feat: + +- Beautify icons +- Add system tray menu ## v0.1.3 -fix: only mac supports `TitleBarStyle` +Fix: Only mac supports `TitleBarStyle` ## v0.1.2 -initialization +Initialization ## v0.1.1 -initialization +Initialization ## v0.1.0 -initialization +Initialization diff --git a/package.json b/package.json index 6f8ca26..2aece26 100644 --- a/package.json +++ b/package.json @@ -12,8 +12,12 @@ "fix:tray": "tr override --json.tauri_systemTray_iconPath=\"icons/tray-icon-light.png\" --json.tauri_systemTray_iconAsTemplate=false", "fix:tray:mac": "tr override --json.tauri_systemTray_iconPath=\"icons/tray-icon.png\" --json.tauri_systemTray_iconAsTemplate=true", "download": "node ./scripts/download.js", + "fmt:rs": "cargo fmt", "tr": "tr", - "tauri": "tauri" + "tauri": "tauri", + "prettier": "prettier -c --write '**/*.{js,md,ts,tsx,yml}'", + "pretty-quick": "pretty-quick --staged", + "prepare": "husky install" }, "license": "MIT", "author": "lencx ", @@ -34,15 +38,22 @@ }, "dependencies": { "@ant-design/icons": "^4.8.0", + "@monaco-editor/react": "^4.4.6", "@tauri-apps/api": "^1.2.0", "antd": "^5.1.0", + "clsx": "^1.2.1", "dayjs": "^1.11.7", + "github-markdown-css": "^5.1.0", "lodash": "^4.17.21", "react": "^18.2.0", "react-dom": "^18.2.0", "react-markdown": "^8.0.4", + "react-resizable-panels": "^0.0.33", "react-router-dom": "^6.4.5", "react-syntax-highlighter": "^15.5.0", + "rehype-raw": "^6.1.1", + "remark-comment-config": "^7.0.1", + "remark-gfm": "^3.0.1", "uuid": "^9.0.0" }, "devDependencies": { @@ -55,6 +66,9 @@ "@types/react-syntax-highlighter": "^15.5.6", "@types/uuid": "^9.0.0", "@vitejs/plugin-react": "^3.0.0", + "husky": "^8.0.3", + "prettier": "^2.8.3", + "pretty-quick": "^3.1.3", "sass": "^1.56.2", "typescript": "^4.9.4", "vite": "^4.0.0", diff --git a/rustfmt.toml b/rustfmt.toml new file mode 100644 index 0000000..37a4436 --- /dev/null +++ b/rustfmt.toml @@ -0,0 +1,14 @@ +max_width = 100 +hard_tabs = false +tab_spaces = 2 +newline_style = "Auto" +use_small_heuristics = "Default" +reorder_imports = true +reorder_modules = true +remove_nested_parens = true +edition = "2021" +merge_derives = true +use_try_shorthand = false +use_field_init_shorthand = false +force_explicit_abi = true +# imports_granularity = "Crate" \ No newline at end of file diff --git a/scripts/download.js b/scripts/download.js index 86a55fe..517ff8e 100644 --- a/scripts/download.js +++ b/scripts/download.js @@ -30,4 +30,4 @@ async function init() { rewrite('README-ZH_CN.md'); } -init().catch(console.error); \ No newline at end of file +init().catch(console.error); diff --git a/src-tauri/Cargo.toml b/src-tauri/Cargo.toml index a7818a9..f0f4147 100644 --- a/src-tauri/Cargo.toml +++ b/src-tauri/Cargo.toml @@ -26,15 +26,10 @@ wry = "0.24.1" dark-light = "1.0.0" serde = { version = "1.0", features = ["derive"] } tokio = { version = "1.23.0", features = ["macros"] } +tauri = { version = "1.2.4", features = ["api-all", "devtools", "global-shortcut", "system-tray", "updater"] } tauri-plugin-positioner = { version = "1.0.4", features = ["system-tray"] } -tauri = { version = "1.2.3", features = ["api-all", "devtools", "global-shortcut", "system-tray", "updater"] } -[dependencies.tauri-plugin-log] -git = "https://github.com/lencx/tauri-plugin-log" -branch = "dev" -features = ["colored"] -[dependencies.tauri-plugin-autostart] -git = "https://github.com/lencx/tauri-plugin-autostart" -branch = "dev" +tauri-plugin-log = { git = "https://github.com/lencx/tauri-plugins-workspace", branch = "dev", features = ["colored"] } +tauri-plugin-autostart = { git = "https://github.com/lencx/tauri-plugins-workspace", branch = "dev" } # sqlx = { version = "0.6.2", features = ["runtime-tokio-rustls", "sqlite"] } @@ -45,10 +40,3 @@ 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 = "s" diff --git a/src-tauri/build.rs b/src-tauri/build.rs index d860e1e..795b9b7 100644 --- a/src-tauri/build.rs +++ b/src-tauri/build.rs @@ -1,3 +1,3 @@ fn main() { - tauri_build::build() + tauri_build::build() } diff --git a/src-tauri/src/app/cmd.rs b/src-tauri/src/app/cmd.rs index b14aebd..64493cf 100644 --- a/src-tauri/src/app/cmd.rs +++ b/src-tauri/src/app/cmd.rs @@ -1,400 +1,108 @@ use crate::{ - app::{fs_extra, window}, - conf::{ChatConfJson, GITHUB_PROMPTS_CSV_URL}, - utils::{self, chat_root, create_file}, + conf::ChatConfJson, + utils::{self, chat_root, create_file}, }; use log::info; -use regex::Regex; -use std::{collections::HashMap, fs, path::PathBuf, vec}; +use std::{fs, path::PathBuf}; use tauri::{api, command, AppHandle, Manager, Theme}; -use walkdir::WalkDir; #[command] pub fn drag_window(app: AppHandle) { - app.get_window("core").unwrap().start_dragging().unwrap(); -} - -#[command] -pub fn dalle2_window(app: AppHandle, query: String) { - window::dalle2_window( - &app.app_handle(), - Some(query), - Some("ChatGPT & DALL·E 2".to_string()), - None, - ); + app.get_window("core").unwrap().start_dragging().unwrap(); } #[command] pub fn fullscreen(app: AppHandle) { - let win = app.get_window("core").unwrap(); - if win.is_fullscreen().unwrap() { - win.set_fullscreen(false).unwrap(); - } else { - win.set_fullscreen(true).unwrap(); - } + let win = app.get_window("core").unwrap(); + if win.is_fullscreen().unwrap() { + win.set_fullscreen(false).unwrap(); + } else { + win.set_fullscreen(true).unwrap(); + } } #[command] pub fn download(_app: AppHandle, name: String, blob: Vec) { - let path = chat_root().join(PathBuf::from(name)); - create_file(&path).unwrap(); - fs::write(&path, blob).unwrap(); - utils::open_file(path); + let path = chat_root().join(PathBuf::from(name)); + create_file(&path).unwrap(); + fs::write(&path, blob).unwrap(); + utils::open_file(path); } #[command] pub fn save_file(_app: AppHandle, name: String, content: String) { - let path = chat_root().join(PathBuf::from(name)); - create_file(&path).unwrap(); - fs::write(&path, content).unwrap(); - utils::open_file(path); + let path = chat_root().join(PathBuf::from(name)); + create_file(&path).unwrap(); + fs::write(&path, content).unwrap(); + utils::open_file(path); } #[command] pub fn open_link(app: AppHandle, url: String) { - api::shell::open(&app.shell_scope(), url, None).unwrap(); + api::shell::open(&app.shell_scope(), url, None).unwrap(); } #[command] pub fn get_chat_conf() -> ChatConfJson { - ChatConfJson::get_chat_conf() -} - -#[command] -pub fn get_theme() -> String { - ChatConfJson::theme().unwrap_or(Theme::Light).to_string() + ChatConfJson::get_chat_conf() } #[command] pub fn reset_chat_conf() -> ChatConfJson { - ChatConfJson::reset_chat_conf() + ChatConfJson::reset_chat_conf() +} + +#[command] +pub fn get_theme() -> String { + ChatConfJson::theme().unwrap_or(Theme::Light).to_string() } #[command] pub fn run_check_update(app: AppHandle, silent: bool, has_msg: Option) { - utils::run_check_update(app, silent, has_msg); + utils::run_check_update(app, silent, has_msg); } #[command] pub fn form_confirm(_app: AppHandle, data: serde_json::Value) { - ChatConfJson::amend(&serde_json::json!(data), None).unwrap(); + ChatConfJson::amend(&serde_json::json!(data), None).unwrap(); } #[command] pub fn form_cancel(app: AppHandle, label: &str, title: &str, msg: &str) { - let win = app.app_handle().get_window(label).unwrap(); - tauri::api::dialog::ask( - app.app_handle().get_window(label).as_ref(), - title, - msg, - move |is_cancel| { - if is_cancel { - win.close().unwrap(); - } - }, - ); + let win = app.app_handle().get_window(label).unwrap(); + tauri::api::dialog::ask( + app.app_handle().get_window(label).as_ref(), + title, + msg, + move |is_cancel| { + if is_cancel { + win.close().unwrap(); + } + }, + ); } #[command] pub fn form_msg(app: AppHandle, label: &str, title: &str, msg: &str) { - let win = app.app_handle().get_window(label); - tauri::api::dialog::message(win.as_ref(), title, msg); + let win = app.app_handle().get_window(label); + tauri::api::dialog::message(win.as_ref(), title, msg); } #[command] pub fn open_file(path: PathBuf) { - utils::open_file(path); + utils::open_file(path); } #[command] -pub fn get_chat_model_cmd() -> serde_json::Value { - let path = utils::chat_root().join("chat.model.cmd.json"); - let content = fs::read_to_string(path).unwrap_or_else(|_| r#"{"data":[]}"#.to_string()); - serde_json::from_str(&content).unwrap() -} - -#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)] -pub struct PromptRecord { - pub cmd: Option, - pub act: String, - pub prompt: String, -} - -#[command] -pub fn parse_prompt(data: String) -> Vec { - let mut rdr = csv::Reader::from_reader(data.as_bytes()); - let mut list = vec![]; - for result in rdr.deserialize() { - let record: PromptRecord = result.unwrap_or_else(|err| { - info!("parse_prompt_error: {}", err); - PromptRecord { - cmd: None, - act: "".to_string(), - prompt: "".to_string(), - } - }); - if !record.act.is_empty() { - list.push(record); - } - } - list -} - -#[command] -pub fn window_reload(app: AppHandle, label: &str) { - app.app_handle() - .get_window(label) - .unwrap() - .eval("window.location.reload()") - .unwrap(); -} - -#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)] -pub struct ModelRecord { - pub cmd: String, - pub act: String, - pub prompt: String, - pub tags: Vec, - pub enable: bool, -} - -#[command] -pub fn cmd_list() -> Vec { - let mut list = vec![]; - for entry in WalkDir::new(utils::chat_root().join("cache_model")) - .into_iter() - .filter_map(|e| e.ok()) - { - let file = fs::read_to_string(entry.path().display().to_string()); - if let Ok(v) = file { - let data: Vec = serde_json::from_str(&v).unwrap_or_else(|_| vec![]); - let enable_list = data.into_iter().filter(|v| v.enable); - list.extend(enable_list) - } - } - // dbg!(&list); - list.sort_by(|a, b| a.cmd.len().cmp(&b.cmd.len())); - list -} - -#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)] -pub struct FileMetadata { - pub name: String, - pub ext: String, - pub created: u64, - pub id: String, -} - -#[tauri::command] -pub fn get_download_list(pathname: &str) -> (Vec, PathBuf) { - info!("get_download_list: {}", pathname); - let download_path = chat_root().join(PathBuf::from(pathname)); - let content = fs::read_to_string(&download_path).unwrap_or_else(|err| { - info!("download_list_error: {}", err); - fs::write(&download_path, "[]").unwrap(); - "[]".to_string() - }); - let list = serde_json::from_str::>(&content).unwrap_or_else(|err| { - info!("download_list_parse_error: {}", err); - vec![] - }); - - (list, download_path) -} - -#[command] -pub fn download_list(pathname: &str, dir: &str, filename: Option, id: Option) { - info!("download_list: {}", pathname); - let data = get_download_list(pathname); - let mut list = vec![]; - let mut idmap = HashMap::new(); - utils::vec_to_hashmap(data.0.into_iter(), "id", &mut idmap); - - for entry in WalkDir::new(utils::chat_root().join(dir)) - .into_iter() - .filter_entry(|e| !utils::is_hidden(e)) - .filter_map(|e| e.ok()) - { - let metadata = entry.metadata().unwrap(); - if metadata.is_file() { - let file_path = entry.path().display().to_string(); - let re = Regex::new(r"(?P[\d\w]+).(?P\w+)$").unwrap(); - let caps = re.captures(&file_path).unwrap(); - let fid = &caps["id"]; - let fext = &caps["ext"]; - - let mut file_data = FileMetadata { - name: fid.to_string(), - id: fid.to_string(), - ext: fext.to_string(), - created: fs_extra::system_time_to_ms(metadata.created()), - }; - - if idmap.get(fid).is_some() { - let name = idmap.get(fid).unwrap().get("name").unwrap().clone(); - match name { - serde_json::Value::String(v) => { - file_data.name = v.clone(); - v - } - _ => "".to_string(), - }; - } - - if filename.is_some() && id.is_some() { - if let Some(ref v) = id { - if fid == v { - if let Some(ref v2) = filename { - file_data.name = v2.to_string(); - } - } - } - } - list.push(serde_json::to_value(file_data).unwrap()); - } - } - - // dbg!(&list); - list.sort_by(|a, b| { - let a1 = a.get("created").unwrap().as_u64().unwrap(); - let b1 = b.get("created").unwrap().as_u64().unwrap(); - a1.cmp(&b1).reverse() - }); - - fs::write(data.1, serde_json::to_string_pretty(&list).unwrap()).unwrap(); -} - -#[command] -pub async fn sync_prompts(app: AppHandle, time: u64) -> Option> { - let res = utils::get_data(GITHUB_PROMPTS_CSV_URL, Some(&app)) - .await - .unwrap(); - - if let Some(v) = res { - let data = parse_prompt(v) - .iter() - .map(move |i| ModelRecord { - cmd: if i.cmd.is_some() { - i.cmd.clone().unwrap() - } else { - utils::gen_cmd(i.act.clone()) - }, - act: i.act.clone(), - prompt: i.prompt.clone(), - tags: vec!["chatgpt-prompts".to_string()], - enable: true, - }) - .collect::>(); - - let data2 = data.clone(); - - let model = utils::chat_root().join("chat.model.json"); - let model_cmd = utils::chat_root().join("chat.model.cmd.json"); - let chatgpt_prompts = utils::chat_root() - .join("cache_model") - .join("chatgpt_prompts.json"); - - if !utils::exists(&model) { - fs::write( - &model, - serde_json::json!({ - "name": "ChatGPT Model", - "link": "https://github.com/lencx/ChatGPT" - }) - .to_string(), - ) - .unwrap(); - } - - // chatgpt_prompts.json - fs::write( - chatgpt_prompts, - serde_json::to_string_pretty(&data).unwrap(), - ) - .unwrap(); - let cmd_data = cmd_list(); - - // chat.model.cmd.json - fs::write( - model_cmd, - serde_json::to_string_pretty(&serde_json::json!({ - "name": "ChatGPT CMD", - "last_updated": time, - "data": cmd_data, - })) - .unwrap(), - ) - .unwrap(); - let mut kv = HashMap::new(); - kv.insert( - "sync_prompts".to_string(), - serde_json::json!({ "id": "chatgpt_prompts", "last_updated": time }), - ); - let model_data = utils::merge( - &serde_json::from_str(&fs::read_to_string(&model).unwrap()).unwrap(), - &kv, - ); - - // chat.model.json - fs::write(model, serde_json::to_string_pretty(&model_data).unwrap()).unwrap(); - - // refresh window - api::dialog::message( - app.get_window("core").as_ref(), - "Sync Prompts", - "ChatGPT Prompts data has been synchronized!", - ); - window_reload(app.clone(), "core"); - window_reload(app, "tray"); - - return Some(data2); - } - - None -} - -#[command] -pub async fn sync_user_prompts(url: String, data_type: String) -> Option> { - let res = utils::get_data(&url, None).await.unwrap_or_else(|err| { - info!("chatgpt_http_error: {}", err); - None - }); - - info!("chatgpt_http_url: {}", url); - - if let Some(v) = res { - let data; - if data_type == "csv" { - info!("chatgpt_http_csv_parse"); - data = parse_prompt(v); - } else if data_type == "json" { - info!("chatgpt_http_json_parse"); - data = serde_json::from_str(&v).unwrap_or_else(|err| { - info!("chatgpt_http_json_parse_error: {}", err); - vec![] - }); - } else { - info!("chatgpt_http_unknown_type"); - data = vec![]; - } - - let data = data - .iter() - .map(move |i| ModelRecord { - cmd: if i.cmd.is_some() { - i.cmd.clone().unwrap() - } else { - utils::gen_cmd(i.act.clone()) - }, - act: i.act.clone(), - prompt: i.prompt.clone(), - tags: vec!["user-sync".to_string()], - enable: true, - }) - .collect::>(); - - return Some(data); - } - +pub async fn get_data(app: AppHandle, url: String, is_msg: Option) -> Option { + let is_msg = is_msg.unwrap_or(false); + let res = if is_msg { + utils::get_data(&url, Some(&app)).await + } else { + utils::get_data(&url, None).await + }; + res.unwrap_or_else(|err| { + info!("chatgpt_client_http_error: {}", err); None + }) } diff --git a/src-tauri/src/app/fs_extra.rs b/src-tauri/src/app/fs_extra.rs index 9ed9cf0..40b3c09 100644 --- a/src-tauri/src/app/fs_extra.rs +++ b/src-tauri/src/app/fs_extra.rs @@ -6,8 +6,8 @@ use serde::{ser::Serializer, Serialize}; use std::{ - path::PathBuf, - time::{SystemTime, UNIX_EPOCH}, + path::PathBuf, + time::{SystemTime, UNIX_EPOCH}, }; use tauri::command; @@ -20,101 +20,102 @@ type Result = std::result::Result; #[derive(Debug, thiserror::Error)] pub enum Error { - #[error(transparent)] - Io(#[from] std::io::Error), + #[error(transparent)] + Io(#[from] std::io::Error), } impl Serialize for Error { - fn serialize(&self, serializer: S) -> std::result::Result - where - S: Serializer, - { - serializer.serialize_str(self.to_string().as_ref()) - } + fn serialize(&self, serializer: S) -> std::result::Result + where + S: Serializer, + { + serializer.serialize_str(self.to_string().as_ref()) + } } #[derive(Serialize)] #[serde(rename_all = "camelCase")] struct Permissions { - readonly: bool, - #[cfg(unix)] - mode: u32, + readonly: bool, + #[cfg(unix)] + mode: u32, } #[cfg(unix)] #[derive(Serialize)] #[serde(rename_all = "camelCase")] struct UnixMetadata { - dev: u64, - ino: u64, - mode: u32, - nlink: u64, - uid: u32, - gid: u32, - rdev: u64, - blksize: u64, - blocks: u64, + dev: u64, + ino: u64, + mode: u32, + nlink: u64, + uid: u32, + gid: u32, + rdev: u64, + blksize: u64, + blocks: u64, } #[derive(Serialize)] #[serde(rename_all = "camelCase")] pub struct Metadata { - accessed_at_ms: u64, - pub created_at_ms: u64, - modified_at_ms: u64, - is_dir: bool, - is_file: bool, - is_symlink: bool, - size: u64, - permissions: Permissions, - #[cfg(unix)] - #[serde(flatten)] - unix: UnixMetadata, - #[cfg(windows)] - file_attributes: u32, + accessed_at_ms: u64, + pub created_at_ms: u64, + modified_at_ms: u64, + is_dir: bool, + is_file: bool, + is_symlink: bool, + size: u64, + permissions: Permissions, + #[cfg(unix)] + #[serde(flatten)] + unix: UnixMetadata, + #[cfg(windows)] + file_attributes: u32, } pub fn system_time_to_ms(time: std::io::Result) -> u64 { - time.map(|t| { - let duration_since_epoch = t.duration_since(UNIX_EPOCH).unwrap(); - duration_since_epoch.as_millis() as u64 + time + .map(|t| { + let duration_since_epoch = t.duration_since(UNIX_EPOCH).unwrap(); + duration_since_epoch.as_millis() as u64 }) .unwrap_or_default() } #[command] pub async fn metadata(path: PathBuf) -> Result { - let metadata = std::fs::metadata(path)?; - let file_type = metadata.file_type(); - let permissions = metadata.permissions(); - Ok(Metadata { - accessed_at_ms: system_time_to_ms(metadata.accessed()), - created_at_ms: system_time_to_ms(metadata.created()), - modified_at_ms: system_time_to_ms(metadata.modified()), - is_dir: file_type.is_dir(), - is_file: file_type.is_file(), - is_symlink: file_type.is_symlink(), - size: metadata.len(), - permissions: Permissions { - readonly: permissions.readonly(), - #[cfg(unix)] - mode: permissions.mode(), - }, - #[cfg(unix)] - unix: UnixMetadata { - dev: metadata.dev(), - ino: metadata.ino(), - mode: metadata.mode(), - nlink: metadata.nlink(), - uid: metadata.uid(), - gid: metadata.gid(), - rdev: metadata.rdev(), - blksize: metadata.blksize(), - blocks: metadata.blocks(), - }, - #[cfg(windows)] - file_attributes: metadata.file_attributes(), - }) + let metadata = std::fs::metadata(path)?; + let file_type = metadata.file_type(); + let permissions = metadata.permissions(); + Ok(Metadata { + accessed_at_ms: system_time_to_ms(metadata.accessed()), + created_at_ms: system_time_to_ms(metadata.created()), + modified_at_ms: system_time_to_ms(metadata.modified()), + is_dir: file_type.is_dir(), + is_file: file_type.is_file(), + is_symlink: file_type.is_symlink(), + size: metadata.len(), + permissions: Permissions { + readonly: permissions.readonly(), + #[cfg(unix)] + mode: permissions.mode(), + }, + #[cfg(unix)] + unix: UnixMetadata { + dev: metadata.dev(), + ino: metadata.ino(), + mode: metadata.mode(), + nlink: metadata.nlink(), + uid: metadata.uid(), + gid: metadata.gid(), + rdev: metadata.rdev(), + blksize: metadata.blksize(), + blocks: metadata.blocks(), + }, + #[cfg(windows)] + file_attributes: metadata.file_attributes(), + }) } // #[command] diff --git a/src-tauri/src/app/gpt.rs b/src-tauri/src/app/gpt.rs new file mode 100644 index 0000000..392c15f --- /dev/null +++ b/src-tauri/src/app/gpt.rs @@ -0,0 +1,294 @@ +use crate::{ + app::{fs_extra, window}, + conf::GITHUB_PROMPTS_CSV_URL, + utils::{self, chat_root}, +}; +use log::info; +use regex::Regex; +use std::{collections::HashMap, fs, path::PathBuf, vec}; +use tauri::{api, command, AppHandle, Manager}; +use walkdir::WalkDir; + +#[command] +pub fn get_chat_model_cmd() -> serde_json::Value { + let path = utils::chat_root().join("chat.model.cmd.json"); + let content = fs::read_to_string(path).unwrap_or_else(|_| r#"{"data":[]}"#.to_string()); + serde_json::from_str(&content).unwrap() +} + +#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)] +pub struct PromptRecord { + pub cmd: Option, + pub act: String, + pub prompt: String, +} + +#[command] +pub fn parse_prompt(data: String) -> Vec { + let mut rdr = csv::Reader::from_reader(data.as_bytes()); + let mut list = vec![]; + for result in rdr.deserialize() { + let record: PromptRecord = result.unwrap_or_else(|err| { + info!("parse_prompt_error: {}", err); + PromptRecord { + cmd: None, + act: "".to_string(), + prompt: "".to_string(), + } + }); + if !record.act.is_empty() { + list.push(record); + } + } + list +} + +#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)] +pub struct ModelRecord { + pub cmd: String, + pub act: String, + pub prompt: String, + pub tags: Vec, + pub enable: bool, +} + +#[command] +pub fn cmd_list() -> Vec { + let mut list = vec![]; + for entry in WalkDir::new(utils::chat_root().join("cache_model")) + .into_iter() + .filter_map(|e| e.ok()) + { + let file = fs::read_to_string(entry.path().display().to_string()); + if let Ok(v) = file { + let data: Vec = serde_json::from_str(&v).unwrap_or_else(|_| vec![]); + let enable_list = data.into_iter().filter(|v| v.enable); + list.extend(enable_list) + } + } + // dbg!(&list); + list.sort_by(|a, b| a.cmd.len().cmp(&b.cmd.len())); + list +} + +#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)] +pub struct FileMetadata { + pub name: String, + pub ext: String, + pub created: u64, + pub id: String, +} + +#[tauri::command] +pub fn get_download_list(pathname: &str) -> (Vec, PathBuf) { + info!("get_download_list: {}", pathname); + let download_path = chat_root().join(PathBuf::from(pathname)); + let content = fs::read_to_string(&download_path).unwrap_or_else(|err| { + info!("download_list_error: {}", err); + fs::write(&download_path, "[]").unwrap(); + "[]".to_string() + }); + let list = serde_json::from_str::>(&content).unwrap_or_else(|err| { + info!("download_list_parse_error: {}", err); + vec![] + }); + + (list, download_path) +} + +#[command] +pub fn download_list(pathname: &str, dir: &str, filename: Option, id: Option) { + info!("download_list: {}", pathname); + let data = get_download_list(pathname); + let mut list = vec![]; + let mut idmap = HashMap::new(); + utils::vec_to_hashmap(data.0.into_iter(), "id", &mut idmap); + + for entry in WalkDir::new(utils::chat_root().join(dir)) + .into_iter() + .filter_entry(|e| !utils::is_hidden(e)) + .filter_map(|e| e.ok()) + { + let metadata = entry.metadata().unwrap(); + if metadata.is_file() { + let file_path = entry.path().display().to_string(); + let re = Regex::new(r"(?P[\d\w]+).(?P\w+)$").unwrap(); + let caps = re.captures(&file_path).unwrap(); + let fid = &caps["id"]; + let fext = &caps["ext"]; + + let mut file_data = FileMetadata { + name: fid.to_string(), + id: fid.to_string(), + ext: fext.to_string(), + created: fs_extra::system_time_to_ms(metadata.created()), + }; + + if idmap.get(fid).is_some() { + let name = idmap.get(fid).unwrap().get("name").unwrap().clone(); + match name { + serde_json::Value::String(v) => { + file_data.name = v.clone(); + v + } + _ => "".to_string(), + }; + } + + if filename.is_some() && id.is_some() { + if let Some(ref v) = id { + if fid == v { + if let Some(ref v2) = filename { + file_data.name = v2.to_string(); + } + } + } + } + list.push(serde_json::to_value(file_data).unwrap()); + } + } + + // dbg!(&list); + list.sort_by(|a, b| { + let a1 = a.get("created").unwrap().as_u64().unwrap(); + let b1 = b.get("created").unwrap().as_u64().unwrap(); + a1.cmp(&b1).reverse() + }); + + fs::write(data.1, serde_json::to_string_pretty(&list).unwrap()).unwrap(); +} + +#[command] +pub async fn sync_prompts(app: AppHandle, time: u64) -> Option> { + let res = utils::get_data(GITHUB_PROMPTS_CSV_URL, Some(&app)) + .await + .unwrap(); + + if let Some(v) = res { + let data = parse_prompt(v) + .iter() + .map(move |i| ModelRecord { + cmd: if i.cmd.is_some() { + i.cmd.clone().unwrap() + } else { + utils::gen_cmd(i.act.clone()) + }, + act: i.act.clone(), + prompt: i.prompt.clone(), + tags: vec!["chatgpt-prompts".to_string()], + enable: true, + }) + .collect::>(); + + let data2 = data.clone(); + + let model = utils::chat_root().join("chat.model.json"); + let model_cmd = utils::chat_root().join("chat.model.cmd.json"); + let chatgpt_prompts = utils::chat_root() + .join("cache_model") + .join("chatgpt_prompts.json"); + + if !utils::exists(&model) { + fs::write( + &model, + serde_json::json!({ + "name": "ChatGPT Model", + "link": "https://github.com/lencx/ChatGPT" + }) + .to_string(), + ) + .unwrap(); + } + + // chatgpt_prompts.json + fs::write( + chatgpt_prompts, + serde_json::to_string_pretty(&data).unwrap(), + ) + .unwrap(); + let cmd_data = cmd_list(); + + // chat.model.cmd.json + fs::write( + model_cmd, + serde_json::to_string_pretty(&serde_json::json!({ + "name": "ChatGPT CMD", + "last_updated": time, + "data": cmd_data, + })) + .unwrap(), + ) + .unwrap(); + let mut kv = HashMap::new(); + kv.insert( + "sync_prompts".to_string(), + serde_json::json!({ "id": "chatgpt_prompts", "last_updated": time }), + ); + let model_data = utils::merge( + &serde_json::from_str(&fs::read_to_string(&model).unwrap()).unwrap(), + &kv, + ); + + // chat.model.json + fs::write(model, serde_json::to_string_pretty(&model_data).unwrap()).unwrap(); + + // refresh window + api::dialog::message( + app.get_window("core").as_ref(), + "Sync Prompts", + "ChatGPT Prompts data has been synchronized!", + ); + window::window_reload(app.clone(), "core"); + window::window_reload(app, "tray"); + + return Some(data2); + } + + None +} + +#[command] +pub async fn sync_user_prompts(url: String, data_type: String) -> Option> { + let res = utils::get_data(&url, None).await.unwrap_or_else(|err| { + info!("chatgpt_http_error: {}", err); + None + }); + + info!("chatgpt_http_url: {}", url); + + if let Some(v) = res { + let data; + if data_type == "csv" { + info!("chatgpt_http_csv_parse"); + data = parse_prompt(v); + } else if data_type == "json" { + info!("chatgpt_http_json_parse"); + data = serde_json::from_str(&v).unwrap_or_else(|err| { + info!("chatgpt_http_json_parse_error: {}", err); + vec![] + }); + } else { + info!("chatgpt_http_unknown_type"); + data = vec![]; + } + + let data = data + .iter() + .map(move |i| ModelRecord { + cmd: if i.cmd.is_some() { + i.cmd.clone().unwrap() + } else { + utils::gen_cmd(i.act.clone()) + }, + act: i.act.clone(), + prompt: i.prompt.clone(), + tags: vec!["user-sync".to_string()], + enable: true, + }) + .collect::>(); + + return Some(data); + } + + None +} diff --git a/src-tauri/src/app/menu.rs b/src-tauri/src/app/menu.rs index 1a0faf7..e507613 100644 --- a/src-tauri/src/app/menu.rs +++ b/src-tauri/src/app/menu.rs @@ -1,11 +1,11 @@ use crate::{ - app::{cmd, window}, - conf::{self, ChatConfJson}, - utils, + app::window, + conf::{self, ChatConfJson}, + utils, }; use tauri::{ - AppHandle, CustomMenuItem, Manager, Menu, MenuItem, Submenu, SystemTray, SystemTrayEvent, - SystemTrayMenu, SystemTrayMenuItem, WindowMenuEvent, + AppHandle, CustomMenuItem, Manager, Menu, MenuItem, Submenu, SystemTray, SystemTrayEvent, + SystemTrayMenu, SystemTrayMenuItem, WindowMenuEvent, }; use tauri_plugin_positioner::{on_tray_event, Position, WindowExt}; @@ -14,461 +14,449 @@ use tauri::AboutMetadata; // --- Menu pub fn init() -> Menu { - let chat_conf = ChatConfJson::get_chat_conf(); - let name = "ChatGPT"; - let app_menu = Submenu::new( - name, - Menu::with_items([ - #[cfg(target_os = "macos")] - MenuItem::About(name.into(), AboutMetadata::default()).into(), - #[cfg(not(target_os = "macos"))] - CustomMenuItem::new("about".to_string(), "About ChatGPT").into(), - CustomMenuItem::new("check_update".to_string(), "Check for Updates").into(), - MenuItem::Services.into(), - MenuItem::Hide.into(), - MenuItem::HideOthers.into(), - MenuItem::ShowAll.into(), - MenuItem::Separator.into(), - MenuItem::Quit.into(), - ]), - ); + let chat_conf = ChatConfJson::get_chat_conf(); + let name = "ChatGPT"; + let app_menu = Submenu::new( + name, + Menu::with_items([ + #[cfg(target_os = "macos")] + MenuItem::About(name.into(), AboutMetadata::default()).into(), + #[cfg(not(target_os = "macos"))] + CustomMenuItem::new("about".to_string(), "About ChatGPT").into(), + CustomMenuItem::new("check_update".to_string(), "Check for Updates").into(), + MenuItem::Services.into(), + MenuItem::Hide.into(), + MenuItem::HideOthers.into(), + MenuItem::ShowAll.into(), + MenuItem::Separator.into(), + MenuItem::Quit.into(), + ]), + ); - let stay_on_top = - CustomMenuItem::new("stay_on_top".to_string(), "Stay On Top").accelerator("CmdOrCtrl+T"); - let stay_on_top_menu = if chat_conf.stay_on_top { - stay_on_top.selected() - } else { - stay_on_top - }; + let stay_on_top = + CustomMenuItem::new("stay_on_top".to_string(), "Stay On Top").accelerator("CmdOrCtrl+T"); + let stay_on_top_menu = if chat_conf.stay_on_top { + stay_on_top.selected() + } else { + stay_on_top + }; - let theme_light = CustomMenuItem::new("theme_light".to_string(), "Light"); - let theme_dark = CustomMenuItem::new("theme_dark".to_string(), "Dark"); - let theme_system = CustomMenuItem::new("theme_system".to_string(), "System"); - let is_dark = chat_conf.theme == "Dark"; - let is_system = chat_conf.theme == "System"; + let theme_light = CustomMenuItem::new("theme_light".to_string(), "Light"); + let theme_dark = CustomMenuItem::new("theme_dark".to_string(), "Dark"); + let theme_system = CustomMenuItem::new("theme_system".to_string(), "System"); + let is_dark = chat_conf.theme == "Dark"; + let is_system = chat_conf.theme == "System"; - let update_prompt = CustomMenuItem::new("update_prompt".to_string(), "Prompt"); - let update_silent = CustomMenuItem::new("update_silent".to_string(), "Silent"); - let _update_disable = CustomMenuItem::new("update_disable".to_string(), "Disable"); + let update_prompt = CustomMenuItem::new("update_prompt".to_string(), "Prompt"); + let update_silent = CustomMenuItem::new("update_silent".to_string(), "Silent"); + let _update_disable = CustomMenuItem::new("update_disable".to_string(), "Disable"); - let popup_search = CustomMenuItem::new("popup_search".to_string(), "Pop-up Search"); - let popup_search_menu = if chat_conf.popup_search { - popup_search.selected() - } else { - popup_search - }; + let popup_search = CustomMenuItem::new("popup_search".to_string(), "Pop-up Search"); + let popup_search_menu = if chat_conf.popup_search { + popup_search.selected() + } else { + popup_search + }; - #[cfg(target_os = "macos")] - let titlebar = - CustomMenuItem::new("titlebar".to_string(), "Titlebar").accelerator("CmdOrCtrl+B"); - #[cfg(target_os = "macos")] - let titlebar_menu = if chat_conf.titlebar { - titlebar.selected() - } else { - titlebar - }; + #[cfg(target_os = "macos")] + let titlebar = CustomMenuItem::new("titlebar".to_string(), "Titlebar").accelerator("CmdOrCtrl+B"); + #[cfg(target_os = "macos")] + let titlebar_menu = if chat_conf.titlebar { + titlebar.selected() + } else { + titlebar + }; - let system_tray = CustomMenuItem::new("system_tray".to_string(), "System Tray"); - let system_tray_menu = if chat_conf.tray { - system_tray.selected() - } else { - system_tray - }; + let system_tray = CustomMenuItem::new("system_tray".to_string(), "System Tray"); + let system_tray_menu = if chat_conf.tray { + system_tray.selected() + } else { + system_tray + }; - let preferences_menu = Submenu::new( - "Preferences", - Menu::with_items([ - CustomMenuItem::new("control_center".to_string(), "Control Center") - .accelerator("CmdOrCtrl+Shift+P") - .into(), - MenuItem::Separator.into(), - stay_on_top_menu.into(), - #[cfg(target_os = "macos")] - titlebar_menu.into(), - #[cfg(target_os = "macos")] - CustomMenuItem::new("hide_dock_icon".to_string(), "Hide Dock Icon").into(), - system_tray_menu.into(), - CustomMenuItem::new("inject_script".to_string(), "Inject Script") - .accelerator("CmdOrCtrl+J") - .into(), - MenuItem::Separator.into(), - Submenu::new( - "Theme", - Menu::new() - .add_item(if is_dark || is_system { - theme_light - } else { - theme_light.selected() - }) - .add_item(if is_dark { - theme_dark.selected() - } else { - theme_dark - }) - .add_item(if is_system { - theme_system.selected() - } else { - theme_system - }), - ) - .into(), - Submenu::new( - "Auto Update", - Menu::new() - .add_item(if chat_conf.auto_update == "Prompt" { - update_prompt.selected() - } else { - update_prompt - }) - .add_item(if chat_conf.auto_update == "Silent" { - update_silent.selected() - } else { - update_silent - }), // .add_item(if chat_conf.auto_update == "Disable" { - // update_disable.selected() - // } else { - // update_disable - // }) - ) - .into(), - MenuItem::Separator.into(), - popup_search_menu.into(), - CustomMenuItem::new("sync_prompts".to_string(), "Sync Prompts").into(), - MenuItem::Separator.into(), - CustomMenuItem::new("go_conf".to_string(), "Go to Config") - .accelerator("CmdOrCtrl+Shift+G") - .into(), - CustomMenuItem::new("clear_conf".to_string(), "Clear Config") - .accelerator("CmdOrCtrl+Shift+D") - .into(), - CustomMenuItem::new("restart".to_string(), "Restart ChatGPT") - .accelerator("CmdOrCtrl+Shift+R") - .into(), - MenuItem::Separator.into(), - CustomMenuItem::new("awesome".to_string(), "Awesome ChatGPT") - .accelerator("CmdOrCtrl+Shift+A") - .into(), - CustomMenuItem::new("buy_coffee".to_string(), "Buy lencx a coffee").into(), - ]), - ); - - let edit_menu = Submenu::new( - "Edit", + let preferences_menu = Submenu::new( + "Preferences", + Menu::with_items([ + CustomMenuItem::new("control_center".to_string(), "Control Center") + .accelerator("CmdOrCtrl+Shift+P") + .into(), + MenuItem::Separator.into(), + stay_on_top_menu.into(), + #[cfg(target_os = "macos")] + titlebar_menu.into(), + #[cfg(target_os = "macos")] + CustomMenuItem::new("hide_dock_icon".to_string(), "Hide Dock Icon").into(), + system_tray_menu.into(), + CustomMenuItem::new("inject_script".to_string(), "Inject Script") + .accelerator("CmdOrCtrl+J") + .into(), + MenuItem::Separator.into(), + Submenu::new( + "Theme", 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", + .add_item(if is_dark || is_system { + theme_light + } else { + theme_light.selected() + }) + .add_item(if is_dark { + theme_dark.selected() + } else { + theme_dark + }) + .add_item(if is_system { + theme_system.selected() + } else { + theme_system + }), + ) + .into(), + Submenu::new( + "Auto Update", 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 window_menu = Submenu::new( - "Window", - Menu::new() - .add_item(CustomMenuItem::new("dalle2".to_string(), "DALL·E 2")) - .add_native_item(MenuItem::Separator) - .add_native_item(MenuItem::Minimize) - .add_native_item(MenuItem::Zoom), - ); - - let help_menu = Submenu::new( - "Help", - Menu::new() - .add_item(CustomMenuItem::new( - "chatgpt_log".to_string(), - "ChatGPT Log", - )) - .add_item(CustomMenuItem::new("update_log".to_string(), "Update Log")) - .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"), - ), - ); + .add_item(if chat_conf.auto_update == "Prompt" { + update_prompt.selected() + } else { + update_prompt + }) + .add_item(if chat_conf.auto_update == "Silent" { + update_silent.selected() + } else { + update_silent + }), // .add_item(if chat_conf.auto_update == "Disable" { + // update_disable.selected() + // } else { + // update_disable + // }) + ) + .into(), + MenuItem::Separator.into(), + popup_search_menu.into(), + CustomMenuItem::new("sync_prompts".to_string(), "Sync Prompts").into(), + MenuItem::Separator.into(), + CustomMenuItem::new("go_conf".to_string(), "Go to Config") + .accelerator("CmdOrCtrl+Shift+G") + .into(), + CustomMenuItem::new("restart".to_string(), "Restart ChatGPT") + .accelerator("CmdOrCtrl+Shift+R") + .into(), + CustomMenuItem::new("clear_conf".to_string(), "Clear Config").into(), + MenuItem::Separator.into(), + CustomMenuItem::new("awesome".to_string(), "Awesome ChatGPT").into(), + CustomMenuItem::new("buy_coffee".to_string(), "Buy lencx a coffee").into(), + ]), + ); + let edit_menu = Submenu::new( + "Edit", Menu::new() - .add_submenu(app_menu) - .add_submenu(preferences_menu) - .add_submenu(window_menu) - .add_submenu(edit_menu) - .add_submenu(view_menu) - .add_submenu(help_menu) + .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 window_menu = Submenu::new( + "Window", + Menu::new() + .add_item(CustomMenuItem::new("dalle2".to_string(), "DALL·E 2")) + .add_native_item(MenuItem::Separator) + .add_native_item(MenuItem::Minimize) + .add_native_item(MenuItem::Zoom), + ); + + let help_menu = Submenu::new( + "Help", + Menu::new() + .add_item(CustomMenuItem::new( + "chatgpt_log".to_string(), + "ChatGPT Log", + )) + .add_item(CustomMenuItem::new("update_log".to_string(), "Update Log")) + .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(preferences_menu) + .add_submenu(window_menu) + .add_submenu(edit_menu) + .add_submenu(view_menu) + .add_submenu(help_menu) } // --- Menu Event pub fn menu_handler(event: WindowMenuEvent) { - let win = Some(event.window()).unwrap(); - let app = win.app_handle(); - let script_path = utils::script_path().to_string_lossy().to_string(); - let menu_id = event.menu_item_id(); - let menu_handle = win.menu_handle(); + let win = Some(event.window()).unwrap(); + let app = win.app_handle(); + let script_path = utils::script_path().to_string_lossy().to_string(); + let menu_id = event.menu_item_id(); + let menu_handle = win.menu_handle(); - match menu_id { - // App - "about" => { - let tauri_conf = utils::get_tauri_conf().unwrap(); - tauri::api::dialog::message( - app.get_window("core").as_ref(), - "ChatGPT", - format!("Version {}", tauri_conf.package.version.unwrap()), - ); - } - "check_update" => { - utils::run_check_update(app, false, None); - } - // Preferences - "control_center" => window::control_window(&app), - "restart" => tauri::api::process::restart(&app.env()), - "inject_script" => open(&app, script_path), - "go_conf" => utils::open_file(utils::chat_root()), - "clear_conf" => utils::clear_conf(&app), - "awesome" => open(&app, conf::AWESOME_URL.to_string()), - "buy_coffee" => open(&app, conf::BUY_COFFEE.to_string()), - "popup_search" => { - let chat_conf = conf::ChatConfJson::get_chat_conf(); - let popup_search = !chat_conf.popup_search; - menu_handle - .get_item(menu_id) - .set_selected(popup_search) - .unwrap(); - ChatConfJson::amend(&serde_json::json!({ "popup_search": popup_search }), None) - .unwrap(); - cmd::window_reload(app.clone(), "core"); - cmd::window_reload(app, "tray"); - } - "sync_prompts" => { - tauri::api::dialog::ask( - app.get_window("core").as_ref(), - "Sync Prompts", - "Data sync will enable all prompts, are you sure you want to sync?", - move |is_restart| { - if is_restart { - app.get_window("core") - .unwrap() - .eval("window.__sync_prompts && window.__sync_prompts()") - .unwrap() - } - }, - ); - } - "hide_dock_icon" => { - ChatConfJson::amend(&serde_json::json!({ "hide_dock_icon": true }), Some(app)).unwrap() - } - "titlebar" => { - let chat_conf = conf::ChatConfJson::get_chat_conf(); - ChatConfJson::amend( - &serde_json::json!({ "titlebar": !chat_conf.titlebar }), - None, - ) - .unwrap(); - tauri::api::process::restart(&app.env()); - } - "system_tray" => { - let chat_conf = conf::ChatConfJson::get_chat_conf(); - ChatConfJson::amend(&serde_json::json!({ "tray": !chat_conf.tray }), None).unwrap(); - tauri::api::process::restart(&app.env()); - } - "theme_light" | "theme_dark" | "theme_system" => { - let theme = match menu_id { - "theme_dark" => "Dark", - "theme_system" => "System", - _ => "Light", - }; - ChatConfJson::amend(&serde_json::json!({ "theme": theme }), Some(app)).unwrap(); - } - "update_prompt" | "update_silent" | "update_disable" => { - // for id in ["update_prompt", "update_silent", "update_disable"] { - for id in ["update_prompt", "update_silent"] { - menu_handle.get_item(id).set_selected(false).unwrap(); - } - let auto_update = match menu_id { - "update_silent" => { - menu_handle - .get_item("update_silent") - .set_selected(true) - .unwrap(); - "Silent" - } - "update_disable" => { - menu_handle - .get_item("update_disable") - .set_selected(true) - .unwrap(); - "Disable" - } - _ => { - menu_handle - .get_item("update_prompt") - .set_selected(true) - .unwrap(); - "Prompt" - } - }; - ChatConfJson::amend(&serde_json::json!({ "auto_update": auto_update }), None).unwrap(); - } - "stay_on_top" => { - let chat_conf = conf::ChatConfJson::get_chat_conf(); - let stay_on_top = !chat_conf.stay_on_top; - menu_handle - .get_item(menu_id) - .set_selected(stay_on_top) - .unwrap(); - win.set_always_on_top(stay_on_top).unwrap(); - ChatConfJson::amend(&serde_json::json!({ "stay_on_top": stay_on_top }), None).unwrap(); - } - // Window - "dalle2" => window::dalle2_window(&app, None, None, Some(false)), - // View - "reload" => win.eval("window.location.reload()").unwrap(), - "go_back" => win.eval("window.history.go(-1)").unwrap(), - "go_forward" => win.eval("window.history.go(1)").unwrap(), - // core: document.querySelector('main .overflow-y-auto') - "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(), - // Help - "chatgpt_log" => utils::open_file(utils::chat_root().join("chatgpt.log")), - "update_log" => open(&app, conf::UPDATE_LOG_URL.to_string()), - "report_bug" => open(&app, conf::ISSUES_URL.to_string()), - "dev_tools" => { - win.open_devtools(); - win.close_devtools(); - } - _ => (), + match menu_id { + // App + "about" => { + let tauri_conf = utils::get_tauri_conf().unwrap(); + tauri::api::dialog::message( + app.get_window("core").as_ref(), + "ChatGPT", + format!("Version {}", tauri_conf.package.version.unwrap()), + ); } + "check_update" => { + utils::run_check_update(app, false, None); + } + // Preferences + "control_center" => window::control_window(app), + "restart" => tauri::api::process::restart(&app.env()), + "inject_script" => open(&app, script_path), + "go_conf" => utils::open_file(utils::chat_root()), + "clear_conf" => utils::clear_conf(&app), + "awesome" => open(&app, conf::AWESOME_URL.to_string()), + "buy_coffee" => open(&app, conf::BUY_COFFEE.to_string()), + "popup_search" => { + let chat_conf = conf::ChatConfJson::get_chat_conf(); + let popup_search = !chat_conf.popup_search; + menu_handle + .get_item(menu_id) + .set_selected(popup_search) + .unwrap(); + ChatConfJson::amend(&serde_json::json!({ "popup_search": popup_search }), None).unwrap(); + window::window_reload(app.clone(), "core"); + window::window_reload(app, "tray"); + } + "sync_prompts" => { + tauri::api::dialog::ask( + app.get_window("core").as_ref(), + "Sync Prompts", + "Data sync will enable all prompts, are you sure you want to sync?", + move |is_restart| { + if is_restart { + app + .get_window("core") + .unwrap() + .eval("window.__sync_prompts && window.__sync_prompts()") + .unwrap() + } + }, + ); + } + "hide_dock_icon" => { + ChatConfJson::amend(&serde_json::json!({ "hide_dock_icon": true }), Some(app)).unwrap() + } + "titlebar" => { + let chat_conf = conf::ChatConfJson::get_chat_conf(); + ChatConfJson::amend( + &serde_json::json!({ "titlebar": !chat_conf.titlebar }), + None, + ) + .unwrap(); + tauri::api::process::restart(&app.env()); + } + "system_tray" => { + let chat_conf = conf::ChatConfJson::get_chat_conf(); + ChatConfJson::amend(&serde_json::json!({ "tray": !chat_conf.tray }), None).unwrap(); + tauri::api::process::restart(&app.env()); + } + "theme_light" | "theme_dark" | "theme_system" => { + let theme = match menu_id { + "theme_dark" => "Dark", + "theme_system" => "System", + _ => "Light", + }; + ChatConfJson::amend(&serde_json::json!({ "theme": theme }), Some(app)).unwrap(); + } + "update_prompt" | "update_silent" | "update_disable" => { + // for id in ["update_prompt", "update_silent", "update_disable"] { + for id in ["update_prompt", "update_silent"] { + menu_handle.get_item(id).set_selected(false).unwrap(); + } + let auto_update = match menu_id { + "update_silent" => { + menu_handle + .get_item("update_silent") + .set_selected(true) + .unwrap(); + "Silent" + } + "update_disable" => { + menu_handle + .get_item("update_disable") + .set_selected(true) + .unwrap(); + "Disable" + } + _ => { + menu_handle + .get_item("update_prompt") + .set_selected(true) + .unwrap(); + "Prompt" + } + }; + ChatConfJson::amend(&serde_json::json!({ "auto_update": auto_update }), None).unwrap(); + } + "stay_on_top" => { + let chat_conf = conf::ChatConfJson::get_chat_conf(); + let stay_on_top = !chat_conf.stay_on_top; + menu_handle + .get_item(menu_id) + .set_selected(stay_on_top) + .unwrap(); + win.set_always_on_top(stay_on_top).unwrap(); + ChatConfJson::amend(&serde_json::json!({ "stay_on_top": stay_on_top }), None).unwrap(); + } + // Window + "dalle2" => window::dalle2_window(&app, None, None, Some(false)), + // View + "reload" => win.eval("window.location.reload()").unwrap(), + "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(), + // Help + "chatgpt_log" => utils::open_file(utils::chat_root().join("chatgpt.log")), + "update_log" => open(&app, conf::UPDATE_LOG_URL.to_string()), + "report_bug" => open(&app, conf::ISSUES_URL.to_string()), + "dev_tools" => { + win.open_devtools(); + win.close_devtools(); + } + _ => (), + } } // --- SystemTray Menu pub fn tray_menu() -> SystemTray { - if cfg!(target_os = "macos") { - SystemTray::new().with_menu( - SystemTrayMenu::new() - .add_item(CustomMenuItem::new( - "control_center".to_string(), - "Control Center", - )) - .add_native_item(SystemTrayMenuItem::Separator) - .add_item(CustomMenuItem::new( - "show_dock_icon".to_string(), - "Show Dock Icon", - )) - .add_item(CustomMenuItem::new( - "hide_dock_icon".to_string(), - "Hide Dock Icon", - )) - .add_item(CustomMenuItem::new("show_core".to_string(), "Show ChatGPT")) - .add_native_item(SystemTrayMenuItem::Separator) - .add_item(CustomMenuItem::new("quit".to_string(), "Quit ChatGPT")), - ) - } else { - SystemTray::new().with_menu( - SystemTrayMenu::new() - .add_item(CustomMenuItem::new( - "control_center".to_string(), - "Control Center", - )) - .add_item(CustomMenuItem::new("show_core".to_string(), "Show ChatGPT")) - .add_native_item(SystemTrayMenuItem::Separator) - .add_item(CustomMenuItem::new("quit".to_string(), "Quit ChatGPT")), - ) - } + if cfg!(target_os = "macos") { + SystemTray::new().with_menu( + SystemTrayMenu::new() + .add_item(CustomMenuItem::new( + "control_center".to_string(), + "Control Center", + )) + .add_native_item(SystemTrayMenuItem::Separator) + .add_item(CustomMenuItem::new( + "show_dock_icon".to_string(), + "Show Dock Icon", + )) + .add_item(CustomMenuItem::new( + "hide_dock_icon".to_string(), + "Hide Dock Icon", + )) + .add_item(CustomMenuItem::new("show_core".to_string(), "Show ChatGPT")) + .add_native_item(SystemTrayMenuItem::Separator) + .add_item(CustomMenuItem::new("quit".to_string(), "Quit ChatGPT")), + ) + } else { + SystemTray::new().with_menu( + SystemTrayMenu::new() + .add_item(CustomMenuItem::new( + "control_center".to_string(), + "Control Center", + )) + .add_item(CustomMenuItem::new("show_core".to_string(), "Show ChatGPT")) + .add_native_item(SystemTrayMenuItem::Separator) + .add_item(CustomMenuItem::new("quit".to_string(), "Quit ChatGPT")), + ) + } } // --- SystemTray Event pub fn tray_handler(handle: &AppHandle, event: SystemTrayEvent) { - on_tray_event(handle, &event); + on_tray_event(handle, &event); - let app = handle.clone(); + let app = handle.clone(); - match event { - SystemTrayEvent::LeftClick { .. } => { - let chat_conf = conf::ChatConfJson::get_chat_conf(); + match event { + SystemTrayEvent::LeftClick { .. } => { + let chat_conf = conf::ChatConfJson::get_chat_conf(); - if !chat_conf.hide_dock_icon { - let core_win = handle.get_window("core").unwrap(); - core_win.minimize().unwrap(); - } + if !chat_conf.hide_dock_icon { + let core_win = handle.get_window("core").unwrap(); + core_win.minimize().unwrap(); + } - let tray_win = handle.get_window("tray").unwrap(); - tray_win.move_window(Position::TrayCenter).unwrap(); + let tray_win = handle.get_window("tray").unwrap(); + tray_win.move_window(Position::TrayCenter).unwrap(); - if tray_win.is_visible().unwrap() { - tray_win.hide().unwrap(); - } else { - tray_win.show().unwrap(); - } - } - SystemTrayEvent::MenuItemClick { id, .. } => match id.as_str() { - "control_center" => window::control_window(&app), - "restart" => tauri::api::process::restart(&handle.env()), - "show_dock_icon" => { - ChatConfJson::amend(&serde_json::json!({ "hide_dock_icon": false }), Some(app)) - .unwrap(); - } - "hide_dock_icon" => { - let chat_conf = conf::ChatConfJson::get_chat_conf(); - if !chat_conf.hide_dock_icon { - ChatConfJson::amend(&serde_json::json!({ "hide_dock_icon": true }), Some(app)) - .unwrap(); - } - } - "show_core" => { - let core_win = app.get_window("core").unwrap(); - let tray_win = app.get_window("tray").unwrap(); - if !core_win.is_visible().unwrap() { - core_win.show().unwrap(); - core_win.set_focus().unwrap(); - tray_win.hide().unwrap(); - } - } - "quit" => std::process::exit(0), - _ => (), - }, - _ => (), + if tray_win.is_visible().unwrap() { + tray_win.hide().unwrap(); + } else { + tray_win.show().unwrap(); + } } + SystemTrayEvent::MenuItemClick { id, .. } => match id.as_str() { + "control_center" => window::control_window(app), + "restart" => tauri::api::process::restart(&handle.env()), + "show_dock_icon" => { + ChatConfJson::amend(&serde_json::json!({ "hide_dock_icon": false }), Some(app)).unwrap(); + } + "hide_dock_icon" => { + let chat_conf = conf::ChatConfJson::get_chat_conf(); + if !chat_conf.hide_dock_icon { + ChatConfJson::amend(&serde_json::json!({ "hide_dock_icon": true }), Some(app)).unwrap(); + } + } + "show_core" => { + let core_win = app.get_window("core").unwrap(); + let tray_win = app.get_window("tray").unwrap(); + if !core_win.is_visible().unwrap() { + core_win.show().unwrap(); + core_win.set_focus().unwrap(); + tray_win.hide().unwrap(); + } + } + "quit" => std::process::exit(0), + _ => (), + }, + _ => (), + } } pub fn open(app: &AppHandle, path: String) { - tauri::api::shell::open(&app.shell_scope(), path, None).unwrap(); + tauri::api::shell::open(&app.shell_scope(), path, None).unwrap(); } diff --git a/src-tauri/src/app/mod.rs b/src-tauri/src/app/mod.rs index 46d47f9..39fac31 100644 --- a/src-tauri/src/app/mod.rs +++ b/src-tauri/src/app/mod.rs @@ -1,5 +1,6 @@ pub mod cmd; pub mod fs_extra; +pub mod gpt; pub mod menu; pub mod setup; pub mod window; diff --git a/src-tauri/src/app/setup.rs b/src-tauri/src/app/setup.rs index 1dcbe76..f596ec3 100644 --- a/src-tauri/src/app/setup.rs +++ b/src-tauri/src/app/setup.rs @@ -4,110 +4,98 @@ use tauri::{utils::config::WindowUrl, window::WindowBuilder, App, GlobalShortcut use wry::application::accelerator::Accelerator; pub fn init(app: &mut App) -> std::result::Result<(), Box> { - info!("stepup"); - let chat_conf = ChatConfJson::get_chat_conf(); - let url = chat_conf.origin.to_string(); - let theme = ChatConfJson::theme(); - let handle = app.app_handle(); + info!("stepup"); + let chat_conf = ChatConfJson::get_chat_conf(); + let url = chat_conf.main_origin.to_string(); + let theme = ChatConfJson::theme(); + let handle = app.app_handle(); + tauri::async_runtime::spawn(async move { + window::tray_window(&handle); + }); + + if let Some(v) = chat_conf.global_shortcut { + info!("global_shortcut: `{}`", v); + match v.parse::() { + Ok(_) => { + info!("global_shortcut_register"); + let handle = app.app_handle(); + let mut shortcut = app.global_shortcut_manager(); + shortcut + .register(&v, move || { + if let Some(w) = handle.get_window("core") { + if w.is_visible().unwrap() { + w.hide().unwrap(); + } else { + w.show().unwrap(); + w.set_focus().unwrap(); + } + } + }) + .unwrap_or_else(|err| { + info!("global_shortcut_register_error: {}", err); + }); + } + Err(err) => { + info!("global_shortcut_parse_error: {}", err); + } + } + } else { + info!("global_shortcut_unregister"); + }; + + if chat_conf.hide_dock_icon { + #[cfg(target_os = "macos")] + app.set_activation_policy(tauri::ActivationPolicy::Accessory); + } else { + let app = app.handle(); tauri::async_runtime::spawn(async move { - window::tray_window(&handle); + let link = if chat_conf.main_dashboard { + "index.html" + } else { + &url + }; + let mut main_win = WindowBuilder::new(&app, "core", WindowUrl::App(link.into())) + .title("ChatGPT") + .resizable(true) + .fullscreen(false) + .inner_size(800.0, 600.0) + .theme(theme) + .always_on_top(chat_conf.stay_on_top) + .initialization_script(&utils::user_script()) + .initialization_script(include_str!("../scripts/core.js")) + .user_agent(&chat_conf.ua_window); + + if cfg!(target_os = "macos") { + main_win = main_win + .title_bar_style(ChatConfJson::titlebar()) + .hidden_title(true); + } + + if url == "https://chat.openai.com" && !chat_conf.main_dashboard { + main_win = main_win + .initialization_script(include_str!("../vendors/floating-ui-core.js")) + .initialization_script(include_str!("../vendors/floating-ui-dom.js")) + .initialization_script(include_str!("../vendors/html2canvas.js")) + .initialization_script(include_str!("../vendors/jspdf.js")) + .initialization_script(include_str!("../vendors/turndown.js")) + .initialization_script(include_str!("../vendors/turndown-plugin-gfm.js")) + .initialization_script(include_str!("../scripts/popup.core.js")) + .initialization_script(include_str!("../scripts/export.js")) + .initialization_script(include_str!("../scripts/markdown.export.js")) + .initialization_script(include_str!("../scripts/cmd.js")) + } + + main_win.build().unwrap(); }); + } - if let Some(v) = chat_conf.global_shortcut { - info!("global_shortcut: `{}`", v); - match v.parse::() { - Ok(_) => { - info!("global_shortcut_register"); - let handle = app.app_handle(); - let mut shortcut = app.global_shortcut_manager(); - shortcut - .register(&v, move || { - if let Some(w) = handle.get_window("core") { - if w.is_visible().unwrap() { - w.hide().unwrap(); - } else { - w.show().unwrap(); - w.set_focus().unwrap(); - } - } - }) - .unwrap_or_else(|err| { - info!("global_shortcut_register_error: {}", err); - }); - } - Err(err) => { - info!("global_shortcut_parse_error: {}", err); - } - } - } else { - info!("global_shortcut_unregister"); - }; + // auto_update + if chat_conf.auto_update != "Disable" { + info!("stepup::run_check_update"); + let app = app.handle(); + utils::run_check_update(app, chat_conf.auto_update == "Silent", None); + } - if chat_conf.hide_dock_icon { - #[cfg(target_os = "macos")] - app.set_activation_policy(tauri::ActivationPolicy::Accessory); - } else { - let app = app.handle(); - tauri::async_runtime::spawn(async move { - #[cfg(target_os = "macos")] - WindowBuilder::new(&app, "core", WindowUrl::App(url.into())) - .title("ChatGPT") - .resizable(true) - .fullscreen(false) - .inner_size(800.0, 600.0) - .hidden_title(true) - .theme(theme) - .always_on_top(chat_conf.stay_on_top) - .title_bar_style(ChatConfJson::titlebar()) - .initialization_script(&utils::user_script()) - .initialization_script(include_str!("../vendors/floating-ui-core.js")) - .initialization_script(include_str!("../vendors/floating-ui-dom.js")) - .initialization_script(include_str!("../vendors/html2canvas.js")) - .initialization_script(include_str!("../vendors/jspdf.js")) - .initialization_script(include_str!("../vendors/turndown.js")) - .initialization_script(include_str!("../vendors/turndown-plugin-gfm.js")) - .initialization_script(include_str!("../scripts/core.js")) - .initialization_script(include_str!("../scripts/popup.core.js")) - .initialization_script(include_str!("../scripts/export.js")) - .initialization_script(include_str!("../scripts/markdown.export.js")) - .initialization_script(include_str!("../scripts/cmd.js")) - .user_agent(&chat_conf.ua_window) - .build() - .unwrap(); - - #[cfg(not(target_os = "macos"))] - WindowBuilder::new(&app, "core", WindowUrl::App(url.into())) - .title("ChatGPT") - .resizable(true) - .fullscreen(false) - .inner_size(800.0, 600.0) - .theme(theme) - .always_on_top(chat_conf.stay_on_top) - .initialization_script(&utils::user_script()) - .initialization_script(include_str!("../vendors/floating-ui-core.js")) - .initialization_script(include_str!("../vendors/floating-ui-dom.js")) - .initialization_script(include_str!("../vendors/html2canvas.js")) - .initialization_script(include_str!("../vendors/jspdf.js")) - .initialization_script(include_str!("../vendors/turndown.js")) - .initialization_script(include_str!("../vendors/turndown-plugin-gfm.js")) - .initialization_script(include_str!("../scripts/core.js")) - .initialization_script(include_str!("../scripts/popup.core.js")) - .initialization_script(include_str!("../scripts/export.js")) - .initialization_script(include_str!("../scripts/markdown.export.js")) - .initialization_script(include_str!("../scripts/cmd.js")) - .user_agent(&chat_conf.ua_window) - .build() - .unwrap(); - }); - } - - // auto_update - if chat_conf.auto_update != "Disable" { - info!("stepup::run_check_update"); - let app = app.handle(); - utils::run_check_update(app, chat_conf.auto_update == "Silent", None); - } - - Ok(()) + Ok(()) } diff --git a/src-tauri/src/app/window.rs b/src-tauri/src/app/window.rs index d1dcf53..cd7cf87 100644 --- a/src-tauri/src/app/window.rs +++ b/src-tauri/src/app/window.rs @@ -4,104 +4,164 @@ use std::time::SystemTime; use tauri::{utils::config::WindowUrl, window::WindowBuilder, Manager}; pub fn tray_window(handle: &tauri::AppHandle) { - let chat_conf = conf::ChatConfJson::get_chat_conf(); - let theme = conf::ChatConfJson::theme(); - let app = handle.clone(); + let chat_conf = conf::ChatConfJson::get_chat_conf(); + let theme = conf::ChatConfJson::theme(); + let app = handle.clone(); - tauri::async_runtime::spawn(async move { - WindowBuilder::new(&app, "tray", WindowUrl::App(chat_conf.origin.into())) - .title("ChatGPT") - .resizable(false) - .fullscreen(false) - .inner_size(360.0, 540.0) - .decorations(false) - .always_on_top(true) - .theme(theme) - .initialization_script(&utils::user_script()) - .initialization_script(include_str!("../vendors/floating-ui-core.js")) - .initialization_script(include_str!("../vendors/floating-ui-dom.js")) - .initialization_script(include_str!("../scripts/core.js")) - .initialization_script(include_str!("../scripts/cmd.js")) - .initialization_script(include_str!("../scripts/popup.core.js")) - .user_agent(&chat_conf.ua_tray) - .build() - .unwrap() - .hide() - .unwrap(); - }); + tauri::async_runtime::spawn(async move { + let link = if chat_conf.tray_dashboard { + "index.html" + } else { + &chat_conf.tray_origin + }; + let mut tray_win = WindowBuilder::new(&app, "tray", WindowUrl::App(link.into())) + .title("ChatGPT") + .resizable(false) + .fullscreen(false) + .inner_size(360.0, 540.0) + .decorations(false) + .always_on_top(true) + .theme(theme) + .initialization_script(&utils::user_script()) + .initialization_script(include_str!("../scripts/core.js")) + .user_agent(&chat_conf.ua_tray); + + if chat_conf.tray_origin == "https://chat.openai.com" && !chat_conf.tray_dashboard { + tray_win = tray_win + .initialization_script(include_str!("../vendors/floating-ui-core.js")) + .initialization_script(include_str!("../vendors/floating-ui-dom.js")) + .initialization_script(include_str!("../scripts/cmd.js")) + .initialization_script(include_str!("../scripts/popup.core.js")) + } + + tray_win.build().unwrap().hide().unwrap(); + }); } pub fn dalle2_window( - handle: &tauri::AppHandle, - query: Option, - title: Option, - is_new: Option, + handle: &tauri::AppHandle, + query: Option, + title: Option, + is_new: Option, ) { - info!("dalle2_query: {:?}", query); - let theme = conf::ChatConfJson::theme(); - let app = handle.clone(); + info!("dalle2_query: {:?}", query); + let theme = conf::ChatConfJson::theme(); + let app = handle.clone(); - let query = if query.is_some() { - format!( - "window.addEventListener('DOMContentLoaded', function() {{\nwindow.__CHATGPT_QUERY__='{}';\n}})", - query.unwrap() - ) - } else { - "".to_string() - }; + let query = if query.is_some() { + format!("window.addEventListener('DOMContentLoaded', function() {{\nwindow.__CHATGPT_QUERY__='{}';\n}})", query.unwrap()) + } else { + "".to_string() + }; - let label = if is_new.unwrap_or(true) { - let timestamp = SystemTime::now() - .duration_since(SystemTime::UNIX_EPOCH) - .unwrap() - .as_secs(); - format!("dalle2_{}", timestamp) - } else { - "dalle2".to_string() - }; + let label = if is_new.unwrap_or(true) { + let timestamp = SystemTime::now() + .duration_since(SystemTime::UNIX_EPOCH) + .unwrap() + .as_secs(); + format!("dalle2_{}", timestamp) + } else { + "dalle2".to_string() + }; - if app.get_window("dalle2").is_none() { - tauri::async_runtime::spawn(async move { - WindowBuilder::new( - &app, - label, - WindowUrl::App("https://labs.openai.com".into()), - ) - .title(title.unwrap_or_else(|| "DALL·E 2".to_string())) - .resizable(true) - .fullscreen(false) - .inner_size(800.0, 600.0) - .always_on_top(false) - .theme(theme) - .initialization_script(include_str!("../scripts/core.js")) - .initialization_script(&query) - .initialization_script(include_str!("../scripts/dalle2.js")) - .build() - .unwrap(); - }); - } else { - let dalle2_win = app.get_window("dalle2").unwrap(); - dalle2_win.show().unwrap(); - dalle2_win.set_focus().unwrap(); - } -} - -pub fn control_window(handle: &tauri::AppHandle) { - let app = handle.clone(); + if app.get_window("dalle2").is_none() { tauri::async_runtime::spawn(async move { - if app.app_handle().get_window("main").is_none() { - WindowBuilder::new(&app, "main", WindowUrl::App("index.html".into())) - .title("Control Center") - .resizable(true) - .fullscreen(false) - .inner_size(800.0, 600.0) - .min_inner_size(800.0, 600.0) - .build() - .unwrap(); - } else { - let main_win = app.app_handle().get_window("main").unwrap(); - main_win.show().unwrap(); - main_win.set_focus().unwrap(); - } + WindowBuilder::new( + &app, + label, + WindowUrl::App("https://labs.openai.com".into()), + ) + .title(title.unwrap_or_else(|| "DALL·E 2".to_string())) + .resizable(true) + .fullscreen(false) + .inner_size(800.0, 600.0) + .always_on_top(false) + .theme(theme) + .initialization_script(include_str!("../scripts/core.js")) + .initialization_script(&query) + .initialization_script(include_str!("../scripts/dalle2.js")) + .build() + .unwrap(); }); + } else { + let dalle2_win = app.get_window("dalle2").unwrap(); + dalle2_win.show().unwrap(); + dalle2_win.set_focus().unwrap(); + } +} + +#[tauri::command] +pub fn dalle2_search_window(app: tauri::AppHandle, query: String) { + dalle2_window( + &app.app_handle(), + Some(query), + Some("ChatGPT & DALL·E 2".to_string()), + None, + ); +} + +#[tauri::command] +pub fn control_window(handle: tauri::AppHandle) { + tauri::async_runtime::spawn(async move { + if handle.get_window("main").is_none() { + WindowBuilder::new( + &handle, + "main", + WindowUrl::App("index.html?type=control".into()), + ) + .title("Control Center") + .resizable(true) + .fullscreen(false) + .inner_size(1200.0, 700.0) + .min_inner_size(1000.0, 600.0) + .build() + .unwrap(); + } else { + let main_win = handle.get_window("main").unwrap(); + main_win.show().unwrap(); + main_win.set_focus().unwrap(); + } + }); +} + +#[tauri::command] +pub async fn wa_window( + app: tauri::AppHandle, + label: String, + title: String, + url: String, + script: Option, +) { + info!("wa_window: {} :=> {}", title, url); + let win = app.get_window(&label); + if win.is_none() { + tauri::async_runtime::spawn(async move { + tauri::WindowBuilder::new(&app, label, tauri::WindowUrl::App(url.parse().unwrap())) + .initialization_script(&script.unwrap_or_default()) + .initialization_script(include_str!("../scripts/core.js")) + .title(title) + .build() + .unwrap(); + }); + } else { + if !win.clone().unwrap().is_visible().unwrap() { + win.clone().unwrap().show().unwrap(); + } + win + .clone() + .unwrap() + .eval("window.location.reload()") + .unwrap(); + win.unwrap().set_focus().unwrap(); + } +} + +#[tauri::command] +pub fn window_reload(app: tauri::AppHandle, label: &str) { + app + .app_handle() + .get_window(label) + .unwrap() + .eval("window.location.reload()") + .unwrap(); } diff --git a/src-tauri/src/conf.rs b/src-tauri/src/conf.rs index 46d5307..e6832be 100644 --- a/src-tauri/src/conf.rs +++ b/src-tauri/src/conf.rs @@ -16,185 +16,185 @@ pub const UPDATE_LOG_URL: &str = "https://github.com/lencx/ChatGPT/blob/main/UPD pub const AWESOME_URL: &str = "https://github.com/lencx/ChatGPT/blob/main/AWESOME.md"; pub const BUY_COFFEE: &str = "https://www.buymeacoffee.com/lencx"; pub const GITHUB_PROMPTS_CSV_URL: &str = - "https://raw.githubusercontent.com/f/awesome-chatgpt-prompts/main/prompts.csv"; + "https://raw.githubusercontent.com/f/awesome-chatgpt-prompts/main/prompts.csv"; pub const DEFAULT_CHAT_CONF: &str = r#"{ - "stay_on_top": false, - "auto_update": "Prompt", - "theme": "Light", - "tray": true, - "titlebar": true, - "popup_search": false, - "global_shortcut": "", - "hide_dock_icon": false, - "default_origin": "https://chat.openai.com", - "origin": "https://chat.openai.com", - "ua_window": "", - "ua_tray": "" + "stay_on_top": false, + "auto_update": "Prompt", + "theme": "Light", + "tray": true, + "titlebar": true, + "popup_search": false, + "global_shortcut": "", + "hide_dock_icon": false, + "main_dashboard": false, + "tray_dashboard": false, + "main_origin": "https://chat.openai.com", + "tray_origin": "https://chat.openai.com", + "default_origin": "https://chat.openai.com", + "ua_window": "", + "ua_tray": "Mozilla/5.0 (iPhone; CPU iPhone OS 16_0 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.0 Mobile/15E148 Safari/604.1" }"#; pub const DEFAULT_CHAT_CONF_MAC: &str = r#"{ - "stay_on_top": false, - "auto_update": "Prompt", - "theme": "Light", - "tray": true, - "titlebar": false, - "popup_search": false, - "global_shortcut": "", - "hide_dock_icon": false, - "default_origin": "https://chat.openai.com", - "origin": "https://chat.openai.com", - "ua_window": "", - "ua_tray": "" + "stay_on_top": false, + "auto_update": "Prompt", + "theme": "Light", + "tray": true, + "titlebar": false, + "popup_search": false, + "global_shortcut": "", + "hide_dock_icon": false, + "main_dashboard": false, + "tray_dashboard": false, + "main_origin": "https://chat.openai.com", + "tray_origin": "https://chat.openai.com", + "default_origin": "https://chat.openai.com", + "ua_window": "", + "ua_tray": "Mozilla/5.0 (iPhone; CPU iPhone OS 16_0 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.0 Mobile/15E148 Safari/604.1" }"#; #[derive(serde::Serialize, serde::Deserialize, Debug, Clone)] pub struct ChatConfJson { - // support macOS only - pub titlebar: bool, - pub hide_dock_icon: bool, + // support macOS only + pub titlebar: bool, + pub hide_dock_icon: bool, - // macOS and Windows, Light/Dark/System - pub theme: String, - // auto update policy, Prompt/Silent/Disable - pub auto_update: String, - pub tray: bool, - pub popup_search: bool, - pub stay_on_top: bool, - pub default_origin: String, - pub origin: String, - pub ua_window: String, - pub ua_tray: String, - pub global_shortcut: Option, + // macOS and Windows, Light/Dark/System + pub theme: String, + // auto update policy, Prompt/Silent/Disable + pub auto_update: String, + pub tray: bool, + pub popup_search: bool, + pub stay_on_top: bool, + pub main_dashboard: bool, + pub tray_dashboard: bool, + pub main_origin: String, + pub tray_origin: String, + pub default_origin: String, + pub ua_window: String, + pub ua_tray: String, + pub global_shortcut: Option, } impl ChatConfJson { - /// init chat.conf.json - /// path: ~/.chatgpt/chat.conf.json - pub fn init() -> PathBuf { - info!("chat_conf_init"); - let conf_file = ChatConfJson::conf_path(); - let content = if cfg!(target_os = "macos") { - DEFAULT_CHAT_CONF_MAC - } else { - DEFAULT_CHAT_CONF - }; + /// init chat.conf.json + /// path: ~/.chatgpt/chat.conf.json + pub fn init() -> PathBuf { + info!("chat_conf_init"); + let conf_file = ChatConfJson::conf_path(); + let content = if cfg!(target_os = "macos") { + DEFAULT_CHAT_CONF_MAC + } else { + DEFAULT_CHAT_CONF + }; - if !exists(&conf_file) { - create_file(&conf_file).unwrap(); - fs::write(&conf_file, content).unwrap(); - return conf_file; + if !exists(&conf_file) { + create_file(&conf_file).unwrap(); + fs::write(&conf_file, content).unwrap(); + return conf_file; + } + + let conf_file = ChatConfJson::conf_path(); + let file_content = fs::read_to_string(&conf_file).unwrap(); + match serde_json::from_str(&file_content) { + Ok(v) => v, + Err(err) => { + if err.to_string() == "invalid type: map, expected unit at line 1 column 0" { + return conf_file; } + fs::write(&conf_file, content).unwrap(); + } + }; - let conf_file = ChatConfJson::conf_path(); - let file_content = fs::read_to_string(&conf_file).unwrap(); - match serde_json::from_str(&file_content) { - Ok(v) => v, - Err(err) => { - if err.to_string() == "invalid type: map, expected unit at line 1 column 0" { - return conf_file; - } - fs::write(&conf_file, content).unwrap(); - } - }; + conf_file + } - conf_file - } + pub fn conf_path() -> PathBuf { + chat_root().join("chat.conf.json") + } - pub fn conf_path() -> PathBuf { - chat_root().join("chat.conf.json") - } + pub fn get_chat_conf() -> Self { + let conf_file = ChatConfJson::conf_path(); + let file_content = fs::read_to_string(&conf_file).unwrap(); + let content = if cfg!(target_os = "macos") { + DEFAULT_CHAT_CONF_MAC + } else { + DEFAULT_CHAT_CONF + }; - pub fn get_chat_conf() -> Self { - let conf_file = ChatConfJson::conf_path(); - let file_content = fs::read_to_string(&conf_file).unwrap(); - let content = if cfg!(target_os = "macos") { - DEFAULT_CHAT_CONF_MAC - } else { - DEFAULT_CHAT_CONF - }; - - match serde_json::from_value(match serde_json::from_str(&file_content) { - Ok(v) => v, - Err(_) => { - fs::write(&conf_file, content).unwrap(); - serde_json::from_str(content).unwrap() - } - }) { - Ok(v) => v, - Err(_) => { - fs::write(&conf_file, content).unwrap(); - serde_json::from_value(serde_json::from_str(content).unwrap()).unwrap() - } - } - } - - pub fn reset_chat_conf() -> Self { - let conf_file = ChatConfJson::conf_path(); - let content = if cfg!(target_os = "macos") { - DEFAULT_CHAT_CONF_MAC - } else { - DEFAULT_CHAT_CONF - }; + match serde_json::from_value(match serde_json::from_str(&file_content) { + Ok(v) => v, + Err(_) => { fs::write(&conf_file, content).unwrap(); serde_json::from_str(content).unwrap() + } + }) { + Ok(v) => v, + Err(_) => { + fs::write(&conf_file, content).unwrap(); + serde_json::from_value(serde_json::from_str(content).unwrap()).unwrap() + } + } + } + + pub fn reset_chat_conf() -> Self { + let conf_file = ChatConfJson::conf_path(); + let content = if cfg!(target_os = "macos") { + DEFAULT_CHAT_CONF_MAC + } else { + DEFAULT_CHAT_CONF + }; + fs::write(&conf_file, content).unwrap(); + serde_json::from_str(content).unwrap() + } + + // https://users.rust-lang.org/t/updating-object-fields-given-dynamic-json/39049/3 + pub fn amend(new_rules: &Value, app: Option) -> Result<()> { + let config = ChatConfJson::get_chat_conf(); + let config: Value = serde_json::to_value(&config)?; + let mut config: BTreeMap = serde_json::from_value(config)?; + let new_rules: BTreeMap = serde_json::from_value(new_rules.clone())?; + + for (k, v) in new_rules { + config.insert(k, v); } - // https://users.rust-lang.org/t/updating-object-fields-given-dynamic-json/39049/3 - pub fn amend(new_rules: &Value, app: Option) -> Result<()> { - let config = ChatConfJson::get_chat_conf(); - let config: Value = serde_json::to_value(&config)?; - let mut config: BTreeMap = serde_json::from_value(config)?; - let new_rules: BTreeMap = serde_json::from_value(new_rules.clone())?; + fs::write( + ChatConfJson::conf_path(), + serde_json::to_string_pretty(&config)?, + )?; - for (k, v) in new_rules { - config.insert(k, v); - } - - fs::write( - ChatConfJson::conf_path(), - serde_json::to_string_pretty(&config)?, - )?; - - if let Some(handle) = app { - tauri::api::process::restart(&handle.env()); - // tauri::api::dialog::ask( - // handle.get_window("core").as_ref(), - // "ChatGPT Restart", - // "Whether to restart immediately?", - // move |is_restart| { - // if is_restart { - // } - // }, - // ); - } - - Ok(()) + if let Some(handle) = app { + tauri::api::process::restart(&handle.env()); } - pub fn theme() -> Option { - let conf = ChatConfJson::get_chat_conf(); - let theme = match conf.theme.as_str() { - "System" => match dark_light::detect() { - // Dark mode - dark_light::Mode::Dark => Theme::Dark, - // Light mode - dark_light::Mode::Light => Theme::Light, - // Unspecified - dark_light::Mode::Default => Theme::Light, - }, - "Dark" => Theme::Dark, - _ => Theme::Light, - }; + Ok(()) + } - Some(theme) - } + pub fn theme() -> Option { + let conf = ChatConfJson::get_chat_conf(); + let theme = match conf.theme.as_str() { + "System" => match dark_light::detect() { + // Dark mode + dark_light::Mode::Dark => Theme::Dark, + // Light mode + dark_light::Mode::Light => Theme::Light, + // Unspecified + dark_light::Mode::Default => Theme::Light, + }, + "Dark" => Theme::Dark, + _ => Theme::Light, + }; - #[cfg(target_os = "macos")] - pub fn titlebar() -> TitleBarStyle { - let conf = ChatConfJson::get_chat_conf(); - if conf.titlebar { - TitleBarStyle::Transparent - } else { - TitleBarStyle::Overlay - } + Some(theme) + } + + #[cfg(target_os = "macos")] + pub fn titlebar() -> TitleBarStyle { + let conf = ChatConfJson::get_chat_conf(); + if conf.titlebar { + TitleBarStyle::Transparent + } else { + TitleBarStyle::Overlay } + } } diff --git a/src-tauri/src/main.rs b/src-tauri/src/main.rs index b41eee9..601d4dc 100644 --- a/src-tauri/src/main.rs +++ b/src-tauri/src/main.rs @@ -1,115 +1,118 @@ #![cfg_attr( - all(not(debug_assertions), target_os = "windows"), - windows_subsystem = "windows" + all(not(debug_assertions), target_os = "windows"), + windows_subsystem = "windows" )] mod app; mod conf; mod utils; -use app::{cmd, fs_extra, menu, setup}; +use app::{cmd, fs_extra, gpt, menu, setup, window}; use conf::ChatConfJson; use tauri::api::path; use tauri_plugin_autostart::MacosLauncher; use tauri_plugin_log::{ - fern::colors::{Color, ColoredLevelConfig}, - LogTarget, LoggerBuilder, + fern::colors::{Color, ColoredLevelConfig}, + LogTarget, }; #[tokio::main] async fn main() { - ChatConfJson::init(); - // If the file does not exist, creating the file will block menu synchronization - utils::create_chatgpt_prompts(); - let context = tauri::generate_context!(); - let colors = ColoredLevelConfig { - error: Color::Red, - warn: Color::Yellow, - debug: Color::Blue, - info: Color::BrightGreen, - trace: Color::Cyan, - }; + ChatConfJson::init(); + // If the file does not exist, creating the file will block menu synchronization + utils::create_chatgpt_prompts(); + let context = tauri::generate_context!(); + let colors = ColoredLevelConfig { + error: Color::Red, + warn: Color::Yellow, + debug: Color::Blue, + info: Color::BrightGreen, + trace: Color::Cyan, + }; - cmd::download_list("chat.download.json", "download", None, None); - cmd::download_list("chat.notes.json", "notes", None, None); + gpt::download_list("chat.download.json", "download", None, None); + gpt::download_list("chat.notes.json", "notes", None, None); - let chat_conf = ChatConfJson::get_chat_conf(); + let chat_conf = ChatConfJson::get_chat_conf(); - let mut builder = tauri::Builder::default() - // https://github.com/tauri-apps/tauri/pull/2736 - .plugin( - LoggerBuilder::new() - .level(log::LevelFilter::Debug) - .with_colors(colors) - .targets([ - // LogTarget::LogDir, - // LOG PATH: ~/.chatgpt/ChatGPT.log - LogTarget::Folder(path::home_dir().unwrap().join(".chatgpt")), - LogTarget::Stdout, - LogTarget::Webview, - ]) - .build(), - ) - .plugin(tauri_plugin_positioner::init()) - .plugin(tauri_plugin_autostart::init( - MacosLauncher::LaunchAgent, - None, - )) - .invoke_handler(tauri::generate_handler![ - cmd::drag_window, - cmd::fullscreen, - cmd::download, - cmd::save_file, - cmd::open_link, - cmd::get_chat_conf, - cmd::get_theme, - cmd::reset_chat_conf, - cmd::run_check_update, - cmd::form_cancel, - cmd::form_confirm, - cmd::form_msg, - cmd::open_file, - cmd::get_chat_model_cmd, - cmd::parse_prompt, - cmd::sync_prompts, - cmd::sync_user_prompts, - cmd::window_reload, - cmd::dalle2_window, - cmd::cmd_list, - cmd::download_list, - cmd::get_download_list, - fs_extra::metadata, + let mut builder = tauri::Builder::default() + // https://github.com/tauri-apps/tauri/pull/2736 + .plugin( + tauri_plugin_log::Builder::default() + .targets([ + // LogTarget::LogDir, + // LOG PATH: ~/.chatgpt/ChatGPT.log + LogTarget::Folder(path::home_dir().unwrap().join(".chatgpt")), + LogTarget::Stdout, + LogTarget::Webview, ]) - .setup(setup::init) - .menu(menu::init()); + .level(log::LevelFilter::Debug) + .with_colors(colors) + .build(), + ) + .plugin(tauri_plugin_positioner::init()) + .plugin(tauri_plugin_autostart::init( + MacosLauncher::LaunchAgent, + None, + )) + .invoke_handler(tauri::generate_handler![ + cmd::drag_window, + cmd::fullscreen, + cmd::download, + cmd::save_file, + cmd::open_link, + cmd::get_chat_conf, + cmd::get_theme, + cmd::reset_chat_conf, + cmd::run_check_update, + cmd::form_cancel, + cmd::form_confirm, + cmd::form_msg, + cmd::open_file, + cmd::get_data, + gpt::get_chat_model_cmd, + gpt::parse_prompt, + gpt::sync_prompts, + gpt::sync_user_prompts, + gpt::cmd_list, + gpt::download_list, + gpt::get_download_list, + window::wa_window, + window::control_window, + window::window_reload, + window::dalle2_search_window, + fs_extra::metadata, + ]) + .setup(setup::init) + .menu(menu::init()); - if chat_conf.tray { - builder = builder.system_tray(menu::tray_menu()); - } + if chat_conf.tray { + builder = builder.system_tray(menu::tray_menu()); + } - builder - .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() { - let win = event.window(); - if win.label() == "core" { - // TODO: https://github.com/tauri-apps/tauri/issues/3084 - // event.window().hide().unwrap(); - // https://github.com/tauri-apps/tao/pull/517 - #[cfg(target_os = "macos")] - event.window().minimize().unwrap(); + builder + .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() { + let win = event.window(); + if win.label() == "core" { + // TODO: https://github.com/tauri-apps/tauri/issues/3084 + // event.window().hide().unwrap(); + // https://github.com/tauri-apps/tao/pull/517 + #[cfg(target_os = "macos")] + event.window().minimize().unwrap(); - // fix: https://github.com/lencx/ChatGPT/issues/93 - #[cfg(not(target_os = "macos"))] - event.window().hide().unwrap(); - } else { - win.close().unwrap(); - } - api.prevent_close(); - } - }) - .run(context) - .expect("error while running ChatGPT application"); + // fix: https://github.com/lencx/ChatGPT/issues/93 + #[cfg(not(target_os = "macos"))] + event.window().hide().unwrap(); + } else { + win.close().unwrap(); + } + api.prevent_close(); + } + }) + .run(context) + .expect("error while running ChatGPT application"); } diff --git a/src-tauri/src/scripts/core.js b/src-tauri/src/scripts/core.js index 74a6f4a..dd930b1 100644 --- a/src-tauri/src/scripts/core.js +++ b/src-tauri/src/scripts/core.js @@ -64,12 +64,14 @@ async function init() { topDom.id = "chatgpt-app-window-top"; document.body.appendChild(topDom); - const nav = document.body.querySelector('nav'); - if (nav) { - const currentPaddingTop = parseInt(window.getComputedStyle(document.querySelector('nav'), null).getPropertyValue('padding-top').replace('px', ''), 10); - const navStyleDom = document.createElement("style"); - navStyleDom.innerHTML = `nav{padding-top:${currentPaddingTop + topDom.clientHeight}px !important}`; - document.head.appendChild(navStyleDom); + if (window.location.host === 'chat.openai.com') { + const nav = document.body.querySelector('nav'); + if (nav) { + const currentPaddingTop = parseInt(window.getComputedStyle(document.querySelector('nav'), null).getPropertyValue('padding-top').replace('px', ''), 10); + const navStyleDom = document.createElement("style"); + navStyleDom.innerHTML = `nav{padding-top:${currentPaddingTop + topDom.clientHeight}px !important}`; + document.head.appendChild(navStyleDom); + } } topDom.addEventListener("mousedown", () => invoke("drag_window")); diff --git a/src-tauri/src/scripts/popup.core.js b/src-tauri/src/scripts/popup.core.js index f45db80..3cbbe18 100644 --- a/src-tauri/src/scripts/popup.core.js +++ b/src-tauri/src/scripts/popup.core.js @@ -33,7 +33,7 @@ async function init() { document.body.addEventListener('mousedown', async (e) => { selectionMenu.style.display = 'none'; if (e.target.id === 'chagpt-selection-menu') { - await invoke('dalle2_window', { query: encodeURIComponent(window.__DALLE2_CONTENT__) }); + await invoke('dalle2_search_window', { query: encodeURIComponent(window.__DALLE2_CONTENT__) }); } else { delete window.__DALLE2_CONTENT__; } diff --git a/src-tauri/src/utils.rs b/src-tauri/src/utils.rs index a5625ec..8afe5bb 100644 --- a/src-tauri/src/utils.rs +++ b/src-tauri/src/utils.rs @@ -3,215 +3,213 @@ use log::info; use regex::Regex; use serde_json::Value; use std::{ - collections::HashMap, - fs::{self, File}, - path::{Path, PathBuf}, - process::Command, + collections::HashMap, + fs::{self, File}, + path::{Path, PathBuf}, + process::Command, }; use tauri::updater::UpdateResponse; use tauri::{utils::config::Config, AppHandle, Manager, Wry}; pub fn chat_root() -> PathBuf { - tauri::api::path::home_dir().unwrap().join(".chatgpt") + tauri::api::path::home_dir().unwrap().join(".chatgpt") } pub fn get_tauri_conf() -> Option { - let config_file = include_str!("../tauri.conf.json"); - let config: Config = - serde_json::from_str(config_file).expect("failed to parse tauri.conf.json"); - Some(config) + let config_file = include_str!("../tauri.conf.json"); + let config: Config = serde_json::from_str(config_file).expect("failed to parse tauri.conf.json"); + Some(config) } pub fn exists(path: &Path) -> bool { - Path::new(path).exists() + Path::new(path).exists() } pub fn create_file(path: &Path) -> Result { - if let Some(p) = path.parent() { - fs::create_dir_all(p)? - } - File::create(path).map_err(Into::into) + if let Some(p) = path.parent() { + fs::create_dir_all(p)? + } + File::create(path).map_err(Into::into) } pub fn create_chatgpt_prompts() { - let sync_file = chat_root().join("cache_model").join("chatgpt_prompts.json"); - if !exists(&sync_file) { - create_file(&sync_file).unwrap(); - fs::write(&sync_file, "[]").unwrap(); - } + let sync_file = chat_root().join("cache_model").join("chatgpt_prompts.json"); + if !exists(&sync_file) { + create_file(&sync_file).unwrap(); + fs::write(&sync_file, "[]").unwrap(); + } } pub fn script_path() -> PathBuf { - 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(); - } + 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(); + } - script_file + script_file } pub fn user_script() -> String { - let user_script_content = fs::read_to_string(script_path()).unwrap_or_else(|_| "".to_string()); - format!( - "window.addEventListener('DOMContentLoaded', function() {{\n{}\n}})", - user_script_content - ) + let user_script_content = fs::read_to_string(script_path()).unwrap_or_else(|_| "".to_string()); + format!( + "window.addEventListener('DOMContentLoaded', function() {{\n{}\n}})", + user_script_content + ) } pub fn open_file(path: PathBuf) { - info!("open_file: {}", path.to_string_lossy()); - #[cfg(target_os = "macos")] - Command::new("open").arg("-R").arg(path).spawn().unwrap(); + let pathname = convert_path(path.to_str().unwrap()); + info!("open_file: {}", pathname); + #[cfg(target_os = "macos")] + Command::new("open") + .arg("-R") + .arg(pathname) + .spawn() + .unwrap(); - #[cfg(target_os = "windows")] - Command::new("explorer") - .arg("/select,") - .arg(path) - .spawn() - .unwrap(); + #[cfg(target_os = "windows")] + Command::new("explorer.exe") + .arg("/select,") + .arg(pathname) + .spawn() + .unwrap(); - // https://askubuntu.com/a/31071 - #[cfg(target_os = "linux")] - Command::new("xdg-open").arg(path).spawn().unwrap(); + // https://askubuntu.com/a/31071 + #[cfg(target_os = "linux")] + Command::new("xdg-open").arg(pathname).spawn().unwrap(); +} + +pub fn convert_path(path_str: &str) -> String { + if cfg!(target_os = "windows") { + path_str.replace('/', "\\") + } else { + String::from(path_str) + } } pub fn clear_conf(app: &tauri::AppHandle) { - let root = chat_root(); - let app2 = app.clone(); - let msg = format!("Path: {}\nAre you sure to clear all ChatGPT configurations? Please backup in advance if necessary!", root.to_string_lossy()); - tauri::api::dialog::ask( - app.get_window("core").as_ref(), - "Clear Config", - msg, - move |is_ok| { - if is_ok { - fs::remove_dir_all(root).unwrap(); - tauri::api::process::restart(&app2.env()); - } - }, - ); + let root = chat_root(); + let msg = format!( + "Path: {}\n + Are you sure you want to clear all ChatGPT configurations? Performing this operation data can not be restored, please back up in advance.\n + Note: The application will exit automatically after the configuration cleanup!", + root.to_string_lossy() + ); + tauri::api::dialog::ask( + app.get_window("core").as_ref(), + "Clear Config", + msg, + move |is_ok| { + if is_ok { + fs::remove_dir_all(root).unwrap(); + std::process::exit(0); + } + }, + ); } pub fn merge(v: &Value, fields: &HashMap) -> Value { - match v { - Value::Object(m) => { - let mut m = m.clone(); - for (k, v) in fields { - m.insert(k.clone(), v.clone()); - } - Value::Object(m) - } - v => v.clone(), + match v { + Value::Object(m) => { + let mut m = m.clone(); + for (k, v) in fields { + m.insert(k.clone(), v.clone()); + } + Value::Object(m) } + v => v.clone(), + } } pub fn gen_cmd(name: String) -> String { - let re = Regex::new(r"[^a-zA-Z0-9]").unwrap(); - re.replace_all(&name, "_").to_lowercase() + let re = Regex::new(r"[^a-zA-Z0-9]").unwrap(); + re.replace_all(&name, "_").to_lowercase() } pub async fn get_data( - url: &str, - app: Option<&tauri::AppHandle>, + url: &str, + app: Option<&tauri::AppHandle>, ) -> Result, reqwest::Error> { - let res = reqwest::get(url).await?; - let is_ok = res.status() == 200; - let body = res.text().await?; + let res = reqwest::get(url).await?; + let is_ok = res.status() == 200; + let body = res.text().await?; - if is_ok { - Ok(Some(body)) - } else { - info!("chatgpt_http_error: {}", body); - if let Some(v) = app { - tauri::api::dialog::message(v.get_window("core").as_ref(), "ChatGPT HTTP", body); - } - Ok(None) + if is_ok { + Ok(Some(body)) + } else { + info!("chatgpt_http_error: {}", body); + if let Some(v) = app { + tauri::api::dialog::message(v.get_window("core").as_ref(), "ChatGPT HTTP", body); } + Ok(None) + } } pub fn run_check_update(app: AppHandle, silent: bool, has_msg: Option) { - info!("run_check_update: silent={} has_msg={:?}", silent, has_msg); - tauri::async_runtime::spawn(async move { - let result = app.updater().check().await; - let update_resp = result.unwrap(); - if update_resp.is_update_available() { - if silent { - tauri::async_runtime::spawn(async move { - silent_install(app, update_resp).await.unwrap(); - }); - } else { - tauri::async_runtime::spawn(async move { - prompt_for_install(app, update_resp).await.unwrap(); - }); - } - } else if let Some(v) = has_msg { - if v { - tauri::api::dialog::message( - app.app_handle().get_window("core").as_ref(), - "ChatGPT", - "Your ChatGPT is up to date", - ); - } - } - }); + info!("run_check_update: silent={} has_msg={:?}", silent, has_msg); + tauri::async_runtime::spawn(async move { + let result = app.updater().check().await; + let update_resp = result.unwrap(); + if update_resp.is_update_available() { + if silent { + tauri::async_runtime::spawn(async move { + silent_install(app, update_resp).await.unwrap(); + }); + } else { + tauri::async_runtime::spawn(async move { + prompt_for_install(app, update_resp).await.unwrap(); + }); + } + } else if let Some(v) = has_msg { + if v { + tauri::api::dialog::message( + app.app_handle().get_window("core").as_ref(), + "ChatGPT", + "Your ChatGPT is up to date", + ); + } + } + }); } // Copy private api in tauri/updater/mod.rs. TODO: refactor to public api // Prompt a dialog asking if the user want to install the new version // Maybe we should add an option to customize it in future versions. pub async fn prompt_for_install(app: AppHandle, update: UpdateResponse) -> Result<()> { - info!("prompt_for_install"); - let windows = app.windows(); - let parent_window = windows.values().next(); - let package_info = app.package_info().clone(); + info!("prompt_for_install"); + let windows = app.windows(); + let parent_window = windows.values().next(); + let package_info = app.package_info().clone(); - let body = update.body().unwrap(); - // todo(lemarier): We should review this and make sure we have - // something more conventional. - let should_install = tauri::api::dialog::blocking::ask( - parent_window, - format!(r#"A new version of {} is available! "#, package_info.name), - format!( - r#"{} {} is now available -- you have {}. + let body = update.body().unwrap(); + // todo(lemarier): We should review this and make sure we have + // something more conventional. + let should_install = tauri::api::dialog::blocking::ask( + parent_window, + format!(r#"A new version of {} is available! "#, package_info.name), + format!( + r#"{} {} is now available -- you have {}. Would you like to install it now? Release Notes: {}"#, - package_info.name, - update.latest_version(), - package_info.version, - body - ), - ); - - if should_install { - // Launch updater download process - // macOS we display the `Ready to restart dialog` asking to restart - // Windows is closing the current App and launch the downloaded MSI when ready (the process stop here) - // Linux we replace the AppImage by launching a new install, it start a new AppImage instance, so we're closing the previous. (the process stop here) - update.download_and_install().await?; - - // Ask user if we need to restart the application - let should_exit = tauri::api::dialog::blocking::ask( - parent_window, - "Ready to Restart", - "The installation was successful, do you want to restart the application now?", - ); - if should_exit { - app.restart(); - } - } - - Ok(()) -} - -pub async fn silent_install(app: AppHandle, update: UpdateResponse) -> Result<()> { - info!("silent_install"); - let windows = app.windows(); - let parent_window = windows.values().next(); + package_info.name, + update.latest_version(), + package_info.version, + body + ), + ); + if should_install { // Launch updater download process // macOS we display the `Ready to restart dialog` asking to restart // Windows is closing the current App and launch the downloaded MSI when ready (the process stop here) @@ -220,33 +218,58 @@ pub async fn silent_install(app: AppHandle, update: UpdateResponse) -> // Ask user if we need to restart the application let should_exit = tauri::api::dialog::blocking::ask( - parent_window, - "Ready to Restart", - "The silent installation was successful, do you want to restart the application now?", + parent_window, + "Ready to Restart", + "The installation was successful, do you want to restart the application now?", ); if should_exit { - app.restart(); + app.restart(); } + } - Ok(()) + Ok(()) +} + +pub async fn silent_install(app: AppHandle, update: UpdateResponse) -> Result<()> { + info!("silent_install"); + let windows = app.windows(); + let parent_window = windows.values().next(); + + // Launch updater download process + // macOS we display the `Ready to restart dialog` asking to restart + // Windows is closing the current App and launch the downloaded MSI when ready (the process stop here) + // Linux we replace the AppImage by launching a new install, it start a new AppImage instance, so we're closing the previous. (the process stop here) + update.download_and_install().await?; + + // Ask user if we need to restart the application + let should_exit = tauri::api::dialog::blocking::ask( + parent_window, + "Ready to Restart", + "The silent installation was successful, do you want to restart the application now?", + ); + if should_exit { + app.restart(); + } + + Ok(()) } pub fn is_hidden(entry: &walkdir::DirEntry) -> bool { - entry - .file_name() - .to_str() - .map(|s| s.starts_with('.')) - .unwrap_or(false) + entry + .file_name() + .to_str() + .map(|s| s.starts_with('.')) + .unwrap_or(false) } pub fn vec_to_hashmap( - vec: impl Iterator, - key: &str, - map: &mut HashMap, + vec: impl Iterator, + key: &str, + map: &mut HashMap, ) { - for v in vec { - if let Some(kval) = v.get(key).and_then(serde_json::Value::as_str) { - map.insert(kval.to_string(), v); - } + for v in vec { + if let Some(kval) = v.get(key).and_then(serde_json::Value::as_str) { + map.insert(kval.to_string(), v); } + } } diff --git a/src-tauri/tauri.conf.json b/src-tauri/tauri.conf.json index ccd136c..970c87f 100644 --- a/src-tauri/tauri.conf.json +++ b/src-tauri/tauri.conf.json @@ -7,7 +7,7 @@ }, "package": { "productName": "ChatGPT", - "version": "0.9.2" + "version": "0.10.0" }, "tauri": { "allowlist": { diff --git a/src/components/FilePath/index.tsx b/src/components/FilePath/index.tsx new file mode 100644 index 0000000..1b9cd09 --- /dev/null +++ b/src/components/FilePath/index.tsx @@ -0,0 +1,41 @@ +import { FC, useEffect, useState } from 'react'; +import clsx from 'clsx'; +import { path, shell } from '@tauri-apps/api'; + +import { chatRoot } from '@/utils'; + +interface FilePathProps { + paths?: string; + label?: string; + className?: string; + content?: string; + url?: string; +} + +const FilePath: FC = ({ className, label = 'PATH', paths = '', url, content }) => { + const [filePath, setPath] = useState(''); + + useEffect(() => { + if (!path && !url) return; + (async () => { + if (url) { + setPath(url); + return; + } + setPath(await path.join(await chatRoot(), ...paths.split('/').filter((i) => !!i))); + })(); + }, [url, paths]); + + return ( + + ); +}; + +export default FilePath; diff --git a/src/components/Markdown/Editor.tsx b/src/components/Markdown/Editor.tsx new file mode 100644 index 0000000..8b36319 --- /dev/null +++ b/src/components/Markdown/Editor.tsx @@ -0,0 +1,48 @@ +import { FC, useEffect, useState } from 'react'; +import Editor from '@monaco-editor/react'; +import { Panel, PanelGroup, PanelResizeHandle } from 'react-resizable-panels'; + +import Markdown from '@/components/Markdown'; +import './index.scss'; + +interface MarkdownEditorProps { + value?: string; + onChange?: (v: string) => void; + mode?: string; +} + +const MarkdownEditor: FC = ({ value = '', onChange, mode = 'split' }) => { + const [content, setContent] = useState(value); + + useEffect(() => { + setContent(value); + onChange && onChange(value); + }, [value]); + + const handleEdit = (e: any) => { + setContent(e); + onChange && onChange(e); + }; + + const isSplit = mode === 'split'; + + return ( +
+ + {['md', 'split'].includes(mode) && ( + + + + )} + {isSplit && } + {['doc', 'split'].includes(mode) && ( + + {content} + + )} + +
+ ); +}; + +export default MarkdownEditor; diff --git a/src/components/Markdown/index.scss b/src/components/Markdown/index.scss new file mode 100644 index 0000000..5c0bc57 --- /dev/null +++ b/src/components/Markdown/index.scss @@ -0,0 +1,33 @@ +.markdown-body { + height: 100%; + overflow: auto; + font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', 'Noto Sans', Helvetica, Arial, + sans-serif, 'Apple Color Emoji', 'Segoe UI Emoji'; + + &.edit-preview { + padding: 10px; + font-size: 14px; + } + + pre, + code { + font-family: monospace, monospace; + } +} + +.md-main { + height: calc(100vh - 130px); + overflow: hidden; +} + +.resize-handle { + width: 0.25rem; + transition: 250ms linear background-color; + background-color: #7f8082; + outline: none; + + &:hover, + &[data-resize-handle-active] { + background-color: #5194eb; + } +} diff --git a/src/components/Markdown/index.tsx b/src/components/Markdown/index.tsx new file mode 100644 index 0000000..785a8f5 --- /dev/null +++ b/src/components/Markdown/index.tsx @@ -0,0 +1,52 @@ +import { FC } from 'react'; +import clsx from 'clsx'; +import ReactMarkdown from 'react-markdown'; +import remarkGfm from 'remark-gfm'; +import rehypeRaw from 'rehype-raw'; +import SyntaxHighlighter from 'react-syntax-highlighter'; +import agate from 'react-syntax-highlighter/dist/esm/styles/hljs/agate'; +import 'github-markdown-css'; + +import './index.scss'; + +interface MarkdownProps { + children: string; + className?: string; +} + +const Markdown: FC = ({ children, className }) => { + return ( +
+
+ + ) : ( + + {children} + + ); + }, + }} + /> +
+
+ ); +}; + +export default Markdown; diff --git a/src/components/SwitchOrigin/index.tsx b/src/components/SwitchOrigin/index.tsx new file mode 100644 index 0000000..e9fd427 --- /dev/null +++ b/src/components/SwitchOrigin/index.tsx @@ -0,0 +1,95 @@ +import { FC } from 'react'; +import { Link } from 'react-router-dom'; +import { Form, Select, Tag, Tooltip, Switch } from 'antd'; +import { QuestionCircleOutlined } from '@ant-design/icons'; + +import useJson from '@/hooks/useJson'; +import { DISABLE_AUTO_COMPLETE, CHAT_AWESOME_JSON } from '@/utils'; +interface SwitchOriginProps { + name: string; +} + +const SwitchOrigin: FC = ({ name }) => { + const { json: list = [] } = useJson(CHAT_AWESOME_JSON); + const form = Form.useFormInstance(); + + const labelName = `(${name === 'main' ? 'Main' : 'SystemTray'})`; + const dashboardName = `${name}_dashboard`; + const originName = `${name}_origin`; + const isEnable = Form.useWatch(dashboardName, form); + + return ( + <> + + Dashboard {labelName}{' '} + +

+ Set Dashboard as the application default window. +

+

+ If this is enabled, the Switch Origin {labelName}{' '} + setting will be invalid. +

+

+ If you want to add a new URL to the dashboard, add it in the{' '} + Awesome menu and make sure it is enabled. +

+ + } + > + + + + } + name={dashboardName} + valuePropName="checked" + > + + + + Switch Origin {labelName}{' '} + +

+ Set a single URL as the application default window. +

+

+ If you need to set a new URL as the application loading window, please add the + URL in the Awesome menu and then select it. +

+ + } + > + + + + } + name={originName} + > + + + + ); +}; + +export default SwitchOrigin; diff --git a/src/components/Tags/index.tsx b/src/components/Tags/index.tsx index b88fa88..09218d9 100644 --- a/src/components/Tags/index.tsx +++ b/src/components/Tags/index.tsx @@ -8,9 +8,11 @@ import { DISABLE_AUTO_COMPLETE } from '@/utils'; interface TagsProps { value?: string[]; onChange?: (v: string[]) => void; + addTxt?: string; + max?: number; } -const Tags: FC = ({ value = [], onChange }) => { +const Tags: FC = ({ max = 99, value = [], onChange, addTxt = 'New Tag' }) => { const [tags, setTags] = useState(value); const [inputVisible, setInputVisible] = useState(false); const [inputValue, setInputValue] = useState(''); @@ -18,7 +20,7 @@ const Tags: FC = ({ value = [], onChange }) => { useEffect(() => { setTags(value); - }, [value]) + }, [value]); useEffect(() => { if (inputVisible) { @@ -86,13 +88,13 @@ const Tags: FC = ({ value = [], onChange }) => { {...DISABLE_AUTO_COMPLETE} /> )} - {!inputVisible && ( + {!inputVisible && tags.length < max && ( - New Tag + {addTxt} )} ); }; -export default Tags; \ No newline at end of file +export default Tags; diff --git a/src/hooks/useChatModel.ts b/src/hooks/useChatModel.ts index 980fe1c..837cce5 100644 --- a/src/hooks/useChatModel.ts +++ b/src/hooks/useChatModel.ts @@ -15,12 +15,12 @@ export default function useChatModel(key: string, file = CHAT_MODEL_JSON) { setModelJson(data); }); - const modelSet = async (data: Record[]|Record) => { + const modelSet = async (data: Record[] | Record) => { const oData = clone(modelJson); oData[key] = data; await writeJSON(file, oData); setModelJson(oData); - } + }; return { modelJson, modelSet, modelData: modelJson?.[key] || [] }; } @@ -40,15 +40,19 @@ export function useCacheModel(file = '') { await writeJSON(newFile ? newFile : file, data, { isRoot: true }); setModelCacheJson(data); await modelCacheCmd(); - } + }; const modelCacheCmd = async () => { // Generate the `chat.model.cmd.json` file and refresh the page for the slash command to take effect. const list = await invoke('cmd_list'); - await writeJSON(CHAT_MODEL_CMD_JSON, { name: 'ChatGPT CMD', last_updated: Date.now(), data: list }); + await writeJSON(CHAT_MODEL_CMD_JSON, { + name: 'ChatGPT CMD', + last_updated: Date.now(), + data: list, + }); await invoke('window_reload', { label: 'core' }); await invoke('window_reload', { label: 'tray' }); }; return { modelCacheJson, modelCacheSet, modelCacheCmd }; -} \ No newline at end of file +} diff --git a/src/hooks/useColumns.tsx b/src/hooks/useColumns.tsx index d5562ff..ab55762 100644 --- a/src/hooks/useColumns.tsx +++ b/src/hooks/useColumns.tsx @@ -5,7 +5,7 @@ import { DISABLE_AUTO_COMPLETE } from '@/utils'; export default function useColumns(columns: any[] = []) { const [opType, setOpType] = useState(''); - const [opRecord, setRecord] = useState | null>(null); + const [opRecord, setRecord] = useState | null>(null); const [opTime, setNow] = useState(null); const [opExtra, setExtra] = useState(null); @@ -58,26 +58,26 @@ export const EditRow: FC = ({ rowKey, row, actions }) => { setEdit(true); }; const handleChange = (e: React.ChangeEvent) => { - setVal(e.target.value) + setVal(e.target.value); }; const handleSave = () => { setEdit(false); row[rowKey] = val?.trim(); - actions?.setRecord(row, 'rowedit') + actions?.setRecord(row, 'rowedit'); }; - return isEdit - ? ( - - ) - : ( -
{val}
- ); + return isEdit ? ( + + ) : ( +
+ {val} +
+ ); }; diff --git a/src/hooks/useData.ts b/src/hooks/useData.ts index 86cf9fa..3a58498 100644 --- a/src/hooks/useData.ts +++ b/src/hooks/useData.ts @@ -8,7 +8,7 @@ export default function useData(oData: any[]) { useEffect(() => { opInit(oData); - }, []) + }, []); const opAdd = (val: any) => { const v = [val, ...opData]; @@ -18,19 +18,25 @@ export default function useData(oData: any[]) { const opInit = (val: any[] = []) => { if (!val || !Array.isArray(val)) return; - const nData = val.map(i => ({ [safeKey]: v4(), ...i })); + const nData = val.map((i) => ({ [safeKey]: v4(), ...i })); setData(nData); }; const opRemove = (id: string) => { - const nData = opData.filter(i => i[safeKey] !== id); + const nData = opData.filter((i) => i[safeKey] !== id); + setData(nData); + return nData; + }; + + const opRemoveItems = (ids: string[]) => { + const nData = opData.filter((i) => !ids.includes(i[safeKey])); setData(nData); return nData; }; const opReplace = (id: string, data: any) => { const nData = [...opData]; - const idx = opData.findIndex(v => v[safeKey] === id); + const idx = opData.findIndex((v) => v[safeKey] === id); nData[idx] = data; setData(nData); return nData; @@ -51,5 +57,14 @@ export default function useData(oData: any[]) { return nData; }; - return { opSafeKey: safeKey, opInit, opReplace, opAdd, opRemove, opData, opReplaceItems }; -} \ No newline at end of file + return { + opSafeKey: safeKey, + opInit, + opReplace, + opAdd, + opRemove, + opRemoveItems, + opData, + opReplaceItems, + }; +} diff --git a/src/hooks/useInit.ts b/src/hooks/useInit.ts index b5438ad..aa58b53 100644 --- a/src/hooks/useInit.ts +++ b/src/hooks/useInit.ts @@ -8,5 +8,5 @@ export default function useInit(callback: () => void) { callback(); isInit.current = false; } - }) -} \ No newline at end of file + }); +} diff --git a/src/hooks/useTable.tsx b/src/hooks/useTable.tsx index 4fc2c65..9469318 100644 --- a/src/hooks/useTable.tsx +++ b/src/hooks/useTable.tsx @@ -7,14 +7,17 @@ import { safeKey } from '@/hooks/useData'; type rowSelectionOptions = { key: 'id' | string; rowType: 'id' | 'row' | 'all'; -} +}; export function useTableRowSelection(options: Partial = {}) { const { key = 'id', rowType = 'id' } = options; const [selectedRowKeys, setSelectedRowKeys] = useState([]); const [selectedRowIDs, setSelectedRowIDs] = useState([]); - const [selectedRows, setSelectedRows] = useState[]>([]); + const [selectedRows, setSelectedRows] = useState[]>([]); - const onSelectChange = (newSelectedRowKeys: React.Key[], newSelectedRows: Record[]) => { + const onSelectChange = ( + newSelectedRowKeys: React.Key[], + newSelectedRows: Record[], + ) => { const keys = newSelectedRows.map((i: any) => i[safeKey] || i[key]); setSelectedRowKeys(newSelectedRowKeys); if (rowType === 'id') { @@ -38,11 +41,7 @@ export function useTableRowSelection(options: Partial = {}) const rowSelection: TableRowSelection> = { selectedRowKeys, onChange: onSelectChange, - selections: [ - Table.SELECTION_ALL, - Table.SELECTION_INVERT, - Table.SELECTION_NONE, - ], + selections: [Table.SELECTION_ALL, Table.SELECTION_INVERT, Table.SELECTION_NONE], }; return { rowSelection, selectedRowIDs, selectedRows, rowReset }; @@ -55,4 +54,4 @@ export const TABLE_PAGINATION = { defaultPageSize: 10, pageSizeOptions: [5, 10, 15, 20], showTotal: (total: number) => Total {total} items, -}; \ No newline at end of file +}; diff --git a/src/icons/SplitIcon.tsx b/src/icons/SplitIcon.tsx new file mode 100644 index 0000000..e53bf9a --- /dev/null +++ b/src/icons/SplitIcon.tsx @@ -0,0 +1,25 @@ +import Icon from '@ant-design/icons'; +import type { CustomIconComponentProps } from '@ant-design/icons/lib/components/Icon'; + +interface IconProps { + onClick: () => void; +} + +export default function SplitIcon(props: Partial) { + return ( + ( + + + + )} + /> + ); +} diff --git a/src/layout/index.scss b/src/layout/index.scss index 07d5cf5..df8c069 100644 --- a/src/layout/index.scss +++ b/src/layout/index.scss @@ -23,6 +23,10 @@ -webkit-user-select: none; } +.ant-layout-sider-children { + overflow-y: auto; +} + .chat-container { padding: 20px; overflow: hidden; @@ -36,4 +40,4 @@ .ant-layout-footer { color: #666 !important; opacity: 0.7; -} \ No newline at end of file +} diff --git a/src/layout/index.tsx b/src/layout/index.tsx index d5397f9..f1a331d 100644 --- a/src/layout/index.tsx +++ b/src/layout/index.tsx @@ -1,4 +1,4 @@ -import { useState } from 'react'; +import { useEffect, useState } from 'react'; import { Layout, Menu, Tooltip, ConfigProvider, theme, Tag } from 'antd'; import { SyncOutlined } from '@ant-design/icons'; import { useNavigate, useLocation } from 'react-router-dom'; @@ -13,77 +13,106 @@ const { Content, Footer, Sider } = Layout; export default function ChatLayout() { const [collapsed, setCollapsed] = useState(false); + const [isDashboard, setDashboard] = useState(null); const [appInfo, setAppInfo] = useState>({}); const location = useLocation(); + const [menuKey, setMenuKey] = useState(location.pathname); const go = useNavigate(); + useEffect(() => { + if (location.search === '?type=control') { + go('/settings'); + } + if (location.search === '?type=preview') { + go('/?type=preview'); + } + setMenuKey(location.pathname); + setDashboard(location.pathname === '/'); + }, [location.pathname]); + useInit(async () => { setAppInfo({ appName: await getName(), appVersion: await getVersion(), - appTheme: await invoke("get_theme"), + appTheme: await invoke('get_theme'), }); - }) + }); const checkAppUpdate = async () => { await invoke('run_check_update', { silent: false, hasMsg: true }); - } + }; - const isDark = appInfo.appTheme === "dark"; + const isDark = appInfo.appTheme === 'dark'; + + if (isDashboard === null) return null; return ( - - setCollapsed(value)} - style={{ - overflow: 'auto', - height: '100vh', - position: 'fixed', - left: 0, - top: 0, - bottom: 0, - zIndex: 999, - }} - > -
-
- {appInfo.appName} - - {appInfo.appVersion} - - - - -
- - go(i.key)} - /> - - - + ) : ( + + setCollapsed(value)} style={{ - overflow: 'inherit' + overflow: 'auto', + height: '100vh', + position: 'fixed', + left: 0, + top: 0, + bottom: 0, + zIndex: 999, }} > - - - +
+ +
+
+ {appInfo.appName} + + {appInfo.appVersion} + + + + + + +
+ + go(i.key)} + /> + + + + + + + - + )} ); -}; \ No newline at end of file +} diff --git a/src/main.scss b/src/main.scss index a2eabaa..c7c6441 100644 --- a/src/main.scss +++ b/src/main.scss @@ -14,11 +14,15 @@ -webkit-text-size-adjust: 100%; } -html, body { +html, +body, +#root { padding: 0; margin: 0; + height: 100%; } +.ant-table-selection-col, .ant-table-selection-column { width: 50px !important; min-width: 50px !important; @@ -56,6 +60,7 @@ html, body { margin-bottom: 5px; } +.chat-tags, .chat-prompts-tags { .ant-tag { margin: 2px; @@ -68,12 +73,11 @@ html, body { } } -.chat-file-path, -.chat-sync-path { +.chat-file-path { font-size: 12px; font-weight: 500; color: #888; - margin-bottom: 5px; + margin-bottom: 3px; line-height: 16px; > div { @@ -81,7 +85,6 @@ html, body { overflow: hidden; text-overflow: ellipsis; white-space: nowrap; - // color: #2a2a2a; } span { @@ -95,4 +98,14 @@ html, body { cursor: pointer; text-decoration: underline; } -} \ No newline at end of file +} + +.chatico { + cursor: pointer; +} + +.awesome-tips { + .ant-tag { + cursor: pointer; + } +} diff --git a/src/main.tsx b/src/main.tsx index 488f0cd..3dad92c 100644 --- a/src/main.tsx +++ b/src/main.tsx @@ -9,8 +9,8 @@ ReactDOM.createRoot(document.getElementById('root') as HTMLElement).render( - + - + , ); diff --git a/src/routes.tsx b/src/routes.tsx index 41527cb..66910fb 100644 --- a/src/routes.tsx +++ b/src/routes.tsx @@ -7,20 +7,26 @@ import { UserOutlined, DownloadOutlined, FormOutlined, + GlobalOutlined, + InfoCircleOutlined, } from '@ant-design/icons'; import type { MenuProps } from 'antd'; -import General from '@view/General'; +import Settings from '@/view/settings'; +import About from '@/view/about'; +import Awesome from '@/view/awesome'; import UserCustom from '@/view/model/UserCustom'; import SyncPrompts from '@/view/model/SyncPrompts'; import SyncCustom from '@/view/model/SyncCustom'; import SyncRecord from '@/view/model/SyncRecord'; import Download from '@/view/download'; import Notes from '@/view/notes'; +import Markdown from '@/view/markdown'; +import Dashboard from '@/view/dashboard'; export type ChatRouteMetaObject = { label: string; - icon?: React.ReactNode, + icon?: React.ReactNode; }; type ChatRouteObject = { @@ -29,17 +35,25 @@ type ChatRouteObject = { hideMenu?: boolean; meta?: ChatRouteMetaObject; children?: ChatRouteObject[]; -} +}; export const routes: Array = [ { - path: '/', - element: , + path: '/settings', + element: , meta: { - label: 'General', + label: 'Settings', icon: , }, }, + { + path: '/awesome', + element: , + meta: { + label: 'Awesome', + icon: , + }, + }, { path: '/notes', element: , @@ -48,6 +62,11 @@ export const routes: Array = [ icon: , }, }, + { + path: '/md/:id', + element: , + hideMenu: true, + }, { path: '/model', meta: { @@ -88,26 +107,39 @@ export const routes: Array = [ ], }, { - path: 'download', + path: '/download', element: , meta: { label: 'Download', icon: , }, }, + { + path: '/about', + element: , + meta: { + label: 'About', + icon: , + }, + }, + { + path: '/', + element: , + hideMenu: true, + }, ]; type MenuItem = Required['items'][number]; export const menuItems: MenuItem[] = routes .filter((j) => !j.hideMenu) - .map(i => ({ + .map((i) => ({ ...i.meta, key: i.path || '', children: i?.children ?.filter((j) => !j.hideMenu) - ?.map((j) => ({ ...j.meta, key: `${i.path}/${j.path}` || ''})), + ?.map((j) => ({ ...j.meta, key: `${i.path}/${j.path}` || '' })), })); export default () => { return useRoutes(routes); -}; \ No newline at end of file +}; diff --git a/src/utils.ts b/src/utils.ts index b90a7bd..7582c68 100644 --- a/src/utils.ts +++ b/src/utils.ts @@ -2,31 +2,36 @@ import { readTextFile, writeTextFile, exists, createDir } from '@tauri-apps/api/ import { homeDir, join, dirname } from '@tauri-apps/api/path'; import dayjs from 'dayjs'; +export const CHAT_CONF_JSON = 'chat.conf.json'; export const CHAT_MODEL_JSON = 'chat.model.json'; export const CHAT_MODEL_CMD_JSON = 'chat.model.cmd.json'; export const CHAT_DOWNLOAD_JSON = 'chat.download.json'; +export const CHAT_AWESOME_JSON = 'chat.awesome.json'; export const CHAT_NOTES_JSON = 'chat.notes.json'; export const CHAT_PROMPTS_CSV = 'chat.prompts.csv'; -export const GITHUB_PROMPTS_CSV_URL = 'https://raw.githubusercontent.com/f/awesome-chatgpt-prompts/main/prompts.csv'; +export const GITHUB_PROMPTS_CSV_URL = + 'https://raw.githubusercontent.com/f/awesome-chatgpt-prompts/main/prompts.csv'; +export const GITHUB_LOG_URL = 'https://raw.githubusercontent.com/lencx/ChatGPT/main/UPDATE_LOG.md'; + export const DISABLE_AUTO_COMPLETE = { autoCapitalize: 'off', autoComplete: 'off', - spellCheck: false + spellCheck: false, }; export const chatRoot = async () => { - return join(await homeDir(), '.chatgpt') -} + return join(await homeDir(), '.chatgpt'); +}; export const chatModelPath = async (): Promise => { return join(await chatRoot(), CHAT_MODEL_JSON); -} +}; export const chatPromptsPath = async (): Promise => { return join(await chatRoot(), CHAT_PROMPTS_CSV); -} +}; -type readJSONOpts = { defaultVal?: Record, isRoot?: boolean, isList?: boolean }; +type readJSONOpts = { defaultVal?: Record; isRoot?: boolean; isList?: boolean }; export const readJSON = async (path: string, opts: readJSONOpts = {}) => { const { defaultVal = {}, isRoot = false, isList = false } = opts; const root = await chatRoot(); @@ -36,26 +41,39 @@ export const readJSON = async (path: string, opts: readJSONOpts = {}) => { file = await join(root, path); } - if (!await exists(file)) { - if (await dirname(file) !== root) { + if (!(await exists(file))) { + if ((await dirname(file)) !== root) { await createDir(await dirname(file), { recursive: true }); } - await writeTextFile(file, isList ? '[]' : JSON.stringify({ - name: 'ChatGPT', - link: 'https://github.com/lencx/ChatGPT', - ...defaultVal, - }, null, 2)) + await writeTextFile( + file, + isList + ? '[]' + : JSON.stringify( + { + name: 'ChatGPT', + link: 'https://github.com/lencx/ChatGPT', + ...defaultVal, + }, + null, + 2, + ), + ); } try { return JSON.parse(await readTextFile(file)); - } catch(e) { + } catch (e) { return {}; } -} +}; -type writeJSONOpts = { dir?: string, isRoot?: boolean }; -export const writeJSON = async (path: string, data: Record, opts: writeJSONOpts = {}) => { +type writeJSONOpts = { dir?: string; isRoot?: boolean }; +export const writeJSON = async ( + path: string, + data: Record, + opts: writeJSONOpts = {}, +) => { const { isRoot = false } = opts; const root = await chatRoot(); let file = path; @@ -64,13 +82,17 @@ export const writeJSON = async (path: string, data: Record, opts: w file = await join(root, path); } - if (isRoot && !await exists(await dirname(file))) { + if (isRoot && !(await exists(await dirname(file)))) { await createDir(await dirname(file), { recursive: true }); } await writeTextFile(file, JSON.stringify(data, null, 2)); -} +}; export const fmtDate = (date: any) => dayjs(date).format('YYYY-MM-DD HH:mm:ss'); -export const genCmd = (act: string) => act.replace(/\s+|\/+/g, '_').replace(/[^\d\w]/g, '').toLocaleLowerCase(); \ No newline at end of file +export const genCmd = (act: string) => + act + .replace(/\s+|\/+/g, '_') + .replace(/[^\d\w]/g, '') + .toLocaleLowerCase(); diff --git a/src/view/General.tsx b/src/view/General.tsx deleted file mode 100644 index 6dfbdab..0000000 --- a/src/view/General.tsx +++ /dev/null @@ -1,182 +0,0 @@ -import { useEffect, useState } from 'react'; -import { Form, Radio, Switch, Input, Button, Space, message, Tooltip } from 'antd'; -import { QuestionCircleOutlined } from '@ant-design/icons'; -import { invoke, shell, path } from '@tauri-apps/api'; -import { platform } from '@tauri-apps/api/os'; -import { ask } from '@tauri-apps/api/dialog'; -import { relaunch } from '@tauri-apps/api/process'; -import { clone, omit, isEqual } from 'lodash'; - -import useInit from '@/hooks/useInit'; -import { DISABLE_AUTO_COMPLETE, chatRoot } from '@/utils'; - -const AutoUpdateLabel = () => { - return ( - - Auto Update -
Auto Update Policy
- Prompt: prompt to install
- Silent: install silently
- {/*Disable: disable auto update
*/} - - )}> - - ) -} - -const OriginLabel = ({ url }: { url: string }) => { - return ( - - Switch Origin - - ) -} - -const PopupSearchLabel = () => { - return ( - - Pop-up Search - {' '} - -
Generate images according to the content: Select the ChatGPT content with the mouse, no more than 400 characters. the DALL·E 2 button appears, and click to jump (Note: because the search content filled by the script cannot trigger the event directly, you need to enter a space in the input box to make the button clickable).
-
The application is built using Tauri, and due to its security restrictions, some of the action buttons will not work, so we recommend going to your browser.
- - )}> - - ) -} - -const GlobalShortcutLabel = () => { - return ( -
- Global Shortcut - {' '} - -
Shortcut definition, modifiers and key separated by "+" e.g. CmdOrControl+Q
-
If empty, the shortcut is disabled.
- https://tauri.app/v1/api/js/globalshortcut -
- )}> - - - - ) -} - -export default function General() { - const [form] = Form.useForm(); - const [jsonPath, setJsonPath] = useState(''); - const [platformInfo, setPlatform] = useState(''); - const [chatConf, setChatConf] = useState(null); - - useInit(async () => { - setJsonPath(await path.join(await chatRoot(), 'chat.conf.json')); - - setPlatform(await platform()); - const chatData = await invoke('get_chat_conf'); - setChatConf(chatData); - }); - - useEffect(() => { - form.setFieldsValue(clone(chatConf)); - }, [chatConf]) - - const onCancel = () => { - form.setFieldsValue(chatConf); - }; - - const onReset = async () => { - const chatData = await invoke('reset_chat_conf'); - setChatConf(chatData); - const isOk = await ask(`Configuration reset successfully, whether to restart?`, { - title: 'ChatGPT Preferences' - }); - if (isOk) { - relaunch(); - return; - } - message.success('Configuration reset successfully'); - }; - - const onFinish = async (values: any) => { - if (!isEqual(omit(chatConf, ['default_origin']), values)) { - await invoke('form_confirm', { data: values, label: 'main' }); - const isOk = await ask(`Configuration saved successfully, whether to restart?`, { - title: 'ChatGPT Preferences' - }); - if (isOk) { - relaunch(); - return; - } - message.success('Configuration saved successfully'); - } - }; - - return ( - <> - -
- - - - {platformInfo === 'darwin' && ( - - - - )} - } name="popup_search" valuePropName="checked"> - - - - - Light - Dark - {["darwin", "windows"].includes(platformInfo) && ( - System - )} - - - } name="auto_update"> - - Prompt - Silent - {/*Disable*/} - - - } name="global_shortcut"> - - - } name="origin"> - - - - - - - - - - - - - - - - -
- - ) -} \ No newline at end of file diff --git a/src/view/about/index.scss b/src/view/about/index.scss new file mode 100644 index 0000000..408924f --- /dev/null +++ b/src/view/about/index.scss @@ -0,0 +1,24 @@ +.about { + .log-tab { + font-size: 14px; + + h2 { + font-size: 16px; + } + } + + .about-tab { + .imgs { + img { + max-width: 200px; + margin-bottom: 20px; + } + } + } + + code { + background-color: rgba(200, 200, 200, 0.4); + padding: 2px 4px; + border-radius: 5px; + } +} diff --git a/src/view/about/index.tsx b/src/view/about/index.tsx new file mode 100644 index 0000000..9a82e62 --- /dev/null +++ b/src/view/about/index.tsx @@ -0,0 +1,88 @@ +import { useState } from 'react'; +import { invoke } from '@tauri-apps/api'; +import { Tabs, Tag } from 'antd'; + +import { GITHUB_LOG_URL } from '@/utils'; +import useInit from '@/hooks/useInit'; +import Markdown from '@/components/Markdown'; +import './index.scss'; + +export default function About() { + const [logContent, setLogContent] = useState(''); + + useInit(async () => { + const data = (await invoke('get_data', { url: GITHUB_LOG_URL })) || ''; + setLogContent(data as string); + }); + + return ( +
+ }, + { label: 'Update Log', key: 'log', children: }, + ]} + /> +
+ ); +} + +const AboutChatGPT = () => { + return ( +
+ ChatGPT Desktop Application (Mac, Windows and Linux) +

+ 🕒 History versions:{' '} + + lencx/ChatGPT/releases + +

+

+ It is just a wrapper for the + + {' '} + OpenAI ChatGPT{' '} + + website, no other data transfer exists (you can check the{' '} + + {' '} + source code{' '} + + ). The development and maintenance of this software has taken up a lot of my time. If it + helps you, you can buy me a cup of coffee (Chinese users can use WeChat to scan the code), + thanks! +

+

+ + Buy Me A Coffee + {' '} +
+ +

+
+ ); +}; + +const LogTab = ({ content }: { content: string }) => { + return ( +
+

+ Ref:{' '} + + lencx/ChatGPT/UPDATE_LOG.md + +

+ +
+ ); +}; diff --git a/src/view/awesome/Form.tsx b/src/view/awesome/Form.tsx new file mode 100644 index 0000000..26b972e --- /dev/null +++ b/src/view/awesome/Form.tsx @@ -0,0 +1,63 @@ +import { useEffect, ForwardRefRenderFunction, useImperativeHandle, forwardRef } from 'react'; +import { Form, Input, Switch } from 'antd'; +import type { FormProps } from 'antd'; + +import Tags from '@comps/Tags'; +import { DISABLE_AUTO_COMPLETE } from '@/utils'; + +interface AwesomeFormProps { + record?: Record | null; +} + +const initFormValue = { + title: '', + url: '', + enable: true, + tags: [], + category: '', +}; + +const AwesomeForm: ForwardRefRenderFunction = ({ record }, ref) => { + const [form] = Form.useForm(); + useImperativeHandle(ref, () => ({ form })); + + useEffect(() => { + if (record) { + form.setFieldsValue(record); + } + }, [record]); + + return ( +
+ + + + + + + + + + + + + + + +
+ ); +}; + +export default forwardRef(AwesomeForm); diff --git a/src/view/awesome/config.tsx b/src/view/awesome/config.tsx new file mode 100644 index 0000000..164c227 --- /dev/null +++ b/src/view/awesome/config.tsx @@ -0,0 +1,74 @@ +import { Tag, Space, Popconfirm, Switch } from 'antd'; +import { open } from '@tauri-apps/api/shell'; + +export const awesomeColumns = () => [ + { + title: 'Title', + dataIndex: 'title', + fixed: 'left', + key: 'title', + width: 160, + }, + { + title: 'URL', + dataIndex: 'url', + key: 'url', + width: 200, + render: (v: string) => open(v)}>{v}, + }, + // { + // title: 'Icon', + // dataIndex: 'icon', + // key: 'icon', + // width: 120, + // }, + { + title: 'Enable', + dataIndex: 'enable', + key: 'enable', + width: 80, + render: (v: boolean = true, row: Record, action: Record) => ( + action.setRecord({ ...row, enable: v }, 'enable')} /> + ), + }, + { + title: 'Category', + dataIndex: 'category', + key: 'category', + width: 120, + render: (v: string) => {v}, + }, + { + title: 'Tags', + dataIndex: 'tags', + key: 'tags', + width: 150, + render: (v: string[]) => ( + + {v?.map((i) => ( + {i} + ))} + + ), + }, + { + title: 'Action', + fixed: 'right', + width: 150, + render: (_: any, row: any, actions: any) => { + return ( + + actions.setRecord(row, 'edit')}>Edit + actions.setRecord(row, 'delete')} + okText="Yes" + cancelText="No" + > + Delete + + + ); + }, + }, +]; diff --git a/src/view/awesome/index.tsx b/src/view/awesome/index.tsx new file mode 100644 index 0000000..654cc1a --- /dev/null +++ b/src/view/awesome/index.tsx @@ -0,0 +1,197 @@ +import { useRef, useEffect, useState } from 'react'; +import { Link, useNavigate } from 'react-router-dom'; +import { Table, Modal, Popconfirm, Button, Tooltip, Tag, message } from 'antd'; +import { QuestionCircleOutlined } from '@ant-design/icons'; +import { invoke } from '@tauri-apps/api'; + +import useJson from '@/hooks/useJson'; +import useData from '@/hooks/useData'; +import useColumns from '@/hooks/useColumns'; +import FilePath from '@/components/FilePath'; +import { CHAT_AWESOME_JSON } from '@/utils'; +import { useTableRowSelection, TABLE_PAGINATION } from '@/hooks/useTable'; +import { awesomeColumns } from './config'; +import AwesomeForm from './Form'; + +export default function Awesome() { + const formRef = useRef(null); + const [isVisible, setVisible] = useState(false); + const { opData, opInit, opAdd, opReplace, opReplaceItems, opRemove, opRemoveItems, opSafeKey } = + useData([]); + const { columns, ...opInfo } = useColumns(awesomeColumns()); + const { rowSelection, selectedRowIDs, rowReset } = useTableRowSelection(); + const { json, updateJson } = useJson(CHAT_AWESOME_JSON); + const selectedItems = rowSelection.selectedRowKeys || []; + + useEffect(() => { + if (!json || json.length <= 0) return; + opInit(json); + }, [json?.length]); + + useEffect(() => { + if (!opInfo.opType) return; + if (['edit', 'new'].includes(opInfo.opType)) { + setVisible(true); + } + if (['delete'].includes(opInfo.opType)) { + const data = opRemove(opInfo?.opRecord?.[opSafeKey]); + updateJson(data); + opInfo.resetRecord(); + } + }, [opInfo.opType, formRef]); + + const hide = () => { + setVisible(false); + opInfo.resetRecord(); + }; + + useEffect(() => { + if (opInfo.opType === 'enable') { + const data = opReplace(opInfo?.opRecord?.[opSafeKey], opInfo?.opRecord); + updateJson(data); + } + }, [opInfo.opTime]); + + const handleDelete = () => { + const data = opRemoveItems(selectedRowIDs); + updateJson(data); + rowReset(); + message.success('All selected URLs have been deleted'); + }; + + const handleOk = () => { + formRef.current?.form?.validateFields().then(async (vals: Record) => { + let idx = opData.findIndex((i) => i.url === vals.url); + if (vals.url === opInfo?.opRecord?.url) { + idx = -1; + } + if (idx === -1) { + if (opInfo.opType === 'new') { + const data = opAdd(vals); + await updateJson(data); + opInit(data); + message.success('Data added successfully'); + } + if (opInfo.opType === 'edit') { + const data = opReplace(opInfo?.opRecord?.[opSafeKey], vals); + await updateJson(data); + message.success('Data updated successfully'); + } + hide(); + } else { + const data = opData[idx]; + message.error( +
+
+ + {data.title}: {data.url} + +
+
This URL already exists, please edit it and try again.
+
, + ); + } + }); + }; + + const handleEnable = (isEnable: boolean) => { + const data = opReplaceItems(selectedRowIDs, { enable: isEnable }); + updateJson(data); + }; + + const handlePreview = () => { + invoke('wa_window', { + label: 'awesome_preview', + url: 'index.html?type=preview', + title: 'Preview Dashboard', + }); + }; + + const modalTitle = `${{ new: 'Create', edit: 'Edit' }[opInfo.opType]} URL`; + + return ( +
+
+
+ + + +
+
+ {selectedItems.length > 0 && ( + <> + + + + + + Selected {selectedItems.length} items + + )} +
+
+ + + + + + + ); +} + +const PreviewTip = () => { + const go = useNavigate(); + const handleGo = (v: string) => { + go(`/settings?type=${v}`); + }; + + return ( + + Click the button to preview, and in + Settings + you can set a single URL or Dashboard as the default window for the app. +
+ handleGo('main_window')} color="blue"> + Main Window + + {'or '} + handleGo('tray_window')} color="blue"> + SystemTray Window + + + } + > + + + ); +}; diff --git a/src/view/dashboard/index.scss b/src/view/dashboard/index.scss new file mode 100644 index 0000000..96c8a7a --- /dev/null +++ b/src/view/dashboard/index.scss @@ -0,0 +1,73 @@ +.dashboard { + position: fixed; + width: calc(100% - 30px); + height: calc(100% - 30px); + overflow-y: auto; + padding: 15px; + + &-no-data { + height: 100%; + display: flex; + align-items: center; + justify-content: center; + text-align: center; + flex-direction: column; + font-weight: bold; + + .icon { + color: #989898; + font-size: 24px; + } + + .txt { + padding: 10px; + font-size: 12px; + line-height: 16px; + + a { + color: #1677ff; + cursor: pointer; + } + } + } + + &.dark { + background-color: #000; + } + + &.has-top-dom { + padding-top: 30px; + } + + &.preview { + padding-top: 15px; + } + + .group-item { + margin-bottom: 20px; + + .title { + font-weight: bold; + font-size: 18px; + margin-bottom: 10px; + } + + .item { + .ant-card-body { + padding: 10px; + text-align: center; + font-weight: 500; + font-size: 14px; + } + + span { + display: block; + height: 100%; + width: 100%; + overflow: hidden; + text-overflow: ellipsis; + white-space: nowrap; + } + } + } +} diff --git a/src/view/dashboard/index.tsx b/src/view/dashboard/index.tsx new file mode 100644 index 0000000..1b97d41 --- /dev/null +++ b/src/view/dashboard/index.tsx @@ -0,0 +1,104 @@ +import { useEffect, useState } from 'react'; +import clsx from 'clsx'; +import { useSearchParams } from 'react-router-dom'; +import { Row, Col, Card } from 'antd'; +import { InboxOutlined } from '@ant-design/icons'; +import { os, invoke } from '@tauri-apps/api'; + +import useInit from '@/hooks/useInit'; +import useJson from '@/hooks/useJson'; +import { CHAT_AWESOME_JSON, CHAT_CONF_JSON, readJSON } from '@/utils'; +import './index.scss'; + +export default function Dashboard() { + const [params] = useSearchParams(); + const { json } = useJson[]>(CHAT_AWESOME_JSON); + const [list, setList] = useState[]]>>(); + const [hasClass, setClass] = useState(false); + const [theme, setTheme] = useState(''); + + useInit(async () => { + const getOS = await os.platform(); + const conf = await readJSON(CHAT_CONF_JSON); + const appTheme = await invoke('get_theme'); + setTheme(appTheme as string); + setClass(!conf?.titlebar && getOS === 'darwin'); + }); + + useEffect(() => { + if (!json) return; + const categories = new Map(); + + json?.forEach((i) => { + if (!i.enable) return; + if (!categories.has(i.category)) { + categories.set(i.category, []); + } + categories.get(i?.category).push(i); + }); + setList(Array.from(categories)); + }, [json?.length]); + + const handleLink = async (item: Record) => { + await invoke('wa_window', { + label: btoa(item.url).replace(/[^a-zA-Z0-9]/g, ''), + title: item.title, + url: item.url, + }); + }; + + if (!list) return null; + if (list?.length === 0) { + return ( +
+
+ +
+ No data +
+
+ Go to invoke('control_window')}>{'Control Center -> Awesome'} to add + data and make sure they are enabled. +
+
+ ); + } + + return ( +
+
+ {list.map((i) => { + return ( +
+ + + {i[1].map((j, idx) => { + return ( +
+ handleLink(j)}> + {j?.title} + + + ); + })} + + + + ); + })} + + + ); +} diff --git a/src/view/download/config.tsx b/src/view/download/config.tsx index 274c208..4f42bb9 100644 --- a/src/view/download/config.tsx +++ b/src/view/download/config.tsx @@ -10,7 +10,7 @@ import { fmtDate, chatRoot } from '@/utils'; const colorMap: any = { pdf: 'blue', png: 'orange', -} +}; export const downloadColumns = () => [ { @@ -61,20 +61,22 @@ export const downloadColumns = () => [ Delete - ) - } - } + ); + }, + }, ]; const RenderPath = ({ row }: any) => { const [filePath, setFilePath] = useState(''); useInit(async () => { - setFilePath(await getPath(row)); - }) + setFilePath(await getPath(row)); + }); return shell.open(filePath)}>{filePath}; }; export const getPath = async (row: any) => { const isImg = ['png'].includes(row?.ext); - return await path.join(await chatRoot(), 'download', isImg ? 'img' : row.ext, row.id) + `.${row.ext}`; -} + return ( + (await path.join(await chatRoot(), 'download', isImg ? 'img' : row.ext, row.id)) + `.${row.ext}` + ); +}; diff --git a/src/view/download/index.tsx b/src/view/download/index.tsx index 0ee3541..5684993 100644 --- a/src/view/download/index.tsx +++ b/src/view/download/index.tsx @@ -1,11 +1,11 @@ import { useEffect, useState } from 'react'; import { Table, Modal, Popconfirm, Button, message } from 'antd'; -import { invoke, path, shell, fs } from '@tauri-apps/api'; +import { invoke, path, fs } from '@tauri-apps/api'; -import useInit from '@/hooks/useInit'; import useJson from '@/hooks/useJson'; import useData from '@/hooks/useData'; import useColumns from '@/hooks/useColumns'; +import FilePath from '@/components/FilePath'; import { useTableRowSelection, TABLE_PAGINATION } from '@/hooks/useTable'; import { chatRoot, CHAT_DOWNLOAD_JSON } from '@/utils'; import { downloadColumns } from './config'; @@ -19,7 +19,6 @@ function renderFile(buff: Uint8Array, type: string) { } export default function Download() { - const [downloadPath, setDownloadPath] = useState(''); const [source, setSource] = useState(''); const [isVisible, setVisible] = useState(false); const { opData, opInit, opReplace, opSafeKey } = useData([]); @@ -28,11 +27,6 @@ export default function Download() { const { json, refreshJson, updateJson } = useJson(CHAT_DOWNLOAD_JSON); const selectedItems = rowSelection.selectedRowKeys || []; - useInit(async () => { - const file = await path.join(await chatRoot(), CHAT_DOWNLOAD_JSON); - setDownloadPath(file); - }); - useEffect(() => { if (!json || json.length <= 0) return; opInit(json); @@ -43,7 +37,12 @@ export default function Download() { (async () => { const record = opInfo?.opRecord; const isImg = ['png'].includes(record?.ext); - const file = await path.join(await chatRoot(), 'download', isImg ? 'img' : record?.ext, `${record?.id}.${record?.ext}`); + const file = await path.join( + await chatRoot(), + 'download', + isImg ? 'img' : record?.ext, + `${record?.id}.${record?.ext}`, + ); if (opInfo.opType === 'preview') { const data = await fs.readBinaryFile(file); const sourceData = renderFile(data, record?.ext); @@ -61,8 +60,8 @@ export default function Download() { message.success('Name has been changed!'); } opInfo.resetRecord(); - })() - }, [opInfo.opType]) + })(); + }, [opInfo.opType]); const handleDelete = async () => { if (opData?.length === selectedRows.length) { @@ -75,10 +74,15 @@ export default function Download() { const rows = selectedRows.map(async (i) => { const isImg = ['png'].includes(i?.ext); - const file = await path.join(await chatRoot(), 'download', isImg ? 'img' : i?.ext, `${i?.id}.${i?.ext}`); + const file = await path.join( + await chatRoot(), + 'download', + isImg ? 'img' : i?.ext, + `${i?.id}.${i?.ext}`, + ); await fs.removeFile(file); return file; - }) + }); Promise.all(rows).then(async () => { await handleRefresh(); message.success('All files selected are cleared!'); @@ -111,18 +115,14 @@ export default function Download() { okText="Yes" cancelText="No" > - + Selected {selectedItems.length} items )} - +
- ) -} \ No newline at end of file + ); +} diff --git a/src/view/markdown/index.scss b/src/view/markdown/index.scss new file mode 100644 index 0000000..964eb74 --- /dev/null +++ b/src/view/markdown/index.scss @@ -0,0 +1,16 @@ +.md-task { + margin-bottom: 5px; + display: flex; + justify-content: space-between; + + .ant-breadcrumb-link { + padding: 3px 5px; + transition: all 300ms ease; + border-radius: 4px; + &:hover { + color: rgba(0, 0, 0, 0.88); + background-color: rgba(0, 0, 0, 0.06); + cursor: pointer; + } + } +} diff --git a/src/view/markdown/index.tsx b/src/view/markdown/index.tsx new file mode 100644 index 0000000..292c7e2 --- /dev/null +++ b/src/view/markdown/index.tsx @@ -0,0 +1,58 @@ +import { useState } from 'react'; +import { useLocation } from 'react-router-dom'; +import { Breadcrumb } from 'antd'; +import { ArrowLeftOutlined } from '@ant-design/icons'; +import MarkdownEditor from '@/components/Markdown/Editor'; +import { fs, shell } from '@tauri-apps/api'; + +import useInit from '@/hooks/useInit'; +import SplitIcon from '@/icons/SplitIcon'; +import { getPath } from '@/view/notes/config'; +import './index.scss'; + +const modeMap: any = { + 0: 'split', + 1: 'md', + 2: 'doc', +}; + +export default function Markdown() { + const [filePath, setFilePath] = useState(''); + const [source, setSource] = useState(''); + const [previewMode, setPreviewMode] = useState(0); + const location = useLocation(); + const state = location?.state; + + useInit(async () => { + const file = await getPath(state); + setFilePath(file); + setSource(await fs.readTextFile(file)); + }); + + const handleChange = async (v: string) => { + await fs.writeTextFile(filePath, v); + }; + + const handlePreview = () => { + let mode = previewMode + 1; + if (mode > 2) mode = 0; + setPreviewMode(mode); + }; + + return ( + <> +
+ + history.go(-1)}> + + + shell.open(filePath)}>{filePath} + +
+ +
+
+ + + ); +} diff --git a/src/view/model/SyncCustom/Form.tsx b/src/view/model/SyncCustom/Form.tsx index adbb4c0..9166643 100644 --- a/src/view/model/SyncCustom/Form.tsx +++ b/src/view/model/SyncCustom/Form.tsx @@ -1,4 +1,10 @@ -import { useEffect, useState, ForwardRefRenderFunction, useImperativeHandle, forwardRef } from 'react'; +import { + useEffect, + useState, + ForwardRefRenderFunction, + useImperativeHandle, + forwardRef, +} from 'react'; import { Form, Input, Select, Tooltip } from 'antd'; import { v4 } from 'uuid'; import type { FormProps } from 'antd'; @@ -7,7 +13,7 @@ import { DISABLE_AUTO_COMPLETE, chatRoot } from '@/utils'; import useInit from '@/hooks/useInit'; interface SyncFormProps { - record?: Record | null; + record?: Record | null; type: string; } @@ -54,10 +60,18 @@ const SyncForm: ForwardRefRenderFunction = ({ record, const jsonTip = ( {JSON.stringify([ - { cmd: '', act: '', prompt: '' }, - { cmd: '', act: '', prompt: '' }, - ], null, 2)}} + title={ + 
+          {JSON.stringify(
+            [
+              { cmd: '', act: '', prompt: '' },
+              { cmd: '', act: '', prompt: '' },
+            ],
+            null,
+            2,
+          )}
+
+ } > JSON @@ -65,10 +79,12 @@ const SyncForm: ForwardRefRenderFunction = ({ record, const csvTip = ( {`"cmd","act","prompt" + title={ + 
{`"cmd","act","prompt"
 "cmd","act","prompt"
 "cmd","act","prompt"
-"cmd","act","prompt"`}
} +"cmd","act","prompt"`} + } > CSV @@ -76,23 +92,19 @@ const SyncForm: ForwardRefRenderFunction = ({ record, return ( <> -
+ - + + rules={[{ required: true, message: 'Please enter the path!' }]} + > = ({ record, {...DISABLE_AUTO_COMPLETE} /> - + + +
-

The file supports only {csvTip} and {jsonTip} formats.

+

+ The file supports only {csvTip} and {jsonTip} formats. +

- ) -} + ); +}; export default forwardRef(SyncForm); diff --git a/src/view/model/SyncCustom/config.tsx b/src/view/model/SyncCustom/config.tsx index d491cfc..bb3721c 100644 --- a/src/view/model/SyncCustom/config.tsx +++ b/src/view/model/SyncCustom/config.tsx @@ -26,7 +26,7 @@ export const syncColumns = () => [ dataIndex: 'path', key: 'path', width: 180, - render: (_: string, row: any) => + render: (_: string, row: any) => , }, { title: 'Last updated', @@ -36,7 +36,7 @@ export const syncColumns = () => [ render: (v: number) => (
- { v ? fmtDate(v) : ''} + {v ? fmtDate(v) : ''}
), }, @@ -56,7 +56,11 @@ export const syncColumns = () => [ > Sync - {row.last_updated && View} + {row.last_updated && ( + + View + + )} actions.setRecord(row, 'edit')}>Edit [ Delete - ) - } - } + ); + }, + }, ]; const RenderPath = ({ row }: any) => { const [filePath, setFilePath] = useState(''); useInit(async () => { - setFilePath(await getPath(row)); - }) - return shell.open(filePath)}>{filePath} + setFilePath(await getPath(row)); + }); + return shell.open(filePath)}>{filePath}; }; export const getPath = async (row: any) => { if (!/^http/.test(row.protocol)) { - return await path.join(await chatRoot(), row.path) + `.${row.ext}`; + return (await path.join(await chatRoot(), row.path)) + `.${row.ext}`; } else { return `${row.protocol}://${row.path}.${row.ext}`; } -} \ No newline at end of file +}; diff --git a/src/view/model/SyncCustom/index.tsx b/src/view/model/SyncCustom/index.tsx index 936283c..8b46faa 100644 --- a/src/view/model/SyncCustom/index.tsx +++ b/src/view/model/SyncCustom/index.tsx @@ -3,14 +3,20 @@ import { Table, Modal, Button, message } from 'antd'; import { invoke, path, fs } from '@tauri-apps/api'; import useData from '@/hooks/useData'; -import useChatModel, { useCacheModel } from '@/hooks/useChatModel'; import useColumns from '@/hooks/useColumns'; import { TABLE_PAGINATION } from '@/hooks/useTable'; +import useChatModel, { useCacheModel } from '@/hooks/useChatModel'; import { CHAT_MODEL_JSON, chatRoot, readJSON, genCmd } from '@/utils'; import { syncColumns, getPath } from './config'; import SyncForm from './Form'; -const fmtData = (data: Record[] = []) => (Array.isArray(data) ? data : []).map((i) => ({ ...i, cmd: i.cmd ? i.cmd : genCmd(i.act), tags: ['user-sync'], enable: true })); +const fmtData = (data: Record[] = []) => + (Array.isArray(data) ? data : []).map((i) => ({ + ...i, + cmd: i.cmd ? i.cmd : genCmd(i.act), + tags: ['user-sync'], + enable: true, + })); export default function SyncCustom() { const [isVisible, setVisible] = useState(false); @@ -37,7 +43,10 @@ export default function SyncCustom() { handleSync(filename).then((isOk: boolean) => { opInfo.resetRecord(); if (!isOk) return; - const data = opReplace(opInfo?.opRecord?.[opSafeKey], { ...opInfo?.opRecord, last_updated: Date.now() }); + const data = opReplace(opInfo?.opRecord?.[opSafeKey], { + ...opInfo?.opRecord, + last_updated: Date.now(), + }); modelSet(data); opInfo.resetRecord(); }); @@ -48,9 +57,13 @@ export default function SyncCustom() { if (['delete'].includes(opInfo.opType)) { (async () => { try { - const file = await path.join(await chatRoot(), 'cache_model', `${opInfo?.opRecord?.id}.json`); + const file = await path.join( + await chatRoot(), + 'cache_model', + `${opInfo?.opRecord?.id}.json`, + ); await fs.removeFile(file); - } catch(e) {} + } catch (e) {} const data = opRemove(opInfo?.opRecord?.[opSafeKey]); modelSet(data); opInfo.resetRecord(); @@ -94,22 +107,25 @@ export default function SyncCustom() { }; const handleOk = () => { - formRef.current?.form?.validateFields() - .then((vals: Record) => { - let data = []; - switch (opInfo.opType) { - case 'new': data = opAdd(vals); break; - case 'edit': data = opReplace(opInfo?.opRecord?.[opSafeKey], vals); break; - default: break; - } + formRef.current?.form?.validateFields().then((vals: Record) => { + if (opInfo.opType === 'new') { + const data = opAdd(vals); modelSet(data); - hide(); - }) + message.success('Data added successfully'); + } + if (opInfo.opType === 'edit') { + const data = opReplace(opInfo?.opRecord?.[opSafeKey], vals); + modelSet(data); + message.success('Data updated successfully'); + } + hide(); + }); }; return (
- ) -} \ No newline at end of file + ); +} diff --git a/src/view/model/SyncPrompts/config.tsx b/src/view/model/SyncPrompts/config.tsx index b8b4ade..b6381de 100644 --- a/src/view/model/SyncPrompts/config.tsx +++ b/src/view/model/SyncPrompts/config.tsx @@ -41,8 +41,6 @@ export const syncColumns = () => [ dataIndex: 'prompt', key: 'prompt', // width: 300, - render: (v: string) => ( - {v} - ), + render: (v: string) => {v}, }, ]; diff --git a/src/view/model/SyncPrompts/index.scss b/src/view/model/SyncPrompts/index.scss index 4e3ba63..0c7fb60 100644 --- a/src/view/model/SyncPrompts/index.scss +++ b/src/view/model/SyncPrompts/index.scss @@ -1,4 +1,5 @@ -.chat-table-tip, .chat-table-btns { +.chat-table-tip, +.chat-table-btns { display: flex; justify-content: space-between; } diff --git a/src/view/model/SyncPrompts/index.tsx b/src/view/model/SyncPrompts/index.tsx index 7db4d2e..e983d37 100644 --- a/src/view/model/SyncPrompts/index.tsx +++ b/src/view/model/SyncPrompts/index.tsx @@ -1,10 +1,11 @@ import { useEffect, useState } from 'react'; import { Table, Button, Popconfirm } from 'antd'; -import { invoke, path, shell } from '@tauri-apps/api'; +import { invoke, path } from '@tauri-apps/api'; import useInit from '@/hooks/useInit'; import useData from '@/hooks/useData'; import useColumns from '@/hooks/useColumns'; +import FilePath from '@/components/FilePath'; import useChatModel, { useCacheModel } from '@/hooks/useChatModel'; import { useTableRowSelection, TABLE_PAGINATION } from '@/hooks/useTable'; import { fmtDate, chatRoot } from '@/utils'; @@ -51,7 +52,7 @@ export default function SyncPrompts() { }, [opInfo.opTime]); const handleEnable = (isEnable: boolean) => { - const data = opReplaceItems(selectedRowIDs, { enable: isEnable }) + const data = opReplaceItems(selectedRowIDs, { enable: isEnable }); modelCacheSet(data); }; @@ -71,7 +72,9 @@ export default function SyncPrompts() {
{selectedItems.length > 0 && ( <> - + Selected {selectedItems.length} items @@ -80,10 +83,14 @@ export default function SyncPrompts() {
- {lastUpdated && Last updated on {fmtDate(lastUpdated)}} + {lastUpdated && ( + + Last updated on {fmtDate(lastUpdated)} + + )}
{record.prompt}
}} + expandable={{ + expandedRowRender: (record) =>
{record.prompt}
, + }} /> - ) -} \ No newline at end of file + ); +} diff --git a/src/view/model/SyncRecord/config.tsx b/src/view/model/SyncRecord/config.tsx index 71321ca..66ffd23 100644 --- a/src/view/model/SyncRecord/config.tsx +++ b/src/view/model/SyncRecord/config.tsx @@ -25,7 +25,11 @@ export const syncColumns = () => [ key: 'tags', // width: 150, render: (v: string[]) => ( - {v?.map(i => {i})} + + {v?.map((i) => ( + {i} + ))} + ), }, { @@ -43,8 +47,6 @@ export const syncColumns = () => [ dataIndex: 'prompt', key: 'prompt', // width: 300, - render: (v: string) => ( - {v} - ), + render: (v: string) => {v}, }, ]; diff --git a/src/view/model/SyncRecord/index.tsx b/src/view/model/SyncRecord/index.tsx index 9f22759..d8b60ca 100644 --- a/src/view/model/SyncRecord/index.tsx +++ b/src/view/model/SyncRecord/index.tsx @@ -2,14 +2,15 @@ import { useEffect, useState } from 'react'; import { useLocation } from 'react-router-dom'; import { ArrowLeftOutlined } from '@ant-design/icons'; import { Table, Button } from 'antd'; -import { shell, path } from '@tauri-apps/api'; +import { path } from '@tauri-apps/api'; -import useColumns from '@/hooks/useColumns'; import useData from '@/hooks/useData'; +import useColumns from '@/hooks/useColumns'; +import FilePath from '@/components/FilePath'; import { useCacheModel } from '@/hooks/useChatModel'; import { useTableRowSelection, TABLE_PAGINATION } from '@/hooks/useTable'; -import { fmtDate, chatRoot } from '@/utils'; import { getPath } from '@/view/model/SyncCustom/config'; +import { fmtDate, chatRoot } from '@/utils'; import { syncColumns } from './config'; import useInit from '@/hooks/useInit'; @@ -29,7 +30,7 @@ export default function SyncRecord() { useInit(async () => { setFilePath(await getPath(state)); setJsonPath(await path.join(await chatRoot(), 'cache_model', `${state?.id}.json`)); - }) + }); useEffect(() => { if (modelCacheJson.length <= 0) return; @@ -44,7 +45,7 @@ export default function SyncRecord() { }, [opInfo.opTime]); const handleEnable = (isEnable: boolean) => { - const data = opReplaceItems(selectedRowIDs, { enable: isEnable }) + const data = opReplaceItems(selectedRowIDs, { enable: isEnable }); modelCacheSet(data); }; @@ -57,7 +58,9 @@ export default function SyncRecord() {
{selectedItems.length > 0 && ( <> - + Selected {selectedItems.length} items @@ -66,10 +69,14 @@ export default function SyncRecord() {
- {state?.last_updated && Last updated on {fmtDate(state?.last_updated)}} + {state?.last_updated && ( + + Last updated on {fmtDate(state?.last_updated)} + + )}
{record.prompt}
}} + expandable={{ + expandedRowRender: (record) =>
{record.prompt}
, + }} /> - ) -} \ No newline at end of file + ); +} diff --git a/src/view/model/UserCustom/Form.tsx b/src/view/model/UserCustom/Form.tsx index 92e7a5b..9d4e306 100644 --- a/src/view/model/UserCustom/Form.tsx +++ b/src/view/model/UserCustom/Form.tsx @@ -6,7 +6,7 @@ import Tags from '@comps/Tags'; import { DISABLE_AUTO_COMPLETE } from '@/utils'; interface UserCustomFormProps { - record?: Record | null; + record?: Record | null; } const initFormValue = { @@ -16,7 +16,10 @@ const initFormValue = { prompt: '', }; -const UserCustomForm: ForwardRefRenderFunction = ({ record }, ref) => { +const UserCustomForm: ForwardRefRenderFunction = ( + { record }, + ref, +) => { const [form] = Form.useForm(); useImperativeHandle(ref, () => ({ form })); @@ -27,24 +30,20 @@ const UserCustomForm: ForwardRefRenderFunction = }, [record]); return ( -
+ - + - + @@ -55,12 +54,12 @@ const UserCustomForm: ForwardRefRenderFunction = - + - ) -} + ); +}; export default forwardRef(UserCustomForm); diff --git a/src/view/model/UserCustom/config.tsx b/src/view/model/UserCustom/config.tsx index 0450043..162dc66 100644 --- a/src/view/model/UserCustom/config.tsx +++ b/src/view/model/UserCustom/config.tsx @@ -7,7 +7,7 @@ export const modelColumns = () => [ fixed: 'left', width: 120, key: 'cmd', - render: (v: string) => /{v} + render: (v: string) => /{v}, }, { title: 'Act', @@ -21,7 +21,11 @@ export const modelColumns = () => [ key: 'tags', width: 150, render: (v: string[]) => ( - {v?.map(i => {i})} + + {v?.map((i) => ( + {i} + ))} + ), }, { @@ -39,9 +43,7 @@ export const modelColumns = () => [ dataIndex: 'prompt', key: 'prompt', width: 300, - render: (v: string) => ( - {v} - ), + render: (v: string) => {v}, }, { title: 'Action', @@ -61,5 +63,5 @@ export const modelColumns = () => [ ), - } + }, ]; diff --git a/src/view/model/UserCustom/index.tsx b/src/view/model/UserCustom/index.tsx index 985fe41..591d35f 100644 --- a/src/view/model/UserCustom/index.tsx +++ b/src/view/model/UserCustom/index.tsx @@ -1,11 +1,12 @@ import { useState, useRef, useEffect } from 'react'; import { Table, Button, Modal, message } from 'antd'; -import { shell, path } from '@tauri-apps/api'; +import { path } from '@tauri-apps/api'; import useInit from '@/hooks/useInit'; import useData from '@/hooks/useData'; -import useChatModel, { useCacheModel } from '@/hooks/useChatModel'; import useColumns from '@/hooks/useColumns'; +import FilePath from '@/components/FilePath'; +import useChatModel, { useCacheModel } from '@/hooks/useChatModel'; import { useTableRowSelection, TABLE_PAGINATION } from '@/hooks/useTable'; import { chatRoot, fmtDate } from '@/utils'; import { modelColumns } from './config'; @@ -44,14 +45,6 @@ export default function LanguageModel() { } }, [opInfo.opType, formRef]); - useEffect(() => { - if (opInfo.opType === 'enable') { - const data = opReplace(opInfo?.opRecord?.[opSafeKey], opInfo?.opRecord); - modelCacheSet(data); - } - }, [opInfo.opTime]) - - useEffect(() => { if (opInfo.opType === 'enable') { const data = opReplace(opInfo?.opRecord?.[opSafeKey], opInfo?.opRecord); @@ -60,7 +53,7 @@ export default function LanguageModel() { }, [opInfo.opTime]); const handleEnable = (isEnable: boolean) => { - const data = opReplaceItems(selectedRowIDs, { enable: isEnable }) + const data = opReplaceItems(selectedRowIDs, { enable: isEnable }); modelCacheSet(data); }; @@ -70,50 +63,64 @@ export default function LanguageModel() { }; const handleOk = () => { - formRef.current?.form?.validateFields() - .then(async (vals: Record) => { - if (modelCacheJson.map((i: any) => i.cmd).includes(vals.cmd) && opInfo?.opRecord?.cmd !== vals.cmd) { - message.warning(`"cmd: /${vals.cmd}" already exists, please change the "${vals.cmd}" name and resubmit.`); - return; - } - let data = []; - switch (opInfo.opType) { - case 'new': data = opAdd(vals); break; - case 'edit': data = opReplace(opInfo?.opRecord?.[opSafeKey], vals); break; - default: break; - } - await modelCacheSet(data); - opInit(data); - modelSet({ - id: 'user_custom', - last_updated: Date.now(), - }); - hide(); - }) + formRef.current?.form?.validateFields().then(async (vals: Record) => { + if ( + modelCacheJson.map((i: any) => i.cmd).includes(vals.cmd) && + opInfo?.opRecord?.cmd !== vals.cmd + ) { + message.warning( + `"cmd: /${vals.cmd}" already exists, please change the "${vals.cmd}" name and resubmit.`, + ); + return; + } + let data = []; + switch (opInfo.opType) { + case 'new': + data = opAdd(vals); + break; + case 'edit': + data = opReplace(opInfo?.opRecord?.[opSafeKey], vals); + break; + default: + break; + } + await modelCacheSet(data); + opInit(data); + modelSet({ + id: 'user_custom', + last_updated: Date.now(), + }); + hide(); + }); }; - const modalTitle = `${({ new: 'Create', edit: 'Edit' })[opInfo.opType]} Model`; + const modalTitle = `${{ new: 'Create', edit: 'Edit' }[opInfo.opType]} Model`; return (
- +
{selectedItems.length > 0 && ( <> - + Selected {selectedItems.length} items )}
- {/*
PATH: {modelPath}
*/}
- - {lastUpdated && Last updated on {fmtDate(lastUpdated)}} + + {lastUpdated && ( + + Last updated on {fmtDate(lastUpdated)} + + )}
{record.prompt}
}} + expandable={{ + expandedRowRender: (record) =>
{record.prompt}
, + }} /> - ) -} \ No newline at end of file + ); +} diff --git a/src/view/notes/config.tsx b/src/view/notes/config.tsx index 0c9b9f9..53ed522 100644 --- a/src/view/notes/config.tsx +++ b/src/view/notes/config.tsx @@ -1,4 +1,5 @@ import { useState } from 'react'; +import { Link } from 'react-router-dom'; import { Space, Popconfirm } from 'antd'; import { path, shell } from '@tauri-apps/api'; @@ -40,7 +41,9 @@ export const notesColumns = () => [ return ( actions.setRecord(row, 'preview')}>Preview - actions.setRecord(row, 'edit')}>Edit + + Edit + actions.setRecord(row, 'delete')} @@ -50,20 +53,20 @@ export const notesColumns = () => [ Delete - ) - } - } + ); + }, + }, ]; const RenderPath = ({ row }: any) => { const [filePath, setFilePath] = useState(''); useInit(async () => { - setFilePath(await getPath(row)); - }) + setFilePath(await getPath(row)); + }); return shell.open(filePath)}>{filePath}; }; export const getPath = async (row: any) => { const isImg = ['png'].includes(row?.ext); - return await path.join(await chatRoot(), 'notes', row.id) + `.${row.ext}`; -} + return (await path.join(await chatRoot(), 'notes', row.id)) + `.${row.ext}`; +}; diff --git a/src/view/notes/index.tsx b/src/view/notes/index.tsx index 38f0ab0..9bbc626 100644 --- a/src/view/notes/index.tsx +++ b/src/view/notes/index.tsx @@ -1,20 +1,17 @@ import { useEffect, useState } from 'react'; import { Table, Modal, Popconfirm, Button, message } from 'antd'; -import { invoke, path, shell, fs } from '@tauri-apps/api'; -import ReactMarkdown from 'react-markdown'; -import { Prism as SyntaxHighlighter } from 'react-syntax-highlighter'; -import { a11yDark } from 'react-syntax-highlighter/dist/esm/styles/prism'; +import { invoke, path, fs } from '@tauri-apps/api'; -import useInit from '@/hooks/useInit'; import useJson from '@/hooks/useJson'; import useData from '@/hooks/useData'; import useColumns from '@/hooks/useColumns'; +import Markdown from '@/components/Markdown'; +import FilePath from '@/components/FilePath'; import { useTableRowSelection, TABLE_PAGINATION } from '@/hooks/useTable'; import { chatRoot, CHAT_NOTES_JSON } from '@/utils'; import { notesColumns } from './config'; export default function Notes() { - const [notesPath, setNotesPath] = useState(''); const [source, setSource] = useState(''); const [isVisible, setVisible] = useState(false); const { opData, opInit, opReplace, opSafeKey } = useData([]); @@ -23,11 +20,6 @@ export default function Notes() { const { json, refreshJson, updateJson } = useJson(CHAT_NOTES_JSON); const selectedItems = rowSelection.selectedRowKeys || []; - useInit(async () => { - const file = await path.join(await chatRoot(), CHAT_NOTES_JSON); - setNotesPath(file); - }); - useEffect(() => { if (!json || json.length <= 0) return; opInit(json); @@ -44,9 +36,6 @@ export default function Notes() { setVisible(true); return; } - if (opInfo.opType === 'edit') { - alert('TODO'); - } if (opInfo.opType === 'delete') { await fs.removeFile(file); await handleRefresh(); @@ -57,8 +46,8 @@ export default function Notes() { message.success('Name has been changed!'); } opInfo.resetRecord(); - })() - }, [opInfo.opType]) + })(); + }, [opInfo.opType]); const handleDelete = async () => { if (opData?.length === selectedRows.length) { @@ -73,7 +62,7 @@ export default function Notes() { const file = await path.join(await chatRoot(), 'notes', `${i?.id}.${i?.ext}`); await fs.removeFile(file); return file; - }) + }); Promise.all(rows).then(async () => { await handleRefresh(); message.success('All files selected are cleared!'); @@ -106,18 +95,14 @@ export default function Notes() { okText="Yes" cancelText="No" > - + Selected {selectedItems.length} items )} - +
- - ) : ( - - {children} - - ) - } - }} - /> + - ) -} \ No newline at end of file + ); +} diff --git a/src/view/settings/General.tsx b/src/view/settings/General.tsx new file mode 100644 index 0000000..6f03a5f --- /dev/null +++ b/src/view/settings/General.tsx @@ -0,0 +1,95 @@ +import { useState } from 'react'; +import { Form, Radio, Switch, Input, Tooltip } from 'antd'; +import { QuestionCircleOutlined } from '@ant-design/icons'; +import { platform } from '@tauri-apps/api/os'; + +import useInit from '@/hooks/useInit'; +import { DISABLE_AUTO_COMPLETE } from '@/utils'; + +export default function General() { + const [platformInfo, setPlatform] = useState(''); + + useInit(async () => { + setPlatform(await platform()); + }); + + return ( + <> + + + + {platformInfo === 'darwin' && ( + + + + )} + {platformInfo === 'darwin' && ( + + + + )} + + + Light + Dark + {['darwin', 'windows'].includes(platformInfo) && System} + + + } name="auto_update"> + + Prompt + Silent + {/*Disable*/} + + + } name="global_shortcut"> + + + + ); +} + +const AutoUpdateLabel = () => { + return ( + + Auto Update{' '} + +
Auto Update Policy
+
+ Prompt: prompt to install +
+
+ Silent: install silently +
+ {/*
Disable: disable auto update
*/} + + } + > + + + + ); +}; + +const GlobalShortcutLabel = () => { + return ( +
+ Global Shortcut{' '} + +
Shortcut definition, modifiers and key separated by "+" e.g. CmdOrControl+Q
+
If empty, the shortcut is disabled.
+ + https://tauri.app/v1/api/js/globalshortcut + +
+ } + > + + + + ); +}; diff --git a/src/view/settings/MainWindow.tsx b/src/view/settings/MainWindow.tsx new file mode 100644 index 0000000..0235ab7 --- /dev/null +++ b/src/view/settings/MainWindow.tsx @@ -0,0 +1,49 @@ +import { Form, Switch, Input, Tooltip } from 'antd'; +import { QuestionCircleOutlined } from '@ant-design/icons'; + +import SwitchOrigin from '@/components/SwitchOrigin'; +import { DISABLE_AUTO_COMPLETE } from '@/utils'; + +const PopupSearchLabel = () => { + return ( + + Pop-up Search{' '} + +
+ Generate images according to the content: Select the ChatGPT content with the mouse, + no more than 400 characters. the DALL·E 2 button appears, and click to jump + (Note: because the search content filled by the script cannot trigger the event + directly, you need to enter a space in the input box to make the button clickable). +
+
+ The application is built using Tauri, and due to its security restrictions, some of + the action buttons will not work, so we recommend going to your browser. +
+ + } + > + + + + ); +}; + +export default function General() { + return ( + <> + } name="popup_search" valuePropName="checked"> + + + + + + + + ); +} diff --git a/src/view/settings/TrayWindow.tsx b/src/view/settings/TrayWindow.tsx new file mode 100644 index 0000000..7415911 --- /dev/null +++ b/src/view/settings/TrayWindow.tsx @@ -0,0 +1,36 @@ +import { Form, Switch, Input, Tooltip } from 'antd'; +import { QuestionCircleOutlined } from '@ant-design/icons'; + +import { DISABLE_AUTO_COMPLETE } from '@/utils'; +import SwitchOrigin from '@/components/SwitchOrigin'; + +const UALabel = () => { + return ( + + User Agent (SystemTray){' '} + For a better experience, we recommend using the Mobile User-Agent.} + > + + + + ); +}; + +export default function General() { + return ( + <> + + + + + } name="ua_tray"> + + + + ); +} diff --git a/src/view/settings/index.tsx b/src/view/settings/index.tsx new file mode 100644 index 0000000..1b41f15 --- /dev/null +++ b/src/view/settings/index.tsx @@ -0,0 +1,117 @@ +import { useEffect, useState } from 'react'; +import { useSearchParams } from 'react-router-dom'; +import { Form, Tabs, Space, Button, Popconfirm, message } from 'antd'; +import { invoke, dialog, process, path, shell } from '@tauri-apps/api'; +import { clone, omit, isEqual } from 'lodash'; + +import useInit from '@/hooks/useInit'; +import FilePath from '@/components/FilePath'; +import { chatRoot, CHAT_CONF_JSON } from '@/utils'; +import General from './General'; +import MainWindow from './MainWindow'; +import TrayWindow from './TrayWindow'; + +export default function Settings() { + const [params] = useSearchParams(); + const [activeKey, setActiveKey] = useState('general'); + const [form] = Form.useForm(); + const [chatConf, setChatConf] = useState(null); + const [filePath, setPath] = useState(''); + const key = params.get('type'); + + useEffect(() => { + setActiveKey(key ? key : 'general'); + }, [key]); + + useInit(async () => { + setChatConf(await invoke('get_chat_conf')); + setPath(await path.join(await chatRoot(), CHAT_CONF_JSON)); + }); + + useEffect(() => { + form.setFieldsValue(clone(chatConf)); + }, [chatConf]); + + const onCancel = () => { + form.setFieldsValue(chatConf); + }; + + const onReset = async () => { + const chatData = await invoke('reset_chat_conf'); + setChatConf(chatData); + const isOk = await dialog.ask(`Configuration reset successfully, whether to restart?`, { + title: 'ChatGPT Preferences', + }); + if (isOk) { + process.relaunch(); + return; + } + message.success('Configuration reset successfully'); + }; + + const onFinish = async (values: any) => { + if (!isEqual(omit(chatConf, ['default_origin']), values)) { + await invoke('form_confirm', { data: values, label: 'main' }); + const isOk = await dialog.ask(`Configuration saved successfully, whether to restart?`, { + title: 'ChatGPT Preferences', + }); + if (isOk) { + process.relaunch(); + return; + } + message.success('Configuration saved successfully'); + } + }; + + const handleTab = (v: string) => { + setActiveKey(v); + }; + + return ( +
+ +
+ }, + { label: 'Main Window', key: 'main_window', children: }, + { label: 'SystemTray Window', key: 'tray_window', children: }, + ]} + /> + + + + + + + Are you sure you want to reset the configuration file + shell.open(filePath)} style={{ margin: '0 5px' }}> + {filePath} + + to the default? +
+ } + onConfirm={onReset} + okText="Yes" + cancelText="No" + > + + + + + + + ); +} diff --git a/tsconfig.json b/tsconfig.json index 474aa73..9505215 100644 --- a/tsconfig.json +++ b/tsconfig.json @@ -20,7 +20,7 @@ "@/*": ["src/*"], "@view/*": ["src/view/*"], "@comps/*": ["src/components/*"], - "@layout/*": ["src/layout/*"], + "@layout/*": ["src/layout/*"] } }, "include": ["src"], diff --git a/vite.config.ts b/vite.config.ts index debf8b2..c69da5d 100644 --- a/vite.config.ts +++ b/vite.config.ts @@ -1,6 +1,6 @@ -import { defineConfig } from "vite"; -import react from "@vitejs/plugin-react"; -import tsconfigPaths from "vite-tsconfig-paths"; +import { defineConfig } from 'vite'; +import react from '@vitejs/plugin-react'; +import tsconfigPaths from 'vite-tsconfig-paths'; // https://vitejs.dev/config/ export default defineConfig({ @@ -16,12 +16,12 @@ export default defineConfig({ }, // to make use of `TAURI_DEBUG` and other env variables // https://tauri.studio/v1/api/config#buildconfig.beforedevcommand - envPrefix: ["VITE_", "TAURI_"], + envPrefix: ['VITE_', 'TAURI_'], build: { // Tauri supports es2021 - target: ["es2021", "chrome100", "safari13"], + target: ['es2021', 'chrome100', 'safari13'], // don't minify for debug builds - minify: !process.env.TAURI_DEBUG ? "esbuild" : false, + minify: !process.env.TAURI_DEBUG ? 'esbuild' : false, // produce sourcemaps for debug builds sourcemap: !!process.env.TAURI_DEBUG, },