From d4bd042b41547cbcde4ae6cefe102e590932deb6 Mon Sep 17 00:00:00 2001
From: Anoncer <anoncer@protonmail.com>
Date: Sun, 14 Jul 2024 14:53:42 +0200
Subject: [PATCH] Fixes and improvements to the pwa update

---
 javascript/server.js | 77 +++++++++++++++++++++++++++++++-------------
 1 file changed, 54 insertions(+), 23 deletions(-)

diff --git a/javascript/server.js b/javascript/server.js
index e04220a..17926d6 100644
--- a/javascript/server.js
+++ b/javascript/server.js
@@ -1,6 +1,5 @@
 const $SERVER = {
     sw: navigator.serviceWorker,
-    cache: caches || undefined,
     version: undefined,
     hash: undefined,
     dialog_key: "dialog-update",
@@ -42,6 +41,10 @@ const $SERVER = {
     }
 };
 
+if(typeof caches !== 'undefined'){
+    $SERVER.cache = caches;
+}
+
 (async () => {
     if ($SERVER.sw) {
         await $SERVER.functions.newRegistration();
@@ -51,11 +54,25 @@ const $SERVER = {
 
         ShowUIVersion();
 
+        HasUpdate();
         OnUpdate();
 
         UpdateSystem();
     }
 
+    async function HasUpdate() {
+        const data = await ParseSWVersion();
+        if (data.ver === $SERVER.version && data.hash === $SERVER.hash) {
+            $SERVER.updatet = false;
+
+            $SERVER.On.complete = true;
+
+            $SERVER.On.subscribe.forEach(event => {
+                event($SERVER.updatet);
+            });
+        }
+    }
+
     // Обрабатывает обновление интерфейса
     function OnUpdate() {
         const url = new URL(window.location.href);
@@ -85,32 +102,37 @@ const $SERVER = {
     async function UpdateSystem() {
         const reg = await $SERVER.functions.getRegistration();
         let newServer = undefined;
+        let type = 'active';
 
         if (reg) {
-            if (reg.installing) {
-                newServer = reg.installing;
-            } else if (reg.waiting) {
-                newServer = reg.waiting;
-                const data = await ParseSWVersion();
-                $SERVER.updatet = true;
-                $('.app-update').css({ display: 'flex' });
-                SetUIVersion(data);
-                $('.app-update').addClass('show');
-                $('.update-progress > .progress').css({ width: '50%' });
-
-                const skipData = await Events().skipWaiting();
-            } else if (reg.active) {
-                newServer = reg.active;
+            newServer = reg.installing || reg.waiting || reg.active;
+            type = reg.installing ? 'installing' : reg.waiting ? 'waiting' : 'active';
+
+            reg.onupdatefound = async () => {
+                const swr = await $SERVER.sw.getRegistration();
+                newServer.removeEventListener('statechange', OnStateChange);
+                newServer = swr.installing || swr.waiting || swr.active;
+                type = swr.installing ? 'installing' : swr.waiting ? 'waiting' : 'active';
+
+                if (type === 'installing') {
+                    Install();
+                }
+
+                newServer.addEventListener('statechange', (ev) => OnStateChange(ev, true));
             }
         }
 
-        $SERVER.On.subscribe.forEach(event => {
-            event($SERVER.updatet);
-        });
+        newServer.addEventListener('statechange', OnStateChange);
+
+        if (type === 'waiting' || type === 'installing') {
+            Install();
+        }
 
-        newServer.addEventListener('statechange', ev => {
-            console.log(ev.target.state);
-            if (ev.target.state === 'activating') {
+        async function OnStateChange(ev) {
+            if (ev.target.state === 'waiting' || ev.target.state === 'installing') {
+                Install();
+            }
+            if (ev.target.state === 'activating' || ev.target.state === 'installed') {
                 $('.update-progress > .progress').css({ width: '80%' });
             }
             if (ev.target.state === 'activated' || ev.target.state === 'redundant') {
@@ -120,7 +142,16 @@ const $SERVER = {
                 url.searchParams.set('hash', $SERVER.hash);
                 window.location.href = url.toString();
             }
-        });
+        }
+
+        async function Install() {
+            $('.app-update').css({ display: 'flex' });
+            const data = await ParseSWVersion();
+            SetUIVersion(data);
+            $('.app-update').addClass('show');
+            $('.update-progress > .progress').css({ width: '50%' });
+            newServer.postMessage({ id: 221 });
+        }
     }
 
     // Отображает версию интерфейса
@@ -129,7 +160,7 @@ const $SERVER = {
         $('.github > .version-hash > .hash').text($SERVER.hash);
 
         let data = JSON.parse(localStorage.getItem($SERVER.dialog_key));
-        if (data === undefined) {
+        if (data === undefined || data === null) {
             data = { show: false, version: $SERVER.version, hash: $SERVER.hash, update: new Date().toJSON() };
             localStorage.setItem($SERVER.dialog_key, JSON.stringify(data));
         }