truss-db-mcp 1.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +69 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +79 -0
- package/dist/index.js.map +1 -0
- package/dist/lib/ai-query.d.ts +7 -0
- package/dist/lib/ai-query.d.ts.map +1 -0
- package/dist/lib/ai-query.js +224 -0
- package/dist/lib/ai-query.js.map +1 -0
- package/dist/lib/config.d.ts +4 -0
- package/dist/lib/config.d.ts.map +1 -0
- package/dist/lib/config.js +30 -0
- package/dist/lib/config.js.map +1 -0
- package/dist/lib/connection.d.ts +14 -0
- package/dist/lib/connection.d.ts.map +1 -0
- package/dist/lib/connection.js +175 -0
- package/dist/lib/connection.js.map +1 -0
- package/dist/lib/license.d.ts +4 -0
- package/dist/lib/license.d.ts.map +1 -0
- package/dist/lib/license.js +94 -0
- package/dist/lib/license.js.map +1 -0
- package/dist/lib/query-optimizer.d.ts +3 -0
- package/dist/lib/query-optimizer.d.ts.map +1 -0
- package/dist/lib/query-optimizer.js +155 -0
- package/dist/lib/query-optimizer.js.map +1 -0
- package/dist/tools/connect.d.ts +3 -0
- package/dist/tools/connect.d.ts.map +1 -0
- package/dist/tools/connect.js +37 -0
- package/dist/tools/connect.js.map +1 -0
- package/dist/tools/describe-table.d.ts +3 -0
- package/dist/tools/describe-table.d.ts.map +1 -0
- package/dist/tools/describe-table.js +60 -0
- package/dist/tools/describe-table.js.map +1 -0
- package/dist/tools/export-data.d.ts +3 -0
- package/dist/tools/export-data.d.ts.map +1 -0
- package/dist/tools/export-data.js +68 -0
- package/dist/tools/export-data.js.map +1 -0
- package/dist/tools/generate-migration.d.ts +3 -0
- package/dist/tools/generate-migration.d.ts.map +1 -0
- package/dist/tools/generate-migration.js +36 -0
- package/dist/tools/generate-migration.js.map +1 -0
- package/dist/tools/list-tables.d.ts +3 -0
- package/dist/tools/list-tables.d.ts.map +1 -0
- package/dist/tools/list-tables.js +45 -0
- package/dist/tools/list-tables.js.map +1 -0
- package/dist/tools/natural-language-query.d.ts +3 -0
- package/dist/tools/natural-language-query.d.ts.map +1 -0
- package/dist/tools/natural-language-query.js +53 -0
- package/dist/tools/natural-language-query.js.map +1 -0
- package/dist/tools/optimize-query.d.ts +3 -0
- package/dist/tools/optimize-query.d.ts.map +1 -0
- package/dist/tools/optimize-query.js +51 -0
- package/dist/tools/optimize-query.js.map +1 -0
- package/dist/tools/query.d.ts +3 -0
- package/dist/tools/query.d.ts.map +1 -0
- package/dist/tools/query.js +79 -0
- package/dist/tools/query.js.map +1 -0
- package/dist/types.d.ts +86 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +3 -0
- package/dist/types.js.map +1 -0
- package/evals/eval-query.ts +356 -0
- package/evals/run-evals.ts +36 -0
- package/evals/test.db +0 -0
- package/glama.json +4 -0
- package/package.json +47 -0
- package/smithery.yaml +18 -0
- package/src/index.ts +99 -0
- package/src/lib/ai-query.ts +274 -0
- package/src/lib/config.ts +33 -0
- package/src/lib/connection.ts +263 -0
- package/src/lib/license.ts +118 -0
- package/src/lib/query-optimizer.ts +191 -0
- package/src/tools/connect.ts +43 -0
- package/src/tools/describe-table.ts +67 -0
- package/src/tools/export-data.ts +81 -0
- package/src/tools/generate-migration.ts +43 -0
- package/src/tools/list-tables.ts +52 -0
- package/src/tools/natural-language-query.ts +61 -0
- package/src/tools/optimize-query.ts +59 -0
- package/src/tools/query.ts +89 -0
- package/src/types.ts +115 -0
- package/tsconfig.json +19 -0
package/LICENSE
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
MIT License
|
|
2
|
+
|
|
3
|
+
Copyright (c) 2026 TRUSS
|
|
4
|
+
|
|
5
|
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
6
|
+
of this software and associated documentation files (the "Software"), to deal
|
|
7
|
+
in the Software without restriction, including without limitation the rights
|
|
8
|
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
9
|
+
copies of the Software, and to permit persons to whom the Software is
|
|
10
|
+
furnished to do so, subject to the following conditions:
|
|
11
|
+
|
|
12
|
+
The above copyright notice and this permission notice shall be included in all
|
|
13
|
+
copies or substantial portions of the Software.
|
|
14
|
+
|
|
15
|
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
16
|
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
17
|
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
18
|
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
19
|
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
20
|
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
21
|
+
SOFTWARE.
|
package/README.md
ADDED
|
@@ -0,0 +1,69 @@
|
|
|
1
|
+
# truss-db-mcp
|
|
2
|
+
|
|
3
|
+
Database query MCP server for Claude Code. Connect to SQLite, PostgreSQL, or MySQL databases directly from your AI assistant.
|
|
4
|
+
|
|
5
|
+
## Quick Start
|
|
6
|
+
|
|
7
|
+
```json
|
|
8
|
+
{
|
|
9
|
+
"mcpServers": {
|
|
10
|
+
"truss-db": {
|
|
11
|
+
"command": "npx",
|
|
12
|
+
"args": ["-y", "truss-db-mcp"]
|
|
13
|
+
}
|
|
14
|
+
}
|
|
15
|
+
}
|
|
16
|
+
```
|
|
17
|
+
|
|
18
|
+
## Free Tier Tools
|
|
19
|
+
|
|
20
|
+
| Tool | Description |
|
|
21
|
+
|------|-------------|
|
|
22
|
+
| `connect` | Connect to a SQLite database file |
|
|
23
|
+
| `list_tables` | List all tables with row counts and column info |
|
|
24
|
+
| `describe_table` | Detailed schema: columns, types, indexes, foreign keys |
|
|
25
|
+
| `query` | Execute read-only SQL queries (SELECT) |
|
|
26
|
+
|
|
27
|
+
## Pro Tier Tools ($25/mo)
|
|
28
|
+
|
|
29
|
+
| Tool | Description |
|
|
30
|
+
|------|-------------|
|
|
31
|
+
| `natural_language_query` | Ask questions in plain English, get SQL + results |
|
|
32
|
+
| `optimize_query` | EXPLAIN analysis with index suggestions |
|
|
33
|
+
| `generate_migration` | Generate up/down SQL migrations from descriptions |
|
|
34
|
+
| `export_data` | Export query results as CSV or JSON |
|
|
35
|
+
|
|
36
|
+
## Environment Variables
|
|
37
|
+
|
|
38
|
+
| Variable | Required | Description |
|
|
39
|
+
|----------|----------|-------------|
|
|
40
|
+
| `TRUSS_LICENSE_KEY` | No | Pro license key |
|
|
41
|
+
| `ANTHROPIC_API_KEY` | Pro | For AI-powered features (natural language query, migration generation) |
|
|
42
|
+
| `OPENAI_API_KEY` | Pro | Alternative AI provider |
|
|
43
|
+
| `DATABASE_URL` | No | Connection URL for PostgreSQL/MySQL (Pro) |
|
|
44
|
+
|
|
45
|
+
## Usage Examples
|
|
46
|
+
|
|
47
|
+
### Connect to a database
|
|
48
|
+
```
|
|
49
|
+
Use the connect tool with path: "./my-app.db"
|
|
50
|
+
```
|
|
51
|
+
|
|
52
|
+
### Query data
|
|
53
|
+
```
|
|
54
|
+
Use the query tool with sql: "SELECT * FROM users LIMIT 10"
|
|
55
|
+
```
|
|
56
|
+
|
|
57
|
+
### Natural language query (Pro)
|
|
58
|
+
```
|
|
59
|
+
Use natural_language_query with question: "What are the top 5 customers by total spend?"
|
|
60
|
+
```
|
|
61
|
+
|
|
62
|
+
### Optimize a slow query (Pro)
|
|
63
|
+
```
|
|
64
|
+
Use optimize_query with sql: "SELECT * FROM orders WHERE status = 'pending'"
|
|
65
|
+
```
|
|
66
|
+
|
|
67
|
+
## License
|
|
68
|
+
|
|
69
|
+
MIT
|
package/dist/index.d.ts
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":""}
|
package/dist/index.js
ADDED
|
@@ -0,0 +1,79 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
3
|
+
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
|
|
4
|
+
// Tool registrations
|
|
5
|
+
import { registerConnect } from './tools/connect.js';
|
|
6
|
+
import { registerListTables } from './tools/list-tables.js';
|
|
7
|
+
import { registerDescribeTable } from './tools/describe-table.js';
|
|
8
|
+
import { registerQuery } from './tools/query.js';
|
|
9
|
+
import { registerNaturalLanguageQuery } from './tools/natural-language-query.js';
|
|
10
|
+
import { registerOptimizeQuery } from './tools/optimize-query.js';
|
|
11
|
+
import { registerGenerateMigration } from './tools/generate-migration.js';
|
|
12
|
+
import { registerExportData } from './tools/export-data.js';
|
|
13
|
+
// Lib
|
|
14
|
+
import { closeConnection } from './lib/connection.js';
|
|
15
|
+
import { getLicenseStatus } from './lib/license.js';
|
|
16
|
+
const server = new McpServer({
|
|
17
|
+
name: 'truss-db-mcp',
|
|
18
|
+
version: '1.0.0',
|
|
19
|
+
});
|
|
20
|
+
// ── Status Tool ─────────────────────────────────────────────────────
|
|
21
|
+
server.tool('db_status', 'Check database connection status and license tier.', {}, async () => {
|
|
22
|
+
const license = await getLicenseStatus();
|
|
23
|
+
return {
|
|
24
|
+
content: [{
|
|
25
|
+
type: 'text',
|
|
26
|
+
text: JSON.stringify({
|
|
27
|
+
license: {
|
|
28
|
+
tier: license.tier,
|
|
29
|
+
valid: license.valid,
|
|
30
|
+
expires_at: license.expiresAt,
|
|
31
|
+
},
|
|
32
|
+
...(license.tier === 'free' ? {
|
|
33
|
+
upgrade_info: {
|
|
34
|
+
url: 'https://truss.dev/pricing',
|
|
35
|
+
price: '$25/mo',
|
|
36
|
+
pro_features: [
|
|
37
|
+
'Natural language to SQL (AI-powered)',
|
|
38
|
+
'Query optimization & EXPLAIN analysis',
|
|
39
|
+
'Migration generation',
|
|
40
|
+
'CSV/JSON data export',
|
|
41
|
+
'Write operations (INSERT/UPDATE/DELETE)',
|
|
42
|
+
'PostgreSQL & MySQL support',
|
|
43
|
+
],
|
|
44
|
+
},
|
|
45
|
+
} : {}),
|
|
46
|
+
}, null, 2),
|
|
47
|
+
}],
|
|
48
|
+
};
|
|
49
|
+
});
|
|
50
|
+
// ── Free Tier Tools ─────────────────────────────────────────────────
|
|
51
|
+
registerConnect(server);
|
|
52
|
+
registerListTables(server);
|
|
53
|
+
registerDescribeTable(server);
|
|
54
|
+
registerQuery(server);
|
|
55
|
+
// ── Pro Tier Tools ──────────────────────────────────────────────────
|
|
56
|
+
registerNaturalLanguageQuery(server);
|
|
57
|
+
registerOptimizeQuery(server);
|
|
58
|
+
registerGenerateMigration(server);
|
|
59
|
+
registerExportData(server);
|
|
60
|
+
// ── Start Server ────────────────────────────────────────────────────
|
|
61
|
+
async function main() {
|
|
62
|
+
const transport = new StdioServerTransport();
|
|
63
|
+
// Clean shutdown
|
|
64
|
+
process.on('SIGINT', () => {
|
|
65
|
+
closeConnection();
|
|
66
|
+
process.exit(0);
|
|
67
|
+
});
|
|
68
|
+
process.on('SIGTERM', () => {
|
|
69
|
+
closeConnection();
|
|
70
|
+
process.exit(0);
|
|
71
|
+
});
|
|
72
|
+
await server.connect(transport);
|
|
73
|
+
}
|
|
74
|
+
main().catch((err) => {
|
|
75
|
+
console.error('Fatal error starting MCP server:', err);
|
|
76
|
+
closeConnection();
|
|
77
|
+
process.exit(1);
|
|
78
|
+
});
|
|
79
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AAEjF,qBAAqB;AACrB,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,EAAE,qBAAqB,EAAE,MAAM,2BAA2B,CAAC;AAClE,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACjD,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AACjF,OAAO,EAAE,qBAAqB,EAAE,MAAM,2BAA2B,CAAC;AAClE,OAAO,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC;AAC1E,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAE5D,MAAM;AACN,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAEpD,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC;IAC3B,IAAI,EAAE,cAAc;IACpB,OAAO,EAAE,OAAO;CACjB,CAAC,CAAC;AAEH,uEAAuE;AAEvE,MAAM,CAAC,IAAI,CACT,WAAW,EACX,oDAAoD,EACpD,EAAE,EACF,KAAK,IAAI,EAAE;IACT,MAAM,OAAO,GAAG,MAAM,gBAAgB,EAAE,CAAC;IAEzC,OAAO;QACL,OAAO,EAAE,CAAC;gBACR,IAAI,EAAE,MAAe;gBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,OAAO,EAAE;wBACP,IAAI,EAAE,OAAO,CAAC,IAAI;wBAClB,KAAK,EAAE,OAAO,CAAC,KAAK;wBACpB,UAAU,EAAE,OAAO,CAAC,SAAS;qBAC9B;oBACD,GAAG,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC;wBAC5B,YAAY,EAAE;4BACZ,GAAG,EAAE,2BAA2B;4BAChC,KAAK,EAAE,QAAQ;4BACf,YAAY,EAAE;gCACZ,sCAAsC;gCACtC,uCAAuC;gCACvC,sBAAsB;gCACtB,sBAAsB;gCACtB,yCAAyC;gCACzC,4BAA4B;6BAC7B;yBACF;qBACF,CAAC,CAAC,CAAC,EAAE,CAAC;iBACR,EAAE,IAAI,EAAE,CAAC,CAAC;aACZ,CAAC;KACH,CAAC;AACJ,CAAC,CACF,CAAC;AAEF,uEAAuE;AAEvE,eAAe,CAAC,MAAM,CAAC,CAAC;AACxB,kBAAkB,CAAC,MAAM,CAAC,CAAC;AAC3B,qBAAqB,CAAC,MAAM,CAAC,CAAC;AAC9B,aAAa,CAAC,MAAM,CAAC,CAAC;AAEtB,uEAAuE;AAEvE,4BAA4B,CAAC,MAAM,CAAC,CAAC;AACrC,qBAAqB,CAAC,MAAM,CAAC,CAAC;AAC9B,yBAAyB,CAAC,MAAM,CAAC,CAAC;AAClC,kBAAkB,CAAC,MAAM,CAAC,CAAC;AAE3B,uEAAuE;AAEvE,KAAK,UAAU,IAAI;IACjB,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAE7C,iBAAiB;IACjB,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE;QACxB,eAAe,EAAE,CAAC;QAClB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;IACH,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;QACzB,eAAe,EAAE,CAAC;QAClB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;AAClC,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACnB,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,GAAG,CAAC,CAAC;IACvD,eAAe,EAAE,CAAC;IAClB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
import type { NaturalLanguageQueryResult } from '../types.js';
|
|
2
|
+
export declare function naturalLanguageToSql(question: string, filterTables?: string[]): Promise<NaturalLanguageQueryResult>;
|
|
3
|
+
export declare function generateMigrationSql(description: string, dialect?: string): Promise<{
|
|
4
|
+
upSql: string;
|
|
5
|
+
downSql: string;
|
|
6
|
+
}>;
|
|
7
|
+
//# sourceMappingURL=ai-query.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ai-query.d.ts","sourceRoot":"","sources":["../../src/lib/ai-query.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,0BAA0B,EAAa,MAAM,aAAa,CAAC;AA0HzE,wBAAsB,oBAAoB,CACxC,QAAQ,EAAE,MAAM,EAChB,YAAY,CAAC,EAAE,MAAM,EAAE,GACtB,OAAO,CAAC,0BAA0B,CAAC,CA6CrC;AAED,wBAAsB,oBAAoB,CACxC,WAAW,EAAE,MAAM,EACnB,OAAO,GAAE,MAAiB,GACzB,OAAO,CAAC;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAAC,CAgG7C"}
|
|
@@ -0,0 +1,224 @@
|
|
|
1
|
+
import { getConfig } from './config.js';
|
|
2
|
+
import { listTables, executeQuery } from './connection.js';
|
|
3
|
+
// ── Schema Summary ──────────────────────────────────────────────────
|
|
4
|
+
function buildSchemaContext(filterTables) {
|
|
5
|
+
const tables = listTables();
|
|
6
|
+
const relevant = filterTables
|
|
7
|
+
? tables.filter(t => filterTables.includes(t.name))
|
|
8
|
+
: tables;
|
|
9
|
+
return relevant.map(t => formatTableSchema(t)).join('\n\n');
|
|
10
|
+
}
|
|
11
|
+
function formatTableSchema(t) {
|
|
12
|
+
const cols = t.columns.map(c => {
|
|
13
|
+
const parts = [` ${c.name} ${c.type}`];
|
|
14
|
+
if (c.primaryKey)
|
|
15
|
+
parts.push('PRIMARY KEY');
|
|
16
|
+
if (!c.nullable)
|
|
17
|
+
parts.push('NOT NULL');
|
|
18
|
+
if (c.defaultValue !== null)
|
|
19
|
+
parts.push(`DEFAULT ${c.defaultValue}`);
|
|
20
|
+
return parts.join(' ');
|
|
21
|
+
}).join(',\n');
|
|
22
|
+
return `TABLE ${t.name} (${t.rowCount} rows):\n${cols}`;
|
|
23
|
+
}
|
|
24
|
+
async function callAnthropic(prompt, apiKey) {
|
|
25
|
+
const response = await fetch('https://api.anthropic.com/v1/messages', {
|
|
26
|
+
method: 'POST',
|
|
27
|
+
headers: {
|
|
28
|
+
'Content-Type': 'application/json',
|
|
29
|
+
'x-api-key': apiKey,
|
|
30
|
+
'anthropic-version': '2023-06-01',
|
|
31
|
+
},
|
|
32
|
+
body: JSON.stringify({
|
|
33
|
+
model: 'claude-sonnet-4-20250514',
|
|
34
|
+
max_tokens: 1024,
|
|
35
|
+
messages: [{
|
|
36
|
+
role: 'user',
|
|
37
|
+
content: prompt,
|
|
38
|
+
}],
|
|
39
|
+
}),
|
|
40
|
+
signal: AbortSignal.timeout(30000),
|
|
41
|
+
});
|
|
42
|
+
if (!response.ok) {
|
|
43
|
+
const body = await response.text();
|
|
44
|
+
throw new Error(`Anthropic API error (${response.status}): ${body}`);
|
|
45
|
+
}
|
|
46
|
+
const data = await response.json();
|
|
47
|
+
const text = data.content.find(c => c.type === 'text')?.text ?? '';
|
|
48
|
+
return parseAiResponse(text);
|
|
49
|
+
}
|
|
50
|
+
async function callOpenAI(prompt, apiKey) {
|
|
51
|
+
const response = await fetch('https://api.openai.com/v1/chat/completions', {
|
|
52
|
+
method: 'POST',
|
|
53
|
+
headers: {
|
|
54
|
+
'Content-Type': 'application/json',
|
|
55
|
+
'Authorization': `Bearer ${apiKey}`,
|
|
56
|
+
},
|
|
57
|
+
body: JSON.stringify({
|
|
58
|
+
model: 'gpt-4o-mini',
|
|
59
|
+
messages: [{
|
|
60
|
+
role: 'user',
|
|
61
|
+
content: prompt,
|
|
62
|
+
}],
|
|
63
|
+
max_tokens: 1024,
|
|
64
|
+
}),
|
|
65
|
+
signal: AbortSignal.timeout(30000),
|
|
66
|
+
});
|
|
67
|
+
if (!response.ok) {
|
|
68
|
+
const body = await response.text();
|
|
69
|
+
throw new Error(`OpenAI API error (${response.status}): ${body}`);
|
|
70
|
+
}
|
|
71
|
+
const data = await response.json();
|
|
72
|
+
const text = data.choices[0]?.message?.content ?? '';
|
|
73
|
+
return parseAiResponse(text);
|
|
74
|
+
}
|
|
75
|
+
function parseAiResponse(text) {
|
|
76
|
+
// Extract SQL from code blocks or raw text
|
|
77
|
+
const sqlMatch = text.match(/```sql\n([\s\S]*?)```/);
|
|
78
|
+
const sql = sqlMatch ? sqlMatch[1].trim() : extractSqlFromText(text);
|
|
79
|
+
// Extract explanation (everything that is not the SQL block)
|
|
80
|
+
const explanation = text
|
|
81
|
+
.replace(/```sql[\s\S]*?```/, '')
|
|
82
|
+
.replace(/^SQL:\s*/mi, '')
|
|
83
|
+
.trim()
|
|
84
|
+
|| 'Query generated from natural language.';
|
|
85
|
+
if (!sql) {
|
|
86
|
+
throw new Error('AI did not produce a valid SQL query. Try rephrasing your question.');
|
|
87
|
+
}
|
|
88
|
+
return { sql, explanation };
|
|
89
|
+
}
|
|
90
|
+
function extractSqlFromText(text) {
|
|
91
|
+
// Try to find a SELECT/WITH statement in the text
|
|
92
|
+
const match = text.match(/((?:WITH|SELECT)\b[\s\S]*?;)/i);
|
|
93
|
+
return match ? match[1].trim() : '';
|
|
94
|
+
}
|
|
95
|
+
// ── Public API ──────────────────────────────────────────────────────
|
|
96
|
+
export async function naturalLanguageToSql(question, filterTables) {
|
|
97
|
+
const config = getConfig();
|
|
98
|
+
if (!config.anthropicApiKey && !config.openaiApiKey) {
|
|
99
|
+
throw new Error('Natural language queries require an AI API key. ' +
|
|
100
|
+
'Set ANTHROPIC_API_KEY or OPENAI_API_KEY in your environment.');
|
|
101
|
+
}
|
|
102
|
+
const schema = buildSchemaContext(filterTables);
|
|
103
|
+
const prompt = `You are a SQL query generator. Given the database schema below and a natural language question, generate the correct SQL query.
|
|
104
|
+
|
|
105
|
+
DATABASE SCHEMA:
|
|
106
|
+
${schema}
|
|
107
|
+
|
|
108
|
+
QUESTION: ${question}
|
|
109
|
+
|
|
110
|
+
Respond with:
|
|
111
|
+
1. The SQL query in a \`\`\`sql code block
|
|
112
|
+
2. A brief explanation of what the query does
|
|
113
|
+
|
|
114
|
+
Important:
|
|
115
|
+
- Use only tables and columns that exist in the schema above
|
|
116
|
+
- Write SQLite-compatible SQL
|
|
117
|
+
- Use appropriate JOINs when data spans multiple tables
|
|
118
|
+
- Always include a LIMIT if the result could be large (default LIMIT 100)`;
|
|
119
|
+
let aiResult;
|
|
120
|
+
if (config.anthropicApiKey) {
|
|
121
|
+
aiResult = await callAnthropic(prompt, config.anthropicApiKey);
|
|
122
|
+
}
|
|
123
|
+
else {
|
|
124
|
+
aiResult = await callOpenAI(prompt, config.openaiApiKey);
|
|
125
|
+
}
|
|
126
|
+
// Execute the generated SQL
|
|
127
|
+
const results = executeQuery(aiResult.sql);
|
|
128
|
+
return {
|
|
129
|
+
sql: aiResult.sql,
|
|
130
|
+
explanation: aiResult.explanation,
|
|
131
|
+
results,
|
|
132
|
+
};
|
|
133
|
+
}
|
|
134
|
+
export async function generateMigrationSql(description, dialect = 'sqlite') {
|
|
135
|
+
const config = getConfig();
|
|
136
|
+
if (!config.anthropicApiKey && !config.openaiApiKey) {
|
|
137
|
+
throw new Error('Migration generation requires an AI API key. ' +
|
|
138
|
+
'Set ANTHROPIC_API_KEY or OPENAI_API_KEY in your environment.');
|
|
139
|
+
}
|
|
140
|
+
let schemaContext = '';
|
|
141
|
+
try {
|
|
142
|
+
schemaContext = buildSchemaContext();
|
|
143
|
+
}
|
|
144
|
+
catch {
|
|
145
|
+
// No connection -- generate migration without existing schema context
|
|
146
|
+
}
|
|
147
|
+
const prompt = `You are a database migration generator. Generate SQL migration statements for the following change.
|
|
148
|
+
|
|
149
|
+
${schemaContext ? `EXISTING SCHEMA:\n${schemaContext}\n\n` : ''}DIALECT: ${dialect}
|
|
150
|
+
CHANGE DESCRIPTION: ${description}
|
|
151
|
+
|
|
152
|
+
Respond with two SQL code blocks:
|
|
153
|
+
|
|
154
|
+
1. UP migration (apply the change):
|
|
155
|
+
\`\`\`sql
|
|
156
|
+
-- up migration here
|
|
157
|
+
\`\`\`
|
|
158
|
+
|
|
159
|
+
2. DOWN migration (revert the change):
|
|
160
|
+
\`\`\`sql
|
|
161
|
+
-- down migration here
|
|
162
|
+
\`\`\`
|
|
163
|
+
|
|
164
|
+
Important:
|
|
165
|
+
- Use IF EXISTS / IF NOT EXISTS where appropriate
|
|
166
|
+
- Make migrations idempotent when possible
|
|
167
|
+
- Use correct syntax for the ${dialect} dialect`;
|
|
168
|
+
let text;
|
|
169
|
+
if (config.anthropicApiKey) {
|
|
170
|
+
const response = await fetch('https://api.anthropic.com/v1/messages', {
|
|
171
|
+
method: 'POST',
|
|
172
|
+
headers: {
|
|
173
|
+
'Content-Type': 'application/json',
|
|
174
|
+
'x-api-key': config.anthropicApiKey,
|
|
175
|
+
'anthropic-version': '2023-06-01',
|
|
176
|
+
},
|
|
177
|
+
body: JSON.stringify({
|
|
178
|
+
model: 'claude-sonnet-4-20250514',
|
|
179
|
+
max_tokens: 2048,
|
|
180
|
+
messages: [{ role: 'user', content: prompt }],
|
|
181
|
+
}),
|
|
182
|
+
signal: AbortSignal.timeout(30000),
|
|
183
|
+
});
|
|
184
|
+
if (!response.ok)
|
|
185
|
+
throw new Error(`Anthropic API error: ${response.status}`);
|
|
186
|
+
const data = await response.json();
|
|
187
|
+
text = data.content.find(c => c.type === 'text')?.text ?? '';
|
|
188
|
+
}
|
|
189
|
+
else {
|
|
190
|
+
const response = await fetch('https://api.openai.com/v1/chat/completions', {
|
|
191
|
+
method: 'POST',
|
|
192
|
+
headers: {
|
|
193
|
+
'Content-Type': 'application/json',
|
|
194
|
+
'Authorization': `Bearer ${config.openaiApiKey}`,
|
|
195
|
+
},
|
|
196
|
+
body: JSON.stringify({
|
|
197
|
+
model: 'gpt-4o-mini',
|
|
198
|
+
messages: [{ role: 'user', content: prompt }],
|
|
199
|
+
max_tokens: 2048,
|
|
200
|
+
}),
|
|
201
|
+
signal: AbortSignal.timeout(30000),
|
|
202
|
+
});
|
|
203
|
+
if (!response.ok)
|
|
204
|
+
throw new Error(`OpenAI API error: ${response.status}`);
|
|
205
|
+
const data = await response.json();
|
|
206
|
+
text = data.choices[0]?.message?.content ?? '';
|
|
207
|
+
}
|
|
208
|
+
// Parse two SQL blocks from response
|
|
209
|
+
const sqlBlocks = [...text.matchAll(/```sql\n([\s\S]*?)```/g)].map(m => m[1].trim());
|
|
210
|
+
if (sqlBlocks.length < 2) {
|
|
211
|
+
// Try to split on common markers
|
|
212
|
+
const upMatch = text.match(/(?:UP|up|Up)[:\s]*\n*```sql\n([\s\S]*?)```/);
|
|
213
|
+
const downMatch = text.match(/(?:DOWN|down|Down)[:\s]*\n*```sql\n([\s\S]*?)```/);
|
|
214
|
+
return {
|
|
215
|
+
upSql: upMatch?.[1]?.trim() || sqlBlocks[0] || '-- Could not generate up migration',
|
|
216
|
+
downSql: downMatch?.[1]?.trim() || sqlBlocks[1] || '-- Could not generate down migration',
|
|
217
|
+
};
|
|
218
|
+
}
|
|
219
|
+
return {
|
|
220
|
+
upSql: sqlBlocks[0],
|
|
221
|
+
downSql: sqlBlocks[1],
|
|
222
|
+
};
|
|
223
|
+
}
|
|
224
|
+
//# sourceMappingURL=ai-query.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ai-query.js","sourceRoot":"","sources":["../../src/lib/ai-query.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAG3D,uEAAuE;AAEvE,SAAS,kBAAkB,CAAC,YAAuB;IACjD,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,MAAM,QAAQ,GAAG,YAAY;QAC3B,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACnD,CAAC,CAAC,MAAM,CAAC;IAEX,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAC9D,CAAC;AAED,SAAS,iBAAiB,CAAC,CAAY;IACrC,MAAM,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;QAC7B,MAAM,KAAK,GAAG,CAAC,KAAK,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;QACxC,IAAI,CAAC,CAAC,UAAU;YAAE,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC5C,IAAI,CAAC,CAAC,CAAC,QAAQ;YAAE,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACxC,IAAI,CAAC,CAAC,YAAY,KAAK,IAAI;YAAE,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC;QACrE,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACzB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAEf,OAAO,SAAS,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,QAAQ,YAAY,IAAI,EAAE,CAAC;AAC1D,CAAC;AASD,KAAK,UAAU,aAAa,CAAC,MAAc,EAAE,MAAc;IACzD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,uCAAuC,EAAE;QACpE,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;YAClC,WAAW,EAAE,MAAM;YACnB,mBAAmB,EAAE,YAAY;SAClC;QACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;YACnB,KAAK,EAAE,0BAA0B;YACjC,UAAU,EAAE,IAAI;YAChB,QAAQ,EAAE,CAAC;oBACT,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,MAAM;iBAChB,CAAC;SACH,CAAC;QACF,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC;KACnC,CAAC,CAAC;IAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,wBAAwB,QAAQ,CAAC,MAAM,MAAM,IAAI,EAAE,CAAC,CAAC;IACvE,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAE/B,CAAC;IAEF,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;IACnE,OAAO,eAAe,CAAC,IAAI,CAAC,CAAC;AAC/B,CAAC;AAED,KAAK,UAAU,UAAU,CAAC,MAAc,EAAE,MAAc;IACtD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,4CAA4C,EAAE;QACzE,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;YAClC,eAAe,EAAE,UAAU,MAAM,EAAE;SACpC;QACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;YACnB,KAAK,EAAE,aAAa;YACpB,QAAQ,EAAE,CAAC;oBACT,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,MAAM;iBAChB,CAAC;YACF,UAAU,EAAE,IAAI;SACjB,CAAC;QACF,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC;KACnC,CAAC,CAAC;IAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,CAAC,MAAM,MAAM,IAAI,EAAE,CAAC,CAAC;IACpE,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAE/B,CAAC;IAEF,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;IACrD,OAAO,eAAe,CAAC,IAAI,CAAC,CAAC;AAC/B,CAAC;AAED,SAAS,eAAe,CAAC,IAAY;IACnC,2CAA2C;IAC3C,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC;IACrD,MAAM,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;IAErE,6DAA6D;IAC7D,MAAM,WAAW,GAAG,IAAI;SACrB,OAAO,CAAC,mBAAmB,EAAE,EAAE,CAAC;SAChC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC;SACzB,IAAI,EAAE;WACJ,wCAAwC,CAAC;IAE9C,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,MAAM,IAAI,KAAK,CAAC,qEAAqE,CAAC,CAAC;IACzF,CAAC;IAED,OAAO,EAAE,GAAG,EAAE,WAAW,EAAE,CAAC;AAC9B,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAY;IACtC,kDAAkD;IAClD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAC;IAC1D,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;AACtC,CAAC;AAED,uEAAuE;AAEvE,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,QAAgB,EAChB,YAAuB;IAEvB,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAE3B,IAAI,CAAC,MAAM,CAAC,eAAe,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;QACpD,MAAM,IAAI,KAAK,CACb,kDAAkD;YAClD,8DAA8D,CAC/D,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAG,kBAAkB,CAAC,YAAY,CAAC,CAAC;IAEhD,MAAM,MAAM,GAAG;;;EAGf,MAAM;;YAEI,QAAQ;;;;;;;;;;0EAUsD,CAAC;IAEzE,IAAI,QAAoB,CAAC;IAEzB,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;QAC3B,QAAQ,GAAG,MAAM,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC;IACjE,CAAC;SAAM,CAAC;QACN,QAAQ,GAAG,MAAM,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,YAAa,CAAC,CAAC;IAC5D,CAAC;IAED,4BAA4B;IAC5B,MAAM,OAAO,GAAG,YAAY,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IAE3C,OAAO;QACL,GAAG,EAAE,QAAQ,CAAC,GAAG;QACjB,WAAW,EAAE,QAAQ,CAAC,WAAW;QACjC,OAAO;KACR,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,WAAmB,EACnB,UAAkB,QAAQ;IAE1B,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAE3B,IAAI,CAAC,MAAM,CAAC,eAAe,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;QACpD,MAAM,IAAI,KAAK,CACb,+CAA+C;YAC/C,8DAA8D,CAC/D,CAAC;IACJ,CAAC;IAED,IAAI,aAAa,GAAG,EAAE,CAAC;IACvB,IAAI,CAAC;QACH,aAAa,GAAG,kBAAkB,EAAE,CAAC;IACvC,CAAC;IAAC,MAAM,CAAC;QACP,sEAAsE;IACxE,CAAC;IAED,MAAM,MAAM,GAAG;;EAEf,aAAa,CAAC,CAAC,CAAC,qBAAqB,aAAa,MAAM,CAAC,CAAC,CAAC,EAAE,YAAY,OAAO;sBAC5D,WAAW;;;;;;;;;;;;;;;;;+BAiBF,OAAO,UAAU,CAAC;IAE/C,IAAI,IAAY,CAAC;IAEjB,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;QAC3B,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,uCAAuC,EAAE;YACpE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,WAAW,EAAE,MAAM,CAAC,eAAe;gBACnC,mBAAmB,EAAE,YAAY;aAClC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,0BAA0B;gBACjC,UAAU,EAAE,IAAI;gBAChB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;aAC9C,CAAC;YACF,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC;SACnC,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,EAAE;YAAE,MAAM,IAAI,KAAK,CAAC,wBAAwB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAC7E,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAwD,CAAC;QACzF,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;IAC/D,CAAC;SAAM,CAAC;QACN,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,4CAA4C,EAAE;YACzE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,eAAe,EAAE,UAAU,MAAM,CAAC,YAAY,EAAE;aACjD;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,aAAa;gBACpB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;gBAC7C,UAAU,EAAE,IAAI;aACjB,CAAC;YACF,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC;SACnC,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,EAAE;YAAE,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAC1E,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAA0D,CAAC;QAC3F,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;IACjD,CAAC;IAED,qCAAqC;IACrC,MAAM,SAAS,GAAG,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IAErF,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,iCAAiC;QACjC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,4CAA4C,CAAC,CAAC;QACzE,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAC;QAEjF,OAAO;YACL,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,SAAS,CAAC,CAAC,CAAC,IAAI,oCAAoC;YACnF,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,SAAS,CAAC,CAAC,CAAC,IAAI,sCAAsC;SAC1F,CAAC;IACJ,CAAC;IAED,OAAO;QACL,KAAK,EAAE,SAAS,CAAC,CAAC,CAAC;QACnB,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC;KACtB,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../src/lib/config.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAW/C,wBAAgB,SAAS,IAAI,WAAW,CAcvC;AAED,wBAAgB,UAAU,IAAI,MAAM,CAEnC"}
|
|
@@ -0,0 +1,30 @@
|
|
|
1
|
+
import { homedir } from 'node:os';
|
|
2
|
+
import { join } from 'node:path';
|
|
3
|
+
import { mkdirSync } from 'node:fs';
|
|
4
|
+
const DATA_DIR = join(homedir(), '.truss');
|
|
5
|
+
// Ensure data directory exists on import
|
|
6
|
+
try {
|
|
7
|
+
mkdirSync(DATA_DIR, { recursive: true });
|
|
8
|
+
}
|
|
9
|
+
catch {
|
|
10
|
+
// ignore -- directory already exists
|
|
11
|
+
}
|
|
12
|
+
export function getConfig() {
|
|
13
|
+
return {
|
|
14
|
+
licenseKey: process.env.TRUSS_LICENSE_KEY,
|
|
15
|
+
databaseUrl: process.env.DATABASE_URL,
|
|
16
|
+
dbHost: process.env.DB_HOST,
|
|
17
|
+
dbPort: process.env.DB_PORT ? parseInt(process.env.DB_PORT, 10) : undefined,
|
|
18
|
+
dbName: process.env.DB_NAME,
|
|
19
|
+
dbUser: process.env.DB_USER,
|
|
20
|
+
dbPassword: process.env.DB_PASSWORD,
|
|
21
|
+
anthropicApiKey: process.env.ANTHROPIC_API_KEY,
|
|
22
|
+
openaiApiKey: process.env.OPENAI_API_KEY,
|
|
23
|
+
trussApiBaseUrl: process.env.TRUSS_API_BASE_URL || 'https://api.truss.dev',
|
|
24
|
+
dataDir: DATA_DIR,
|
|
25
|
+
};
|
|
26
|
+
}
|
|
27
|
+
export function getDataDir() {
|
|
28
|
+
return DATA_DIR;
|
|
29
|
+
}
|
|
30
|
+
//# sourceMappingURL=config.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/lib/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAGpC,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,QAAQ,CAAC,CAAC;AAE3C,yCAAyC;AACzC,IAAI,CAAC;IACH,SAAS,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;AAC3C,CAAC;AAAC,MAAM,CAAC;IACP,qCAAqC;AACvC,CAAC;AAED,MAAM,UAAU,SAAS;IACvB,OAAO;QACL,UAAU,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;QACzC,WAAW,EAAE,OAAO,CAAC,GAAG,CAAC,YAAY;QACrC,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,OAAO;QAC3B,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS;QAC3E,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,OAAO;QAC3B,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,OAAO;QAC3B,UAAU,EAAE,OAAO,CAAC,GAAG,CAAC,WAAW;QACnC,eAAe,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;QAC9C,YAAY,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;QACxC,eAAe,EAAE,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,uBAAuB;QAC1E,OAAO,EAAE,QAAQ;KAClB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,UAAU;IACxB,OAAO,QAAQ,CAAC;AAClB,CAAC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import Database from 'better-sqlite3';
|
|
2
|
+
import type { DatabaseDialect, ConnectionInfo, TableInfo, TableSchema, QueryResult } from '../types.js';
|
|
3
|
+
export declare function connectSqlite(dbPath: string): ConnectionInfo;
|
|
4
|
+
export declare function getDb(): Database.Database;
|
|
5
|
+
export declare function getDialect(): DatabaseDialect;
|
|
6
|
+
export declare function isConnected(): boolean;
|
|
7
|
+
export declare function closeConnection(): void;
|
|
8
|
+
export declare function listTables(): TableInfo[];
|
|
9
|
+
export declare function describeTable(tableName: string): TableSchema;
|
|
10
|
+
export declare function isWriteQuery(sql: string): boolean;
|
|
11
|
+
export declare function executeQuery(sql: string, params?: unknown[]): QueryResult;
|
|
12
|
+
export declare function executeReadOnlyQuery(sql: string, params?: unknown[]): QueryResult;
|
|
13
|
+
export declare function explainQuery(sql: string): string;
|
|
14
|
+
//# sourceMappingURL=connection.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"connection.d.ts","sourceRoot":"","sources":["../../src/lib/connection.ts"],"names":[],"mappings":"AAEA,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,KAAK,EACV,eAAe,EACf,cAAc,EACd,SAAS,EACT,WAAW,EAIX,WAAW,EACZ,MAAM,aAAa,CAAC;AAUrB,wBAAgB,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,cAAc,CA8B5D;AAID,wBAAgB,KAAK,IAAI,QAAQ,CAAC,QAAQ,CAKzC;AAED,wBAAgB,UAAU,IAAI,eAAe,CAE5C;AAED,wBAAgB,WAAW,IAAI,OAAO,CAErC;AAED,wBAAgB,eAAe,IAAI,IAAI,CAMtC;AAYD,wBAAgB,UAAU,IAAI,SAAS,EAAE,CAiBxC;AAED,wBAAgB,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,WAAW,CA8B5D;AAuED,wBAAgB,YAAY,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAEjD;AAED,wBAAgB,YAAY,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,GAAE,OAAO,EAAO,GAAG,WAAW,CAyB7E;AAED,wBAAgB,oBAAoB,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,GAAE,OAAO,EAAO,GAAG,WAAW,CAQrF;AAID,wBAAgB,YAAY,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAUhD"}
|