Update app.js

app.js

@@ -10,6 +10,8 @@ const NOTDIAMOND_HEADERS = {
   'Content-Type': 'application/json',
   'next-action': '8189eb37107121e024940f588629a394a594e6a4'
 };
+const SUPABASE_URL = 'https://spuckhogycrxcbomznwo.supabase.co';
+const SUPABASE_KEY = 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6InNwdWNraG9neWNyeGNib216bndvIiwicm9sZSI6ImFub24iLCJpYXQiOjE3MDcyNDYwMzksImV4cCI6MjAyMjgyMjAzOX0.tvlGT7NZY8bijMjNIu1WhAtPnSKuDeYhtveo4DRt6xg';
 
 const DEFAULT_MODEL = 'gpt-4-turbo';
 
@@ -60,10 +62,42 @@ const MODEL_MAPPING = {
     }
 };
 
-function getAuthKey(req) {
+async function getAuthCookie(req) {
   const authHeader = req.headers['authorization'];
   if (authHeader && authHeader.startsWith('Bearer ')) {
-    return authHeader.slice(7);
+    const [email, password] = authHeader.slice(7).split('|');
+    if (!email || !password) {
+      throw new Error('Invalid authentication format');
+    }
+
+    const response = await fetch(`${SUPABASE_URL}/auth/v1/token?grant_type=password`, {
+      method: 'POST',
+      headers: {
+        'accept': '*/*',
+        'accept-language': 'zh-CN,zh;q=0.9',
+        'apikey': SUPABASE_KEY,
+        'authorization': `Bearer ${SUPABASE_KEY}`,
+        'content-type': 'application/json;charset=UTF-8',
+        'origin': 'https://chat.notdiamond.ai',
+        'referer': 'https://chat.notdiamond.ai/',
+        'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36',
+        'x-client-info': 'supabase-ssr/0.4.0',
+        'x-supabase-api-version': '2024-01-01'
+      },
+      body: JSON.stringify({
+        email: email,
+        password: password,
+        gotrue_meta_security: {}
+      })
+    });
+
+    if (!response.ok) {
+      throw new Error('Authentication failed');
+    }
+
+    const responseData = await response.text();
+    const baseCookie = Buffer.from(responseData).toString('base64');
+    return `sb-spuckhogycrxcbomznwo-auth-token=base64-${baseCookie}`;
   }
   return null;
 }
@@ -188,26 +222,30 @@ app.get('/', (req, res) => {
   });
 });
 
-app.get('/ai/v1/models', (req, res) => {
-  const authKey = getAuthKey(req);
-  if (!authKey) {
-    return res.status(401).json({ error: 'Unauthorized' });
-  }
+app.get('/ai/v1/models', async (req, res) => {
+  try {
+    const authCookie = await getAuthCookie(req);
+    if (!authCookie) {
+      return res.status(401).json({ error: 'Unauthorized' });
+    }
 
-  res.json({
-    object: "list",
-    data: Object.entries(MODEL_MAPPING).map(([id, info]) => ({
-      id: id,
-      object: "model",
-      created: Math.floor(Date.now() / 1000),
-      owned_by: info.provider,
-      permission: [],
-      root: id,
-      parent: null,
-    }))
-  });
+    res.json({
+      object: "list",
+      data: Object.entries(MODEL_MAPPING).map(([id, info]) => ({
+        id: id,
+        object: "model",
+        created: Math.floor(Date.now() / 1000),
+        owned_by: info.provider,
+        permission: [],
+        root: id,
+        parent: null,
+      }))
+    });
 
-  console.log(`${new Date().toISOString()} - ${req.ip || req.connection.remoteAddress} - GET /ai/v1/models - 200`);
+    console.log(`${new Date().toISOString()} - ${req.ip || req.connection.remoteAddress} - GET /ai/v1/models - 200`);
+  } catch (error) {
+    res.status(401).json({ error: 'Authentication failed', details: error.message });
+  }
 });
 
 app.post('/ai/v1/chat/completions', async (req, res) => {
@@ -217,8 +255,8 @@ app.post('/ai/v1/chat/completions', async (req, res) => {
   let status = 200;
 
   try {
-    const authKey = getAuthKey(req);
-    if (!authKey) {
+    const authCookie = await getAuthCookie(req);
+    if (!authCookie) {
       status = 401;
       throw new Error('Unauthorized');
     }
@@ -251,7 +289,7 @@ app.post('/ai/v1/chat/completions', async (req, res) => {
     const headers = {
       'Content-Type': 'application/json',
       'next-action': '4e63dabc37fef18cae74cbfd41d1bace49acf47e',
-      'Authorization': `Bearer ${authKey}`
+      'Cookie': authCookie
     };
 
     const response = await fetch(NOTDIAMOND_URL, {