index.js

const express = require("express");
const puppeteer = require("puppeteer-core");
const fs = require("fs");

const app = express();
const PORT = process.env.PORT || 7860;

app.get("/scrape-token", async (req, res) => {
  try {
    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");

    await page.waitForFunction(() => {
      const token = window.localStorage.getItem("token");
      return token && token.startsWith("Bearer");
    }, { timeout: 10000 });

    const token = await page.evaluate(() => window.localStorage.getItem("token"));
    await browser.close();

    res.json({ token });
  } catch (error) {
    console.error(error);
    res.status(500).json({ error: "Failed to extract token" });
  }
});

app.listen(PORT, () => {
  console.log(`Server running on http://localhost:${PORT}`);
});