Create main.ts

main.ts

@@ -0,0 +1,298 @@
+const COOKIES = `__gads=ID=fa119d4bd4f0deb4:T=1767802127:RT=1769915891:S=ALNI_MadOwOvOyuYGa63fhmTXAOSMuMPzA; __gpi=UID=000011dec0f55e42:T=1767802127:RT=1769915891:S=ALNI_MbmxAr7exkcHsXJpxvp5Jda40UEhQ; __eoi=ID=1b168e49673ee98a:T=1767802127:RT=1769915891:S=AA-Afjb1OtBK1G-c16acoNoP6uKS; NEXT_LOCALE=zh; u_device_id=d403ec97-7698-4c1a-a5ad-4121004056df; home_chat_id=f2b45c03-aeee-4c9e-862f-50aa2e7ba549; __Secure-authjs.callback-url=https%3A%2F%2Fapp.unlimitedai.chat; __Host-authjs.csrf-token=85724d6088bf62366e70be6dca13329f84a248a2e682b9469ea9c4edea86294b%7Cb6d8881d70b2a9fcb7b3bdc9c67cec59c17caba2225b24909cddbc1d6092d9c3; _clck=1ehq05y%5E2%5Eg5e%5E0%5E2302; _clsk=1j5vo0j%5E1776785008641%5E1%5E1%5Ez.clarity.ms%2Fcollect`;
+
+// 目标 API 地址
+const TARGET_API = "https://app.unlimitedai.chat/api/chat";
+
+// 模拟的设备 ID (保持会话连续性)
+const DEVICE_ID = "d403ec97-7698-4c1a-a5ad-4121004056de"; 
+
+// --- 核心逻辑 ---
+
+async function handler(req: Request): Promise<Response> {
+  const url = new URL(req.url);
+  
+  // CORS 处理
+  if (req.method === 'OPTIONS') {
+    return new Response(null, { 
+      status: 204, 
+      headers: corsHeaders() 
+    });
+  }
+
+  // 静态页面
+  if (url.pathname === '/') {
+    return new Response(getUI(), {
+      headers: { 'Content-Type': 'text/html; charset=utf-8' }
+    });
+  }
+
+  // 模型列表
+  if (url.pathname === '/v1/models' && req.method === 'GET') {
+    return new Response(JSON.stringify({
+      object: "list",
+      data: [
+        { id: "chat-model-reasoning", object: "model", created: Date.now(), owned_by: "unlimitedai" }
+      ]
+    }), {
+      headers: { 'Content-Type': 'application/json', ...corsHeaders() }
+    });
+  }
+
+  // 聊天补全
+  if (url.pathname === '/v1/chat/completions' && req.method === 'POST') {
+    return handleChatCompletion(req);
+  }
+
+  return new Response("Not Found", { status: 404 });
+}
+
+async function handleChatCompletion(req: Request): Promise<Response> {
+  try {
+    const body = await req.json();
+    const model = body.model || "chat-model-reasoning";
+    const openAiMessages = body.messages || [];
+
+    // 1. 构建 UnlimitedAI 格式的请求体
+    // 需要为每条消息生成 ID 和时间戳
+    const now = new Date().toISOString();
+    const uaiMessages = openAiMessages.map((msg: any, index: number) => ({
+      id: crypto.randomUUID(),
+      role: msg.role,
+      content: msg.content,
+      parts: [{ type: "text", text: msg.content }], // UnlimitedAI 特有的 parts 结构
+      createdAt: now
+    }));
+
+    // 生成一个新的会话 ID (实际生产中可能需要根据历史消息复用 ID)
+    // 这里为了简单演示，每次请求生成一个新的 Chat ID
+    const chatId = crypto.randomUUID();
+
+    const payload = {
+      chatId: chatId,
+      messages: uaiMessages,
+      selectedChatModel: model,
+      selectedCharacter: null,
+      selectedStory: null,
+      deviceId: DEVICE_ID,
+      locale: "zh"
+    };
+
+    // 2. 构建请求头 (完全模拟抓包数据)
+    const headers: Record<string, string> = {
+      "accept": "*/*",
+      "content-type": "application/json",
+      "cookie": COOKIES,
+      "origin": "https://app.unlimitedai.chat",
+      "referer": "https://app.unlimitedai.chat/zh",
+      "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/147.0.0.0 Safari/537.36 Edg/147.0.0.0",
+      "x-next-intl-locale": "zh",
+      "sec-ch-ua": '"Microsoft Edge";v="147", "Not.A/Brand";v="8", "Chromium";v="147"',
+      "sec-ch-ua-mobile": "?0",
+      "sec-ch-ua-platform": '"Windows"',
+      "sec-fetch-dest": "empty",
+      "sec-fetch-mode": "cors",
+      "sec-fetch-site": "same-origin"
+    };
+
+    // 3. 发起上游请求
+    const upstreamRes = await fetch(TARGET_API, {
+      method: "POST",
+      headers,
+      body: JSON.stringify(payload)
+    });
+
+    if (!upstreamRes.ok) {
+      const errText = await upstreamRes.text();
+      return new Response(JSON.stringify({ error: { message: `Upstream Error ${upstreamRes.status}: ${errText}` } }), {
+        status: upstreamRes.status,
+        headers: { 'Content-Type': 'application/json' }
+      });
+    }
+
+    // 4. 流式转换处理 (NDJSON -> SSE)
+    const { readable, writable } = new TransformStream();
+    const writer = writable.getWriter();
+    const encoder = new TextEncoder();
+    
+    // 用于跟踪会话 ID
+    const openAiChatId = crypto.randomUUID();
+    
+    (async () => {
+      try {
+        const reader = upstreamRes.body?.getReader();
+        if (!reader) throw new Error("No body");
+        const decoder = new TextDecoder();
+        let buffer = "";
+
+        while (true) {
+          const { done, value } = await reader.read();
+          if (done) break;
+
+          buffer += decoder.decode(value, { stream: true });
+          const lines = buffer.split('\n');
+          buffer = lines.pop() || ""; // 保留最后一个可能不完整的块
+
+          for (const line of lines) {
+            if (!line.trim()) continue;
+            try {
+              const json = JSON.parse(line);
+              
+              // UnlimitedAI 返回格式: {"type":"delta","delta":"你好！"}
+              if (json.type === 'delta' && json.delta) {
+                const content = json.delta;
+                
+                // 转换为 OpenAI SSE 格式
+                const sseChunk = {
+                  id: openAiChatId,
+                  object: "chat.completion.chunk",
+                  created: Math.floor(Date.now() / 1000),
+                  model: model,
+                  choices: [{
+                    index: 0,
+                    delta: { content: content },
+                    finish_reason: null
+                  }]
+                };
+                
+                writer.write(encoder.encode(`data: ${JSON.stringify(sseChunk)}\n\n`));
+              }
+            } catch (e) {
+              console.error("Parse error", e);
+            }
+          }
+        }
+
+        // 发送结束标记
+        writer.write(encoder.encode(`data: ${JSON.stringify({
+          id: openAiChatId,
+          object: "chat.completion.chunk",
+          created: Math.floor(Date.now() / 1000),
+          model: model,
+          choices: [{
+            index: 0,
+            delta: {},
+            finish_reason: "stop"
+          }]
+        })}\n\n`));
+        
+        writer.write(encoder.encode('data: [DONE]\n\n'));
+
+      } catch (err: any) {
+        console.error("Stream processing error:", err);
+        writer.write(encoder.encode(`data: ${JSON.stringify({ error: { message: err.message } })}\n\n`));
+      } finally {
+        writer.close();
+      }
+    })();
+
+    return new Response(readable, {
+      headers: {
+        "Content-Type": "text/event-stream; charset=utf-8",
+        "Cache-Control": "no-cache",
+        "Connection": "keep-alive",
+        ...corsHeaders()
+      }
+    });
+
+  } catch (error: any) {
+    return new Response(JSON.stringify({ error: { message: error.message } }), {
+      status: 500,
+      headers: { 'Content-Type': 'application/json' }
+    });
+  }
+}
+
+// --- 辅助函数 ---
+
+function corsHeaders() {
+  return {
+    'Access-Control-Allow-Origin': '*',
+    'Access-Control-Allow-Methods': 'GET, POST, OPTIONS',
+    'Access-Control-Allow-Headers': 'Content-Type, Authorization',
+    'Access-Control-Max-Age': '86400',
+  };
+}
+
+// --- 简单的测试 UI ---
+
+function getUI() {
+  return `<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <title>UnlimitedAI Proxy Test</title>
+    <style>
+        body { font-family: sans-serif; max-width: 800px; margin: 0 auto; padding: 20px; background: #f5f5f5; }
+        .container { background: white; padding: 20px; border-radius: 10px; box-shadow: 0 2px 10px rgba(0,0,0,0.1); }
+        textarea { width: 100%; height: 100px; margin-bottom: 10px; padding: 10px; border-radius: 5px; border: 1px solid #ccc; }
+        button { padding: 10px 20px; background: #007AFF; color: white; border: none; border-radius: 5px; cursor: pointer; }
+        button:disabled { background: #ccc; }
+        #output { margin-top: 20px; background: #333; color: #fff; padding: 15px; border-radius: 5px; min-height: 100px; white-space: pre-wrap; font-family: monospace; }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <h2>UnlimitedAI Proxy (OpenAI 兼容)</h2>
+        <p>API Endpoint: <code>http://localhost:8000/v1/chat/completions</code></p>
+        <textarea id="prompt" placeholder="输入问题...">介绍一下人工智能</textarea>
+        <button id="sendBtn" onclick="send()">发送</button>
+        <div id="output">等待响应...</div>
+    </div>
+
+    <script>
+        async function send() {
+            const btn = document.getElementById('sendBtn');
+            const out = document.getElementById('output');
+            const prompt = document.getElementById('prompt').value;
+            
+            btn.disabled = true;
+            out.textContent = '';
+            
+            try {
+                const res = await fetch('/v1/chat/completions', {
+                    method: 'POST',
+                    headers: { 'Content-Type': 'application/json' },
+                    body: JSON.stringify({
+                        model: 'chat-model-reasoning',
+                        messages: [{ role: 'user', content: prompt }],
+                        stream: true
+                    })
+                });
+
+                const reader = res.body.getReader();
+                const decoder = new TextDecoder();
+
+                while (true) {
+                    const { done, value } = await reader.read();
+                    if (done) break;
+                    const chunk = decoder.decode(value);
+                    const lines = chunk.split('\\n');
+                    
+                    for (const line of lines) {
+                        if (line.startsWith('data: ')) {
+                            const dataStr = line.substring(6);
+                            if (dataStr === '[DONE]') continue;
+                            try {
+                                const data = JSON.parse(dataStr);
+                                const content = data.choices?.[0]?.delta?.content || '';
+                                out.textContent += content;
+                            } catch (e) {}
+                        }
+                    }
+                }
+            } catch (err) {
+                out.textContent = 'Error: ' + err.message;
+            } finally {
+                btn.disabled = false;
+            }
+        }
+    </script>
+</body>
+</html>`;
+}
+
+// --- 启动服务 ---
+
+const PORT = 7860;
+console.log(`UnlimitedAI Proxy running on http://localhost:${PORT}`);
+await Deno.serve({ port: PORT }, handler);