main.js

const puppeteer = require("puppeteer");
const { speak } = require("./tts");
const https = require("https");

const videoUrl = "https://www.youtube.com/watch?v=CODE";

let continuation = null;
let clientVersion = null;
const seenMessageIds = new Set();

const messageQueue = [];
let isSpeaking = false;

async function processQueue() {
  if (isSpeaking) return;
  if (!messageQueue.length) return;

  isSpeaking = true;
  const message = messageQueue.shift();

  await new Promise((resolve) => {
    speak(message, { voice: "Microsoft Zira Desktop", speed: 1.0 });

    const duration = Math.max(message.length * 150, 3000);
    setTimeout(resolve, duration);
  });

  isSpeaking = false;
  if (messageQueue.length) processQueue();
}

async function initContinuation() {
  const browser = await puppeteer.launch({ headless: true });
  const page = await browser.newPage();
  await page.goto(videoUrl, { waitUntil: "networkidle2" });

  const html = await page.content();
  await browser.close();

  const contMatch = html.match(/"continuation":"(.*?)"/);
  const versionMatch = html.match(/"clientVersion":"(.*?)"/);

  if (!contMatch || !versionMatch) {
    console.error("Failed to get continuation or clientVersion");
    process.exit(1);
  }

  continuation = contMatch[1];
  clientVersion = versionMatch[1];

  console.log("Initialization complete");
}

function postJSON(url, data) {
  return new Promise((resolve, reject) => {
    const parsedUrl = new URL(url);
    const options = {
      hostname: parsedUrl.hostname,
      path: parsedUrl.pathname + parsedUrl.search,
      method: "POST",
      headers: {
        "Content-Type": "application/json",
        "Content-Length": Buffer.byteLength(JSON.stringify(data)),
        "User-Agent": "Mozilla/5.0",
      },
    };

    const req = https.request(options, (res) => {
      let body = "";
      res.on("data", (chunk) => (body += chunk));
      res.on("end", () => resolve(JSON.parse(body)));
    });

    req.on("error", (err) => reject(err));
    req.write(JSON.stringify(data));
    req.end();
  });
}

async function skipOldMessages() {
  const url =
    "https://www.youtube.com/youtubei/v1/live_chat/get_live_chat?prettyPrint=false";

  try {
    const payload = {
      context: { client: { clientName: "WEB", clientVersion } },
      continuation,
    };

    const data = await postJSON(url, payload);
    const actions =
      data.continuationContents.liveChatContinuation.actions || [];

    for (const action of actions) {
      try {
        const msgData =
          action.addChatItemAction.item.liveChatTextMessageRenderer;
        const messageId = msgData.id;
        seenMessageIds.add(messageId);
      } catch {}
    }

    const continuations =
      data.continuationContents.liveChatContinuation.continuations[0];
    continuation =
      continuations.invalidationContinuationData?.continuation ||
      continuations.timedContinuationData?.continuation;

    console.log(
      `Skipped ${seenMessageIds.size} old messages, now listening for new messages...`
    );
  } catch (err) {
    console.error("Error skipping old messages:", err);
  }
}

async function pollChat() {
  const url =
    "https://www.youtube.com/youtubei/v1/live_chat/get_live_chat?prettyPrint=false";

  while (true) {
    try {
      const payload = {
        context: { client: { clientName: "WEB", clientVersion } },
        continuation,
      };

      const data = await postJSON(url, payload);

      const continuations =
        data.continuationContents.liveChatContinuation.continuations[0];
      continuation =
        continuations.invalidationContinuationData?.continuation ||
        continuations.timedContinuationData?.continuation;

      const actions =
        data.continuationContents.liveChatContinuation.actions || [];
      for (const action of actions) {
        try {
          const msgData =
            action.addChatItemAction.item.liveChatTextMessageRenderer;
          const messageId = msgData.id;

          if (seenMessageIds.has(messageId)) continue;
          seenMessageIds.add(messageId);

          const author = msgData.authorName.simpleText;
          const message = msgData.message.runs[0].text;

          console.log(`${author} says: ${message}`);

          messageQueue.push(`${author} says: ${message}`);
          processQueue();
        } catch {}
      }

      await new Promise((resolve) => setTimeout(resolve, 1500));
    } catch (err) {
      console.error("Error fetching chat:", err);
      await new Promise((resolve) => setTimeout(resolve, 5000));
    }
  }
}

(async () => {
  await initContinuation();
  await skipOldMessages();
  pollChat();
})();