arela/dist/rag/server.js

AI-powered CTO with multi-agent orchestration, code summarization, visual testing (web + mobile) for blazing fast development.

import http from "node:http";
import { URL } from "node:url";
import pc from "picocolors";
import { search } from "./index.js";
/**
 * Check if port is available
 */
async function isPortAvailable(port) {
    return new Promise((resolve) => {
        const server = http.createServer();
        server.once("error", () => resolve(false));
        server.once("listening", () => {
            server.close();
            resolve(true);
        });
        server.listen(port);
    });
}
/**
 * Find available port starting from base port
 */
async function findAvailablePort(basePort, maxAttempts = 10) {
    for (let i = 0; i < maxAttempts; i++) {
        const port = basePort + i;
        if (await isPortAvailable(port)) {
            return port;
        }
        if (i > 0) {
            console.log(pc.yellow(`Port ${basePort + i - 1} in use, trying ${port}...`));
        }
    }
    throw new Error(`No available ports found in range ${basePort}-${basePort + maxAttempts - 1}`);
}
/**
 * Start RAG HTTP server for AI assistants
 */
export async function startServer(config) {
    const { cwd, port: requestedPort = 3456, model = "nomic-embed-text", autoPort = false } = config;
    // Find available port if autoPort is enabled
    let port = requestedPort;
    if (autoPort) {
        port = await findAvailablePort(requestedPort);
        if (port !== requestedPort) {
            console.log(pc.green(`✅ Using port ${port}`));
        }
    }
    const server = http.createServer(async (req, res) => {
        // CORS headers for browser/IDE access
        res.setHeader("Access-Control-Allow-Origin", "*");
        res.setHeader("Access-Control-Allow-Methods", "GET, POST, OPTIONS");
        res.setHeader("Access-Control-Allow-Headers", "Content-Type");
        // Handle preflight
        if (req.method === "OPTIONS") {
            res.writeHead(200);
            res.end();
            return;
        }
        const url = new URL(req.url || "/", `http://localhost:${port}`);
        // Health check
        if (url.pathname === "/health") {
            res.writeHead(200, { "Content-Type": "application/json" });
            res.end(JSON.stringify({ status: "ok", model, cwd }));
            return;
        }
        // Search endpoint
        if (url.pathname === "/search") {
            try {
                const query = url.searchParams.get("q") || url.searchParams.get("query");
                const topK = parseInt(url.searchParams.get("top") || "5");
                if (!query) {
                    res.writeHead(400, { "Content-Type": "application/json" });
                    res.end(JSON.stringify({ error: "Missing query parameter" }));
                    return;
                }
                const startTime = Date.now();
                const results = await search(query, { cwd, model }, topK);
                const took = Date.now() - startTime;
                const response = {
                    results,
                    query,
                    took,
                };
                res.writeHead(200, { "Content-Type": "application/json" });
                res.end(JSON.stringify(response, null, 2));
                // Log request
                console.log(pc.dim(`[${new Date().toISOString()}] Search: "${query}" (${results.length} results, ${took}ms)`));
            }
            catch (error) {
                console.error(pc.red(`Search error: ${error.message}`));
                res.writeHead(500, { "Content-Type": "application/json" });
                res.end(JSON.stringify({ error: error.message }));
            }
            return;
        }
        // 404
        res.writeHead(404, { "Content-Type": "application/json" });
        res.end(JSON.stringify({ error: "Not found" }));
    });
    return new Promise((resolve, reject) => {
        server.listen(port, () => {
            console.log(pc.bold(pc.green(`\n🚀 Arela RAG Server running\n`)));
            console.log(pc.cyan(`  Local:   http://localhost:${port}`));
            console.log(pc.dim(`  Model:   ${model}`));
            console.log(pc.dim(`  CWD:     ${cwd}`));
            console.log("");
            console.log(pc.dim("Endpoints:"));
            console.log(pc.dim(`  GET /health`));
            console.log(pc.dim(`  GET /search?q=<query>&top=5`));
            console.log("");
            console.log(pc.dim("Press Ctrl+C to stop"));
            console.log("");
            resolve(server);
        });
        server.on("error", (error) => {
            if (error.code === "EADDRINUSE") {
                console.error(pc.red(`Port ${port} is already in use`));
                console.log(pc.dim("Try a different port: npx arela serve --port 3457"));
            }
            else {
                console.error(pc.red(`Server error: ${error.message}`));
            }
            reject(error);
        });
    });
}
/**
 * Stop RAG server
 */
export async function stopServer(server) {
    return new Promise((resolve) => {
        server.close(() => {
            console.log(pc.yellow("\n✓ Server stopped"));
            resolve();
        });
    });
}
//# sourceMappingURL=server.js.map