@kiyeonjeon21/datacontext 0.3.1 → 0.3.3

package/src/cli/index.ts

@@ -34,9 +34,14 @@ function createAdapter(connectionString: string): DatabaseAdapter {
     return createPostgresAdapter(connectionString);
   } else if (url.startsWith('mysql://') || url.startsWith('mariadb://')) {
     return createMySQLAdapter(connectionString);
-  } else if (url.startsWith('sqlite://') || url.endsWith('.sqlite') || url.endsWith('.sqlite3') || url.endsWith('.db') || url === ':memory:') {
-    // SQLite: sqlite://path/to/db.sqlite or ./mydb.sqlite or :memory:
-    const filePath = url.startsWith('sqlite://') ? url.replace('sqlite://', '') : connectionString;
+  } else if (url.startsWith('sqlite://') || url.startsWith('sqlite:') || url.endsWith('.sqlite') || url.endsWith('.sqlite3') || url.endsWith('.db') || url === ':memory:') {
+    // SQLite: sqlite://path or sqlite:path or ./mydb.sqlite or :memory:
+    let filePath = connectionString;
+    if (url.startsWith('sqlite://')) {
+      filePath = connectionString.replace(/^sqlite:\/\//i, '');
+    } else if (url.startsWith('sqlite:')) {
+      filePath = connectionString.replace(/^sqlite:/i, '');
+    }
     return createSQLiteAdapter(filePath);
   } else {
     throw new Error(
@@ -65,20 +70,37 @@ program
   .option('--allowed-tables <tables...>', 'Only allow access to these tables')
   .action(async (connectionString: string, options) => {
     try {
-      // Parse connection string
-      const config = parseConnectionString(connectionString);
-      
       // Create adapter (auto-detect database type)
       const adapter = createAdapter(connectionString);
       await adapter.connect();
 
+      // Determine database ID for knowledge store
+      let databaseId: string;
+      const url = connectionString.toLowerCase();
+      
+      if (url.startsWith('sqlite://') || url.startsWith('sqlite:') || url.endsWith('.sqlite') || url.endsWith('.sqlite3') || url.endsWith('.db') || url === ':memory:') {
+        // SQLite: use filename as database ID
+        let filePath = connectionString;
+        if (url.startsWith('sqlite://')) {
+          filePath = connectionString.replace(/^sqlite:\/\//i, '');
+        } else if (url.startsWith('sqlite:')) {
+          filePath = connectionString.replace(/^sqlite:/i, '');
+        }
+        const fileName = filePath.split('/').pop() || 'sqlite';
+        databaseId = `sqlite_${fileName.replace(/[^a-zA-Z0-9]/g, '_')}`;
+      } else {
+        // URL-based databases (postgres, mysql)
+        const config = parseConnectionString(connectionString);
+        databaseId = `${config.host}_${config.database}`;
+      }
+
       // Create knowledge store
-      const databaseId = `${config.host}_${config.database}`;
       const knowledge = createKnowledgeStore(databaseId);
       await knowledge.load();
 
-      // Update schema hash
-      const schema = await adapter.getSchema(options.schema);
+      // Update schema hash (SQLite uses 'main' schema)
+      const schemaName = url.startsWith('sqlite') || url.endsWith('.db') ? 'main' : options.schema;
+      const schema = await adapter.getSchema(schemaName);
       knowledge.updateSchemaHash(schema.schemaHash);
       await knowledge.save();
 
@@ -836,21 +858,40 @@ program
     try {
       console.log('[datacontext] Starting REST API server...');
       
-      // Parse connection string
-      const config = parseConnectionString(connectionString);
-      
       // Create adapter (auto-detect database type)
       const adapter = createAdapter(connectionString);
       await adapter.connect();
-      console.log(`[datacontext] Connected to ${config.database}`);
+      
+      // Determine database ID for knowledge store
+      let databaseId: string;
+      const url = connectionString.toLowerCase();
+      
+      if (url.startsWith('sqlite://') || url.startsWith('sqlite:') || url.endsWith('.sqlite') || url.endsWith('.sqlite3') || url.endsWith('.db') || url === ':memory:') {
+        // SQLite: use filename as database ID
+        let filePath = connectionString;
+        if (url.startsWith('sqlite://')) {
+          filePath = connectionString.replace(/^sqlite:\/\//i, '');
+        } else if (url.startsWith('sqlite:')) {
+          filePath = connectionString.replace(/^sqlite:/i, '');
+        }
+        const fileName = filePath.split('/').pop() || 'sqlite';
+        databaseId = `sqlite_${fileName.replace(/[^a-zA-Z0-9]/g, '_')}`;
+        console.log(`[datacontext] Connected to SQLite: ${filePath}`);
+      } else {
+        // URL-based databases (postgres, mysql)
+        const config = parseConnectionString(connectionString);
+        databaseId = `${config.host}_${config.database}`;
+        console.log(`[datacontext] Connected to ${config.database}`);
+      }
 
       // Create knowledge store
-      const databaseId = `${config.host}_${config.database}`;
       const knowledge = createKnowledgeStore(databaseId);
       await knowledge.load();
 
-      // Update schema hash
-      const schema = await adapter.getSchema(options.schema);
+      // Update schema hash (SQLite uses 'main' schema)
+      const isSqlite = url.startsWith('sqlite') || url.endsWith('.db') || url.endsWith('.sqlite') || url.endsWith('.sqlite3');
+      const schemaName = isSqlite ? 'main' : options.schema;
+      const schema = await adapter.getSchema(schemaName);
       knowledge.updateSchemaHash(schema.schemaHash);
       await knowledge.save();
 

package/src/knowledge/store.ts

@@ -48,7 +48,14 @@ export class KnowledgeStore {
     if (existsSync(this.dataPath)) {
       try {
         const content = await readFile(this.dataPath, 'utf-8');
-        this.data = JSON.parse(content);
+        const parsed = JSON.parse(content) as KnowledgeData;
+        
+        // Migration: ensure businessTerms array exists (for old data files)
+        if (!parsed.businessTerms) {
+          parsed.businessTerms = [];
+        }
+        
+        this.data = parsed;
       } catch (error) {
         console.error(`Failed to load knowledge store: ${error}`);
         this.data = this.createEmptyData();
@@ -387,6 +394,11 @@ export class KnowledgeStore {
       throw new Error('Knowledge store not loaded');
     }
 
+    // Initialize businessTerms if not exists (migration for old data)
+    if (!this.data.businessTerms) {
+      this.data.businessTerms = [];
+    }
+
     // Check for duplicates
     const existing = this.data.businessTerms.find(
       t => t.term.toLowerCase() === term.toLowerCase()

package/cursor-mcp-config.json.example

@@ -1,29 +0,0 @@
-{
-  "mcpServers": {
-    "datacontext": {
-      "command": "node",
-      "args": [
-        "/Users/kiyeonjeon/Projects/business/DataContext-HQ/datacontext-cli/dist/cli/index.js",
-        "connect",
-        "sqlite:///Users/kiyeonjeon/Projects/business/DataContext-HQ/datacontext-cli/test-mcp.db"
-      ],
-      "env": {
-        "ANTHROPIC_API_KEY": "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",
-        "NODE_PATH": "/Users/kiyeonjeon/Projects/business/DataContext-HQ/datacontext-cli/node_modules"
-      }
-    }
-  }
-}
-
-
-"datacontext": {
-    "command": "node",
-    "args": [
-    "/Users/kiyeonjeon/Projects/business/DataContext-HQ/datacontext-cli/dist/cli/index.js",
-    "connect",
-    "postgres://postgres:postgres@localhost:5432/datacontext_test"
-    ],
-    "env": {
-    "NODE_PATH": "/Users/kiyeonjeon/Projects/business/DataContext-HQ/datacontext-cli/node_modules"
-    }
-}