cohere-db 1.0.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Nyan Lin Maung
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,383 @@
+# Cohere
+
+> AI-powered database schema extraction and documentation generator for Claude, Cursor, and other AI coding assistants
+
+[![npm version](https://img.shields.io/npm/v/cohere-db.svg)](https://www.npmjs.com/package/cohere-db)
+[![License: MIT](https://img.shields.io/badge/License-MIT-blue.svg)](LICENSE)
+
+## The Problem
+
+AI coding assistants struggle with databases because they lack context about:
+- **Schema structure** - Tables, columns, types, and constraints
+- **Relationships** - Foreign keys, joins, and cardinalities
+- **Data types** - Proper mapping between SQL and your ORM
+- **Valid queries** - Common patterns and edge cases
+
+This leads to:
+- ❌ Incorrect SQL generation (wrong table/column names)
+- ❌ Missing relationships (N+1 query problems)
+- ❌ Type mismatches and runtime errors
+- ❌ Context drift in long conversations
+
+## The Solution
+
+`cohere` extracts database schema information and generates AI-optimized context files:
+- `CLAUDE.md` - Structured context for Claude Code
+- `AGENTS.md` - Context for Cursor/Windsurf
+- `DATABASE.md` - Human-readable schema documentation
+- Type definitions and query templates
+
+## Features
+
+### 🎯 Core Commands
+
+```bash
+# Initialize in your project
+cohere init
+
+# Generate context from your database
+cohere generate --url "postgresql://user:pass@localhost:5432/mydb"
+
+# Generate from ORM schema files
+cohere generate --prisma
+cohere generate --drizzle
+
+# Watch for schema changes and auto-regenerate
+cohere watch
+
+# Validate generated context matches database
+cohere validate
+
+# Show current database state
+cohere show
+```
+
+### 🗄️ Supported Databases
+
+| Database | Status | Notes |
+|----------|--------|-------|
+| PostgreSQL | ✅ | Full support with indexes, FKs, constraints |
+| MySQL | ✅ | Full support with InnoDB features |
+| SQLite | ✅ | Full support with PRAGMA introspection |
+| MongoDB | ✅ | Document sampling and type inference |
+| Firebase Firestore | ✅ | Collection sampling with field detection |
+
+### 🛠️ Supported ORMs & Schema Tools
+
+| Tool | Status | Notes |
+|------|--------|-------|
+| Prisma | ✅ | Parse `schema.prisma` files |
+| Drizzle ORM | ✅ | Parse schema files |
+| Kysely | 🔄 | Planned |
+| TypeORM | 🔄 | Planned |
+
+## Installation
+
+```bash
+npm install -g cohere-db
+```
+
+Or use with npx:
+```bash
+npx cohere-db generate --url "postgresql://..."
+```
+
+## Quick Start
+
+### 1. Initialize
+```bash
+cd my-project
+cohere init
+```
+
+### 2. Generate Context
+
+**From a Live Database:**
+```bash
+cohere generate --url "postgresql://localhost:5432/mydb"
+```
+
+**From Prisma Schema:**
+```bash
+cohere generate --prisma
+```
+
+**From Drizzle Schema:**
+```bash
+cohere generate --drizzle src/db/schema.ts
+```
+
+### 3. Generated Files
+
+After running `generate`, you'll have:
+
+```
+.ai/
+├── CLAUDE.md       # Claude-optimized context
+├── AGENTS.md       # Cursor/Windsurf context
+├── DATABASE.md     # Human-readable docs
+└── queries/        # Example query templates
+```
+
+### 4. Use with AI Tools
+
+**Claude Code:**
+- Claude automatically reads `.ai/CLAUDE.md` in your project root
+
+**Cursor/Windsurf:**
+- Add `.ai/AGENTS.md` to your `.cursorrules` or `.windsurfrules`
+
+**Gemini CLI:**
+- Reference `.ai/DATABASE.md` in your context
+
+## Example Output
+
+```markdown
+<!-- CLAUDE.md excerpt -->
+## Database Schema
+
+### users
+**Type:** table  
+**Engine:** InnoDB (MySQL)
+
+#### Columns
+- `id` - int, PRIMARY KEY, AUTO_INCREMENT
+- `email` - varchar(255), UNIQUE, NOT NULL
+- `full_name` - varchar(255), NULL
+- `organization_id` - int, FOREIGN KEY → organizations.id
+- `created_at` - timestamp, DEFAULT CURRENT_TIMESTAMP
+
+#### Indexes
+- PRIMARY KEY (`id`)
+- UNIQUE INDEX `idx_email` (`email`)
+- INDEX `idx_org` (`organization_id`)
+
+#### Foreign Keys
+- `fk_user_org` → organizations(id) ON DELETE CASCADE
+
+### Relationships
+- users → organizations (many-to-one)
+- users → posts (one-to-many)
+
+### Query Examples
+
+**Get user with organization:**
+\`\`\`sql
+SELECT u.*, o.name as org_name
+FROM users u
+JOIN organizations o ON u.organization_id = o.id
+WHERE u.id = ?
+\`\`\`
+```
+
+## Advanced Usage
+
+### Watch Mode
+
+Automatically regenerate when schema changes:
+```bash
+cohere watch
+```
+
+This monitors:
+- Prisma schema files
+- Database migrations
+- Drizzle schema files
+
+### MongoDB Sampling
+
+For NoSQL databases, Cohere samples documents to infer schema:
+
+```bash
+cohere generate --url "mongodb://localhost:27017/mydb" --sample-size 1000
+```
+
+### Firebase/Firestore
+
+Requires service account credentials:
+
+```bash
+export GOOGLE_APPLICATION_CREDENTIALS="path/to/service-account.json"
+cohere generate --firebase my-project-id
+```
+
+Or provide the key directly:
+```bash
+cohere generate --firebase my-project-id --firebase-key service-account.json
+```
+
+### Validation
+
+Ensure your generated docs match the actual database:
+
+```bash
+cohere validate --url "postgresql://..."
+```
+
+This checks:
+- ✅ All tables documented
+- ✅ Column types match
+- ✅ Indexes present
+- ✅ Foreign keys valid
+
+## Configuration
+
+Create `.cohererc.json` in your project root:
+
+```json
+{
+  "outputDir": ".ai",
+  "databases": {
+    "development": "postgresql://localhost:5432/dev",
+    "production": "postgresql://prod-server:5432/prod"
+  },
+  "include": ["users", "posts", "comments"],
+  "exclude": ["migrations", "sessions"],
+  "watch": {
+    "enabled": true,
+    "paths": ["prisma/schema.prisma", "src/db/**/*.ts"]
+  },
+  "templates": {
+    "claude": "custom/claude-template.md"
+  }
+}
+```
+
+## CLI Reference
+
+### `cohere init`
+Initialize Cohere in your project
+
+**Options:**
+- `-u, --url <url>` - Database connection URL
+- `-d, --dir <dir>` - Output directory (default: `.ai`)
+
+### `cohere generate`
+Generate context files from database or ORM
+
+**Options:**
+- `-u, --url <url>` - Database connection URL
+- `--prisma [path]` - Use Prisma schema
+- `--drizzle [path]` - Use Drizzle schema  
+- `--firebase <projectId>` - Firestore project ID
+- `--firebase-key <path>` - Service account key path
+- `--mongo-sample <n>` - MongoDB sample size (default: 100)
+- `-o, --output <dir>` - Output directory
+
+### `cohere watch`
+Watch for schema changes and auto-regenerate
+
+**Options:**
+- `-u, --url <url>` - Database connection URL
+- `--interval <ms>` - Check interval (default: 5000)
+
+### `cohere validate`
+Validate generated docs against database
+
+**Options:**
+- `-u, --url <url>` - Database connection URL
+
+### `cohere show`
+Display current database state
+
+**Options:**
+- `-u, --url <url>` - Database connection URL
+- `--json` - Output as JSON
+
+## Architecture
+
+```
+cohere/
+├── src/
+│   ├── cli.ts              # CLI entry point
+│   ├── commands/           # Command implementations
+│   │   ├── generate.ts     # Schema extraction & generation
+│   │   ├── init.ts         # Project initialization
+│   │   ├── validate.ts     # Schema validation
+│   │   ├── watch.ts        # File watching
+│   │   └── show.ts         # Display schema
+│   ├── extractors/         # Database-specific extractors
+│   │   ├── postgres.ts     # PostgreSQL extractor
+│   │   ├── mysql.ts        # MySQL extractor
+│   │   ├── sqlite.ts       # SQLite extractor
+│   │   ├── mongodb.ts      # MongoDB extractor
+│   │   ├── firebase.ts     # Firestore extractor
+│   │   ├── prisma.ts       # Prisma schema parser
+│   │   └── drizzle.ts      # Drizzle schema parser
+│   ├── generators/         # Context file generators
+│   │   ├── claude.ts       # CLAUDE.md generator
+│   │   ├── cursor.ts       # AGENTS.md generator
+│   │   └── markdown.ts     # DATABASE.md generator
+│   └── utils/              # Shared utilities
+└── templates/              # Output templates
+    ├── claude.md
+    ├── cursor.md
+    └── queries/
+```
+
+## Testing
+
+Cohere includes comprehensive unit tests for all extractors:
+
+```bash
+# Run all tests
+npm test
+
+# Run tests with coverage
+npm run test:coverage
+
+# Watch mode
+npm run test:watch
+```
+
+Test coverage:
+- ✅ PostgreSQL extractor (tables, indexes, foreign keys)
+- ✅ MySQL extractor (InnoDB features)
+- ✅ SQLite extractor (PRAGMA queries)
+- ✅ MongoDB extractor (document sampling)
+- ✅ Firebase extractor (collection inference)
+- ✅ Prisma parser (schema.prisma files)
+- ✅ Drizzle parser (schema definitions)
+
+## Contributing
+
+Contributions are welcome! Please:
+
+1. Fork the repository
+2. Create a feature branch (`git checkout -b feature/amazing-feature`)
+3. Write tests for your changes
+4. Ensure all tests pass (`npm test`)
+5. Commit your changes (`git commit -m 'Add amazing feature'`)
+6. Push to the branch (`git push origin feature/amazing-feature`)
+7. Open a Pull Request
+
+## Roadmap
+
+See [ROADMAP.md](docs/ROADMAP.md) for planned features.
+
+## FAQ
+
+**Q: Does Cohere send data to external services?**  
+A: No. All schema extraction happens locally. Your database credentials and schema never leave your machine.
+
+**Q: Can I use this with private/commercial databases?**  
+A: Yes. Cohere is MIT licensed and can be used commercially.
+
+**Q: How often should I regenerate context?**  
+A: Use `cohere watch` during development. Regenerate after schema migrations in production.
+
+**Q: Does this work with multi-tenant databases?**  
+A: Yes. Cohere extracts the schema structure. Tenant-specific data is not included.
+
+**Q: Can I customize the output templates?**  
+A: Yes. Create custom templates in `.cohererc.json` or the `templates/` directory.
+
+## License
+
+MIT © Cohere Team
+
+## Acknowledgments
+
+- [Prisma](https://prisma.io) - Inspiration for type-safe database access
+- [Drizzle ORM](https://orm.drizzle.team) - Lightweight ORM patterns
+- [Supabase CLI](https://supabase.com/docs/reference/cli) - CLI design patterns

package/dist/cli.d.ts

@@ -0,0 +1,8 @@
+#!/usr/bin/env node
+/**
+ * Cohere CLI
+ *
+ * Generate AI-friendly database context documentation for coding assistants.
+ */
+export {};
+//# sourceMappingURL=cli.d.ts.map

package/dist/cli.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.d.ts","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":";AAEA;;;;GAIG"}

package/dist/cli.js

@@ -0,0 +1,104 @@
+#!/usr/bin/env node
+/**
+ * Cohere CLI
+ *
+ * Generate AI-friendly database context documentation for coding assistants.
+ */
+import { fileURLToPath } from 'url';
+import { dirname, join } from 'path';
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+import { Command } from 'commander';
+import inquirer from 'inquirer';
+import { generate } from './commands/generate.js';
+import { init } from './commands/init.js';
+import { validate } from './commands/validate.js';
+import { watch } from './commands/watch.js';
+import { handoff } from './commands/handoff.js';
+import * as fs from 'fs';
+const pkg = JSON.parse(fs.readFileSync(join(__dirname, '..', 'package.json'), 'utf-8'));
+const version = pkg.version;
+const program = new Command();
+program
+    .name('cohere')
+    .description('Generate AI-friendly database context for coding assistants')
+    .version(version);
+// Commands
+program
+    .command('init')
+    .description('Initialize Cohere in your project')
+    .option('-u, --url <url>', 'Database connection URL')
+    .option('-d, --dir <dir>', 'Output directory', '.ai')
+    .action(init);
+program
+    .command('generate')
+    .description('Generate context files from database or ORM schema')
+    .option('-u, --url <url>', 'Database connection URL')
+    .option('--orm <orm>', 'ORM type (prisma, drizzle)')
+    .option('--mysql', 'Use MySQL extractor')
+    .option('--sqlite <path>', 'Use SQLite extractor with file path')
+    .option('--prisma [path]', 'Use Prisma extractor (optional path)')
+    .option('--drizzle [path]', 'Use Drizzle extractor (optional path)')
+    .option('-o, --output <dir>', 'Output directory')
+    .option('--format <format>', 'Output format (markdown, json)', 'markdown')
+    .action(generate);
+program
+    .command('validate')
+    .description('Validate generated context against database')
+    .option('--strict', 'Fail on any mismatches')
+    .action(validate);
+program
+    .command('watch')
+    .description('Watch for schema changes and regenerate')
+    .option('-d, --debounce <ms>', 'Debounce time in ms', '2000')
+    .action(watch);
+program
+    .command('show')
+    .description('Show current database schema')
+    .option('-f, --format <format>', 'Output format', 'markdown')
+    .action(async (options) => {
+    const { show } = await import('./commands/show.js');
+    await show(options);
+});
+program
+    .command('handoff')
+    .description('Multi-agent handoff management')
+    .option('--record', 'Record current session state')
+    .option('--resume <id>', 'Resume from previous session')
+    .option('--list', 'List available sessions')
+    .option('-o, --output <dir>', 'Output directory', '.ai')
+    .option('--status <status>', 'Session status', 'in_progress')
+    .option('--agentId <id>', 'Agent identifier')
+    .action(handoff);
+// Interactive mode
+program
+    .command('interactive')
+    .alias('i')
+    .description('Interactive mode for quick operations')
+    .action(async () => {
+    const { action } = await inquirer.prompt([
+        {
+            type: 'list',
+            name: 'action',
+            message: 'What would you like to do?',
+            choices: [
+                { name: 'Generate context from database', value: 'generate' },
+                { name: 'Validate existing context', value: 'validate' },
+                { name: 'Watch for changes', value: 'watch' },
+                { name: 'Show schema', value: 'show' },
+            ],
+        },
+    ]);
+    const command = program.commands.find((c) => c.name() === action);
+    if (command) {
+        const argv = [process.argv[0], program.name(), action];
+        await command.parseAsync(argv);
+    }
+});
+// Parse arguments
+program.parse();
+// Show help if no command
+if (!process.argv.slice(2).length) {
+    program.outputHelp();
+}
+//# sourceMappingURL=cli.js.map

package/dist/cli.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.js","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":";AAEA;;;;GAIG;AAEH,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAErC,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;AAEtC,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,QAAQ,MAAM,UAAU,CAAC;AAChC,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAClD,OAAO,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAC1C,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAClD,OAAO,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,MAAM,uBAAuB,CAAC;AAChD,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AAEzB,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;AACxF,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;AAE5B,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;AAE9B,OAAO;KACJ,IAAI,CAAC,QAAQ,CAAC;KACd,WAAW,CAAC,6DAA6D,CAAC;KAC1E,OAAO,CAAC,OAAO,CAAC,CAAC;AAEpB,WAAW;AACX,OAAO;KACJ,OAAO,CAAC,MAAM,CAAC;KACf,WAAW,CAAC,mCAAmC,CAAC;KAChD,MAAM,CAAC,iBAAiB,EAAE,yBAAyB,CAAC;KACpD,MAAM,CAAC,iBAAiB,EAAE,kBAAkB,EAAE,KAAK,CAAC;KACpD,MAAM,CAAC,IAAI,CAAC,CAAC;AAEhB,OAAO;KACJ,OAAO,CAAC,UAAU,CAAC;KACnB,WAAW,CAAC,oDAAoD,CAAC;KACjE,MAAM,CAAC,iBAAiB,EAAE,yBAAyB,CAAC;KACpD,MAAM,CAAC,aAAa,EAAE,4BAA4B,CAAC;KACnD,MAAM,CAAC,SAAS,EAAE,qBAAqB,CAAC;KACxC,MAAM,CAAC,iBAAiB,EAAE,qCAAqC,CAAC;KAChE,MAAM,CAAC,iBAAiB,EAAE,sCAAsC,CAAC;KACjE,MAAM,CAAC,kBAAkB,EAAE,uCAAuC,CAAC;KACnE,MAAM,CAAC,oBAAoB,EAAE,kBAAkB,CAAC;KAChD,MAAM,CAAC,mBAAmB,EAAE,gCAAgC,EAAE,UAAU,CAAC;KACzE,MAAM,CAAC,QAAQ,CAAC,CAAC;AAEpB,OAAO;KACJ,OAAO,CAAC,UAAU,CAAC;KACnB,WAAW,CAAC,6CAA6C,CAAC;KAC1D,MAAM,CAAC,UAAU,EAAE,wBAAwB,CAAC;KAC5C,MAAM,CAAC,QAAQ,CAAC,CAAC;AAEpB,OAAO;KACJ,OAAO,CAAC,OAAO,CAAC;KAChB,WAAW,CAAC,yCAAyC,CAAC;KACtD,MAAM,CAAC,qBAAqB,EAAE,qBAAqB,EAAE,MAAM,CAAC;KAC5D,MAAM,CAAC,KAAK,CAAC,CAAC;AAEjB,OAAO;KACJ,OAAO,CAAC,MAAM,CAAC;KACf,WAAW,CAAC,8BAA8B,CAAC;KAC3C,MAAM,CAAC,uBAAuB,EAAE,eAAe,EAAE,UAAU,CAAC;KAC5D,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;IACxB,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,oBAAoB,CAAC,CAAC;IACpD,MAAM,IAAI,CAAC,OAAO,CAAC,CAAC;AACtB,CAAC,CAAC,CAAC;AAEL,OAAO;KACJ,OAAO,CAAC,SAAS,CAAC;KAClB,WAAW,CAAC,gCAAgC,CAAC;KAC7C,MAAM,CAAC,UAAU,EAAE,8BAA8B,CAAC;KAClD,MAAM,CAAC,eAAe,EAAE,8BAA8B,CAAC;KACvD,MAAM,CAAC,QAAQ,EAAE,yBAAyB,CAAC;KAC3C,MAAM,CAAC,oBAAoB,EAAE,kBAAkB,EAAE,KAAK,CAAC;KACvD,MAAM,CAAC,mBAAmB,EAAE,gBAAgB,EAAE,aAAa,CAAC;KAC5D,MAAM,CAAC,gBAAgB,EAAE,kBAAkB,CAAC;KAC5C,MAAM,CAAC,OAAO,CAAC,CAAC;AAEnB,mBAAmB;AACnB,OAAO;KACJ,OAAO,CAAC,aAAa,CAAC;KACtB,KAAK,CAAC,GAAG,CAAC;KACV,WAAW,CAAC,uCAAuC,CAAC;KACpD,MAAM,CAAC,KAAK,IAAI,EAAE;IACjB,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC;QACvC;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,4BAA4B;YACrC,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,gCAAgC,EAAE,KAAK,EAAE,UAAU,EAAE;gBAC7D,EAAE,IAAI,EAAE,2BAA2B,EAAE,KAAK,EAAE,UAAU,EAAE;gBACxD,EAAE,IAAI,EAAE,mBAAmB,EAAE,KAAK,EAAE,OAAO,EAAE;gBAC7C,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,EAAE;aACvC;SACF;KACF,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,MAAM,CAAC,CAAC;IAClE,IAAI,OAAO,EAAE,CAAC;QACZ,MAAM,IAAI,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,IAAI,EAAE,EAAE,MAAM,CAAC,CAAC;QACvD,MAAM,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;AACH,CAAC,CAAC,CAAC;AAEL,kBAAkB;AAClB,OAAO,CAAC,KAAK,EAAE,CAAC;AAEhB,0BAA0B;AAC1B,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAClC,OAAO,CAAC,UAAU,EAAE,CAAC;AACvB,CAAC"}

package/dist/commands/generate.d.ts

@@ -0,0 +1,21 @@
+/**
+ * Generate Command
+ * Generates CLAUDE.md and AGENTS.md from database schema
+ */
+interface GenerateOptions {
+    url?: string;
+    orm?: string;
+    output?: string;
+    format?: string;
+    schema?: string;
+    mysql?: boolean;
+    sqlite?: string;
+    prisma?: string;
+    drizzle?: string;
+    mongodb?: string;
+    firebaseKey?: string;
+    firebaseProject?: string;
+}
+export declare function generate(options: GenerateOptions): Promise<void>;
+export {};
+//# sourceMappingURL=generate.d.ts.map

package/dist/commands/generate.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate.d.ts","sourceRoot":"","sources":["../../src/commands/generate.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAOH,UAAU,eAAe;IACvB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,wBAAsB,QAAQ,CAAC,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC,CAwHtE"}

package/dist/commands/generate.js

@@ -0,0 +1,131 @@
+/**
+ * Generate Command
+ * Generates CLAUDE.md and AGENTS.md from database schema
+ */
+import * as fs from 'fs';
+import * as path from 'path';
+import { createExtractor, UnifiedSchemaConverter } from '../extractors/index.js';
+import { TemplateGenerator } from '../generators/templates.js';
+export async function generate(options) {
+    const outputDir = options.output || '.ai';
+    const schemaName = options.schema || 'public';
+    console.log(`🔍 Generating database context...`);
+    console.log(`   Output directory: ${outputDir}`);
+    let extractorType = 'postgresql';
+    let connectionString = options.url || process.env.DATABASE_URL || '';
+    let schemaPath;
+    // Determine extractor type
+    if (options.mysql) {
+        extractorType = 'mysql';
+        if (!connectionString) {
+            throw new Error('MySQL connection URL required. Use --url or DATABASE_URL env var.');
+        }
+    }
+    else if (options.sqlite) {
+        extractorType = 'sqlite';
+        connectionString = options.sqlite;
+    }
+    else if (options.prisma) {
+        extractorType = 'prisma';
+        schemaPath = typeof options.prisma === 'string' ? options.prisma : 'prisma/schema.prisma';
+        connectionString = 'dummy'; // Not used for Prisma
+    }
+    else if (options.drizzle) {
+        extractorType = 'drizzle';
+        schemaPath = typeof options.drizzle === 'string' ? options.drizzle : 'src/db/schema.ts';
+        connectionString = 'dummy'; // Not used for Drizzle
+    }
+    else if (options.orm === 'prisma') {
+        extractorType = 'prisma';
+        schemaPath = 'prisma/schema.prisma';
+        connectionString = 'dummy';
+    }
+    else if (options.orm === 'drizzle') {
+        extractorType = 'drizzle';
+        schemaPath = 'src/db/schema.ts';
+        connectionString = 'dummy';
+    }
+    else if (connectionString) {
+        if (connectionString.startsWith('mysql')) {
+            extractorType = 'mysql';
+        }
+        else if (connectionString.startsWith('file:') || connectionString.endsWith('.db') || connectionString.endsWith('.sqlite')) {
+            extractorType = 'sqlite';
+        }
+        else {
+            extractorType = 'postgresql';
+        }
+    }
+    else {
+        // Try to auto-detect
+        if (fs.existsSync('prisma/schema.prisma')) {
+            console.log('👀 Detected Prisma schema, using Prisma extractor...');
+            extractorType = 'prisma';
+            schemaPath = 'prisma/schema.prisma';
+            connectionString = 'dummy';
+        }
+        else if (fs.existsSync('drizzle.config.ts') || fs.existsSync('src/db/schema.ts')) {
+            console.log('👀 Detected Drizzle project, using Drizzle extractor...');
+            extractorType = 'drizzle';
+            schemaPath = 'src/db/schema.ts';
+            connectionString = 'dummy';
+        }
+        else if (options.mongodb || (connectionString && (connectionString.startsWith('mongodb://') || connectionString.startsWith('mongodb+srv://')))) {
+            console.log('👀 Detected MongoDB connection...');
+            extractorType = 'mongodb';
+            connectionString = options.mongodb || connectionString;
+            if (!connectionString) {
+                throw new Error('MongoDB connection URL required. Use --url or DATABASE_URL env var.');
+            }
+        }
+        else if (options.firebaseProject || process.env.GOOGLE_APPLICATION_CREDENTIALS) {
+            console.log('👀 Detected Firebase project...');
+            extractorType = 'firebase';
+            connectionString = 'dummy'; // Not used for Firebase
+        }
+        else {
+            throw new Error('Database connection not found. Use --url, --sqlite, --prisma, --drizzle, or --mongodb.');
+        }
+    }
+    console.log(`📦 Using extractor: ${extractorType}`);
+    if (schemaPath)
+        console.log(`   Schema path: ${schemaPath}`);
+    const extractor = await createExtractor(extractorType, connectionString, {
+        schemaPath,
+        projectId: options.firebaseProject,
+        serviceAccountPath: options.firebaseKey
+    });
+    let unifiedSchema;
+    try {
+        const rawSchema = await extractor.extract();
+        rawSchema.databaseType = extractorType;
+        rawSchema.schemaName = rawSchema.schemaName || schemaName;
+        rawSchema.source = rawSchema.source || schemaPath || connectionString;
+        // Convert to unified format using generic convert
+        unifiedSchema = UnifiedSchemaConverter.convert(rawSchema);
+        // Add source info
+        unifiedSchema.source = schemaPath || connectionString.replace(/:[^:]*@/, ':***@'); // Hide password
+    }
+    finally {
+        if (extractor.close) {
+            await extractor.close();
+        }
+    }
+    // Generate and save templates
+    console.log('📝 Generating context files...');
+    const generator = new TemplateGenerator('templates', outputDir);
+    await generator.save(outputDir, unifiedSchema);
+    // Summary
+    console.log('\n✅ Generation complete!');
+    console.log(`   Tables: ${unifiedSchema.tables.length}`);
+    console.log(`   Output: ${path.resolve(outputDir)}`);
+    console.log('\n📄 Generated files:');
+    console.log(`   - ${outputDir}/CLAUDE.md`);
+    console.log(`   - ${outputDir}/AGENTS.md`);
+    console.log(`   - ${outputDir}/queries/`);
+    console.log('\n💡 Next steps:');
+    console.log('   1. Review the generated files');
+    console.log('   2. Add custom business rules');
+    console.log('   3. Commit to version control');
+}
+//# sourceMappingURL=generate.js.map