From 64a17abfe2f84e9dc205aa417f6ce9a05c7c1cd2 Mon Sep 17 00:00:00 2001
From: H0llyW00dzZ <priv8@btz.pm>
Date: Tue, 3 Oct 2023 08:08:11 +0700
Subject: [PATCH] Client App [Notification]

[+] feat(global.d.ts): add support for window.__TAURI__.notification methods
[+] feat(update.ts): add notification for new version availability
[+] fix(Cargo.toml): add tauri feature "notification-all" to enable notifications
[+] fix(tauri.conf.json): enable all notification features in tauri configuration
---
 app/global.d.ts           |  5 +++++
 app/store/update.ts       | 29 +++++++++++++++++++++++++++++
 src-tauri/Cargo.toml      |  2 +-
 src-tauri/tauri.conf.json |  6 ++++++
 4 files changed, 41 insertions(+), 1 deletion(-)

diff --git a/app/global.d.ts b/app/global.d.ts
index 524ce77d..dc1d5265 100644
--- a/app/global.d.ts
+++ b/app/global.d.ts
@@ -13,5 +13,10 @@ declare module "*.svg";
 declare interface Window {
   __TAURI__?: {
     writeText(text: string): Promise<void>;
+    notification:{
+      requestPermission(): Promise<Permission>;
+      isPermissionGranted(): Promise<boolean>;
+      sendNotification(options: string | Options): void;
+    };
   };
 }
diff --git a/app/store/update.ts b/app/store/update.ts
index 42b86586..facb5732 100644
--- a/app/store/update.ts
+++ b/app/store/update.ts
@@ -2,8 +2,10 @@ import { FETCH_COMMIT_URL, FETCH_TAG_URL, StoreKey } from "../constant";
 import { api } from "../client/api";
 import { getClientConfig } from "../config/client";
 import { createPersistStore } from "../utils/store";
+import ChatGptIcon from "../icons/chatgpt.png";
 
 const ONE_MINUTE = 60 * 1000;
+const isApp = !!getClientConfig()?.isApp;
 
 function formatVersionDate(t: string) {
   const d = new Date(+t);
@@ -80,6 +82,33 @@ export const useUpdateStore = createPersistStore(
         set(() => ({
           remoteVersion: remoteId,
         }));
+        if (window.__TAURI__?.notification && isApp) {
+          // Check if notification permission is granted
+          await window.__TAURI__?.notification.isPermissionGranted().then((granted) => {
+            if (!granted) {
+              // Send a notification without waiting for permission (because we don't neeed a permisison once client is already click "check")
+              window.__TAURI__?.notification.sendNotification({
+                title: "ChatGPT Next Web",
+                body: `A new version (${remoteId}) is available.`,
+                icon: `${ChatGptIcon.src}`,
+                sound: "Default"
+              });
+            } else {
+              // Request permission to show notifications
+              window.__TAURI__?.notification.requestPermission().then((permission) => {
+                if (permission === 'granted') {
+                  // Show a notification using Tauri
+                  window.__TAURI__?.notification.sendNotification({
+                    title: "ChatGPT Next Web",
+                    body: `A new version (${remoteId}) is available.`,
+                    icon: `${ChatGptIcon.src}`,
+                    sound: "Default"
+                  });
+                }
+              });
+            }
+          });
+        }
         console.log("[Got Upstream] ", remoteId);
       } catch (error) {
         console.error("[Fetch Upstream Commit Id]", error);
diff --git a/src-tauri/Cargo.toml b/src-tauri/Cargo.toml
index ac5d04e8..fee1c860 100644
--- a/src-tauri/Cargo.toml
+++ b/src-tauri/Cargo.toml
@@ -17,7 +17,7 @@ tauri-build = { version = "1.3.0", features = [] }
 [dependencies]
 serde_json = "1.0"
 serde = { version = "1.0", features = ["derive"] }
-tauri = { version = "1.3.0", features = ["clipboard-all", "dialog-all", "shell-open", "updater", "window-close", "window-hide", "window-maximize", "window-minimize", "window-set-icon", "window-set-ignore-cursor-events", "window-set-resizable", "window-show", "window-start-dragging", "window-unmaximize", "window-unminimize"] }
+tauri = { version = "1.3.0", features = ["notification-all", "fs-all", "clipboard-all", "dialog-all", "shell-open", "updater", "window-close", "window-hide", "window-maximize", "window-minimize", "window-set-icon", "window-set-ignore-cursor-events", "window-set-resizable", "window-show", "window-start-dragging", "window-unmaximize", "window-unminimize"] }
 tauri-plugin-window-state = { git = "https://github.com/tauri-apps/plugins-workspace", branch = "v1" }
 
 [features]
diff --git a/src-tauri/tauri.conf.json b/src-tauri/tauri.conf.json
index 77b02a3b..147fc994 100644
--- a/src-tauri/tauri.conf.json
+++ b/src-tauri/tauri.conf.json
@@ -44,6 +44,12 @@
         "startDragging": true,
         "unmaximize": true,
         "unminimize": true
+      },
+      "fs": {
+        "all": true
+      },
+      "notification": {
+        "all": true
       }
     },
     "bundle": {