Update index.js

index.js

@@ -1,50 +1,38 @@
 const express = require("express");
 const puppeteer = require("puppeteer-core");
-const cookies = require("./cookies.json");
+const fs = require("fs");
 
 const app = express();
-const BROWSERLESS_TOKEN = "2SSAwtJD82odF6zf96a9dc3bf70bf1aeb62f89e673e3ebb5a";
-
-app.get("/", (req, res) => {
-  res.send("Visit /scrape-token to get the Bearer token.");
-});
+const PORT = process.env.PORT || 7860;
 
 app.get("/scrape-token", async (req, res) => {
   try {
-    const browser = await puppeteer.connect({
-      browserWSEndpoint: `wss://chrome.browserless.io?token=${BROWSERLESS_TOKEN}`,
+    const cookies = JSON.parse(fs.readFileSync("./cookies.json", "utf8"));
+    const browser = await puppeteer.launch({
+      headless: true,
+      executablePath: "/usr/bin/google-chrome", // required on HF Spaces
+      args: ["--no-sandbox", "--disable-setuid-sandbox"],
     });
 
     const page = await browser.newPage();
     await page.setCookie(...cookies);
+    await page.goto("https://chatglm.cn");
 
-    let bearerToken = null;
-
-    page.on("request", (request) => {
-      const headers = request.headers();
-      if (headers.authorization?.startsWith("Bearer")) {
-        bearerToken = headers.authorization;
-      }
-    });
-
-    await page.goto("https://chatglm.cn/video?lang=zh", {
-      waitUntil: "networkidle2",
-    });
+    await page.waitForFunction(() => {
+      const token = window.localStorage.getItem("token");
+      return token && token.startsWith("Bearer");
+    }, { timeout: 10000 });
 
-    await page.waitForTimeout(8000);
+    const token = await page.evaluate(() => window.localStorage.getItem("token"));
     await browser.close();
 
-    if (bearerToken) {
-      res.json({ token: bearerToken });
-    } else {
-      res.status(500).json({ error: "Bearer token not found" });
-    }
-  } catch (err) {
-    console.error(err);
-    res.status(500).json({ error: err.message });
+    res.json({ token });
+  } catch (error) {
+    console.error(error);
+    res.status(500).json({ error: "Failed to extract token" });
   }
 });
 
-app.listen(7860, () => {
-  console.log("✅ App is running on http://localhost:7860");
+app.listen(PORT, () => {
+  console.log(`Server running on http://localhost:${PORT}`);
 });