mod(driver): add google retry

Author: PIKACHUIM

public/index.html

@@ -155,6 +155,7 @@ <h4>隐私政策和使用条款</h4>
 <script src="/static/spoid.js"></script>
 <script src="/static/input.js"></script>
 <script>
+    let callback_flag = false;
     // 获取select元素和输入框元素
     const driver_txt_input = document.getElementById('driver-txt-input');
     const direct_url_input = document.getElementById('direct-url-input');

public/static/event.js

@@ -104,6 +104,7 @@ function onSelect() {
     // Google弹出隐私政策和使用条款
     if (driver_txt_input.value === "googleui_go") {
         if (server_use_input.checked)
-            showGoogleConsent();
+            if (!callback_flag)
+                showGoogleConsent();
     }
 }

public/static/token.js

@@ -4,7 +4,6 @@ async function getToken() {
             const json_byte = Uint8Array.from(atob(url_hash), c => c.charCodeAt(0));
             const json_text = new TextDecoder().decode(json_byte);
             const json_data = JSON.parse(json_text);
-            console.log(json_data);
             const server_use = json_data.server_use;
             const client_uid = json_data.client_uid;
             const secret_key = json_data.secret_key;
@@ -13,6 +12,7 @@ async function getToken() {
             const access_key = json_data.access_token;
             const refresh_ui = json_data.refresh_token;
             const message_ui = json_data.message_err;
+            callback_flag = true;
             // 从历史记录清除#号部分，避免隐私信息泄漏
             // 这只会在正常解析JSON后执行，其他的hash不会被清除
             // window.history.replaceState(null, null, window.location.pathname + window.location.search);

src/driver/googleui_oa.ts

@@ -88,10 +88,21 @@ export async function oneToken(c: Context) {
     // 执行请求 ===========================================================================
     try {
         const paramsString = new URLSearchParams(params_all).toString();
-        const response: Response = await fetch(server_url, {
-            method: 'POST', body: paramsString,
-            headers: {'Content-Type': 'application/x-www-form-urlencoded',},
-        });
+        let try_time: number = 5;
+        let response: Response | null = null;
+        while (try_time > 0) {
+            try {
+                response = await fetch(server_url, {
+                    method: 'POST', body: paramsString,
+                    headers: {'Content-Type': 'application/x-www-form-urlencoded',},
+                });
+                break;
+            } catch (error) {
+                try_time -= 1;
+            }
+        }
+        if (!try_time || !response) return c.redirect(
+            showErr("多次尝试获取Token失败"));
         if (server_use == "false") {
             local.deleteCookie(c, 'client_uid');
             local.deleteCookie(c, 'client_key');