youtube-po-token-generator/index.js
Fijxu 7c3f7294cc
Some checks failed
Docker build image / build (push) Failing after 56s
use http instead of redis
2024-12-14 17:24:29 -03:00

65 lines
1.7 KiB
JavaScript

const { env } = require('process');
const { createTask } = require('./lib/task')
const { fetchVisitorData } = require('./lib/workflow')
const http = require('http');
const port = env.PORT || "3000"
const address = env.ADDRESS || "0.0.0.0"
let tokens = {};
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 refreshTokens = async () => {
console.log("[DEBUG] generateTokens called")
try {
const { visitorData, poToken } = await generate();
tokens.visitorData = visitorData;
tokens.poToken = poToken;
} catch (e) {
console.error(`refreshTokens: Error ${e}`)
}
}
const server = http.Server(async (req, res) => {
switch (req.url) {
case "/generate":
if (tokens.visitorData == undefined || tokens.poToken == undefined) {
res.writeHead(500)
return res.end()
}
const json = {
potoken: tokens.poToken,
visitorData: tokens.visitorData,
};
res.writeHead(200, { 'Content-Type': 'text/json' })
res.write(JSON.stringify(json))
return res.end()
case "/update":
await refreshTokens();
res.writeHead(200, { 'Content-Type': 'text/plain' })
res.write("Tokens updated")
return res.end()
default:
return res.end()
}
})
refreshTokens();
setInterval(refreshTokens, (env.RENEW_INTERVAL || 60) * 1000)
server.listen(port, address, () => {
console.log("Listening at port", port);
})
module.exports = { generate }