Update server.js

server.js

@@ -1,10 +1,19 @@
 const express = require("express");
 const { chromium } = require("playwright");
 const bodyParser = require("body-parser");
+const fs = require("fs");
+const path = require("path");
+const os = require("os");
 
 const app = express();
 app.use(bodyParser.text({ type: "text/html" }));
 
+const tmpDir = os.tmpdir();
+
+function randomFilename() {
+  return Math.random().toString(36).slice(2, 6) + ".png";
+}
+
 app.post("/v1/start", async (req, res) => {
   const html = req.body;
   if (!html) return res.status(400).send("HTML required");
@@ -12,11 +21,31 @@ app.post("/v1/start", async (req, res) => {
   const browser = await chromium.launch({ args: ["--no-sandbox"] });
   const page = await browser.newPage();
   await page.setContent(html);
-  const screenshot = await page.screenshot({ type: "png" });
+
+  const filename = randomFilename();
+  const filepath = path.join(tmpDir, filename);
+  await page.screenshot({ path: filepath });
 
   await browser.close();
-  res.setHeader("Content-Type", "image/png");
-  res.send(screenshot);
+
+  res.json({ url: `${req.protocol}://${req.get("host")}/t/${filename}` });
 });
 
-app.listen(7860, () => console.log("Server running on port 7860"));
+app.use("/t", (req, res, next) => {
+  fs.readdir(tmpDir, (err, files) => {
+    if (err) return next();
+    const now = Date.now();
+    files.forEach((file) => {
+      const filePath = path.join(tmpDir, file);
+      fs.stat(filePath, (err, stats) => {
+        if (err) return;
+        if (now - stats.ctimeMs > 5 * 60 * 1000) {
+          fs.unlink(filePath, () => {});
+        }
+      });
+    });
+  });
+  next();
+}, express.static(tmpDir));
+
+app.listen(process.env.PORT || 7860);