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 });

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

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")
}

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

module.exports = { generate }