Update app.js

app.js

@@ -1,100 +1,79 @@
 // server.js
-// Node + Express server to accept Snapshotter payloads from the Roblox plugin.
-// Save as server.js and run: node server.js
-// Installs: npm install express body-parser cors node-fetch
-
 import express from "express";
 import bodyParser from "body-parser";
 import cors from "cors";
 import fs from "fs";
 import path from "path";
 import { fileURLToPath } from "url";
-import fetch from "node-fetch";
 
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
-
-const PORT = process.env.PORT ? Number(process.env.PORT) : 7860;
-const STORAGE_DIR = path.join(__dirname, "data");
-
-if (!fs.existsSync(STORAGE_DIR)) fs.mkdirSync(STORAGE_DIR, { recursive: true });
+const PORT = process.env.PORT || 7860;
 
 const app = express();
 app.use(cors());
-// Increase limit to 50mb to handle large complex workspaces
 app.use(bodyParser.json({ limit: "50mb" }));
 
-// Logging middleware
-app.use((req, res, next) => {
-  console.log(new Date().toISOString(), req.method, req.url);
-  next();
-});
-
-// Health check
-app.get("/ping", (req, res) => res.status(200).send("pong"));
+// === DEMO CONTROLLER ===
+// CHANGE THIS TO TEST DIFFERENT INTERACTIONS
+// Options: 'NONE' (Just save), 'BUILD_PART' (Send Lua Code), 'FETCH_SCRIPT' (Ask for Source)
+const DEMO_MODE = 'BUILD_PART'; 
 
-// Main endpoint for the Plugin
 app.post("/api/ai-build", async (req, res) => {
   try {
-    const payload = req.body;
-    if (!payload) return res.status(400).json({ success: false, error: "empty payload" });
-
-    // Generate ID
-    const id = "asset_" + Math.random().toString(36).slice(2, 10);
-    const filename = path.join(STORAGE_DIR, `${id}.json`);
-
-    // Log what we received
-    console.log(`📥 Received snapshot from: ${payload.sender || "Unknown"}`);
-    console.log(`📝 Instruction: "${payload.instruction || "No instruction provided"}"`);
-    console.log(`Target Root: ${payload.selectionName}`);
+    const { instruction, state, scriptContext } = req.body;
+    
+    console.log("📥 Received Request");
+    console.log("   Instruction:", instruction);
+    
+    // 1. Log Payload Size (Check token efficiency)
+    if(state && state.flatList) {
+        console.log(`   Hierarchy Size: ${state.flatList.length} items (Flat Format)`);
+    }
 
-    // Save payload to disk (Game State + Instruction)
-    fs.writeFileSync(filename, JSON.stringify({ receivedAt: Date.now(), payload }, null, 2), "utf8");
+    // 2. IF WE RECEIVED SCRIPT SOURCE (From a previous fetch request)
+    if (scriptContext) {
+        console.log("   📘 Received Script Source for:", scriptContext.name);
+        console.log("   --- START SOURCE ---\n", scriptContext.source.substring(0, 100) + "...", "\n   --- END SOURCE ---");
+        // In a real app, you would now feed this to the AI to edit.
+        return res.json({ success: true, message: "Source received, analyzing..." });
+    }
 
-    // FUTURE: Here is where you would trigger the Gemini API call using 'payload.instruction' and 'payload.state'
+    // 3. DEMO RESPONSES
+    if (DEMO_MODE === 'BUILD_PART') {
+        console.log("   👉 Sending Code Demo");
+        const luaCode = `
+        ```lua
+        -- AI Generated Demo
+        local p = Instance.new("Part")
+        p.Name = "AI_Generated_Cube"
+        p.Color = Color3.fromRGB(0, 255, 100)
+        p.Size = Vector3.new(4, 4, 4)
+        p.Position = Vector3.new(0, 10, 0)
+        p.Anchored = true
+        p.Parent = workspace
+        print("Hello from the Server! I made a part.")
+        ```
+        `;
+        return res.send(luaCode); // Sending raw markdown/text
+    } 
     
-    // If client provided a callback URL, fire-and-forget
-    if (payload.callbackUrl && typeof payload.callbackUrl === "string") {
-      const cbBody = { assetId: id, status: "processed", ts: Date.now() };
-      (async () => {
-        try {
-          if (typeof fetch === "function") {
-            await fetch(payload.callbackUrl, {
-              method: "POST",
-              headers: { "Content-Type": "application/json" },
-              body: JSON.stringify(cbBody)
-            });
-            console.log("Callback sent to", payload.callbackUrl);
-          }
-        } catch (err) {
-          console.warn("Callback failed:", err.message);
-        }
-      })();
+    else if (DEMO_MODE === 'FETCH_SCRIPT') {
+        console.log("   👉 Sending Fetch Request Demo");
+        // This tells the plugin to find a script named "BikeLogic" and send it back
+        return res.json({
+            action: "read_script",
+            targetName: "BikeLogic" // CHANGE THIS to a script name that actually exists in your studio
+        });
     }
 
-    return res.json({ success: true, assetId: id, message: "Snapshot received successfully" });
-
-  } catch (err) {
-    console.error("/api/ai-build error:", err);
-    return res.status(500).json({ success: false, error: String(err) });
-  }
-});
+    // Default: Just acknowledge
+    return res.json({ success: true, assetId: "req_" + Date.now() });
 
-// Retrieve stored assets
-app.get("/asset/:id", (req, res) => {
-  try {
-    const id = req.params.id;
-    const filename = path.join(STORAGE_DIR, `${id}.json`);
-    if (!fs.existsSync(filename)) return res.status(404).json({ found: false });
-    const content = fs.readFileSync(filename, "utf8");
-    return res.json({ found: true, data: JSON.parse(content) });
   } catch (err) {
-    console.error("/asset error", err);
-    return res.status(500).json({ success: false, error: String(err) });
+    console.error("Server Error:", err);
+    res.status(500).send("Server Error");
   }
 });
 
-app.listen(PORT, () => {
-  console.log(`🚀 Plugin backend listening on http://localhost:${PORT}`);
-  console.log(`📂 Storage: ${STORAGE_DIR}`);
-});
+app.listen(PORT, () => console.log(`🚀 Server running on port ${PORT}. Demo Mode: ${DEMO_MODE}`));