import { createReadStream, promises as fs } from "fs" import express from "express" import { VideoTask, VideoSequenceRequest } from "./types.mts" import { parseVideoRequest } from "./utils/parseVideoRequest.mts" import { savePendingTask } from "./database/savePendingTask.mts" import { getTask } from "./database/getTask.mts" import { main } from "./main.mts" main() const app = express() const port = 7860 app.use(express.json()) app.post("/", async (req, res) => { const request = req.body as VideoSequenceRequest const token = `${request.token || ""}` if (token !== process.env.VS_SECRET_ACCESS_TOKEN) { console.log("couldn't find access token in the query") res.status(401) res.write(JSON.stringify({ error: "invalid token" })) res.end() return } let task: VideoTask = null console.log(`creating task from request..`) try { task = await parseVideoRequest(request) } catch (err) { console.error(`failed to create task: ${task}`) res.status(400) res.write(JSON.stringify({ error: "query seems to be malformed" })) res.end() return } console.log(`saving task ${task.id}`) try { await savePendingTask(task) res.status(200) res.write(JSON.stringify(task)) res.end() } catch (err) { console.error(err) res.status(500) res.write(JSON.stringify({ error: "couldn't save the task" })) res.end() } }) app.get("/:id", async (req, res) => { try { const task = await getTask(req.params.id) delete task.finalFilePath delete task.tmpFilePath res.status(200) res.write(JSON.stringify(task)) res.end() } catch (err) { console.error(err) res.status(404) res.write(JSON.stringify({ error: "couldn't find this task" })) res.end() } }) app.get("/video/:id\.mp4", async (req, res) => { if (!req.params.id) { res.status(400) res.write(JSON.stringify({ error: "please provide a valid video id" })) res.end() return } let task: VideoTask = null try { task = await getTask(req.params.id) console.log("returning result to user..") } catch (err) { res.status(404) res.write(JSON.stringify({ error: "this video doesn't exist" })) res.end() return } const filePath = task.finalFilePath || task.tmpFilePath || '' if (!filePath) { res.status(400) res.write(JSON.stringify({ error: "video exists, but cannot be previewed yet" })) res.end() return } // file path exists, let's try to read it try { // do we need this? // res.status(200) // res.setHeader("Content-Type", "media/mp4") console.log(`creating a video read stream from ${filePath}`) const stream = createReadStream(filePath) stream.on('close', () => { console.log(`finished streaming the video`) res.end() }) stream.pipe(res) } catch (err) { console.error(`failed to read the video file at ${filePath}: ${err}`) res.status(500) res.write(JSON.stringify({ error: "failed to read the video file" })) res.end() } }) app.listen(port, () => { console.log(`Open http://localhost:${port}`) })