bonecode 1.0.0

package/src/server.ts

@@ -0,0 +1,201 @@
+/**
+ * BoneCode Server Entry Point
+ *
+ * Wires together:
+ *   1. The BoneScript-generated domain routes (session, agent, rag, workspace)
+ *   2. The OpenCode v2 compatibility adapter (for existing TUI/Desktop/VS Code clients)
+ *   3. The RAG ingestion worker (background job processor)
+ *   4. The durable event bus worker
+ *   5. WebSocket server for realtime channels
+ */
+
+import "dotenv/config";
+import express from "express";
+import { createServer } from "http";
+import cors from "cors";
+import helmet from "helmet";
+import rateLimit from "express-rate-limit";
+import { compatRouter } from "../compat/opencode_adapter";
+import { ragWorker } from "./rag_worker";
+import { loadConfig } from "./config";
+
+// ─── Import generated domain routes ──────────────────────────────────────────
+// These are generated by `npm run compile` into bone/output/*/src/
+
+import { pool } from "../bone/output/session/src/db";
+import { eventBus } from "../bone/output/session/src/events";
+import { logger } from "../bone/output/session/src/logger";
+import { authMiddleware } from "../bone/output/session/src/auth";
+import { setupWebSocketServer, broadcastToChannel } from "../bone/output/session/src/websocket";
+
+// Generated routers
+import { projectsRouter } from "../bone/output/session/src/routes/project";
+import { sessionsRouter } from "../bone/output/session/src/routes/session";
+import { messagesRouter } from "../bone/output/session/src/routes/message";
+import { partsRouter } from "../bone/output/session/src/routes/part";
+import { permissionsRouter } from "../bone/output/session/src/routes/permission";
+
+import { agentInstancesRouter } from "../bone/output/agent/src/routes/agent_instance";
+import { tasksRouter } from "../bone/output/agent/src/routes/task";
+import { toolCallsRouter } from "../bone/output/agent/src/routes/tool_call";
+import { plansRouter } from "../bone/output/agent/src/routes/plan";
+import { buildStepsRouter } from "../bone/output/agent/src/routes/build_step";
+
+import { knowledgeBasesRouter } from "../bone/output/rag/src/routes/knowledge_base";
+import { codeFilesRouter } from "../bone/output/rag/src/routes/code_file";
+import { codeChunksRouter } from "../bone/output/rag/src/routes/code_chunk";
+import { memoryEntrysRouter } from "../bone/output/rag/src/routes/memory_entry";
+import { indexingJobsRouter } from "../bone/output/rag/src/routes/indexing_job";
+
+import { workspacesRouter } from "../bone/output/workspace/src/routes/workspace";
+import { snapshotsRouter } from "../bone/output/workspace/src/routes/snapshot";
+import { codebasesRouter } from "../bone/output/workspace/src/routes/codebase";
+
+// ─── App Setup ────────────────────────────────────────────────────────────────
+
+const config = loadConfig();
+const app = express();
+const httpServer = createServer(app);
+const PORT = config.port;
+
+// ─── Middleware ───────────────────────────────────────────────────────────────
+
+app.use(helmet());
+app.set("trust proxy", 1); // trust first proxy for rate limiting
+
+const allowedOrigins = config.allowedOrigins;
+app.use(cors({
+  origin: allowedOrigins.length > 0
+    ? (origin, cb) => {
+        if (!origin || allowedOrigins.includes(origin)) cb(null, true);
+        else cb(new Error("Not allowed by CORS"));
+      }
+    : false,
+  credentials: true,
+}));
+
+app.use(express.json({ limit: "4mb" }));
+app.use(express.urlencoded({ extended: false, limit: "1mb" }));
+app.use(authMiddleware); // populate req.auth for all routes
+
+// Global rate limit
+app.use(rateLimit({ windowMs: 60_000, max: 1000, standardHeaders: true, legacyHeaders: false }));
+
+// Request timeout
+app.use((req: any, res: any, next: any) => {
+  const timer = setTimeout(() => {
+    if (!res.headersSent) {
+      res.status(503).json({ error: { code: "REQUEST_TIMEOUT", message: "Request timed out" } });
+    }
+  }, config.requestTimeoutMs);
+  res.on("finish", () => clearTimeout(timer));
+  next();
+});
+
+// ─── WebSocket server (generated by BoneScript) ───────────────────────────────
+setupWebSocketServer(httpServer);
+
+// ─── OpenCode v2 Compatibility Routes (must come first — clients use these) ──
+
+app.use(compatRouter);
+
+// ─── BoneCode Domain Routes ───────────────────────────────────────────────────
+
+// Session domain
+app.use("/projects", projectsRouter);
+app.use("/sessions", sessionsRouter);
+app.use("/messages", messagesRouter);
+app.use("/parts", partsRouter);
+app.use("/permissions", permissionsRouter);
+
+// Agent domain
+app.use("/agents", agentInstancesRouter);
+app.use("/tasks", tasksRouter);
+app.use("/tool-calls", toolCallsRouter);
+app.use("/plans", plansRouter);
+app.use("/build-steps", buildStepsRouter);
+
+// RAG domain
+app.use("/knowledge-bases", knowledgeBasesRouter);
+app.use("/code-files", codeFilesRouter);
+app.use("/code-chunks", codeChunksRouter);
+app.use("/memory-entries", memoryEntrysRouter);
+app.use("/indexing-jobs", indexingJobsRouter);
+
+// Workspace domain
+app.use("/workspaces", workspacesRouter);
+app.use("/snapshots", snapshotsRouter);
+app.use("/codebases", codebasesRouter);
+
+// ─── 404 Handler ─────────────────────────────────────────────────────────────
+
+app.use((_req: any, res: any) => {
+  res.status(404).json({ error: { code: "NOT_FOUND", message: "Route not found" } });
+});
+
+// ─── Error Handler ────────────────────────────────────────────────────────────
+
+app.use((err: any, _req: any, res: any, _next: any) => {
+  logger.error("unhandled_error", { event: "error", metadata: { message: err.message, stack: err.stack } });
+  res.status(500).json({ error: { code: "INTERNAL_ERROR", message: "Internal server error" } });
+});
+
+// ─── Start ────────────────────────────────────────────────────────────────────
+
+httpServer.listen(PORT, () => {
+  logger.info("server_started", { event: "startup", metadata: { port: PORT, node_env: process.env.NODE_ENV } });
+
+  console.log(`\n[BoneCode] Server running on port ${PORT}`);
+  console.log(`  OpenCode v2 API:  http://localhost:${PORT}/v2/session`);
+  console.log(`  Health:           http://localhost:${PORT}/health`);
+  console.log(`  Domain routes:    /sessions /agents /knowledge-bases /workspaces`);
+  console.log(`  WebSocket:        ws://localhost:${PORT}/ws`);
+
+  // Start durable event worker (no-op in in_process mode)
+  eventBus.startWorker(config.eventWorkerIntervalMs);
+
+  // Bridge eventBus events → WebSocket channels
+  // session_events channel: session lifecycle + message events
+  const sessionEventTypes = ["SessionCreated", "SessionStateChanged", "MessageAdded", "SessionCompacted"];
+  for (const type of sessionEventTypes) {
+    eventBus.subscribe(type, async (event) => {
+      broadcastToChannel("session_events", { type: type.toLowerCase().replace(/([A-Z])/g, ".$1").slice(1), ...event.payload });
+    });
+  }
+  // part_stream channel: realtime part deltas for live streaming
+  eventBus.subscribe("PartUpdated", async (event) => {
+    broadcastToChannel("part_stream", { type: "part.updated", ...event.payload });
+  });
+  // agent_events: tool calls, agent state changes
+  const agentEventTypes = ["AgentStarted", "AgentCompleted", "AgentFailed", "ToolCallRequested", "ToolCallCompleted", "PermissionRequired"];
+  for (const type of agentEventTypes) {
+    eventBus.subscribe(type, async (event) => {
+      broadcastToChannel("session_events", { type: type.toLowerCase().replace(/([A-Z])/g, ".$1").slice(1), ...event.payload });
+    });
+  }
+
+  // Start RAG ingestion worker
+  ragWorker.start();
+});
+
+// ─── Graceful Shutdown ────────────────────────────────────────────────────────
+
+const shutdown = async (signal: string) => {
+  logger.info("server_stopping", { event: "shutdown", metadata: { signal } });
+  ragWorker.stop();
+  httpServer.close(async () => {
+    try {
+      await pool.end();
+      logger.info("server_stopped", { event: "shutdown", status: "success" });
+    } catch (e: any) {
+      logger.error("shutdown_error", { event: "shutdown", metadata: { error: e.message } });
+    }
+    process.exit(0);
+  });
+  setTimeout(() => process.exit(1), 10_000);
+};
+
+process.on("SIGTERM", () => shutdown("SIGTERM"));
+process.on("SIGINT", () => shutdown("SIGINT"));
+
+export { app, httpServer };