Update index.js

index.js

@@ -4,10 +4,10 @@ const path = require('path');
 const { WebSocketServer } = require('ws');
 
 // =============== 修改這裡 ===============
-const UUID = 'b831381d-6324-4d53-ad4f-8cda48b30811';   // ← 改成你自己的 UUID
-const DOMAIN = 'wky-tty.hf.space';               // ← 改成你的 HF Space 域名
-const WSPATH = 'ws';                                   // 路徑，可改
-const SUB_PATH = 'sub';                                // 訂閱路徑，建議改成較複雜的
+const UUID = 'b831381d-6324-4d53-ad4f-8cda48b30811';   // 你的 UUID
+const DOMAIN = 'wky-tty.hf.space';                     // 你的 HF 域名
+const WSPATH = 'ws';                                   // WS 路徑
+const SUB_PATH = 'sub';                                // 訂閱路徑
 const PORT = 7860;
 // ========================================
 
@@ -15,7 +15,7 @@ const requestHandler = (req, res) => {
     if (req.url === '/') {
         fs.readFile(path.join(__dirname, 'index.html'), (err, content) => {
             res.writeHead(200, { 'Content-Type': 'text/html; charset=utf-8' });
-            res.end(err ? '<h1>Hello from HF Spaces</h1>' : content);
+            res.end(err ? '<h1>Hugging Face Proxy</h1>' : content);
         });
     } 
     else if (req.url === `/${SUB_PATH}`) {
@@ -24,7 +24,7 @@ const requestHandler = (req, res) => {
         
         res.writeHead(200, { 'Content-Type': 'text/plain' });
         res.end(base64);
-        console.log('Subscription requested');
+        console.log('✅ 訂閱被請求');
     } 
     else {
         res.writeHead(404);
@@ -35,21 +35,33 @@ const requestHandler = (req, res) => {
 const server = http.createServer(requestHandler);
 const wss = new WebSocketServer({ server });
 
-wss.on('connection', (ws) => {
-    console.log('New connection');
+wss.on('connection', (ws, req) => {
+    console.log('New WS Connection');
+
     ws.once('message', (data) => {
-        // 簡單 VLESS + Trojan 處理（基本版）
-        const firstByte = data[0];
-        if (firstByte === 0 || firstByte === 1) {
-            // VLESS
-            ws.write(Buffer.from([0, 0])); // 回應
-        } else {
-            // Trojan 簡易處理
-            ws.write(Buffer.from([0x01, 0x00]));
+        const buf = Buffer.from(data);
+        
+        // VLESS 協議簡單回應（版本 0）
+        if (buf[0] === 0) {
+            const response = Buffer.from([0x00, 0x00]); // VLESS 成功回應
+            ws.send(response);
+            console.log('VLESS Handshake OK');
+        } 
+        // Trojan 簡易處理
+        else if (buf.length > 56) {
+            ws.send(Buffer.from([0x01, 0x00]));
+            console.log('Trojan Handshake');
         }
     });
 
-    ws.on('error', () => ws.terminate());
+    ws.on('error', (err) => {
+        console.log('WS Error:', err.message);
+        ws.terminate();
+    });
+
+    ws.on('close', () => {
+        console.log('Connection closed');
+    });
 });
 
 server.listen(PORT, () => {