Update app/api/run/route.ts

app/api/run/route.ts

@@ -1,39 +1,34 @@
 import { NextResponse } from 'next/server';
 import { exec } from 'child_process';
 import { promisify } from 'util';
-import path from 'path';
 
 const execAsync = promisify(exec);
 
 export async function POST(req: Request) {
   try {
-    const { inputData } = await req.json();
+    const { command } = await req.json();
 
-    // 1. Locate the script (Needs to work both locally and in Docker)
-    // process.cwd() is usually /app in Docker
-    const scriptPath = path.join(process.cwd(), 'scripts', 'logic.py');
-
-    // 2. The Move 37 Command: Next.js executing Python
-    // We sanitize inputData to avoid shell injection (basic level)
-    const safeInput = inputData.replace(/[^a-zA-Z0-9 ]/g, ""); 
-    const command = `python3 ${scriptPath} "${safeInput}"`;
-
-    // 3. Run the shell command
-    const { stdout, stderr } = await execAsync(command);
-
-    if (stderr) {
-      console.error('Python Error:', stderr);
+    if (!command) {
+      return NextResponse.json({ error: "No command provided" }, { status: 400 });
     }
 
-    // 4. Parse output and return
-    const result = JSON.parse(stdout);
+    // SECURITY WARNING: This allows executing ANY command on the container.
+    // Ideally, for a real app, you would whitelist commands here.
+    
+    // Execute the custom command in the shell
+    // timeout: 10000ms (10 seconds) to prevent hanging
+    const { stdout, stderr } = await execAsync(command, { timeout: 10000 });
 
     return NextResponse.json({
-      server: "Next.js Controller",
-      python_response: result
+      output: stdout || stderr, // Return whichever has content
+      type: stderr ? "stderr" : "stdout"
     });
 
   } catch (error: any) {
-    return NextResponse.json({ error: error.message }, { status: 500 });
+    // If the command fails (e.g. syntax error in shell), return that
+    return NextResponse.json({ 
+      output: error.message || error.stderr || "Unknown Error",
+      type: "error"
+    });
   }
 }