Update index.ts

index.ts

@@ -1,31 +1,66 @@
-import { connect } from 'cloudflare:sockets';
-const passwd='X_PASSWORD';
-
-export default{
-  async fetch(request){
-    const upgradeHeader=request.headers.get("Upgrade");
-    if(upgradeHeader!=="websocket") return new Response("hello world!", {status:200});
-    const [client, server]=Object.values(new WebSocketPair());
-    server.accept();
-    server.addEventListener('message',({data})=>{
-      try{
-        const {hostname,port,psw}=JSON.parse(data);
-        if(passwd!=psw) throw 'Illegal-User';
-        const socket=connect({hostname,port});
-        new ReadableStream({
-          start(controller){
-            server.onmessage= ({data})=>controller.enqueue(data);
-            server.onerror=e=>controller.error(e);
-            server.onclose=e=>controller.close(e);
-          },
-          cancel(reason){server.close();}
-        }).pipeTo(socket.writable);
-        socket.readable.pipeTo(new WritableStream({
-          start(controller){server.onerror=e=>controller.error(e);},
-          write(chunk){server.send(chunk);}
-        }));
-      }catch(error){ server.close(); }
-    },{once:true});
-    return new Response(null, {status:101, webSocket:client});
-  }
+import { connect } from 'cloudflare:sockets';
+var TOKEN_URL = "X_TOKEN_URL";
+var WPAD_PATH = "X_WPAD_URL";
+var CLI_PATH = "X_CLI_PATH"
+
+var tokenmap = {};
+var tokenSyncTime = 0;
+
+var syncTokens = async ()=>{
+       let res = await fetch(TOKEN_URL, {}).then(res=>{
+	   return res.json()
+       })
+       tokenSyncTime = new Date().getTime();
+       Object.assign(tokenmap, res);
+       console.log(tokenmap);
+}
+
+export default{
+  async fetch(request, env){
+     if (Object.keys(tokenmap).length == 0){
+        await syncTokens();
+     } else if(new Date().getTime() - 10*1000 > tokenSyncTime) {
+         syncTokens();
+     }
+
+    const upgradeHeader=request.headers.get("Upgrade");
+    var url = request.url.split("token=")[0];
+    var token = request.url.split("token=")[1];
+    if (!token || !tokenmap[token] || tokenmap[token] < new Date().getTime()) {
+        return new Response("Forbidden", {status: 403});
+    }
+    try {
+        console.log(url);
+        if(url.length > 9 && url.substring(url.length-9) === "/wpad.dat") {
+             return await fetch(WPAD_PATH, {})
+        } else if(url.length > 7 && url.substring(url.length-7) === "/cli.js") {
+             return await fetch(CLI_PATH, {})
+        } 
+    } catch(e) {
+        return new Response("Server error", {status: 500});
+    }
+    if(upgradeHeader!=="websocket") return new Response("可以<a href='./cli.js'>连接</a>了", {status:200});
+    const [client, server]=Object.values(new WebSocketPair());
+    server.accept();
+    server.addEventListener('message',({data})=>{
+      try{
+        const {hostname,port,psw}=JSON.parse(data);
+        if(passwd!=psw) throw 'Illegal-User';
+        const socket=connect({hostname,port});
+        new ReadableStream({
+          start(controller){
+            server.onmessage= ({data})=>controller.enqueue(data);
+            server.onerror=e=>controller.error(e);
+            server.onclose=e=>controller.close(e);
+          },
+          cancel(reason){server.close();}
+        }).pipeTo(socket.writable);
+        socket.readable.pipeTo(new WritableStream({
+          start(controller){server.onerror=e=>controller.error(e);},
+          write(chunk){server.send(chunk);}
+        }));
+      }catch(error){ server.close(); }
+    },{once:true});
+    return new Response(null, {status:101, webSocket:client});
+  }
 }