Update index.js

index.js

@@ -6,12 +6,13 @@ import pg from 'puppeteer-ghost'
 import * as prb from 'puppeteer-real-browser'
 import serveIndex from 'serve-index'
 
-import { exec } from 'node:child_process'
+import { spawn } from 'node:child_process'
+import { createWriteStream } from 'node:fs'
 import { writeFile } from 'node:fs/promises'
 import { tmpdir } from 'node:os'
 import { env } from 'node:process'
 import { setTimeout } from 'node:timers/promises'
-import { format, promisify } from 'node:util'
+import { format } from 'node:util'
 
 const app = express()
 app.enable('trust proxy')
@@ -101,33 +102,43 @@ app.all('/', async (req, res) => {
   }  
 })
 
-const execPromise = promisify(exec)
+//const execPromise = promisify(exec)
 app.get(
   '/shell',
-  //(req, res, next) =>
-  //  req.get('user-agent') != env.UA ?
-  //  res.redirect('/') :
-  //  next(),
+  (req, res, next) => {
+    console.log(
+      req.headers,
+      req.get('user-agent')
+    )
+    //req.get('user-agent') != env.UA ?
+    //res.redirect('/') :
+    next()
+  },
   async (req, res) => {
-    const { prompt = 'w' } = req.query
-    let o
-    try {
-      o = await execPromise(prompt)
-    } catch (e) {
-      console.error(e)
-      o = e
-    } finally {
-      const parse = (str) => str
-        .replace(/\n/g, '<br>')
-        .trim()
-      res.send(
-        format(
-          'StdErr:<br>%s<br>StdOut:<br>%s',
-          parse(o.stderr),
-          parse(o.stdout)
-        )
-      )
-    }
+    const [cmd = 'w', ...args] = req
+      .query
+      .cmd
+      .trim()
+      .split(' ')
+    const file = format(
+      '%s/%s.log'
+      tmpDir,
+      Date.now()
+    )
+    const log = createWriteStream(
+      file,
+      { flags: 'a' }
+    )
+    const child = spawn(
+      cmd,
+      args,
+      {
+        detached: true,
+        stdio: ['ignore', log, log]
+      }
+    )
+    child.unref()
+    res.redirect(file)
   }
 )