Update index.js

index.js

@@ -34,24 +34,42 @@ app.all(
 )
 
 app.get('/ss', async (req, res) => {
-  const { url = process.env.SPACE_HOST } = req.query
+  const {
+    delay = 0,
+    full = false,
+    url = 'https://hf.co'
+  } = req.query
   const browser = await puppeteer.launch({
     executablePath: process.env.CHROME_BIN,
     headless: 'new'
   })
   try {
     const page = await browser.newPage()
-    await page.goto(url)
-    const name =
-      tmpDir + '/' +
-      Math.random().toString(36).slice(2) + '.jpg'
-    const ss = await page.screenshot({ path: name })
-    res.status(200).redirect(name)
+    await page.goto(
+      url,
+      { waitUntil: 'networkidle0'}
+    )
+    const name = format(
+      '%s/%s.png',
+      tmpDir,
+      Math.random().toString(36).slice(2)
+    )
+    if (/^\d$/.test(delay))
+      await page.waitForTimeout(+delay)
+    await page.screenshot({
+      fullPage: full,
+      path: name,
+      omitBackground: true
+    })
+    res.redirect(name)
   } catch (e) {
     console.error(e)
     res
       .status(500)
-      .json({ error: true, msg: format(e) })
+      .json({
+        err: true,
+        msg: format(e?.message || e)
+      })
   } finally {
     await browser.close()
   }