@rangerchaz/aimem 0.1.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 +380 -0
- package/dist/cli/commands/git.d.ts +6 -0
- package/dist/cli/commands/git.d.ts.map +1 -0
- package/dist/cli/commands/git.js +298 -0
- package/dist/cli/commands/git.js.map +1 -0
- package/dist/cli/commands/hook-session-end.d.ts +7 -0
- package/dist/cli/commands/hook-session-end.d.ts.map +1 -0
- package/dist/cli/commands/hook-session-end.js +109 -0
- package/dist/cli/commands/hook-session-end.js.map +1 -0
- package/dist/cli/commands/hook-session-start.d.ts +7 -0
- package/dist/cli/commands/hook-session-start.d.ts.map +1 -0
- package/dist/cli/commands/hook-session-start.js +116 -0
- package/dist/cli/commands/hook-session-start.js.map +1 -0
- package/dist/cli/commands/import.d.ts +14 -0
- package/dist/cli/commands/import.d.ts.map +1 -0
- package/dist/cli/commands/import.js +527 -0
- package/dist/cli/commands/import.js.map +1 -0
- package/dist/cli/commands/init.d.ts +2 -0
- package/dist/cli/commands/init.d.ts.map +1 -0
- package/dist/cli/commands/init.js +32 -0
- package/dist/cli/commands/init.js.map +1 -0
- package/dist/cli/commands/mcp-serve.d.ts +2 -0
- package/dist/cli/commands/mcp-serve.d.ts.map +1 -0
- package/dist/cli/commands/mcp-serve.js +5 -0
- package/dist/cli/commands/mcp-serve.js.map +1 -0
- package/dist/cli/commands/query.d.ts +8 -0
- package/dist/cli/commands/query.d.ts.map +1 -0
- package/dist/cli/commands/query.js +83 -0
- package/dist/cli/commands/query.js.map +1 -0
- package/dist/cli/commands/setup.d.ts +10 -0
- package/dist/cli/commands/setup.d.ts.map +1 -0
- package/dist/cli/commands/setup.js +504 -0
- package/dist/cli/commands/setup.js.map +1 -0
- package/dist/cli/commands/start.d.ts +8 -0
- package/dist/cli/commands/start.d.ts.map +1 -0
- package/dist/cli/commands/start.js +90 -0
- package/dist/cli/commands/start.js.map +1 -0
- package/dist/cli/commands/status.d.ts +2 -0
- package/dist/cli/commands/status.d.ts.map +1 -0
- package/dist/cli/commands/status.js +85 -0
- package/dist/cli/commands/status.js.map +1 -0
- package/dist/cli/commands/stop.d.ts +7 -0
- package/dist/cli/commands/stop.d.ts.map +1 -0
- package/dist/cli/commands/stop.js +46 -0
- package/dist/cli/commands/stop.js.map +1 -0
- package/dist/cli/commands/visualize.d.ts +8 -0
- package/dist/cli/commands/visualize.d.ts.map +1 -0
- package/dist/cli/commands/visualize.js +96 -0
- package/dist/cli/commands/visualize.js.map +1 -0
- package/dist/cli/index.d.ts +3 -0
- package/dist/cli/index.d.ts.map +1 -0
- package/dist/cli/index.js +114 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/db/index.d.ts +55 -0
- package/dist/db/index.d.ts.map +1 -0
- package/dist/db/index.js +464 -0
- package/dist/db/index.js.map +1 -0
- package/dist/db/schema.d.ts +4 -0
- package/dist/db/schema.d.ts.map +1 -0
- package/dist/db/schema.js +200 -0
- package/dist/db/schema.js.map +1 -0
- package/dist/extractor/index.d.ts +27 -0
- package/dist/extractor/index.d.ts.map +1 -0
- package/dist/extractor/index.js +227 -0
- package/dist/extractor/index.js.map +1 -0
- package/dist/git/extractor.d.ts +30 -0
- package/dist/git/extractor.d.ts.map +1 -0
- package/dist/git/extractor.js +126 -0
- package/dist/git/extractor.js.map +1 -0
- package/dist/git/hooks.d.ts +36 -0
- package/dist/git/hooks.d.ts.map +1 -0
- package/dist/git/hooks.js +142 -0
- package/dist/git/hooks.js.map +1 -0
- package/dist/git/index.d.ts +69 -0
- package/dist/git/index.d.ts.map +1 -0
- package/dist/git/index.js +250 -0
- package/dist/git/index.js.map +1 -0
- package/dist/indexer/index.d.ts +20 -0
- package/dist/indexer/index.d.ts.map +1 -0
- package/dist/indexer/index.js +173 -0
- package/dist/indexer/index.js.map +1 -0
- package/dist/indexer/parsers/base.d.ts +19 -0
- package/dist/indexer/parsers/base.d.ts.map +1 -0
- package/dist/indexer/parsers/base.js +46 -0
- package/dist/indexer/parsers/base.js.map +1 -0
- package/dist/indexer/parsers/cpp.d.ts +3 -0
- package/dist/indexer/parsers/cpp.d.ts.map +1 -0
- package/dist/indexer/parsers/cpp.js +180 -0
- package/dist/indexer/parsers/cpp.js.map +1 -0
- package/dist/indexer/parsers/go.d.ts +3 -0
- package/dist/indexer/parsers/go.d.ts.map +1 -0
- package/dist/indexer/parsers/go.js +98 -0
- package/dist/indexer/parsers/go.js.map +1 -0
- package/dist/indexer/parsers/java.d.ts +3 -0
- package/dist/indexer/parsers/java.d.ts.map +1 -0
- package/dist/indexer/parsers/java.js +204 -0
- package/dist/indexer/parsers/java.js.map +1 -0
- package/dist/indexer/parsers/javascript.d.ts +3 -0
- package/dist/indexer/parsers/javascript.d.ts.map +1 -0
- package/dist/indexer/parsers/javascript.js +157 -0
- package/dist/indexer/parsers/javascript.js.map +1 -0
- package/dist/indexer/parsers/kotlin.d.ts +3 -0
- package/dist/indexer/parsers/kotlin.d.ts.map +1 -0
- package/dist/indexer/parsers/kotlin.js +182 -0
- package/dist/indexer/parsers/kotlin.js.map +1 -0
- package/dist/indexer/parsers/php.d.ts +3 -0
- package/dist/indexer/parsers/php.d.ts.map +1 -0
- package/dist/indexer/parsers/php.js +190 -0
- package/dist/indexer/parsers/php.js.map +1 -0
- package/dist/indexer/parsers/python.d.ts +3 -0
- package/dist/indexer/parsers/python.d.ts.map +1 -0
- package/dist/indexer/parsers/python.js +101 -0
- package/dist/indexer/parsers/python.js.map +1 -0
- package/dist/indexer/parsers/ruby.d.ts +3 -0
- package/dist/indexer/parsers/ruby.d.ts.map +1 -0
- package/dist/indexer/parsers/ruby.js +92 -0
- package/dist/indexer/parsers/ruby.js.map +1 -0
- package/dist/indexer/parsers/rust.d.ts +3 -0
- package/dist/indexer/parsers/rust.d.ts.map +1 -0
- package/dist/indexer/parsers/rust.js +190 -0
- package/dist/indexer/parsers/rust.js.map +1 -0
- package/dist/indexer/watcher-daemon.d.ts +2 -0
- package/dist/indexer/watcher-daemon.d.ts.map +1 -0
- package/dist/indexer/watcher-daemon.js +27 -0
- package/dist/indexer/watcher-daemon.js.map +1 -0
- package/dist/indexer/watcher.d.ts +7 -0
- package/dist/indexer/watcher.d.ts.map +1 -0
- package/dist/indexer/watcher.js +77 -0
- package/dist/indexer/watcher.js.map +1 -0
- package/dist/mcp/server.d.ts +2 -0
- package/dist/mcp/server.d.ts.map +1 -0
- package/dist/mcp/server.js +241 -0
- package/dist/mcp/server.js.map +1 -0
- package/dist/proxy/interceptor-mockttp.d.ts +27 -0
- package/dist/proxy/interceptor-mockttp.d.ts.map +1 -0
- package/dist/proxy/interceptor-mockttp.js +274 -0
- package/dist/proxy/interceptor-mockttp.js.map +1 -0
- package/dist/proxy/proxy-daemon.d.ts +5 -0
- package/dist/proxy/proxy-daemon.d.ts.map +1 -0
- package/dist/proxy/proxy-daemon.js +26 -0
- package/dist/proxy/proxy-daemon.js.map +1 -0
- package/dist/query/index.d.ts +32 -0
- package/dist/query/index.d.ts.map +1 -0
- package/dist/query/index.js +135 -0
- package/dist/query/index.js.map +1 -0
- package/dist/types/index.d.ts +89 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +3 -0
- package/dist/types/index.js.map +1 -0
- package/dist/visualize/index.d.ts +144 -0
- package/dist/visualize/index.d.ts.map +1 -0
- package/dist/visualize/index.js +707 -0
- package/dist/visualize/index.js.map +1 -0
- package/dist/visualize/server.d.ts +7 -0
- package/dist/visualize/server.d.ts.map +1 -0
- package/dist/visualize/server.js +77 -0
- package/dist/visualize/server.js.map +1 -0
- package/dist/visualize/template.d.ts +3 -0
- package/dist/visualize/template.d.ts.map +1 -0
- package/dist/visualize/template.js +3465 -0
- package/dist/visualize/template.js.map +1 -0
- package/package.json +56 -0
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
// Watcher daemon - runs in background
|
|
2
|
+
import { startWatcher } from './watcher.js';
|
|
3
|
+
console.log('[aimem watcher] Starting...');
|
|
4
|
+
const stop = startWatcher({
|
|
5
|
+
onIndex: (projectId, path, structures) => {
|
|
6
|
+
console.log(`[aimem watcher] Indexed ${path} (${structures} structures)`);
|
|
7
|
+
},
|
|
8
|
+
onDelete: (projectId, path) => {
|
|
9
|
+
console.log(`[aimem watcher] Removed ${path}`);
|
|
10
|
+
},
|
|
11
|
+
onError: (error) => {
|
|
12
|
+
console.error('[aimem watcher] Error:', error.message);
|
|
13
|
+
},
|
|
14
|
+
});
|
|
15
|
+
// Handle shutdown
|
|
16
|
+
process.on('SIGTERM', () => {
|
|
17
|
+
console.log('[aimem watcher] Shutting down...');
|
|
18
|
+
stop();
|
|
19
|
+
process.exit(0);
|
|
20
|
+
});
|
|
21
|
+
process.on('SIGINT', () => {
|
|
22
|
+
console.log('[aimem watcher] Shutting down...');
|
|
23
|
+
stop();
|
|
24
|
+
process.exit(0);
|
|
25
|
+
});
|
|
26
|
+
console.log('[aimem watcher] Ready');
|
|
27
|
+
//# sourceMappingURL=watcher-daemon.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"watcher-daemon.js","sourceRoot":"","sources":["../../src/indexer/watcher-daemon.ts"],"names":[],"mappings":"AAAA,sCAAsC;AACtC,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAE5C,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;AAE3C,MAAM,IAAI,GAAG,YAAY,CAAC;IACxB,OAAO,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE;QACvC,OAAO,CAAC,GAAG,CAAC,2BAA2B,IAAI,KAAK,UAAU,cAAc,CAAC,CAAC;IAC5E,CAAC;IACD,QAAQ,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,EAAE;QAC5B,OAAO,CAAC,GAAG,CAAC,2BAA2B,IAAI,EAAE,CAAC,CAAC;IACjD,CAAC;IACD,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;QACjB,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IACzD,CAAC;CACF,CAAC,CAAC;AAEH,kBAAkB;AAClB,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;IACzB,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;IAChD,IAAI,EAAE,CAAC;IACP,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE;IACxB,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;IAChD,IAAI,EAAE,CAAC;IACP,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
export interface WatcherOptions {
|
|
2
|
+
onIndex?: (projectId: number, path: string, structures: number) => void;
|
|
3
|
+
onDelete?: (projectId: number, path: string) => void;
|
|
4
|
+
onError?: (error: Error) => void;
|
|
5
|
+
}
|
|
6
|
+
export declare function startWatcher(options?: WatcherOptions): () => void;
|
|
7
|
+
//# sourceMappingURL=watcher.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"watcher.d.ts","sourceRoot":"","sources":["../../src/indexer/watcher.ts"],"names":[],"mappings":"AAqBA,MAAM,WAAW,cAAc;IAC7B,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,KAAK,IAAI,CAAC;IACxE,QAAQ,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;IACrD,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CAClC;AAED,wBAAgB,YAAY,CAAC,OAAO,GAAE,cAAmB,GAAG,MAAM,IAAI,CA8DrE"}
|
|
@@ -0,0 +1,77 @@
|
|
|
1
|
+
import { watch } from 'chokidar';
|
|
2
|
+
import { getAllProjects, getFileByPath, deleteFile } from '../db/index.js';
|
|
3
|
+
import { indexFile, getSupportedExtensions } from './index.js';
|
|
4
|
+
const IGNORE_PATTERNS = [
|
|
5
|
+
'**/node_modules/**',
|
|
6
|
+
'**/.git/**',
|
|
7
|
+
'**/dist/**',
|
|
8
|
+
'**/build/**',
|
|
9
|
+
'**/__pycache__/**',
|
|
10
|
+
'**/.venv/**',
|
|
11
|
+
'**/venv/**',
|
|
12
|
+
'**/vendor/**',
|
|
13
|
+
'**/*.min.js',
|
|
14
|
+
'**/*.bundle.js',
|
|
15
|
+
'**/coverage/**',
|
|
16
|
+
'**/.next/**',
|
|
17
|
+
'**/.nuxt/**',
|
|
18
|
+
];
|
|
19
|
+
export function startWatcher(options = {}) {
|
|
20
|
+
const projects = getAllProjects();
|
|
21
|
+
const watchers = [];
|
|
22
|
+
const extensions = getSupportedExtensions();
|
|
23
|
+
const extPattern = extensions.map(e => e.slice(1)).join(',');
|
|
24
|
+
for (const project of projects) {
|
|
25
|
+
const watcher = watch(`**/*.{${extPattern}}`, {
|
|
26
|
+
cwd: project.path,
|
|
27
|
+
ignored: IGNORE_PATTERNS,
|
|
28
|
+
persistent: true,
|
|
29
|
+
ignoreInitial: true,
|
|
30
|
+
awaitWriteFinish: {
|
|
31
|
+
stabilityThreshold: 300,
|
|
32
|
+
pollInterval: 100,
|
|
33
|
+
},
|
|
34
|
+
});
|
|
35
|
+
watcher.on('add', async (path) => {
|
|
36
|
+
try {
|
|
37
|
+
const count = await indexFile(project.id, project.path, path);
|
|
38
|
+
options.onIndex?.(project.id, path, count);
|
|
39
|
+
}
|
|
40
|
+
catch (err) {
|
|
41
|
+
options.onError?.(err);
|
|
42
|
+
}
|
|
43
|
+
});
|
|
44
|
+
watcher.on('change', async (path) => {
|
|
45
|
+
try {
|
|
46
|
+
const count = await indexFile(project.id, project.path, path);
|
|
47
|
+
options.onIndex?.(project.id, path, count);
|
|
48
|
+
}
|
|
49
|
+
catch (err) {
|
|
50
|
+
options.onError?.(err);
|
|
51
|
+
}
|
|
52
|
+
});
|
|
53
|
+
watcher.on('unlink', async (path) => {
|
|
54
|
+
try {
|
|
55
|
+
const file = getFileByPath(project.id, path);
|
|
56
|
+
if (file) {
|
|
57
|
+
deleteFile(file.id);
|
|
58
|
+
options.onDelete?.(project.id, path);
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
catch (err) {
|
|
62
|
+
options.onError?.(err);
|
|
63
|
+
}
|
|
64
|
+
});
|
|
65
|
+
watcher.on('error', (err) => {
|
|
66
|
+
options.onError?.(err instanceof Error ? err : new Error(String(err)));
|
|
67
|
+
});
|
|
68
|
+
watchers.push(watcher);
|
|
69
|
+
}
|
|
70
|
+
// Return cleanup function
|
|
71
|
+
return () => {
|
|
72
|
+
for (const watcher of watchers) {
|
|
73
|
+
watcher.close();
|
|
74
|
+
}
|
|
75
|
+
};
|
|
76
|
+
}
|
|
77
|
+
//# sourceMappingURL=watcher.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"watcher.js","sourceRoot":"","sources":["../../src/indexer/watcher.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,UAAU,CAAC;AAEjC,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC3E,OAAO,EAAE,SAAS,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AAE/D,MAAM,eAAe,GAAG;IACtB,oBAAoB;IACpB,YAAY;IACZ,YAAY;IACZ,aAAa;IACb,mBAAmB;IACnB,aAAa;IACb,YAAY;IACZ,cAAc;IACd,aAAa;IACb,gBAAgB;IAChB,gBAAgB;IAChB,aAAa;IACb,aAAa;CACd,CAAC;AAQF,MAAM,UAAU,YAAY,CAAC,UAA0B,EAAE;IACvD,MAAM,QAAQ,GAAG,cAAc,EAAE,CAAC;IAClC,MAAM,QAAQ,GAA+B,EAAE,CAAC;IAEhD,MAAM,UAAU,GAAG,sBAAsB,EAAE,CAAC;IAC5C,MAAM,UAAU,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAE7D,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,UAAU,GAAG,EAAE;YAC5C,GAAG,EAAE,OAAO,CAAC,IAAI;YACjB,OAAO,EAAE,eAAe;YACxB,UAAU,EAAE,IAAI;YAChB,aAAa,EAAE,IAAI;YACnB,gBAAgB,EAAE;gBAChB,kBAAkB,EAAE,GAAG;gBACvB,YAAY,EAAE,GAAG;aAClB;SACF,CAAC,CAAC;QAEH,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAC/B,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAC9D,OAAO,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;YAC7C,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,OAAO,EAAE,CAAC,GAAY,CAAC,CAAC;YAClC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAClC,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAC9D,OAAO,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;YAC7C,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,OAAO,EAAE,CAAC,GAAY,CAAC,CAAC;YAClC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAClC,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,aAAa,CAAC,OAAO,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;gBAC7C,IAAI,IAAI,EAAE,CAAC;oBACT,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBACpB,OAAO,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;gBACvC,CAAC;YACH,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,OAAO,EAAE,CAAC,GAAY,CAAC,CAAC;YAClC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAY,EAAE,EAAE;YACnC,OAAO,CAAC,OAAO,EAAE,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACzB,CAAC;IAED,0BAA0B;IAC1B,OAAO,GAAG,EAAE;QACV,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,OAAO,CAAC,KAAK,EAAE,CAAC;QAClB,CAAC;IACH,CAAC,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/mcp/server.ts"],"names":[],"mappings":"AAsBA,wBAAsB,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC,CAqQpD"}
|
|
@@ -0,0 +1,241 @@
|
|
|
1
|
+
import { Server } from '@modelcontextprotocol/sdk/server/index.js';
|
|
2
|
+
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
|
|
3
|
+
import { CallToolRequestSchema, ListToolsRequestSchema, } from '@modelcontextprotocol/sdk/types.js';
|
|
4
|
+
import { searchStructures, searchConversations, structureExists, fileExists, searchExtractions, getStructuresByName, getLinksTo, getExtraction, getFile, getConversationById, searchFullConversations, findProjectForPath, searchCommits, } from '../db/index.js';
|
|
5
|
+
export async function startMcpServer() {
|
|
6
|
+
const server = new Server({
|
|
7
|
+
name: 'aimem',
|
|
8
|
+
version: '0.1.0',
|
|
9
|
+
}, {
|
|
10
|
+
capabilities: {
|
|
11
|
+
tools: {},
|
|
12
|
+
},
|
|
13
|
+
});
|
|
14
|
+
// List available tools
|
|
15
|
+
server.setRequestHandler(ListToolsRequestSchema, async () => ({
|
|
16
|
+
tools: [
|
|
17
|
+
{
|
|
18
|
+
name: 'aimem_query',
|
|
19
|
+
description: 'Search code, conversations, decisions, and commits',
|
|
20
|
+
inputSchema: {
|
|
21
|
+
type: 'object',
|
|
22
|
+
properties: {
|
|
23
|
+
query: { type: 'string', description: 'Function name, class name, or keyword' },
|
|
24
|
+
type: { type: 'string', enum: ['all', 'structures', 'conversations', 'decisions', 'commits'], default: 'all' },
|
|
25
|
+
limit: { type: 'number', default: 10 },
|
|
26
|
+
},
|
|
27
|
+
required: ['query'],
|
|
28
|
+
},
|
|
29
|
+
},
|
|
30
|
+
{
|
|
31
|
+
name: 'aimem_verify',
|
|
32
|
+
description: 'Check if a function, class, or file exists',
|
|
33
|
+
inputSchema: {
|
|
34
|
+
type: 'object',
|
|
35
|
+
properties: {
|
|
36
|
+
name: { type: 'string', description: 'Name to verify' },
|
|
37
|
+
type: { type: 'string', enum: ['structure', 'file'], default: 'structure' },
|
|
38
|
+
},
|
|
39
|
+
required: ['name'],
|
|
40
|
+
},
|
|
41
|
+
},
|
|
42
|
+
{
|
|
43
|
+
name: 'aimem_conversations',
|
|
44
|
+
description: 'Search past AI conversation history',
|
|
45
|
+
inputSchema: {
|
|
46
|
+
type: 'object',
|
|
47
|
+
properties: {
|
|
48
|
+
query: { type: 'string', description: 'Search keywords' },
|
|
49
|
+
id: { type: 'number', description: 'Get by ID' },
|
|
50
|
+
limit: { type: 'number', default: 5 },
|
|
51
|
+
},
|
|
52
|
+
},
|
|
53
|
+
},
|
|
54
|
+
],
|
|
55
|
+
}));
|
|
56
|
+
// Handle tool calls
|
|
57
|
+
server.setRequestHandler(CallToolRequestSchema, async (request) => {
|
|
58
|
+
const { name, arguments: args } = request.params;
|
|
59
|
+
try {
|
|
60
|
+
switch (name) {
|
|
61
|
+
case 'aimem_query': {
|
|
62
|
+
const query = args?.query;
|
|
63
|
+
const type = args?.type || 'all';
|
|
64
|
+
const limit = args?.limit || 10;
|
|
65
|
+
const results = {};
|
|
66
|
+
if (type === 'all' || type === 'structures') {
|
|
67
|
+
const structures = searchStructures(query, limit);
|
|
68
|
+
results.structures = structures.map(s => ({
|
|
69
|
+
type: s.type,
|
|
70
|
+
name: s.name,
|
|
71
|
+
file: getFile(s.file_id)?.path,
|
|
72
|
+
line: s.line_start,
|
|
73
|
+
signature: s.signature,
|
|
74
|
+
author: s.last_author || undefined,
|
|
75
|
+
commit: s.last_commit_hash?.slice(0, 7) || undefined,
|
|
76
|
+
}));
|
|
77
|
+
}
|
|
78
|
+
if (type === 'all' || type === 'conversations') {
|
|
79
|
+
const conversations = searchConversations(query, limit);
|
|
80
|
+
results.conversations = conversations.map(c => ({
|
|
81
|
+
id: c.id,
|
|
82
|
+
timestamp: c.timestamp,
|
|
83
|
+
model: c.model,
|
|
84
|
+
summary: c.summary || c.raw_content.slice(0, 200),
|
|
85
|
+
}));
|
|
86
|
+
}
|
|
87
|
+
if (type === 'all' || type === 'decisions') {
|
|
88
|
+
const decisions = [];
|
|
89
|
+
const seen = new Set();
|
|
90
|
+
// First: try to find linked decisions via structure graph
|
|
91
|
+
const structures = getStructuresByName(query);
|
|
92
|
+
for (const s of structures) {
|
|
93
|
+
const links = getLinksTo('structure', s.id);
|
|
94
|
+
for (const link of links) {
|
|
95
|
+
if (link.source_type === 'extraction') {
|
|
96
|
+
const extraction = getExtraction(link.source_id);
|
|
97
|
+
if (extraction && !seen.has(extraction.content)) {
|
|
98
|
+
seen.add(extraction.content);
|
|
99
|
+
decisions.push({
|
|
100
|
+
type: extraction.type,
|
|
101
|
+
content: extraction.content,
|
|
102
|
+
source: 'linked',
|
|
103
|
+
});
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
}
|
|
107
|
+
}
|
|
108
|
+
// Fallback: keyword search in extraction content
|
|
109
|
+
const keywordResults = searchExtractions(query, limit);
|
|
110
|
+
for (const e of keywordResults) {
|
|
111
|
+
if (!seen.has(e.content)) {
|
|
112
|
+
seen.add(e.content);
|
|
113
|
+
decisions.push({
|
|
114
|
+
type: e.type,
|
|
115
|
+
content: e.content,
|
|
116
|
+
source: 'keyword',
|
|
117
|
+
});
|
|
118
|
+
}
|
|
119
|
+
}
|
|
120
|
+
results.decisions = decisions.slice(0, limit);
|
|
121
|
+
}
|
|
122
|
+
if (type === 'all' || type === 'commits') {
|
|
123
|
+
const project = findProjectForPath(process.cwd());
|
|
124
|
+
const commits = searchCommits(query, limit, project?.id);
|
|
125
|
+
results.commits = commits.map(c => ({
|
|
126
|
+
hash: c.short_hash || c.hash.slice(0, 7),
|
|
127
|
+
author: c.author_name,
|
|
128
|
+
date: c.timestamp.split('T')[0],
|
|
129
|
+
subject: c.subject,
|
|
130
|
+
}));
|
|
131
|
+
}
|
|
132
|
+
return {
|
|
133
|
+
content: [{ type: 'text', text: JSON.stringify(results, null, 2) }],
|
|
134
|
+
};
|
|
135
|
+
}
|
|
136
|
+
case 'aimem_verify': {
|
|
137
|
+
const entityName = args?.name;
|
|
138
|
+
const entityType = args?.type || 'structure';
|
|
139
|
+
let exists = false;
|
|
140
|
+
if (entityType === 'structure') {
|
|
141
|
+
exists = structureExists(entityName);
|
|
142
|
+
}
|
|
143
|
+
else if (entityType === 'file') {
|
|
144
|
+
exists = fileExists(entityName);
|
|
145
|
+
}
|
|
146
|
+
return {
|
|
147
|
+
content: [{
|
|
148
|
+
type: 'text',
|
|
149
|
+
text: JSON.stringify({
|
|
150
|
+
name: entityName,
|
|
151
|
+
type: entityType,
|
|
152
|
+
exists,
|
|
153
|
+
message: exists
|
|
154
|
+
? `${entityType} "${entityName}" exists in the codebase`
|
|
155
|
+
: `${entityType} "${entityName}" was NOT found in the codebase`,
|
|
156
|
+
}, null, 2),
|
|
157
|
+
}],
|
|
158
|
+
};
|
|
159
|
+
}
|
|
160
|
+
case 'aimem_conversations': {
|
|
161
|
+
const query = args?.query;
|
|
162
|
+
const id = args?.id;
|
|
163
|
+
const limit = args?.limit || 5;
|
|
164
|
+
// Get specific conversation by ID
|
|
165
|
+
if (id !== undefined) {
|
|
166
|
+
const conversation = getConversationById(id);
|
|
167
|
+
if (!conversation) {
|
|
168
|
+
return {
|
|
169
|
+
content: [{ type: 'text', text: `No conversation found with ID: ${id}` }],
|
|
170
|
+
};
|
|
171
|
+
}
|
|
172
|
+
return {
|
|
173
|
+
content: [{
|
|
174
|
+
type: 'text',
|
|
175
|
+
text: JSON.stringify({
|
|
176
|
+
id: conversation.id,
|
|
177
|
+
timestamp: conversation.timestamp,
|
|
178
|
+
model: conversation.model,
|
|
179
|
+
tool: conversation.tool,
|
|
180
|
+
summary: conversation.summary,
|
|
181
|
+
content: conversation.raw_content,
|
|
182
|
+
}, null, 2),
|
|
183
|
+
}],
|
|
184
|
+
};
|
|
185
|
+
}
|
|
186
|
+
// Search conversations
|
|
187
|
+
if (!query) {
|
|
188
|
+
return {
|
|
189
|
+
content: [{ type: 'text', text: 'Please provide a query to search conversations, or an id to get a specific conversation' }],
|
|
190
|
+
};
|
|
191
|
+
}
|
|
192
|
+
// Try to scope to current project
|
|
193
|
+
const cwd = process.cwd();
|
|
194
|
+
const project = findProjectForPath(cwd);
|
|
195
|
+
const projectId = project?.id;
|
|
196
|
+
const conversations = searchFullConversations(query, limit, projectId);
|
|
197
|
+
if (conversations.length === 0) {
|
|
198
|
+
return {
|
|
199
|
+
content: [{ type: 'text', text: `No conversations found matching: ${query}` }],
|
|
200
|
+
};
|
|
201
|
+
}
|
|
202
|
+
const results = conversations.map(c => ({
|
|
203
|
+
id: c.id,
|
|
204
|
+
timestamp: c.timestamp,
|
|
205
|
+
model: c.model,
|
|
206
|
+
tool: c.tool,
|
|
207
|
+
summary: c.summary,
|
|
208
|
+
content: c.raw_content.length > 2000
|
|
209
|
+
? c.raw_content.slice(0, 2000) + '... [truncated - use id to get full content]'
|
|
210
|
+
: c.raw_content,
|
|
211
|
+
}));
|
|
212
|
+
return {
|
|
213
|
+
content: [{
|
|
214
|
+
type: 'text',
|
|
215
|
+
text: JSON.stringify({
|
|
216
|
+
query,
|
|
217
|
+
count: results.length,
|
|
218
|
+
conversations: results,
|
|
219
|
+
}, null, 2),
|
|
220
|
+
}],
|
|
221
|
+
};
|
|
222
|
+
}
|
|
223
|
+
default:
|
|
224
|
+
return {
|
|
225
|
+
content: [{ type: 'text', text: `Unknown tool: ${name}` }],
|
|
226
|
+
isError: true,
|
|
227
|
+
};
|
|
228
|
+
}
|
|
229
|
+
}
|
|
230
|
+
catch (error) {
|
|
231
|
+
return {
|
|
232
|
+
content: [{ type: 'text', text: `Error: ${error.message}` }],
|
|
233
|
+
isError: true,
|
|
234
|
+
};
|
|
235
|
+
}
|
|
236
|
+
});
|
|
237
|
+
// Start server
|
|
238
|
+
const transport = new StdioServerTransport();
|
|
239
|
+
await server.connect(transport);
|
|
240
|
+
}
|
|
241
|
+
//# sourceMappingURL=server.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/mcp/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EACL,qBAAqB,EACrB,sBAAsB,GACvB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EACL,gBAAgB,EAChB,mBAAmB,EACnB,eAAe,EACf,UAAU,EACV,iBAAiB,EACjB,mBAAmB,EACnB,UAAU,EACV,aAAa,EACb,OAAO,EACP,mBAAmB,EACnB,uBAAuB,EACvB,kBAAkB,EAClB,aAAa,GACd,MAAM,gBAAgB,CAAC;AAExB,MAAM,CAAC,KAAK,UAAU,cAAc;IAClC,MAAM,MAAM,GAAG,IAAI,MAAM,CACvB;QACE,IAAI,EAAE,OAAO;QACb,OAAO,EAAE,OAAO;KACjB,EACD;QACE,YAAY,EAAE;YACZ,KAAK,EAAE,EAAE;SACV;KACF,CACF,CAAC;IAEF,uBAAuB;IACvB,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;QAC5D,KAAK,EAAE;YACL;gBACE,IAAI,EAAE,aAAa;gBACnB,WAAW,EAAE,oDAAoD;gBACjE,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,uCAAuC,EAAE;wBAC/E,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,WAAW,EAAE,SAAS,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE;wBAC9G,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE;qBACvC;oBACD,QAAQ,EAAE,CAAC,OAAO,CAAC;iBACpB;aACF;YACD;gBACE,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,4CAA4C;gBACzD,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,gBAAgB,EAAE;wBACvD,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,WAAW,EAAE,MAAM,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE;qBAC5E;oBACD,QAAQ,EAAE,CAAC,MAAM,CAAC;iBACnB;aACF;YACD;gBACE,IAAI,EAAE,qBAAqB;gBAC3B,WAAW,EAAE,qCAAqC;gBAClD,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,iBAAiB,EAAE;wBACzD,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,WAAW,EAAE;wBAChD,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,EAAE;qBACtC;iBACF;aACF;SACF;KACF,CAAC,CAAC,CAAC;IAEJ,oBAAoB;IACpB,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAChE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QAEjD,IAAI,CAAC;YACH,QAAQ,IAAI,EAAE,CAAC;gBACb,KAAK,aAAa,CAAC,CAAC,CAAC;oBACnB,MAAM,KAAK,GAAG,IAAI,EAAE,KAAe,CAAC;oBACpC,MAAM,IAAI,GAAI,IAAI,EAAE,IAAe,IAAI,KAAK,CAAC;oBAC7C,MAAM,KAAK,GAAI,IAAI,EAAE,KAAgB,IAAI,EAAE,CAAC;oBAE5C,MAAM,OAAO,GAAsG,EAAE,CAAC;oBAEtH,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;wBAC5C,MAAM,UAAU,GAAG,gBAAgB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;wBAClD,OAAO,CAAC,UAAU,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;4BACxC,IAAI,EAAE,CAAC,CAAC,IAAI;4BACZ,IAAI,EAAE,CAAC,CAAC,IAAI;4BACZ,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI;4BAC9B,IAAI,EAAE,CAAC,CAAC,UAAU;4BAClB,SAAS,EAAE,CAAC,CAAC,SAAS;4BACtB,MAAM,EAAE,CAAC,CAAC,WAAW,IAAI,SAAS;4BAClC,MAAM,EAAE,CAAC,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,SAAS;yBACrD,CAAC,CAAC,CAAC;oBACN,CAAC;oBAED,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,eAAe,EAAE,CAAC;wBAC/C,MAAM,aAAa,GAAG,mBAAmB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;wBACxD,OAAO,CAAC,aAAa,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;4BAC9C,EAAE,EAAE,CAAC,CAAC,EAAE;4BACR,SAAS,EAAE,CAAC,CAAC,SAAS;4BACtB,KAAK,EAAE,CAAC,CAAC,KAAK;4BACd,OAAO,EAAE,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;yBAClD,CAAC,CAAC,CAAC;oBACN,CAAC;oBAED,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;wBAC3C,MAAM,SAAS,GAA6D,EAAE,CAAC;wBAC/E,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;wBAE/B,0DAA0D;wBAC1D,MAAM,UAAU,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC;wBAC9C,KAAK,MAAM,CAAC,IAAI,UAAU,EAAE,CAAC;4BAC3B,MAAM,KAAK,GAAG,UAAU,CAAC,WAAW,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;4BAC5C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gCACzB,IAAI,IAAI,CAAC,WAAW,KAAK,YAAY,EAAE,CAAC;oCACtC,MAAM,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;oCACjD,IAAI,UAAU,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;wCAChD,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;wCAC7B,SAAS,CAAC,IAAI,CAAC;4CACb,IAAI,EAAE,UAAU,CAAC,IAAI;4CACrB,OAAO,EAAE,UAAU,CAAC,OAAO;4CAC3B,MAAM,EAAE,QAAQ;yCACjB,CAAC,CAAC;oCACL,CAAC;gCACH,CAAC;4BACH,CAAC;wBACH,CAAC;wBAED,iDAAiD;wBACjD,MAAM,cAAc,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;wBACvD,KAAK,MAAM,CAAC,IAAI,cAAc,EAAE,CAAC;4BAC/B,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;gCACzB,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gCACpB,SAAS,CAAC,IAAI,CAAC;oCACb,IAAI,EAAE,CAAC,CAAC,IAAI;oCACZ,OAAO,EAAE,CAAC,CAAC,OAAO;oCAClB,MAAM,EAAE,SAAS;iCAClB,CAAC,CAAC;4BACL,CAAC;wBACH,CAAC;wBAED,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;oBAChD,CAAC;oBAED,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;wBACzC,MAAM,OAAO,GAAG,kBAAkB,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;wBAClD,MAAM,OAAO,GAAG,aAAa,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;wBACzD,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;4BAClC,IAAI,EAAE,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;4BACxC,MAAM,EAAE,CAAC,CAAC,WAAW;4BACrB,IAAI,EAAE,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;4BAC/B,OAAO,EAAE,CAAC,CAAC,OAAO;yBACnB,CAAC,CAAC,CAAC;oBACN,CAAC;oBAED,OAAO;wBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;qBACpE,CAAC;gBACJ,CAAC;gBAED,KAAK,cAAc,CAAC,CAAC,CAAC;oBACpB,MAAM,UAAU,GAAG,IAAI,EAAE,IAAc,CAAC;oBACxC,MAAM,UAAU,GAAI,IAAI,EAAE,IAAe,IAAI,WAAW,CAAC;oBAEzD,IAAI,MAAM,GAAG,KAAK,CAAC;oBACnB,IAAI,UAAU,KAAK,WAAW,EAAE,CAAC;wBAC/B,MAAM,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC;oBACvC,CAAC;yBAAM,IAAI,UAAU,KAAK,MAAM,EAAE,CAAC;wBACjC,MAAM,GAAG,UAAU,CAAC,UAAU,CAAC,CAAC;oBAClC,CAAC;oBAED,OAAO;wBACL,OAAO,EAAE,CAAC;gCACR,IAAI,EAAE,MAAM;gCACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oCACnB,IAAI,EAAE,UAAU;oCAChB,IAAI,EAAE,UAAU;oCAChB,MAAM;oCACN,OAAO,EAAE,MAAM;wCACb,CAAC,CAAC,GAAG,UAAU,KAAK,UAAU,0BAA0B;wCACxD,CAAC,CAAC,GAAG,UAAU,KAAK,UAAU,iCAAiC;iCAClE,EAAE,IAAI,EAAE,CAAC,CAAC;6BACZ,CAAC;qBACH,CAAC;gBACJ,CAAC;gBAED,KAAK,qBAAqB,CAAC,CAAC,CAAC;oBAC3B,MAAM,KAAK,GAAG,IAAI,EAAE,KAA2B,CAAC;oBAChD,MAAM,EAAE,GAAG,IAAI,EAAE,EAAwB,CAAC;oBAC1C,MAAM,KAAK,GAAI,IAAI,EAAE,KAAgB,IAAI,CAAC,CAAC;oBAE3C,kCAAkC;oBAClC,IAAI,EAAE,KAAK,SAAS,EAAE,CAAC;wBACrB,MAAM,YAAY,GAAG,mBAAmB,CAAC,EAAE,CAAC,CAAC;wBAC7C,IAAI,CAAC,YAAY,EAAE,CAAC;4BAClB,OAAO;gCACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kCAAkC,EAAE,EAAE,EAAE,CAAC;6BAC1E,CAAC;wBACJ,CAAC;wBACD,OAAO;4BACL,OAAO,EAAE,CAAC;oCACR,IAAI,EAAE,MAAM;oCACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;wCACnB,EAAE,EAAE,YAAY,CAAC,EAAE;wCACnB,SAAS,EAAE,YAAY,CAAC,SAAS;wCACjC,KAAK,EAAE,YAAY,CAAC,KAAK;wCACzB,IAAI,EAAE,YAAY,CAAC,IAAI;wCACvB,OAAO,EAAE,YAAY,CAAC,OAAO;wCAC7B,OAAO,EAAE,YAAY,CAAC,WAAW;qCAClC,EAAE,IAAI,EAAE,CAAC,CAAC;iCACZ,CAAC;yBACH,CAAC;oBACJ,CAAC;oBAED,uBAAuB;oBACvB,IAAI,CAAC,KAAK,EAAE,CAAC;wBACX,OAAO;4BACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,yFAAyF,EAAE,CAAC;yBAC7H,CAAC;oBACJ,CAAC;oBAED,kCAAkC;oBAClC,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;oBAC1B,MAAM,OAAO,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAC;oBACxC,MAAM,SAAS,GAAG,OAAO,EAAE,EAAE,CAAC;oBAE9B,MAAM,aAAa,GAAG,uBAAuB,CAAC,KAAK,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;oBAEvE,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;wBAC/B,OAAO;4BACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,oCAAoC,KAAK,EAAE,EAAE,CAAC;yBAC/E,CAAC;oBACJ,CAAC;oBAED,MAAM,OAAO,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;wBACtC,EAAE,EAAE,CAAC,CAAC,EAAE;wBACR,SAAS,EAAE,CAAC,CAAC,SAAS;wBACtB,KAAK,EAAE,CAAC,CAAC,KAAK;wBACd,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,OAAO,EAAE,CAAC,CAAC,OAAO;wBAClB,OAAO,EAAE,CAAC,CAAC,WAAW,CAAC,MAAM,GAAG,IAAI;4BAClC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,8CAA8C;4BAC/E,CAAC,CAAC,CAAC,CAAC,WAAW;qBAClB,CAAC,CAAC,CAAC;oBAEJ,OAAO;wBACL,OAAO,EAAE,CAAC;gCACR,IAAI,EAAE,MAAM;gCACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oCACnB,KAAK;oCACL,KAAK,EAAE,OAAO,CAAC,MAAM;oCACrB,aAAa,EAAE,OAAO;iCACvB,EAAE,IAAI,EAAE,CAAC,CAAC;6BACZ,CAAC;qBACH,CAAC;gBACJ,CAAC;gBAED;oBACE,OAAO;wBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,IAAI,EAAE,EAAE,CAAC;wBAC1D,OAAO,EAAE,IAAI;qBACd,CAAC;YACN,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAW,KAAe,CAAC,OAAO,EAAE,EAAE,CAAC;gBACvE,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,eAAe;IACf,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;AAClC,CAAC"}
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* mockttp-based interceptor for aimem v2.0
|
|
3
|
+
* Capture-only: no injection, just captures conversations and extracts decisions
|
|
4
|
+
*/
|
|
5
|
+
interface ProxyOptions {
|
|
6
|
+
port?: number;
|
|
7
|
+
projectId?: number;
|
|
8
|
+
}
|
|
9
|
+
export declare class AimemProxy {
|
|
10
|
+
private server;
|
|
11
|
+
private projectId;
|
|
12
|
+
private dataDir;
|
|
13
|
+
private certPath;
|
|
14
|
+
private keyPath;
|
|
15
|
+
private requestMap;
|
|
16
|
+
constructor(options?: ProxyOptions);
|
|
17
|
+
private getToolFromHost;
|
|
18
|
+
private getOrCreateCert;
|
|
19
|
+
private extractAssistantContent;
|
|
20
|
+
private parseSSEContent;
|
|
21
|
+
private storeConversation;
|
|
22
|
+
start(port?: number): Promise<void>;
|
|
23
|
+
stop(): Promise<void>;
|
|
24
|
+
getCertPath(): string;
|
|
25
|
+
}
|
|
26
|
+
export {};
|
|
27
|
+
//# sourceMappingURL=interceptor-mockttp.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"interceptor-mockttp.d.ts","sourceRoot":"","sources":["../../src/proxy/interceptor-mockttp.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAwBH,UAAU,YAAY;IACpB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,qBAAa,UAAU;IACrB,OAAO,CAAC,MAAM,CAAgC;IAC9C,OAAO,CAAC,SAAS,CAAuB;IACxC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,UAAU,CAA2D;gBAEjE,OAAO,GAAE,YAAiB;IAOtC,OAAO,CAAC,eAAe;YAeT,eAAe;IAwB7B,OAAO,CAAC,uBAAuB;IAgD/B,OAAO,CAAC,eAAe;IA0CvB,OAAO,CAAC,iBAAiB;IA+CnB,KAAK,CAAC,IAAI,GAAE,MAAa,GAAG,OAAO,CAAC,IAAI,CAAC;IA2DzC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAQ3B,WAAW,IAAI,MAAM;CAGtB"}
|