@studioindia/designx 1.0.2 → 1.1.0

package/server.cjs

@@ -46216,7 +46216,7 @@ var __dirname = import_node_path.default.dirname(__filename);
 var app = (0, import_express.default)();
 var IS_PROD = process.env.NODE_ENV === "production";
 var PORT = parseInt(process.env.PORT ?? "3001", 10);
-var DESIGNX_DATA_DIR = process.env.DESIGNX_DATA_DIR ?? import_node_path.default.join((0, import_node_os.homedir)(), ".designx");
+var DESIGNX_DATA_DIR = process.env.DESIGNX_DATA_DIR ?? import_node_path.default.join((0, import_node_os.homedir)(), "DesignX");
 var DIST_DIR = import_node_path.default.join(__dirname, IS_PROD ? "dist" : "../dist");
 var COPILOT_API_URL = "https://api.githubcopilot.com";
 var AUTO_PINNED_MODEL = "claude-sonnet-4.6";
@@ -47727,6 +47727,276 @@ app.get("/api/data/projects/:slug/export", (req, res) => {
     res.status(500).json({ error: String(err) });
   }
 });
+var AUTH_FILE = import_node_path.default.join(DESIGNX_DATA_DIR, "auth.json");
+app.get("/api/auth", (_req, res) => {
+  try {
+    if (!import_node_fs.default.existsSync(AUTH_FILE)) {
+      res.json({});
+      return;
+    }
+    res.json(readJson(AUTH_FILE, {}));
+  } catch {
+    res.json({});
+  }
+});
+app.put("/api/auth", (req, res) => {
+  try {
+    writeJson(AUTH_FILE, req.body);
+    try {
+      import_node_fs.default.chmodSync(AUTH_FILE, 384);
+    } catch {
+    }
+    res.json({ ok: true });
+  } catch (err) {
+    res.status(500).json({ error: String(err) });
+  }
+});
+app.delete("/api/auth", (_req, res) => {
+  try {
+    if (import_node_fs.default.existsSync(AUTH_FILE)) import_node_fs.default.unlinkSync(AUTH_FILE);
+    res.json({ ok: true });
+  } catch (err) {
+    res.status(500).json({ error: String(err) });
+  }
+});
+var TOKEN_STATS_FILE = import_node_path.default.join(KB_JOBS_DIR, "token-stats.json");
+app.get("/api/data/token-stats", (_req, res) => {
+  try {
+    if (!import_node_fs.default.existsSync(TOKEN_STATS_FILE)) {
+      res.json(null);
+      return;
+    }
+    res.json(readJson(TOKEN_STATS_FILE, null));
+  } catch {
+    res.json(null);
+  }
+});
+app.put("/api/data/token-stats", (req, res) => {
+  try {
+    writeJson(TOKEN_STATS_FILE, req.body);
+    res.json({ ok: true });
+  } catch (err) {
+    res.status(500).json({ error: String(err) });
+  }
+});
+app.get("/api/kb/:slug/context", (req, res) => {
+  const { slug } = req.params;
+  if (!safeSlug(slug)) {
+    res.status(400).json({ error: "Invalid slug" });
+    return;
+  }
+  try {
+    const uploadDir = import_node_path.default.join(projectDir(slug), "uploads");
+    if (!import_node_fs.default.existsSync(uploadDir)) {
+      res.json({ entries: [], totalChars: 0 });
+      return;
+    }
+    const TEXT_EXTS = /* @__PURE__ */ new Set([
+      ".txt",
+      ".md",
+      ".mdx",
+      ".json",
+      ".yaml",
+      ".yml",
+      ".csv",
+      ".html",
+      ".htm",
+      ".xml",
+      ".js",
+      ".ts",
+      ".jsx",
+      ".tsx",
+      ".py",
+      ".java",
+      ".cs",
+      ".go",
+      ".rs",
+      ".rb",
+      ".php",
+      ".sh",
+      ".bash",
+      ".zsh",
+      ".env",
+      ".conf",
+      ".ini",
+      ".toml"
+    ]);
+    const PER_FILE_CAP = 8e4;
+    const TOTAL_CAP = 2e5;
+    let totalChars = 0;
+    const entries = [];
+    for (const name of import_node_fs.default.readdirSync(uploadDir)) {
+      if (totalChars >= TOTAL_CAP) break;
+      const filePath = import_node_path.default.join(uploadDir, name);
+      if (import_node_fs.default.statSync(filePath).isDirectory()) continue;
+      const ext = import_node_path.default.extname(name).toLowerCase();
+      if (ext && !TEXT_EXTS.has(ext)) continue;
+      try {
+        const raw = import_node_fs.default.readFileSync(filePath, "utf8");
+        const limit = Math.min(PER_FILE_CAP, TOTAL_CAP - totalChars);
+        const truncated = raw.length > limit;
+        const content = truncated ? raw.slice(0, limit) : raw;
+        entries.push({ name, content, truncated });
+        totalChars += content.length;
+      } catch {
+      }
+    }
+    res.json({ entries, totalChars });
+  } catch (err) {
+    res.status(500).json({ error: String(err) });
+  }
+});
+app.post("/api/byok/chat", async (req, res) => {
+  const provider = req.headers["x-byok-provider"];
+  const apiKey = req.headers["x-byok-key"];
+  if (typeof provider !== "string" || typeof apiKey !== "string" || !provider || !apiKey) {
+    res.status(400).json({ error: "x-byok-provider and x-byok-key headers required." });
+    return;
+  }
+  if (provider !== "openai" && provider !== "anthropic") {
+    res.status(400).json({ error: 'x-byok-provider must be "openai" or "anthropic".' });
+    return;
+  }
+  const body = req.body;
+  try {
+    if (provider === "openai") {
+      const upstream2 = await fetch("https://api.openai.com/v1/chat/completions", {
+        method: "POST",
+        headers: { "Content-Type": "application/json", Authorization: `Bearer ${apiKey}` },
+        body: JSON.stringify(body),
+        signal: AbortSignal.timeout(12e4)
+      });
+      res.status(upstream2.status);
+      const ct = upstream2.headers.get("content-type");
+      if (ct) res.setHeader("Content-Type", ct);
+      if (!upstream2.body) {
+        res.end();
+        return;
+      }
+      const reader2 = upstream2.body.getReader();
+      while (true) {
+        const { done, value } = await reader2.read();
+        if (done) {
+          res.end();
+          break;
+        }
+        res.write(Buffer.from(value));
+      }
+      return;
+    }
+    const inMessages = body.messages ?? [];
+    const sysMsg = inMessages.find((m) => m.role === "system");
+    const chatMsgs = inMessages.filter((m) => m.role !== "system");
+    const anthropicBody = {
+      model: body.model ?? "claude-3-5-sonnet-20241022",
+      messages: chatMsgs,
+      max_tokens: body.max_tokens ?? 4096,
+      stream: true
+    };
+    if (sysMsg) anthropicBody.system = sysMsg.content;
+    const upstream = await fetch("https://api.anthropic.com/v1/messages", {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        "x-api-key": apiKey,
+        "anthropic-version": "2023-06-01"
+      },
+      body: JSON.stringify(anthropicBody)
+    });
+    if (!upstream.ok) {
+      const errText = await upstream.text().catch(() => "");
+      res.status(upstream.status).json({ error: errText.slice(0, 500) });
+      return;
+    }
+    res.setHeader("Content-Type", "text/event-stream");
+    res.setHeader("Cache-Control", "no-cache");
+    res.setHeader("Connection", "keep-alive");
+    if (!upstream.body) {
+      res.end();
+      return;
+    }
+    const reader = upstream.body.getReader();
+    const decoder = new TextDecoder();
+    let sseBuffer = "";
+    const msgId = (0, import_node_crypto.randomUUID)();
+    const modelId = String(body.model ?? "anthropic");
+    while (true) {
+      const { done, value } = await reader.read();
+      if (done) {
+        res.write("data: [DONE]\n\n");
+        res.end();
+        break;
+      }
+      sseBuffer += decoder.decode(value, { stream: true });
+      const lines = sseBuffer.split("\n");
+      sseBuffer = lines.pop() ?? "";
+      for (const line of lines) {
+        if (!line.startsWith("data: ")) continue;
+        const raw = line.slice(6).trim();
+        if (raw === "" || raw === "[DONE]") continue;
+        try {
+          const evt = JSON.parse(raw);
+          if (evt.type === "content_block_delta" && evt.delta?.type === "text_delta" && evt.delta.text) {
+            res.write(`data: ${JSON.stringify({ id: msgId, object: "chat.completion.chunk", model: modelId, choices: [{ index: 0, delta: { content: evt.delta.text }, finish_reason: null }] })}
+
+`);
+          } else if (evt.type === "message_delta" && evt.usage) {
+            const outTok = evt.usage.output_tokens ?? 0;
+            res.write(`data: ${JSON.stringify({ id: msgId, object: "chat.completion.chunk", model: modelId, choices: [{ index: 0, delta: {}, finish_reason: "stop" }], usage: { prompt_tokens: 0, completion_tokens: outTok, total_tokens: outTok } })}
+
+`);
+          }
+        } catch {
+        }
+      }
+    }
+  } catch (err) {
+    if (!res.headersSent) {
+      res.status(502).json({ error: err instanceof Error ? err.message : "BYOK request failed." });
+    }
+  }
+});
+app.post("/api/byok/test", async (req, res) => {
+  const provider = req.headers["x-byok-provider"];
+  const apiKey = req.headers["x-byok-key"];
+  if (typeof provider !== "string" || typeof apiKey !== "string" || !provider || !apiKey) {
+    res.status(400).json({ ok: false, error: "x-byok-provider and x-byok-key headers required." });
+    return;
+  }
+  try {
+    if (provider === "openai") {
+      const r = await fetch("https://api.openai.com/v1/chat/completions", {
+        method: "POST",
+        headers: { "Content-Type": "application/json", Authorization: `Bearer ${apiKey}` },
+        body: JSON.stringify({ model: "gpt-4o-mini", messages: [{ role: "user", content: "Hi" }], max_tokens: 1, stream: false }),
+        signal: AbortSignal.timeout(1e4)
+      });
+      if (r.ok) {
+        res.json({ ok: true });
+        return;
+      }
+      const t = await r.text().catch(() => "");
+      res.json({ ok: false, error: `OpenAI ${r.status}: ${t.slice(0, 200)}` });
+    } else if (provider === "anthropic") {
+      const r = await fetch("https://api.anthropic.com/v1/messages", {
+        method: "POST",
+        headers: { "Content-Type": "application/json", "x-api-key": apiKey, "anthropic-version": "2023-06-01" },
+        body: JSON.stringify({ model: "claude-3-5-haiku-20241022", messages: [{ role: "user", content: "Hi" }], max_tokens: 1, stream: false }),
+        signal: AbortSignal.timeout(1e4)
+      });
+      if (r.ok) {
+        res.json({ ok: true });
+        return;
+      }
+      const t = await r.text().catch(() => "");
+      res.json({ ok: false, error: `Anthropic ${r.status}: ${t.slice(0, 200)}` });
+    } else {
+      res.status(400).json({ ok: false, error: "Unknown provider." });
+    }
+  } catch (err) {
+    res.json({ ok: false, error: err instanceof Error ? err.message : "Test failed." });
+  }
+});
 if (IS_PROD) {
   app.use(import_express.default.static(DIST_DIR));
   app.use((_req, res) => {