codesearch-mcp 0.0.1-alpha

package/dist/src/setup.js

@@ -0,0 +1,145 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.runSetup = runSetup;
+const readline = __importStar(require("readline"));
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const os = __importStar(require("os"));
+const child_process_1 = require("child_process");
+const CONFIG_DIR = path.join(os.homedir(), '.config', 'codesearch-mcp');
+const CONFIG_FILE = path.join(CONFIG_DIR, '.env');
+const DOCKER_COMPOSE_SRC = path.join(__dirname, '..', 'docker', 'docker-compose.yml');
+const DOCKER_COMPOSE_DST = path.join(CONFIG_DIR, 'docker-compose.yml');
+function ask(rl, question) {
+    return new Promise(resolve => rl.question(question, resolve));
+}
+async function prompt(rl, label, defaultVal) {
+    const answer = await ask(rl, `${label} (${defaultVal}): `);
+    return answer.trim() || defaultVal;
+}
+function writeEnv(vars) {
+    fs.mkdirSync(CONFIG_DIR, { recursive: true });
+    const content = Object.entries(vars)
+        .map(([k, v]) => `${k}=${v}`)
+        .join('\n') + '\n';
+    fs.writeFileSync(CONFIG_FILE, content, { mode: 0o600 });
+    console.log(`\nConfig saved to ${CONFIG_FILE}`);
+}
+async function checkRemoteDb(host, port, user, pass, db) {
+    try {
+        const { Client } = await Promise.resolve().then(() => __importStar(require('pg')));
+        const client = new Client({ host, port: parseInt(port), user, password: pass, database: db, connectionTimeoutMillis: 5000 });
+        await client.connect();
+        await client.end();
+        return true;
+    }
+    catch (err) {
+        console.error(`  Connection failed: ${err.message}`);
+        return false;
+    }
+}
+async function setupLocalDocker(rl) {
+    console.log('\nCopying docker-compose.yml to', DOCKER_COMPOSE_DST);
+    fs.mkdirSync(CONFIG_DIR, { recursive: true });
+    fs.copyFileSync(DOCKER_COMPOSE_SRC, DOCKER_COMPOSE_DST);
+    console.log('\nStarting Docker containers...');
+    const upResult = (0, child_process_1.spawnSync)('docker', ['compose', '-f', DOCKER_COMPOSE_DST, 'up', '-d'], { stdio: 'inherit' });
+    if (upResult.status !== 0) {
+        console.error('Failed to start Docker containers. Make sure Docker is running.');
+        process.exit(1);
+    }
+    console.log('\nWaiting for Ollama to be ready...');
+    await new Promise(r => setTimeout(r, 5000));
+    console.log('Pulling bge-m3 model (this may take a while)...');
+    (0, child_process_1.spawnSync)('docker', ['exec', 'codesearch-mcp-ollama-1', 'ollama', 'pull', 'bge-m3'], { stdio: 'inherit' });
+    return {
+        DB_HOST: 'localhost',
+        DB_PORT: '5433',
+        DB_USER: 'dev',
+        DB_PASS: 'dev',
+        DB_NAME: 'rag',
+    };
+}
+async function setupRemoteDb(rl) {
+    let vars = {};
+    while (true) {
+        vars.DB_HOST = await prompt(rl, 'DB host', 'localhost');
+        vars.DB_PORT = await prompt(rl, 'DB port', '5432');
+        vars.DB_USER = await prompt(rl, 'DB user', 'postgres');
+        vars.DB_PASS = await prompt(rl, 'DB password', '');
+        vars.DB_NAME = await prompt(rl, 'DB name', 'rag');
+        console.log('\nTesting connection...');
+        const ok = await checkRemoteDb(vars.DB_HOST, vars.DB_PORT, vars.DB_USER, vars.DB_PASS, vars.DB_NAME);
+        if (ok) {
+            console.log('Connection successful.');
+            break;
+        }
+        const retry = await ask(rl, 'Retry? (y/n): ');
+        if (retry.trim().toLowerCase() !== 'y')
+            break;
+    }
+    return vars;
+}
+async function runSetup() {
+    const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+    console.log('=== codesearch-mcp setup ===\n');
+    const dbChoice = await ask(rl, 'Database type:\n  1. Local Docker (auto-start)\n  2. Remote (Supabase, Railway, self-hosted)\n\nEnter choice (1): ');
+    const useDocker = (dbChoice.trim() || '1') === '1';
+    const dbVars = useDocker
+        ? await setupLocalDocker(rl)
+        : await setupRemoteDb(rl);
+    const workspaceRoot = await prompt(rl, 'Workspace root path', path.join(os.homedir(), 'Projects'));
+    const ollamaHost = await prompt(rl, 'Ollama API URL', 'http://localhost:11434');
+    const ollamaModel = await prompt(rl, 'Embedding model', 'bge-m3');
+    writeEnv({
+        ...dbVars,
+        WORKSPACE_ROOT: workspaceRoot,
+        OLLAMA_HOST: ollamaHost,
+        OLLAMA_MODEL: ollamaModel,
+    });
+    rl.close();
+    console.log('\nInitializing database schema...');
+    // Re-apply env from the file we just wrote so initDbWithSchema picks up the new values
+    const dotenv = await Promise.resolve().then(() => __importStar(require('dotenv')));
+    dotenv.config({ path: CONFIG_FILE, override: true });
+    const { initDbWithSchema } = await Promise.resolve().then(() => __importStar(require('./db')));
+    await initDbWithSchema();
+    console.log('\nSetup complete!');
+    console.log('\nNext steps:');
+    console.log(`  rag-index ${workspaceRoot}`);
+    console.log('  claude mcp add codesearch-mcp -s user -- rag-mcp');
+    process.exit(0);
+}

package/dist/vitest.config.d.ts

@@ -0,0 +1,2 @@
+declare const _default: UserConfig;
+export default _default;

package/dist/vitest.config.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const config_1 = require("vitest/config");
+exports.default = (0, config_1.defineConfig)({
+    test: {
+        environment: 'node',
+        include: ['tests/**/*.test.ts'],
+    },
+});

package/docker/docker-compose.yml

@@ -0,0 +1,19 @@
+services:
+  db:
+    image: pgvector/pgvector:pg17
+    environment:
+      POSTGRES_USER: dev
+      POSTGRES_PASSWORD: dev
+      POSTGRES_DB: rag
+    ports:
+      - "5433:5432"
+
+  ollama:
+    image: ollama/ollama
+    ports:
+      - "11434:11434"
+    volumes:
+      - ollama_data:/root/.ollama
+
+volumes:
+  ollama_data:

package/package.json

@@ -0,0 +1,49 @@
+{
+  "name": "codesearch-mcp",
+  "version": "0.0.1-alpha",
+  "description": "Local semantic codebase search via MCP — indexes projects into pgvector, exposes search to Claude",
+  "main": "dist/app.js",
+  "bin": {
+    "rag-index": "dist/app.js",
+    "rag-mcp":   "dist/app.js",
+    "rag-setup": "dist/app.js"
+  },
+  "files": [
+    "dist/",
+    "docker/",
+    "README.md"
+  ],
+  "scripts": {
+    "build":           "tsc",
+    "postbuild":       "node -e \"const fs=require('fs');const f='dist/app.js';fs.writeFileSync(f,'#!/usr/bin/env node\\n'+fs.readFileSync(f,'utf8'));fs.chmodSync(f,0o755);\"",
+    "prepublishOnly":  "npm run build && npm run typecheck",
+    "start:mcp":       "ts-node app.ts mcp",
+    "index:workspace": "ts-node app.ts index",
+    "index:reindex":   "ts-node app.ts index --reindex",
+    "typecheck":       "tsc --noEmit",
+    "test":            "vitest run",
+    "test:watch":      "vitest"
+  },
+  "dependencies": {
+    "@inquirer/prompts": "^8.4.1",
+    "@langchain/textsplitters": "^0.1.0",
+    "@modelcontextprotocol/sdk": "^1.0.0",
+    "cli-progress": "^3.12.0",
+    "dotenv": "^17.4.2",
+    "ignore": "^7.0.5",
+    "ollama": "^0.5.0",
+    "pg": "^8.13.0",
+    "reflect-metadata": "^0.2.2",
+    "ts-morph": "^28.0.0",
+    "typeorm": "^0.3.20",
+    "zod": "^4.3.6"
+  },
+  "devDependencies": {
+    "@types/cli-progress": "^3.11.6",
+    "@types/node": "^22.0.0",
+    "@types/pg": "^8.20.0",
+    "ts-node": "^10.9.2",
+    "typescript": "^5.6.0",
+    "vitest": "^4.1.4"
+  }
+}