From 515d0e93ce117fbe788fe7d4c8002ddafe7c841f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=81=B5?= <132723284+shulng@users.noreply.github.com>
Date: Thu, 28 Nov 2024 04:28:47 -0800
Subject: [PATCH] Add files via upload

---
 src/worker.js | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/src/worker.js b/src/worker.js
index e7afe5582..ab722ccbc 100644
--- a/src/worker.js
+++ b/src/worker.js
@@ -2,7 +2,7 @@
 import { vlessOverWSHandler } from './protocols/vless';
 import { trojanOverWSHandler } from './protocols/trojan';
 import { updateWarpConfigs } from './kv/handlers';
-import { logout, resetPassword, login } from './authentication/auth';
+import { logout, resetPassword, login, checkSecrets } from './authentication/auth';
 import { renderErrorPage } from './pages/error';
 import { getXrayCustomConfigs, getXrayWarpConfigs } from './cores-configs/xray';
 import { getSingBoxCustomConfig, getSingBoxWarpConfig } from './cores-configs/sing-box';
@@ -10,6 +10,7 @@ import { getClashNormalConfig, getClashWarpConfig } from './cores-configs/clash'
 import { getNormalConfigs } from './cores-configs/normalConfigs';
 import { initializeParams, userID, client, pathName } from './helpers/init';
 import { fallback, getMyIP, handlePanel } from './helpers/helpers';
+import { renderSecretsPage } from './pages/secrets';
 
 export default {
     async fetch(request, env) {
@@ -19,43 +20,56 @@ export default {
             if (!upgradeHeader || upgradeHeader !== 'websocket') {            
                 switch (pathName) {                    
                     case '/update-warp':
+                        checkSecrets(env);
                         return await updateWarpConfigs(request, env);
 
                     case `/sub/${userID}`:
+                        checkSecrets(env);
                         if (client === 'sfa') return await getSingBoxCustomConfig(request, env, false);
                         if (client === 'clash') return await getClashNormalConfig(request, env);
                         if (client === 'xray') return await getXrayCustomConfigs(request, env, false);
                         return await getNormalConfigs(request, env);                        
 
                     case `/fragsub/${userID}`:
+                        checkSecrets(env);
                         return client === 'hiddify'
                             ? await getSingBoxCustomConfig(request, env, true)
                             : await getXrayCustomConfigs(request, env, true);
 
                     case `/warpsub/${userID}`:
+                        checkSecrets(env);
                         if (client === 'clash') return await getClashWarpConfig(request, env);   
                         if (client === 'singbox' || client === 'hiddify') return await getSingBoxWarpConfig(request, env, client);
                         return await getXrayWarpConfigs(request, env, client);
 
                     case '/panel':
+                        checkSecrets(env);
                         return await handlePanel(request, env);
                                                       
                     case '/login':
+                        checkSecrets(env);
                         return await login(request, env);
                     
-                    case '/logout':                        
+                    case '/logout':
+                        checkSecrets(env);                        
                         return logout();        
 
                     case '/panel/password':
+                        checkSecrets(env);
                         return await resetPassword(request, env);
                     
                     case '/my-ip':
+                        checkSecrets(env);
                         return await getMyIP(request);
 
+                    case '/secrets':
+                        return await renderSecretsPage(request, env);
+
                     default:
                         return await fallback(request);
                 }
             } else {
+                checkSecrets(env);
                 return pathName.startsWith('/tr') 
                     ? await trojanOverWSHandler(request, env) 
                     : await vlessOverWSHandler(request, env);