From aba0f6b5e2bf809d6d4b5ead15461c0b82ff5d09 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Tue, 17 Jun 2025 23:27:49 +0800
Subject: [PATCH] :bug: Fixes the callback page

---
 .../Pages/Auth/Callback.cshtml                | 41 +++++++++----------
 1 file changed, 19 insertions(+), 22 deletions(-)

diff --git a/DysonNetwork.Sphere/Pages/Auth/Callback.cshtml b/DysonNetwork.Sphere/Pages/Auth/Callback.cshtml
index 527558c..9dd82d6 100644
--- a/DysonNetwork.Sphere/Pages/Auth/Callback.cshtml
+++ b/DysonNetwork.Sphere/Pages/Auth/Callback.cshtml
@@ -14,36 +14,33 @@
 
 @section Scripts {
     <script>
-        (function() {
+        (function () {
             const urlParams = new URLSearchParams(window.location.search);
             const token = urlParams.get('token');
 
-            if (token) {
-                console.log("Authentication token received.");
+            console.log("Authentication token received.");
 
-                // For WebView2/UWP apps that can handle window.external.notify
-                if (window.external && typeof window.external.notify === 'function') {
-                    try {
-                        window.external.notify(token);
-                        console.log("Token sent via window.external.notify.");
-                        return; // Exit after successful notification
-                    } catch (e) {
-                        console.error("Failed to send token via window.external.notify:", e);
-                    }
-                }
-
-                // For mobile apps that use custom URI schemes
+            // For WebView2/UWP apps that can handle window.external.notify
+            if (window.external && typeof window.external.notify === 'function') {
                 try {
-                    const customSchemeUrl = `dyson://auth?token=${encodeURIComponent(token)}`;
-                    window.location.href = customSchemeUrl;
-                    console.log("Attempting to redirect to custom scheme:", customSchemeUrl);
+                    if (!token)
+                        window.external.notify('done');
+                    else
+                        window.external.notify(token);
+                    console.log("Token sent via window.external.notify.");
+                    return; // Exit after successful notification
                 } catch (e) {
-                    console.error("Failed to redirect to custom scheme:", e);
+                    console.error("Failed to send token via window.external.notify:", e);
                 }
+            }
 
-            } else {
-                console.error("Authentication token not found in URL.");
-                document.querySelector('p').innerText = "Authentication failed: No token was provided.";
+            // For mobile apps that use custom URI schemes
+            try {
+                const customSchemeUrl = `solian://auth/callback?token=${encodeURIComponent(token ?? 'done')}`;
+                window.location.href = customSchemeUrl;
+                console.log("Attempting to redirect to custom scheme:", customSchemeUrl);
+            } catch (e) {
+                console.error("Failed to redirect to custom scheme:", e);
             }
         })();
     </script>