Update index.js

index.js

@@ -1,17 +1,21 @@
+import { format as formatBytes } from 'bytes'
 import express from 'express'
+import { fileTypeFromBuffer } from 'file-type'
 import morgan from 'morgan'
 import pg from 'puppeteer-ghost'
 import * as prb from 'puppeteer-real-browser'
 import serveIndex from 'serve-index'
 
+import { writeFile } from 'node:fs/promises'
 import { tmpdir } from 'node:os'
 import { env } from 'node:process'
 import { setTimeout } from 'node:timers/promises'
 import { format } from 'node:util'
 
 const app = express()
+app.enable('trust proxy')
 
-const limitSize = '200mb'
+const limitSize = '1000mb'
 app.set('json spaces', 2)
 app.use(express.json({ limit: limitSize }))
 app.use(express.urlencoded({
@@ -30,10 +34,55 @@ app.use(
   )
 )
 
-app.all(
-  '/',
-  (_, res) => res.json({ msg: 'Hello World' })
-)
+const isBase64 = (str) => {
+  try {
+    return btoa(atob(str)) === str
+  } catch {
+    return false
+  }
+}
+
+app.all('/', async (req, res) => {
+  if (req.method !== 'POST') return res
+    .status(405)
+    .json({ msg: 'Hello World' })
+
+  const { file } = req.body
+  if (!isBase64(file)) return res
+    .status(400)
+    .json({ msg: 'Bad Request' })
+
+  const buffer = Buffer.from(file, 'base64')
+  const type = await fileTypeFromBuffer(buffer) ||
+    { mime: 'file', ext: 'bin' }
+
+  const name = format(
+    '%s/%s-%s.%s',
+    tmpDir,
+    type.mime.split('/')[0],
+    Math.random().toString(36).slice(2),
+    type.ext
+  )
+  await writeFile(name, buffer)
+
+  const bytes = buffer.length
+  res.json({
+    name: name.split('/').pop(),
+    size: {
+      bytes,
+      readable: formatBytes(
+        bytes,
+        { unitSeparator: ' ' }
+      )
+    },
+    type,
+    url: format(
+      '%s://%s',
+      req.protocol,
+      env.SPACE_HOST + name
+    )
+  })   
+}
 
 const executablePath = env.CHROME_BIN
 app.get('/ss', async (req, res) => {