youtube-po-token-generator/index.js

const { createTask } = require('./lib/task')
const { fetchVisitorData } = require('./lib/workflow')
const { Redis } = require('ioredis')

const redis = new Redis({ host: process.env.HOST || config.host });
const sub = new Redis({ host: process.env.HOST || config.host });

redis.on('error', (error) => {
  console.error('Could not connect to Redis:', error);
  process.exit(1);
});

sub.on('error', (error) => {
  console.error('Could not connect to Redis:', error);
  process.exit(1);
});

sub.subscribe("generate-token", (err, count) => {
  if (err) {
    console.error("Failed to subscribe: %s", err.message);
  } else {
    console.log(
      `Subscribed successfully! This client is currently subscribed to ${count} channels.`
    );
  }
});

sub.on("message", (channel, user) => {
  console.log(`${channel}`);
  console.log(`Generating po_token and visitor_data for ${user}`);
  userGen(user);
});

const generate = async () => {
  console.debug("[DEBUG] generate called")
  try {
    const visitorData = await fetchVisitorData()
    const task = await createTask(visitorData)
    const { poToken } = await task.start()
    return { visitorData, poToken }
  } catch (e) {
    console.error(`generate: Error ${e}`)
  }
}

const run = async () => {
  try {
    const { visitorData, poToken } = await generate();
    redis.set("invidious:po_token", poToken)
    redis.set("invidious:visitor_data", visitorData)
  } catch(e) {
    console.error(`run: Error ${e}`)
  }
  console.log("New tokens sent to redis")
}

const userGen = async (user) => {
  try {
    const { visitorData, poToken } = await generate();
    redis.set(`invidious:${user}:po_token`, poToken)
    redis.set(`invidious:${user}:visitor_data`, visitorData)
  } catch(e) {
    console.error(`userGen: Error ${e}`)
  }
  console.log(`Tokens generated successfully for user: ${user}`)
}

run();
setInterval(run, (process.env.RENEW_INTERVAL || 60) * 1000)

module.exports = { generate }