codevault 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/.env.example +37 -0
- package/LICENSE +21 -0
- package/README.md +174 -0
- package/dist/chunking/file-grouper.d.ts +39 -0
- package/dist/chunking/file-grouper.d.ts.map +1 -0
- package/dist/chunking/file-grouper.js +181 -0
- package/dist/chunking/file-grouper.js.map +1 -0
- package/dist/chunking/semantic-chunker.d.ts +37 -0
- package/dist/chunking/semantic-chunker.d.ts.map +1 -0
- package/dist/chunking/semantic-chunker.js +172 -0
- package/dist/chunking/semantic-chunker.js.map +1 -0
- package/dist/chunking/token-counter.d.ts +28 -0
- package/dist/chunking/token-counter.d.ts.map +1 -0
- package/dist/chunking/token-counter.js +207 -0
- package/dist/chunking/token-counter.js.map +1 -0
- package/dist/cli/commands/context.d.ts +3 -0
- package/dist/cli/commands/context.d.ts.map +1 -0
- package/dist/cli/commands/context.js +98 -0
- package/dist/cli/commands/context.js.map +1 -0
- package/dist/cli.d.ts +3 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +222 -0
- package/dist/cli.js.map +1 -0
- package/dist/codemap/io.d.ts +5 -0
- package/dist/codemap/io.d.ts.map +1 -0
- package/dist/codemap/io.js +30 -0
- package/dist/codemap/io.js.map +1 -0
- package/dist/context/packs.d.ts +33 -0
- package/dist/context/packs.d.ts.map +1 -0
- package/dist/context/packs.js +180 -0
- package/dist/context/packs.js.map +1 -0
- package/dist/core/indexer.d.ts +3 -0
- package/dist/core/indexer.d.ts.map +1 -0
- package/dist/core/indexer.js +610 -0
- package/dist/core/indexer.js.map +1 -0
- package/dist/core/metadata.d.ts +19 -0
- package/dist/core/metadata.d.ts.map +1 -0
- package/dist/core/metadata.js +161 -0
- package/dist/core/metadata.js.map +1 -0
- package/dist/core/search.d.ts +6 -0
- package/dist/core/search.d.ts.map +1 -0
- package/dist/core/search.js +482 -0
- package/dist/core/search.js.map +1 -0
- package/dist/core/symbol-extractor.d.ts +3 -0
- package/dist/core/symbol-extractor.d.ts.map +1 -0
- package/dist/core/symbol-extractor.js +78 -0
- package/dist/core/symbol-extractor.js.map +1 -0
- package/dist/core/types.d.ts +96 -0
- package/dist/core/types.d.ts.map +1 -0
- package/dist/core/types.js +2 -0
- package/dist/core/types.js.map +1 -0
- package/dist/database/db.d.ts +63 -0
- package/dist/database/db.d.ts.map +1 -0
- package/dist/database/db.js +205 -0
- package/dist/database/db.js.map +1 -0
- package/dist/indexer/merkle.d.ts +13 -0
- package/dist/indexer/merkle.d.ts.map +1 -0
- package/dist/indexer/merkle.js +86 -0
- package/dist/indexer/merkle.js.map +1 -0
- package/dist/indexer/update.d.ts +19 -0
- package/dist/indexer/update.d.ts.map +1 -0
- package/dist/indexer/update.js +40 -0
- package/dist/indexer/update.js.map +1 -0
- package/dist/indexer/watch.d.ts +21 -0
- package/dist/indexer/watch.d.ts.map +1 -0
- package/dist/indexer/watch.js +204 -0
- package/dist/indexer/watch.js.map +1 -0
- package/dist/languages/rules.d.ts +11 -0
- package/dist/languages/rules.d.ts.map +1 -0
- package/dist/languages/rules.js +371 -0
- package/dist/languages/rules.js.map +1 -0
- package/dist/languages/tree-sitter-loader.d.ts +27 -0
- package/dist/languages/tree-sitter-loader.d.ts.map +1 -0
- package/dist/languages/tree-sitter-loader.js +75 -0
- package/dist/languages/tree-sitter-loader.js.map +1 -0
- package/dist/mcp/tools/use-context-pack.d.ts +57 -0
- package/dist/mcp/tools/use-context-pack.d.ts.map +1 -0
- package/dist/mcp/tools/use-context-pack.js +91 -0
- package/dist/mcp/tools/use-context-pack.js.map +1 -0
- package/dist/mcp-server.d.ts +3 -0
- package/dist/mcp-server.d.ts.map +1 -0
- package/dist/mcp-server.js +363 -0
- package/dist/mcp-server.js.map +1 -0
- package/dist/providers/base.d.ts +34 -0
- package/dist/providers/base.d.ts.map +1 -0
- package/dist/providers/base.js +160 -0
- package/dist/providers/base.js.map +1 -0
- package/dist/providers/index.d.ts +6 -0
- package/dist/providers/index.d.ts.map +1 -0
- package/dist/providers/index.js +22 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/providers/ollama.d.ts +13 -0
- package/dist/providers/ollama.d.ts.map +1 -0
- package/dist/providers/ollama.js +50 -0
- package/dist/providers/ollama.js.map +1 -0
- package/dist/providers/openai.d.ts +13 -0
- package/dist/providers/openai.d.ts.map +1 -0
- package/dist/providers/openai.js +59 -0
- package/dist/providers/openai.js.map +1 -0
- package/dist/providers/token-counter.d.ts +2 -0
- package/dist/providers/token-counter.d.ts.map +1 -0
- package/dist/providers/token-counter.js +18 -0
- package/dist/providers/token-counter.js.map +1 -0
- package/dist/ranking/api-reranker.d.ts +18 -0
- package/dist/ranking/api-reranker.d.ts.map +1 -0
- package/dist/ranking/api-reranker.js +148 -0
- package/dist/ranking/api-reranker.js.map +1 -0
- package/dist/ranking/symbol-boost.d.ts +15 -0
- package/dist/ranking/symbol-boost.d.ts.map +1 -0
- package/dist/ranking/symbol-boost.js +154 -0
- package/dist/ranking/symbol-boost.js.map +1 -0
- package/dist/search/bm25.d.ts +17 -0
- package/dist/search/bm25.d.ts.map +1 -0
- package/dist/search/bm25.js +56 -0
- package/dist/search/bm25.js.map +1 -0
- package/dist/search/hybrid.d.ts +21 -0
- package/dist/search/hybrid.d.ts.map +1 -0
- package/dist/search/hybrid.js +50 -0
- package/dist/search/hybrid.js.map +1 -0
- package/dist/search/scope.d.ts +5 -0
- package/dist/search/scope.d.ts.map +1 -0
- package/dist/search/scope.js +107 -0
- package/dist/search/scope.js.map +1 -0
- package/dist/storage/encrypted-chunks.d.ts +40 -0
- package/dist/storage/encrypted-chunks.d.ts.map +1 -0
- package/dist/storage/encrypted-chunks.js +236 -0
- package/dist/storage/encrypted-chunks.js.map +1 -0
- package/dist/symbols/extract.d.ts +15 -0
- package/dist/symbols/extract.d.ts.map +1 -0
- package/dist/symbols/extract.js +187 -0
- package/dist/symbols/extract.js.map +1 -0
- package/dist/symbols/graph.d.ts +3 -0
- package/dist/symbols/graph.d.ts.map +1 -0
- package/dist/symbols/graph.js +89 -0
- package/dist/symbols/graph.js.map +1 -0
- package/dist/types/ast.d.ts +3 -0
- package/dist/types/ast.d.ts.map +1 -0
- package/dist/types/ast.js +2 -0
- package/dist/types/ast.js.map +1 -0
- package/dist/types/codemap.d.ts +58 -0
- package/dist/types/codemap.d.ts.map +1 -0
- package/dist/types/codemap.js +224 -0
- package/dist/types/codemap.js.map +1 -0
- package/dist/types/context-pack.d.ts +47 -0
- package/dist/types/context-pack.d.ts.map +1 -0
- package/dist/types/context-pack.js +44 -0
- package/dist/types/context-pack.js.map +1 -0
- package/dist/types/search.d.ts +15 -0
- package/dist/types/search.d.ts.map +1 -0
- package/dist/types/search.js +11 -0
- package/dist/types/search.js.map +1 -0
- package/dist/utils/rate-limiter.d.ts +31 -0
- package/dist/utils/rate-limiter.d.ts.map +1 -0
- package/dist/utils/rate-limiter.js +168 -0
- package/dist/utils/rate-limiter.js.map +1 -0
- package/package.json +92 -0
|
@@ -0,0 +1,482 @@
|
|
|
1
|
+
import fs from 'fs';
|
|
2
|
+
import path from 'path';
|
|
3
|
+
import { createEmbeddingProvider } from '../providers/index.js';
|
|
4
|
+
import { Database } from '../database/db.js';
|
|
5
|
+
import { readCodemap } from '../codemap/io.js';
|
|
6
|
+
import { normalizeScopeFilters, applyScope } from '../search/scope.js';
|
|
7
|
+
import { BM25Index } from '../search/bm25.js';
|
|
8
|
+
import { reciprocalRankFusion } from '../search/hybrid.js';
|
|
9
|
+
import { rerankWithAPI } from '../ranking/api-reranker.js';
|
|
10
|
+
import { applySymbolBoost } from '../ranking/symbol-boost.js';
|
|
11
|
+
import { readChunkFromDisk } from '../storage/encrypted-chunks.js';
|
|
12
|
+
const bm25IndexCache = new Map();
|
|
13
|
+
const chunkTextCache = new Map();
|
|
14
|
+
const RERANKER_MAX_CANDIDATES = Number.parseInt(process.env.CODEVAULT_RERANKER_MAX || '50', 10);
|
|
15
|
+
function normalizeQuery(query) {
|
|
16
|
+
return query
|
|
17
|
+
.toLowerCase()
|
|
18
|
+
.trim()
|
|
19
|
+
.replace(/[¿?]/g, '')
|
|
20
|
+
.replace(/\s+/g, ' ');
|
|
21
|
+
}
|
|
22
|
+
function cosineSimilarity(a, b) {
|
|
23
|
+
if (a.length !== b.length)
|
|
24
|
+
return 0;
|
|
25
|
+
let dotProduct = 0;
|
|
26
|
+
let normA = 0;
|
|
27
|
+
let normB = 0;
|
|
28
|
+
for (let i = 0; i < a.length; i++) {
|
|
29
|
+
dotProduct += a[i] * b[i];
|
|
30
|
+
normA += a[i] * a[i];
|
|
31
|
+
normB += b[i] * b[i];
|
|
32
|
+
}
|
|
33
|
+
return dotProduct / (Math.sqrt(normA) * Math.sqrt(normB));
|
|
34
|
+
}
|
|
35
|
+
function getBm25CacheKey(basePath, providerName, dimensions) {
|
|
36
|
+
return `${basePath}::${providerName}::${dimensions}`;
|
|
37
|
+
}
|
|
38
|
+
function getChunkCacheKey(basePath, sha) {
|
|
39
|
+
return `${basePath}::${sha}`;
|
|
40
|
+
}
|
|
41
|
+
function readChunkTextCached(sha, chunkDir, basePath) {
|
|
42
|
+
if (!sha) {
|
|
43
|
+
return null;
|
|
44
|
+
}
|
|
45
|
+
const cacheKey = getChunkCacheKey(basePath, sha);
|
|
46
|
+
if (chunkTextCache.has(cacheKey)) {
|
|
47
|
+
return chunkTextCache.get(cacheKey);
|
|
48
|
+
}
|
|
49
|
+
try {
|
|
50
|
+
const result = readChunkFromDisk({ chunkDir, sha });
|
|
51
|
+
const code = result ? result.code : null;
|
|
52
|
+
chunkTextCache.set(cacheKey, code);
|
|
53
|
+
return code;
|
|
54
|
+
}
|
|
55
|
+
catch (error) {
|
|
56
|
+
chunkTextCache.set(cacheKey, null);
|
|
57
|
+
return null;
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
function buildBm25Document(chunk, codeText) {
|
|
61
|
+
if (!chunk) {
|
|
62
|
+
return '';
|
|
63
|
+
}
|
|
64
|
+
const parts = [
|
|
65
|
+
chunk.symbol,
|
|
66
|
+
chunk.file_path,
|
|
67
|
+
chunk.codevault_description,
|
|
68
|
+
chunk.codevault_intent,
|
|
69
|
+
codeText
|
|
70
|
+
].filter(value => typeof value === 'string' && value.trim().length > 0);
|
|
71
|
+
return parts.join('\n');
|
|
72
|
+
}
|
|
73
|
+
function ensureBm25IndexForChunks(basePath, chunkDir, providerName, dimensions, chunks) {
|
|
74
|
+
if (!Array.isArray(chunks) || chunks.length === 0) {
|
|
75
|
+
return null;
|
|
76
|
+
}
|
|
77
|
+
const key = getBm25CacheKey(basePath, providerName, dimensions);
|
|
78
|
+
let entry = bm25IndexCache.get(key);
|
|
79
|
+
if (!entry) {
|
|
80
|
+
entry = { index: new BM25Index(), added: new Set() };
|
|
81
|
+
bm25IndexCache.set(key, entry);
|
|
82
|
+
}
|
|
83
|
+
const toAdd = [];
|
|
84
|
+
for (const chunk of chunks) {
|
|
85
|
+
if (!chunk || !chunk.id || entry.added.has(chunk.id)) {
|
|
86
|
+
continue;
|
|
87
|
+
}
|
|
88
|
+
const codeText = readChunkTextCached(chunk.sha, chunkDir, basePath);
|
|
89
|
+
const docText = buildBm25Document(chunk, codeText);
|
|
90
|
+
if (docText && docText.trim().length > 0) {
|
|
91
|
+
toAdd.push({ id: chunk.id, text: docText });
|
|
92
|
+
}
|
|
93
|
+
entry.added.add(chunk.id);
|
|
94
|
+
}
|
|
95
|
+
if (toAdd.length > 0) {
|
|
96
|
+
entry.index.addDocuments(toAdd);
|
|
97
|
+
}
|
|
98
|
+
entry.index.consolidate();
|
|
99
|
+
return entry.index;
|
|
100
|
+
}
|
|
101
|
+
export async function searchCode(query, limit = 10, provider = 'auto', workingPath = '.', scopeOptions = {}) {
|
|
102
|
+
const basePath = path.resolve(workingPath);
|
|
103
|
+
const dbPath = path.join(basePath, '.codevault/codevault.db');
|
|
104
|
+
const chunkDir = path.join(basePath, '.codevault/chunks');
|
|
105
|
+
const codemapPath = path.join(basePath, 'codevault.codemap.json');
|
|
106
|
+
if (!query || !query.trim()) {
|
|
107
|
+
return getOverview(limit, workingPath);
|
|
108
|
+
}
|
|
109
|
+
const normalizedScope = normalizeScopeFilters(scopeOptions);
|
|
110
|
+
const effectiveProvider = normalizedScope.provider || provider;
|
|
111
|
+
const hybridEnabled = normalizedScope.hybrid !== false;
|
|
112
|
+
const bm25Enabled = normalizedScope.bm25 !== false;
|
|
113
|
+
const symbolBoostEnabled = normalizedScope.symbol_boost !== false;
|
|
114
|
+
const embeddingProvider = createEmbeddingProvider(effectiveProvider);
|
|
115
|
+
try {
|
|
116
|
+
if (!fs.existsSync(dbPath)) {
|
|
117
|
+
return {
|
|
118
|
+
success: false,
|
|
119
|
+
error: 'database_not_found',
|
|
120
|
+
message: `Database not found at ${dbPath}. Project needs to be indexed first.`,
|
|
121
|
+
suggestion: `Run index_project on directory: ${workingPath}`,
|
|
122
|
+
provider: embeddingProvider.getName(),
|
|
123
|
+
scope: normalizedScope,
|
|
124
|
+
hybrid: { enabled: hybridEnabled, bm25Enabled },
|
|
125
|
+
symbolBoost: { enabled: symbolBoostEnabled, boosted: false },
|
|
126
|
+
reranker: normalizedScope.reranker,
|
|
127
|
+
results: []
|
|
128
|
+
};
|
|
129
|
+
}
|
|
130
|
+
const db = new Database(dbPath);
|
|
131
|
+
const chunks = await db.getChunks(embeddingProvider.getName(), embeddingProvider.getDimensions());
|
|
132
|
+
const codemapData = readCodemap(codemapPath);
|
|
133
|
+
if (chunks.length === 0) {
|
|
134
|
+
db.close();
|
|
135
|
+
return {
|
|
136
|
+
success: false,
|
|
137
|
+
error: 'no_chunks_found',
|
|
138
|
+
message: `No indexed chunks found with ${embeddingProvider.getName()} in ${basePath}`,
|
|
139
|
+
suggestion: `Run: codevault index --provider ${effectiveProvider} from ${basePath}`,
|
|
140
|
+
provider: embeddingProvider.getName(),
|
|
141
|
+
scope: normalizedScope,
|
|
142
|
+
hybrid: { enabled: hybridEnabled, bm25Enabled },
|
|
143
|
+
reranker: normalizedScope.reranker,
|
|
144
|
+
results: []
|
|
145
|
+
};
|
|
146
|
+
}
|
|
147
|
+
const scopedChunks = applyScope(chunks, normalizedScope);
|
|
148
|
+
const chunkInfoById = new Map();
|
|
149
|
+
const results = [];
|
|
150
|
+
let queryEmbedding = null;
|
|
151
|
+
if (scopedChunks.length > 0) {
|
|
152
|
+
if (embeddingProvider.init) {
|
|
153
|
+
await embeddingProvider.init();
|
|
154
|
+
}
|
|
155
|
+
queryEmbedding = await embeddingProvider.generateEmbedding(query);
|
|
156
|
+
}
|
|
157
|
+
for (const chunk of scopedChunks) {
|
|
158
|
+
const embedding = JSON.parse(chunk.embedding.toString());
|
|
159
|
+
const vectorSimilarity = queryEmbedding ? cosineSimilarity(queryEmbedding, embedding) : 0;
|
|
160
|
+
let boostScore = 0;
|
|
161
|
+
if (chunk.codevault_intent && query.toLowerCase().includes(chunk.codevault_intent.toLowerCase())) {
|
|
162
|
+
boostScore += 0.2;
|
|
163
|
+
}
|
|
164
|
+
if (chunk.codevault_tags) {
|
|
165
|
+
try {
|
|
166
|
+
const tags = JSON.parse(chunk.codevault_tags || '[]');
|
|
167
|
+
const queryLower = query.toLowerCase();
|
|
168
|
+
tags.forEach((tag) => {
|
|
169
|
+
if (typeof tag === 'string' && queryLower.includes(tag.toLowerCase())) {
|
|
170
|
+
boostScore += 0.1;
|
|
171
|
+
}
|
|
172
|
+
});
|
|
173
|
+
}
|
|
174
|
+
catch (error) {
|
|
175
|
+
// Ignore tag parsing errors
|
|
176
|
+
}
|
|
177
|
+
}
|
|
178
|
+
let docBoost = 0;
|
|
179
|
+
const filePath = chunk.file_path.toLowerCase();
|
|
180
|
+
if (filePath.includes('readme') ||
|
|
181
|
+
filePath.includes('/docs/') ||
|
|
182
|
+
filePath.startsWith('docs/') ||
|
|
183
|
+
filePath.includes('changelog') ||
|
|
184
|
+
filePath.includes('contributing') ||
|
|
185
|
+
filePath.endsWith('.md')) {
|
|
186
|
+
docBoost = 0.15;
|
|
187
|
+
}
|
|
188
|
+
const finalScore = Math.min(vectorSimilarity + boostScore + docBoost, 1.0);
|
|
189
|
+
const info = {
|
|
190
|
+
id: chunk.id,
|
|
191
|
+
file_path: chunk.file_path,
|
|
192
|
+
symbol: chunk.symbol,
|
|
193
|
+
sha: chunk.sha,
|
|
194
|
+
lang: chunk.lang,
|
|
195
|
+
chunk_type: chunk.chunk_type,
|
|
196
|
+
codevault_intent: chunk.codevault_intent,
|
|
197
|
+
codevault_description: chunk.codevault_description,
|
|
198
|
+
score: finalScore,
|
|
199
|
+
vectorScore: vectorSimilarity,
|
|
200
|
+
boostScore: boostScore
|
|
201
|
+
};
|
|
202
|
+
chunkInfoById.set(chunk.id, info);
|
|
203
|
+
results.push(info);
|
|
204
|
+
}
|
|
205
|
+
if (symbolBoostEnabled) {
|
|
206
|
+
try {
|
|
207
|
+
applySymbolBoost(results, { query, codemap: codemapData });
|
|
208
|
+
}
|
|
209
|
+
catch (error) {
|
|
210
|
+
// Symbol boost fails silently
|
|
211
|
+
}
|
|
212
|
+
}
|
|
213
|
+
const sortedResults = results.sort((a, b) => b.score - a.score);
|
|
214
|
+
const remainingSlots = limit;
|
|
215
|
+
let vectorResults = [];
|
|
216
|
+
let bm25Fused = false;
|
|
217
|
+
let bm25CandidateCount = 0;
|
|
218
|
+
if (remainingSlots > 0) {
|
|
219
|
+
const selectionBudget = Math.max(remainingSlots, 60);
|
|
220
|
+
const vectorPool = sortedResults.slice(0, selectionBudget);
|
|
221
|
+
if (hybridEnabled && bm25Enabled) {
|
|
222
|
+
const bm25Index = ensureBm25IndexForChunks(basePath, chunkDir, embeddingProvider.getName(), embeddingProvider.getDimensions(), scopedChunks);
|
|
223
|
+
if (bm25Index) {
|
|
224
|
+
const allowedIds = new Set(scopedChunks.map((chunk) => chunk.id));
|
|
225
|
+
const bm25RawResults = bm25Index.search(query, selectionBudget);
|
|
226
|
+
const bm25Results = bm25RawResults.filter(result => allowedIds.has(result.id));
|
|
227
|
+
bm25CandidateCount = bm25Results.length;
|
|
228
|
+
if (bm25Results.length > 0) {
|
|
229
|
+
const fused = reciprocalRankFusion({
|
|
230
|
+
vectorResults: vectorPool.map((item) => ({ id: item.id, score: item.score })),
|
|
231
|
+
bm25Results: bm25Results.map(item => ({ id: item.id, score: item.score })),
|
|
232
|
+
limit: selectionBudget,
|
|
233
|
+
k: 60
|
|
234
|
+
});
|
|
235
|
+
if (fused.length > 0) {
|
|
236
|
+
bm25Fused = true;
|
|
237
|
+
vectorResults = fused
|
|
238
|
+
.map(entry => {
|
|
239
|
+
const info = chunkInfoById.get(entry.id);
|
|
240
|
+
if (!info) {
|
|
241
|
+
return null;
|
|
242
|
+
}
|
|
243
|
+
info.hybridScore = entry.score;
|
|
244
|
+
info.bm25Score = entry.bm25Score;
|
|
245
|
+
info.bm25Rank = entry.bm25Rank;
|
|
246
|
+
info.vectorRank = entry.vectorRank;
|
|
247
|
+
return info;
|
|
248
|
+
})
|
|
249
|
+
.filter(Boolean);
|
|
250
|
+
}
|
|
251
|
+
}
|
|
252
|
+
}
|
|
253
|
+
}
|
|
254
|
+
if (vectorResults.length === 0) {
|
|
255
|
+
vectorResults = vectorPool;
|
|
256
|
+
}
|
|
257
|
+
const hasSymbolBoost = symbolBoostEnabled && vectorResults.some((candidate) => typeof candidate.symbolBoost === 'number' && candidate.symbolBoost > 0);
|
|
258
|
+
if (hasSymbolBoost && vectorResults.length > 1) {
|
|
259
|
+
vectorResults.sort((a, b) => {
|
|
260
|
+
const scoreA = typeof a.score === 'number' ? a.score : 0;
|
|
261
|
+
const scoreB = typeof b.score === 'number' ? b.score : 0;
|
|
262
|
+
if (scoreB !== scoreA) {
|
|
263
|
+
return scoreB - scoreA;
|
|
264
|
+
}
|
|
265
|
+
const boostA = typeof a.symbolBoost === 'number' ? a.symbolBoost : 0;
|
|
266
|
+
const boostB = typeof b.symbolBoost === 'number' ? b.symbolBoost : 0;
|
|
267
|
+
if (boostB !== boostA) {
|
|
268
|
+
return boostB - boostA;
|
|
269
|
+
}
|
|
270
|
+
const hybridA = typeof a.hybridScore === 'number' ? a.hybridScore : Number.NEGATIVE_INFINITY;
|
|
271
|
+
const hybridB = typeof b.hybridScore === 'number' ? b.hybridScore : Number.NEGATIVE_INFINITY;
|
|
272
|
+
return hybridB - hybridA;
|
|
273
|
+
});
|
|
274
|
+
}
|
|
275
|
+
vectorResults = vectorResults.slice(0, remainingSlots);
|
|
276
|
+
if (vectorResults.length > 1 && normalizedScope.reranker === 'api') {
|
|
277
|
+
try {
|
|
278
|
+
const reranked = await rerankWithAPI(query, vectorResults, {
|
|
279
|
+
max: Math.min(RERANKER_MAX_CANDIDATES, vectorResults.length),
|
|
280
|
+
getText: (candidate) => {
|
|
281
|
+
const codeText = readChunkTextCached(candidate.sha, chunkDir, basePath) || '';
|
|
282
|
+
return buildBm25Document(candidate, codeText);
|
|
283
|
+
}
|
|
284
|
+
});
|
|
285
|
+
if (Array.isArray(reranked) && reranked.length === vectorResults.length) {
|
|
286
|
+
vectorResults = reranked;
|
|
287
|
+
}
|
|
288
|
+
}
|
|
289
|
+
catch (error) {
|
|
290
|
+
// Silent fallback when reranker is unavailable
|
|
291
|
+
}
|
|
292
|
+
}
|
|
293
|
+
}
|
|
294
|
+
const vectorSearchType = bm25Fused ? 'hybrid' : 'vector';
|
|
295
|
+
const combinedResults = vectorResults.map((result) => {
|
|
296
|
+
const rawScore = typeof result.score === 'number' ? result.score : 0;
|
|
297
|
+
const meta = {
|
|
298
|
+
id: result.id,
|
|
299
|
+
symbol: result.symbol,
|
|
300
|
+
score: Math.min(1, rawScore),
|
|
301
|
+
intent: result.codevault_intent,
|
|
302
|
+
description: result.codevault_description,
|
|
303
|
+
searchType: vectorSearchType,
|
|
304
|
+
vectorScore: result.vectorScore
|
|
305
|
+
};
|
|
306
|
+
if (typeof result.hybridScore === 'number') {
|
|
307
|
+
meta.hybridScore = result.hybridScore;
|
|
308
|
+
}
|
|
309
|
+
if (typeof result.bm25Score === 'number') {
|
|
310
|
+
meta.bm25Score = result.bm25Score;
|
|
311
|
+
}
|
|
312
|
+
if (typeof result.bm25Rank === 'number') {
|
|
313
|
+
meta.bm25Rank = result.bm25Rank;
|
|
314
|
+
}
|
|
315
|
+
if (typeof result.vectorRank === 'number') {
|
|
316
|
+
meta.vectorRank = result.vectorRank;
|
|
317
|
+
}
|
|
318
|
+
if (typeof result.rerankerScore === 'number') {
|
|
319
|
+
meta.rerankerScore = result.rerankerScore;
|
|
320
|
+
}
|
|
321
|
+
if (typeof result.rerankerRank === 'number') {
|
|
322
|
+
meta.rerankerRank = result.rerankerRank;
|
|
323
|
+
}
|
|
324
|
+
if (typeof result.symbolBoost === 'number' && result.symbolBoost > 0) {
|
|
325
|
+
meta.symbolBoost = result.symbolBoost;
|
|
326
|
+
if (Array.isArray(result.symbolBoostSources) && result.symbolBoostSources.length > 0) {
|
|
327
|
+
meta.symbolBoostSources = result.symbolBoostSources;
|
|
328
|
+
}
|
|
329
|
+
}
|
|
330
|
+
if (typeof rawScore === 'number' && rawScore > 1) {
|
|
331
|
+
meta.scoreRaw = rawScore;
|
|
332
|
+
}
|
|
333
|
+
return {
|
|
334
|
+
type: 'code',
|
|
335
|
+
lang: result.lang,
|
|
336
|
+
path: result.file_path,
|
|
337
|
+
sha: result.sha,
|
|
338
|
+
data: null,
|
|
339
|
+
meta
|
|
340
|
+
};
|
|
341
|
+
});
|
|
342
|
+
combinedResults.sort((a, b) => {
|
|
343
|
+
const hasRerankerA = typeof a.meta?.rerankerScore === 'number';
|
|
344
|
+
const hasRerankerB = typeof b.meta?.rerankerScore === 'number';
|
|
345
|
+
if (hasRerankerA && hasRerankerB) {
|
|
346
|
+
return b.meta.rerankerScore - a.meta.rerankerScore;
|
|
347
|
+
}
|
|
348
|
+
const scoreA = a.meta?.score ?? 0;
|
|
349
|
+
const scoreB = b.meta?.score ?? 0;
|
|
350
|
+
return scoreB - scoreA;
|
|
351
|
+
});
|
|
352
|
+
if (combinedResults.length === 0) {
|
|
353
|
+
db.close();
|
|
354
|
+
return {
|
|
355
|
+
success: false,
|
|
356
|
+
error: 'no_relevant_matches',
|
|
357
|
+
message: `No relevant matches found for "${query}"`,
|
|
358
|
+
suggestion: 'Try broader search terms or check if the project is properly indexed',
|
|
359
|
+
provider: embeddingProvider.getName(),
|
|
360
|
+
scope: normalizedScope,
|
|
361
|
+
hybrid: { enabled: hybridEnabled, bm25Enabled },
|
|
362
|
+
symbolBoost: { enabled: symbolBoostEnabled, boosted: false },
|
|
363
|
+
reranker: normalizedScope.reranker,
|
|
364
|
+
results: []
|
|
365
|
+
};
|
|
366
|
+
}
|
|
367
|
+
if (symbolBoostEnabled && combinedResults.length > 0 && combinedResults[0].meta.score > 0.8) {
|
|
368
|
+
await db.recordIntention(normalizeQuery(query), query, combinedResults[0].sha, combinedResults[0].meta.score);
|
|
369
|
+
}
|
|
370
|
+
const pattern = query
|
|
371
|
+
.toLowerCase()
|
|
372
|
+
.replace(/\b[\w-]+Session\b/gi, '[SESSION]')
|
|
373
|
+
.replace(/\bstripe\b/gi, '[PAYMENT_PROVIDER]')
|
|
374
|
+
.replace(/\b\w+Service\b/gi, '[SERVICE]')
|
|
375
|
+
.replace(/\b\w+Controller\b/gi, '[CONTROLLER]')
|
|
376
|
+
.trim();
|
|
377
|
+
await db.recordQueryPattern(pattern);
|
|
378
|
+
db.close();
|
|
379
|
+
return {
|
|
380
|
+
success: true,
|
|
381
|
+
query,
|
|
382
|
+
searchType: bm25Fused ? 'hybrid' : 'vector',
|
|
383
|
+
vectorResults: vectorResults.length,
|
|
384
|
+
provider: embeddingProvider.getName(),
|
|
385
|
+
scope: normalizedScope,
|
|
386
|
+
reranker: normalizedScope.reranker,
|
|
387
|
+
hybrid: {
|
|
388
|
+
enabled: hybridEnabled,
|
|
389
|
+
bm25Enabled,
|
|
390
|
+
fused: bm25Fused,
|
|
391
|
+
bm25Candidates: bm25CandidateCount
|
|
392
|
+
},
|
|
393
|
+
symbolBoost: {
|
|
394
|
+
enabled: symbolBoostEnabled,
|
|
395
|
+
boosted: symbolBoostEnabled && vectorResults.some((result) => typeof result.symbolBoost === 'number' && result.symbolBoost > 0)
|
|
396
|
+
},
|
|
397
|
+
results: combinedResults
|
|
398
|
+
};
|
|
399
|
+
}
|
|
400
|
+
catch (error) {
|
|
401
|
+
console.error('Error in searchCode:', error);
|
|
402
|
+
return {
|
|
403
|
+
success: false,
|
|
404
|
+
error: 'search_error',
|
|
405
|
+
message: error.message,
|
|
406
|
+
provider: embeddingProvider.getName(),
|
|
407
|
+
scope: normalizedScope,
|
|
408
|
+
hybrid: { enabled: hybridEnabled, bm25Enabled },
|
|
409
|
+
symbolBoost: { enabled: symbolBoostEnabled, boosted: false },
|
|
410
|
+
reranker: normalizedScope.reranker,
|
|
411
|
+
results: []
|
|
412
|
+
};
|
|
413
|
+
}
|
|
414
|
+
}
|
|
415
|
+
export async function getOverview(limit = 20, workingPath = '.') {
|
|
416
|
+
const basePath = path.resolve(workingPath);
|
|
417
|
+
const dbPath = path.join(basePath, '.codevault/codevault.db');
|
|
418
|
+
try {
|
|
419
|
+
if (!fs.existsSync(dbPath)) {
|
|
420
|
+
return {
|
|
421
|
+
success: false,
|
|
422
|
+
error: 'database_not_found',
|
|
423
|
+
message: `Database not found at ${dbPath}. Project needs to be indexed first.`,
|
|
424
|
+
suggestion: `Run index_project on directory: ${workingPath}`,
|
|
425
|
+
provider: 'unknown',
|
|
426
|
+
results: []
|
|
427
|
+
};
|
|
428
|
+
}
|
|
429
|
+
const db = new Database(dbPath);
|
|
430
|
+
const chunks = await db.getOverviewChunks(limit);
|
|
431
|
+
db.close();
|
|
432
|
+
const results = chunks.map(chunk => ({
|
|
433
|
+
type: 'code',
|
|
434
|
+
lang: chunk.lang,
|
|
435
|
+
path: chunk.file_path,
|
|
436
|
+
sha: chunk.sha,
|
|
437
|
+
data: null,
|
|
438
|
+
meta: {
|
|
439
|
+
id: chunk.id,
|
|
440
|
+
symbol: chunk.symbol,
|
|
441
|
+
score: 1.0
|
|
442
|
+
}
|
|
443
|
+
}));
|
|
444
|
+
return {
|
|
445
|
+
success: true,
|
|
446
|
+
provider: 'overview',
|
|
447
|
+
results
|
|
448
|
+
};
|
|
449
|
+
}
|
|
450
|
+
catch (error) {
|
|
451
|
+
return {
|
|
452
|
+
success: false,
|
|
453
|
+
error: 'overview_error',
|
|
454
|
+
message: error.message,
|
|
455
|
+
provider: 'overview',
|
|
456
|
+
results: []
|
|
457
|
+
};
|
|
458
|
+
}
|
|
459
|
+
}
|
|
460
|
+
export async function getChunk(sha, workingPath = '.') {
|
|
461
|
+
const basePath = path.resolve(workingPath);
|
|
462
|
+
const chunkDir = path.join(basePath, '.codevault/chunks');
|
|
463
|
+
try {
|
|
464
|
+
const result = readChunkFromDisk({ chunkDir, sha });
|
|
465
|
+
if (!result) {
|
|
466
|
+
const plainPath = path.join(chunkDir, `${sha}.gz`);
|
|
467
|
+
const encryptedPath = path.join(chunkDir, `${sha}.gz.enc`);
|
|
468
|
+
throw new Error(`Chunk ${sha} not found at ${plainPath} or ${encryptedPath}`);
|
|
469
|
+
}
|
|
470
|
+
return { success: true, code: result.code };
|
|
471
|
+
}
|
|
472
|
+
catch (error) {
|
|
473
|
+
if (error && error.code === 'ENCRYPTION_KEY_REQUIRED') {
|
|
474
|
+
return {
|
|
475
|
+
success: false,
|
|
476
|
+
error: `Chunk ${sha} is encrypted. Configure CODEVAULT_ENCRYPTION_KEY to decrypt.`
|
|
477
|
+
};
|
|
478
|
+
}
|
|
479
|
+
return { success: false, error: error.message };
|
|
480
|
+
}
|
|
481
|
+
}
|
|
482
|
+
//# sourceMappingURL=search.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"search.js","sourceRoot":"","sources":["../../src/core/search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,qBAAqB,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AACvE,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAC3D,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAE9D,OAAO,EAAE,iBAAiB,EAAE,MAAM,gCAAgC,CAAC;AAKnE,MAAM,cAAc,GAAG,IAAI,GAAG,EAAoD,CAAC;AACnF,MAAM,cAAc,GAAG,IAAI,GAAG,EAAyB,CAAC;AACxD,MAAM,uBAAuB,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;AAEhG,SAAS,cAAc,CAAC,KAAa;IACnC,OAAO,KAAK;SACT,WAAW,EAAE;SACb,IAAI,EAAE;SACN,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;SACpB,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;AAC1B,CAAC;AAED,SAAS,gBAAgB,CAAC,CAAW,EAAE,CAAW;IAChD,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM;QAAE,OAAO,CAAC,CAAC;IAEpC,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,KAAK,GAAG,CAAC,CAAC;IAEd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,UAAU,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1B,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACvB,CAAC;IAED,OAAO,UAAU,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;AAC5D,CAAC;AAED,SAAS,eAAe,CAAC,QAAgB,EAAE,YAAoB,EAAE,UAAkB;IACjF,OAAO,GAAG,QAAQ,KAAK,YAAY,KAAK,UAAU,EAAE,CAAC;AACvD,CAAC;AAED,SAAS,gBAAgB,CAAC,QAAgB,EAAE,GAAW;IACrD,OAAO,GAAG,QAAQ,KAAK,GAAG,EAAE,CAAC;AAC/B,CAAC;AAED,SAAS,mBAAmB,CAAC,GAAW,EAAE,QAAgB,EAAE,QAAgB;IAC1E,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,QAAQ,GAAG,gBAAgB,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;IACjD,IAAI,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;QACjC,OAAO,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC;IACvC,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,iBAAiB,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;QACpD,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;QACzC,cAAc,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACnC,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,cAAc,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACnC,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAU,EAAE,QAAuB;IAC5D,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,KAAK,GAAG;QACZ,KAAK,CAAC,MAAM;QACZ,KAAK,CAAC,SAAS;QACf,KAAK,CAAC,qBAAqB;QAC3B,KAAK,CAAC,gBAAgB;QACtB,QAAQ;KACT,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAExE,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,SAAS,wBAAwB,CAC/B,QAAgB,EAChB,QAAgB,EAChB,YAAoB,EACpB,UAAkB,EAClB,MAAa;IAEb,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAClD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,GAAG,GAAG,eAAe,CAAC,QAAQ,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;IAChE,IAAI,KAAK,GAAG,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAEpC,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,KAAK,GAAG,EAAE,KAAK,EAAE,IAAI,SAAS,EAAE,EAAE,KAAK,EAAE,IAAI,GAAG,EAAE,EAAE,CAAC;QACrD,cAAc,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IACjC,CAAC;IAED,MAAM,KAAK,GAAwC,EAAE,CAAC;IAEtD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC;YACrD,SAAS;QACX,CAAC;QAED,MAAM,QAAQ,GAAG,mBAAmB,CAAC,KAAK,CAAC,GAAG,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACpE,MAAM,OAAO,GAAG,iBAAiB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QAEnD,IAAI,OAAO,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAC9C,CAAC;QAED,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC5B,CAAC;IAED,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACrB,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;IAC1B,OAAO,KAAK,CAAC,KAAK,CAAC;AACrB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,KAAa,EACb,KAAK,GAAG,EAAE,EACV,QAAQ,GAAG,MAAM,EACjB,WAAW,GAAG,GAAG,EACjB,eAA6B,EAAE;IAE/B,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,yBAAyB,CAAC,CAAC;IAC9D,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,mBAAmB,CAAC,CAAC;IAC1D,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,wBAAwB,CAAC,CAAC;IAElE,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;QAC5B,OAAO,WAAW,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;IACzC,CAAC;IAED,MAAM,eAAe,GAAG,qBAAqB,CAAC,YAAY,CAAC,CAAC;IAC5D,MAAM,iBAAiB,GAAG,eAAe,CAAC,QAAQ,IAAI,QAAQ,CAAC;IAC/D,MAAM,aAAa,GAAG,eAAe,CAAC,MAAM,KAAK,KAAK,CAAC;IACvD,MAAM,WAAW,GAAG,eAAe,CAAC,IAAI,KAAK,KAAK,CAAC;IACnD,MAAM,kBAAkB,GAAG,eAAe,CAAC,YAAY,KAAK,KAAK,CAAC;IAElE,MAAM,iBAAiB,GAAG,uBAAuB,CAAC,iBAAiB,CAAC,CAAC;IAErE,IAAI,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3B,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,oBAAoB;gBAC3B,OAAO,EAAE,yBAAyB,MAAM,sCAAsC;gBAC9E,UAAU,EAAE,mCAAmC,WAAW,EAAE;gBAC5D,QAAQ,EAAE,iBAAiB,CAAC,OAAO,EAAE;gBACrC,KAAK,EAAE,eAAe;gBACtB,MAAM,EAAE,EAAE,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE;gBAC/C,WAAW,EAAE,EAAE,OAAO,EAAE,kBAAkB,EAAE,OAAO,EAAE,KAAK,EAAE;gBAC5D,QAAQ,EAAE,eAAe,CAAC,QAAQ;gBAClC,OAAO,EAAE,EAAE;aACZ,CAAC;QACJ,CAAC;QAED,MAAM,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;QAChC,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,SAAS,CAAC,iBAAiB,CAAC,OAAO,EAAE,EAAE,iBAAiB,CAAC,aAAa,EAAE,CAAC,CAAC;QAClG,MAAM,WAAW,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;QAE7C,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,EAAE,CAAC,KAAK,EAAE,CAAC;YACX,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,iBAAiB;gBACxB,OAAO,EAAE,gCAAgC,iBAAiB,CAAC,OAAO,EAAE,OAAO,QAAQ,EAAE;gBACrF,UAAU,EAAE,mCAAmC,iBAAiB,SAAS,QAAQ,EAAE;gBACnF,QAAQ,EAAE,iBAAiB,CAAC,OAAO,EAAE;gBACrC,KAAK,EAAE,eAAe;gBACtB,MAAM,EAAE,EAAE,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE;gBAC/C,QAAQ,EAAE,eAAe,CAAC,QAAQ;gBAClC,OAAO,EAAE,EAAE;aACZ,CAAC;QACJ,CAAC;QAED,MAAM,YAAY,GAAG,UAAU,CAAC,MAAM,EAAE,eAAe,CAAoB,CAAC;QAC5E,MAAM,aAAa,GAAG,IAAI,GAAG,EAAe,CAAC;QAC7C,MAAM,OAAO,GAAU,EAAE,CAAC;QAE1B,IAAI,cAAc,GAAoB,IAAI,CAAC;QAC3C,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,IAAI,iBAAiB,CAAC,IAAI,EAAE,CAAC;gBAC3B,MAAM,iBAAiB,CAAC,IAAI,EAAE,CAAC;YACjC,CAAC;YACD,cAAc,GAAG,MAAM,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACpE,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,YAAY,EAAE,CAAC;YACjC,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC;YACzD,MAAM,gBAAgB,GAAG,cAAc,CAAC,CAAC,CAAC,gBAAgB,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAE1F,IAAI,UAAU,GAAG,CAAC,CAAC;YAEnB,IAAI,KAAK,CAAC,gBAAgB,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,gBAAgB,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC;gBACjG,UAAU,IAAI,GAAG,CAAC;YACpB,CAAC;YAED,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;gBACzB,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,cAAc,IAAI,IAAI,CAAC,CAAC;oBACtD,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;oBACvC,IAAI,CAAC,OAAO,CAAC,CAAC,GAAW,EAAE,EAAE;wBAC3B,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC;4BACtE,UAAU,IAAI,GAAG,CAAC;wBACpB,CAAC;oBACH,CAAC,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,4BAA4B;gBAC9B,CAAC;YACH,CAAC;YAED,IAAI,QAAQ,GAAG,CAAC,CAAC;YACjB,MAAM,QAAQ,GAAG,KAAK,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC;YAC/C,IAAI,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBAC3B,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBAC3B,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC;gBAC5B,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC;gBAC9B,QAAQ,CAAC,QAAQ,CAAC,cAAc,CAAC;gBACjC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC7B,QAAQ,GAAG,IAAI,CAAC;YAClB,CAAC;YAED,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,gBAAgB,GAAG,UAAU,GAAG,QAAQ,EAAE,GAAG,CAAC,CAAC;YAE3E,MAAM,IAAI,GAAG;gBACX,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,GAAG,EAAE,KAAK,CAAC,GAAG;gBACd,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,UAAU,EAAE,KAAK,CAAC,UAAU;gBAC5B,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;gBACxC,qBAAqB,EAAE,KAAK,CAAC,qBAAqB;gBAClD,KAAK,EAAE,UAAU;gBACjB,WAAW,EAAE,gBAAgB;gBAC7B,UAAU,EAAE,UAAU;aACvB,CAAC;YAEF,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YAClC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrB,CAAC;QAED,IAAI,kBAAkB,EAAE,CAAC;YACvB,IAAI,CAAC;gBACH,gBAAgB,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;YAC7D,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,8BAA8B;YAChC,CAAC;QACH,CAAC;QAED,MAAM,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QAChE,MAAM,cAAc,GAAG,KAAK,CAAC;QAC7B,IAAI,aAAa,GAAU,EAAE,CAAC;QAC9B,IAAI,SAAS,GAAG,KAAK,CAAC;QACtB,IAAI,kBAAkB,GAAG,CAAC,CAAC;QAE3B,IAAI,cAAc,GAAG,CAAC,EAAE,CAAC;YACvB,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;YACrD,MAAM,UAAU,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC;YAE3D,IAAI,aAAa,IAAI,WAAW,EAAE,CAAC;gBACjC,MAAM,SAAS,GAAG,wBAAwB,CACxC,QAAQ,EACR,QAAQ,EACR,iBAAiB,CAAC,OAAO,EAAE,EAC3B,iBAAiB,CAAC,aAAa,EAAE,EACjC,YAAY,CACb,CAAC;gBAEF,IAAI,SAAS,EAAE,CAAC;oBACd,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,KAAU,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;oBACvE,MAAM,cAAc,GAAG,SAAS,CAAC,MAAM,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;oBAChE,MAAM,WAAW,GAAG,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;oBAC/E,kBAAkB,GAAG,WAAW,CAAC,MAAM,CAAC;oBAExC,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAC3B,MAAM,KAAK,GAAG,oBAAoB,CAAC;4BACjC,aAAa,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;4BAClF,WAAW,EAAE,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;4BAC1E,KAAK,EAAE,eAAe;4BACtB,CAAC,EAAE,EAAE;yBACN,CAAC,CAAC;wBAEH,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;4BACrB,SAAS,GAAG,IAAI,CAAC;4BACjB,aAAa,GAAG,KAAK;iCAClB,GAAG,CAAC,KAAK,CAAC,EAAE;gCACX,MAAM,IAAI,GAAG,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gCACzC,IAAI,CAAC,IAAI,EAAE,CAAC;oCACV,OAAO,IAAI,CAAC;gCACd,CAAC;gCAED,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC;gCAC/B,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC;gCACjC,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;gCAC/B,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC,UAAU,CAAC;gCACnC,OAAO,IAAI,CAAC;4BACd,CAAC,CAAC;iCACD,MAAM,CAAC,OAAO,CAAC,CAAC;wBACrB,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC/B,aAAa,GAAG,UAAU,CAAC;YAC7B,CAAC;YAED,MAAM,cAAc,GAAG,kBAAkB,IAAI,aAAa,CAAC,IAAI,CAC7D,CAAC,SAAc,EAAE,EAAE,CAAC,OAAO,SAAS,CAAC,WAAW,KAAK,QAAQ,IAAI,SAAS,CAAC,WAAW,GAAG,CAAC,CAC3F,CAAC;YAEF,IAAI,cAAc,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC/C,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;oBAC1B,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;oBACzD,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;oBACzD,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;wBACtB,OAAO,MAAM,GAAG,MAAM,CAAC;oBACzB,CAAC;oBAED,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;oBACrE,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;oBACrE,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;wBACtB,OAAO,MAAM,GAAG,MAAM,CAAC;oBACzB,CAAC;oBAED,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC;oBAC7F,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC;oBAC7F,OAAO,OAAO,GAAG,OAAO,CAAC;gBAC3B,CAAC,CAAC,CAAC;YACL,CAAC;YAED,aAAa,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC;YAEvD,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,IAAI,eAAe,CAAC,QAAQ,KAAK,KAAK,EAAE,CAAC;gBACnE,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,KAAK,EAAE,aAAa,EAAE;wBACzD,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,uBAAuB,EAAE,aAAa,CAAC,MAAM,CAAC;wBAC5D,OAAO,EAAE,CAAC,SAAc,EAAE,EAAE;4BAC1B,MAAM,QAAQ,GAAG,mBAAmB,CAAC,SAAS,CAAC,GAAG,EAAE,QAAQ,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC;4BAC9E,OAAO,iBAAiB,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;wBAChD,CAAC;qBACF,CAAC,CAAC;oBAEH,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,MAAM,KAAK,aAAa,CAAC,MAAM,EAAE,CAAC;wBACxE,aAAa,GAAG,QAAQ,CAAC;oBAC3B,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,+CAA+C;gBACjD,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,gBAAgB,GAAG,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC;QACzD,MAAM,eAAe,GAAmB,aAAa,CAAC,GAAG,CAAC,CAAC,MAAW,EAAE,EAAE;YACxE,MAAM,QAAQ,GAAG,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YACrE,MAAM,IAAI,GAAQ;gBAChB,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,CAAC;gBAC5B,MAAM,EAAE,MAAM,CAAC,gBAAgB;gBAC/B,WAAW,EAAE,MAAM,CAAC,qBAAqB;gBACzC,UAAU,EAAE,gBAAgB;gBAC5B,WAAW,EAAE,MAAM,CAAC,WAAW;aAChC,CAAC;YAEF,IAAI,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ,EAAE,CAAC;gBAC3C,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;YACxC,CAAC;YAED,IAAI,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;gBACzC,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;YACpC,CAAC;YAED,IAAI,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBACxC,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;YAClC,CAAC;YAED,IAAI,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;gBAC1C,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;YACtC,CAAC;YAED,IAAI,OAAO,MAAM,CAAC,aAAa,KAAK,QAAQ,EAAE,CAAC;gBAC7C,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC;YAC5C,CAAC;YAED,IAAI,OAAO,MAAM,CAAC,YAAY,KAAK,QAAQ,EAAE,CAAC;gBAC5C,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;YAC1C,CAAC;YAED,IAAI,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ,IAAI,MAAM,CAAC,WAAW,GAAG,CAAC,EAAE,CAAC;gBACrE,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;gBACtC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,kBAAkB,CAAC,IAAI,MAAM,CAAC,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACrF,IAAI,CAAC,kBAAkB,GAAG,MAAM,CAAC,kBAAkB,CAAC;gBACtD,CAAC;YACH,CAAC;YAED,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;gBACjD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;YAC3B,CAAC;YAED,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,IAAI,EAAE,MAAM,CAAC,SAAS;gBACtB,GAAG,EAAE,MAAM,CAAC,GAAG;gBACf,IAAI,EAAE,IAAI;gBACV,IAAI;aACL,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YAC5B,MAAM,YAAY,GAAG,OAAO,CAAC,CAAC,IAAI,EAAE,aAAa,KAAK,QAAQ,CAAC;YAC/D,MAAM,YAAY,GAAG,OAAO,CAAC,CAAC,IAAI,EAAE,aAAa,KAAK,QAAQ,CAAC;YAE/D,IAAI,YAAY,IAAI,YAAY,EAAE,CAAC;gBACjC,OAAO,CAAC,CAAC,IAAI,CAAC,aAAc,GAAG,CAAC,CAAC,IAAI,CAAC,aAAc,CAAC;YACvD,CAAC;YAED,MAAM,MAAM,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,IAAI,CAAC,CAAC;YAClC,MAAM,MAAM,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,IAAI,CAAC,CAAC;YAClC,OAAO,MAAM,GAAG,MAAM,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,EAAE,CAAC,KAAK,EAAE,CAAC;YACX,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,qBAAqB;gBAC5B,OAAO,EAAE,kCAAkC,KAAK,GAAG;gBACnD,UAAU,EAAE,sEAAsE;gBAClF,QAAQ,EAAE,iBAAiB,CAAC,OAAO,EAAE;gBACrC,KAAK,EAAE,eAAe;gBACtB,MAAM,EAAE,EAAE,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE;gBAC/C,WAAW,EAAE,EAAE,OAAO,EAAE,kBAAkB,EAAE,OAAO,EAAE,KAAK,EAAE;gBAC5D,QAAQ,EAAE,eAAe,CAAC,QAAQ;gBAClC,OAAO,EAAE,EAAE;aACZ,CAAC;QACJ,CAAC;QAED,IAAI,kBAAkB,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,IAAI,eAAe,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,GAAG,GAAG,EAAE,CAAC;YAC5F,MAAM,EAAE,CAAC,eAAe,CAAC,cAAc,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChH,CAAC;QAED,MAAM,OAAO,GAAG,KAAK;aAClB,WAAW,EAAE;aACb,OAAO,CAAC,qBAAqB,EAAE,WAAW,CAAC;aAC3C,OAAO,CAAC,cAAc,EAAE,oBAAoB,CAAC;aAC7C,OAAO,CAAC,kBAAkB,EAAE,WAAW,CAAC;aACxC,OAAO,CAAC,qBAAqB,EAAE,cAAc,CAAC;aAC9C,IAAI,EAAE,CAAC;QAEV,MAAM,EAAE,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;QAErC,EAAE,CAAC,KAAK,EAAE,CAAC;QAEX,OAAO;YACL,OAAO,EAAE,IAAI;YACb,KAAK;YACL,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ;YAC3C,aAAa,EAAE,aAAa,CAAC,MAAM;YACnC,QAAQ,EAAE,iBAAiB,CAAC,OAAO,EAAE;YACrC,KAAK,EAAE,eAAe;YACtB,QAAQ,EAAE,eAAe,CAAC,QAAQ;YAClC,MAAM,EAAE;gBACN,OAAO,EAAE,aAAa;gBACtB,WAAW;gBACX,KAAK,EAAE,SAAS;gBAChB,cAAc,EAAE,kBAAkB;aACnC;YACD,WAAW,EAAE;gBACX,OAAO,EAAE,kBAAkB;gBAC3B,OAAO,EAAE,kBAAkB,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC,MAAW,EAAE,EAAE,CAAC,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ,IAAI,MAAM,CAAC,WAAW,GAAG,CAAC,CAAC;aACrI;YACD,OAAO,EAAE,eAAe;SACzB,CAAC;IAEJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;QAC7C,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,cAAc;YACrB,OAAO,EAAG,KAAe,CAAC,OAAO;YACjC,QAAQ,EAAE,iBAAiB,CAAC,OAAO,EAAE;YACrC,KAAK,EAAE,eAAe;YACtB,MAAM,EAAE,EAAE,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE;YAC/C,WAAW,EAAE,EAAE,OAAO,EAAE,kBAAkB,EAAE,OAAO,EAAE,KAAK,EAAE;YAC5D,QAAQ,EAAE,eAAe,CAAC,QAAQ;YAClC,OAAO,EAAE,EAAE;SACZ,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,KAAK,GAAG,EAAE,EAAE,WAAW,GAAG,GAAG;IAC7D,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,yBAAyB,CAAC,CAAC;IAE9D,IAAI,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3B,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,oBAAoB;gBAC3B,OAAO,EAAE,yBAAyB,MAAM,sCAAsC;gBAC9E,UAAU,EAAE,mCAAmC,WAAW,EAAE;gBAC5D,QAAQ,EAAE,SAAS;gBACnB,OAAO,EAAE,EAAE;aACZ,CAAC;QACJ,CAAC;QAED,MAAM,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;QAChC,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACjD,EAAE,CAAC,KAAK,EAAE,CAAC;QAEX,MAAM,OAAO,GAAmB,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACnD,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,IAAI,EAAE,KAAK,CAAC,SAAS;YACrB,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,IAAI,EAAE,IAAI;YACV,IAAI,EAAE;gBACJ,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,KAAK,EAAE,GAAG;aACX;SACF,CAAC,CAAC,CAAC;QAEJ,OAAO;YACL,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,UAAU;YACpB,OAAO;SACR,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,gBAAgB;YACvB,OAAO,EAAG,KAAe,CAAC,OAAO;YACjC,QAAQ,EAAE,UAAU;YACpB,OAAO,EAAE,EAAE;SACZ,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAC,GAAW,EAAE,WAAW,GAAG,GAAG;IAC3D,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAC3C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,mBAAmB,CAAC,CAAC;IAE1D,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,iBAAiB,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;QACpD,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,GAAG,KAAK,CAAC,CAAC;YACnD,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,GAAG,SAAS,CAAC,CAAC;YAC3D,MAAM,IAAI,KAAK,CAAC,SAAS,GAAG,iBAAiB,SAAS,OAAO,aAAa,EAAE,CAAC,CAAC;QAChF,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC;IAC9C,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,yBAAyB,EAAE,CAAC;YACtD,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,SAAS,GAAG,+DAA+D;aACnF,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;IAClD,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"symbol-extractor.d.ts","sourceRoot":"","sources":["../../src/core/symbol-extractor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAEtD,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,cAAc,EAAE,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAiFrF"}
|
|
@@ -0,0 +1,78 @@
|
|
|
1
|
+
export function extractSymbolName(node, source) {
|
|
2
|
+
if (node.type === 'function_declaration' || node.type === 'function_definition') {
|
|
3
|
+
for (let i = 0; i < node.childCount; i++) {
|
|
4
|
+
const child = node.child(i);
|
|
5
|
+
if (child && child.type === 'identifier') {
|
|
6
|
+
return source.slice(child.startIndex, child.endIndex);
|
|
7
|
+
}
|
|
8
|
+
}
|
|
9
|
+
}
|
|
10
|
+
if (node.type === 'method_declaration' || node.type === 'method_definition') {
|
|
11
|
+
function findMethodName(n) {
|
|
12
|
+
if (n.type === 'name' || n.type === 'identifier' || n.type === 'property_identifier') {
|
|
13
|
+
const text = source.slice(n.startIndex, n.endIndex);
|
|
14
|
+
if (!['public', 'private', 'protected', 'static', 'function', 'abstract', 'final'].includes(text)) {
|
|
15
|
+
return text;
|
|
16
|
+
}
|
|
17
|
+
}
|
|
18
|
+
for (let i = 0; i < n.childCount; i++) {
|
|
19
|
+
const child = n.child(i);
|
|
20
|
+
if (child) {
|
|
21
|
+
const result = findMethodName(child);
|
|
22
|
+
if (result)
|
|
23
|
+
return result;
|
|
24
|
+
}
|
|
25
|
+
}
|
|
26
|
+
return null;
|
|
27
|
+
}
|
|
28
|
+
const methodName = findMethodName(node);
|
|
29
|
+
if (methodName)
|
|
30
|
+
return methodName;
|
|
31
|
+
}
|
|
32
|
+
if (node.type === 'class_declaration') {
|
|
33
|
+
for (let i = 0; i < node.childCount; i++) {
|
|
34
|
+
const child = node.child(i);
|
|
35
|
+
if (child && (child.type === 'identifier' || child.type === 'type_identifier' || child.type === 'name')) {
|
|
36
|
+
const text = source.slice(child.startIndex, child.endIndex);
|
|
37
|
+
if (text !== 'class') {
|
|
38
|
+
return text;
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
function findAnyIdentifier(n) {
|
|
44
|
+
if (n.type === 'identifier' || n.type === 'name' || n.type === 'property_identifier') {
|
|
45
|
+
const text = source.slice(n.startIndex, n.endIndex);
|
|
46
|
+
if (!['public', 'private', 'protected', 'static', 'function', 'class', 'abstract', 'final', 'const', 'var', 'let'].includes(text)) {
|
|
47
|
+
return text;
|
|
48
|
+
}
|
|
49
|
+
}
|
|
50
|
+
for (let i = 0; i < n.childCount; i++) {
|
|
51
|
+
const child = n.child(i);
|
|
52
|
+
if (child) {
|
|
53
|
+
const result = findAnyIdentifier(child);
|
|
54
|
+
if (result)
|
|
55
|
+
return result;
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
return null;
|
|
59
|
+
}
|
|
60
|
+
const anyIdentifier = findAnyIdentifier(node);
|
|
61
|
+
if (anyIdentifier)
|
|
62
|
+
return anyIdentifier;
|
|
63
|
+
const code = source.slice(node.startIndex, node.endIndex);
|
|
64
|
+
const phpMethodMatch = code.match(/(?:public|private|protected)?\s*(?:static)?\s*function\s+([a-zA-Z_][a-zA-Z0-9_]*)/);
|
|
65
|
+
if (phpMethodMatch)
|
|
66
|
+
return phpMethodMatch[1];
|
|
67
|
+
const jsFunctionMatch = code.match(/function\s+([a-zA-Z_][a-zA-Z0-9_]*)/);
|
|
68
|
+
if (jsFunctionMatch)
|
|
69
|
+
return jsFunctionMatch[1];
|
|
70
|
+
const jsMethodMatch = code.match(/([a-zA-Z_][a-zA-Z0-9_]*)\s*\([^)]*\)\s*\{/);
|
|
71
|
+
if (jsMethodMatch)
|
|
72
|
+
return jsMethodMatch[1];
|
|
73
|
+
const classMatch = code.match(/class\s+([a-zA-Z_][a-zA-Z0-9_]*)/);
|
|
74
|
+
if (classMatch)
|
|
75
|
+
return classMatch[1];
|
|
76
|
+
return `${node.type}_${node.startIndex}`;
|
|
77
|
+
}
|
|
78
|
+
//# sourceMappingURL=symbol-extractor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"symbol-extractor.js","sourceRoot":"","sources":["../../src/core/symbol-extractor.ts"],"names":[],"mappings":"AAEA,MAAM,UAAU,iBAAiB,CAAC,IAAoB,EAAE,MAAc;IACpE,IAAI,IAAI,CAAC,IAAI,KAAK,sBAAsB,IAAI,IAAI,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;QAChF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAC5B,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACzC,OAAO,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,IAAI,CAAC,IAAI,KAAK,oBAAoB,IAAI,IAAI,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;QAC5E,SAAS,cAAc,CAAC,CAAiB;YACvC,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,KAAK,YAAY,IAAI,CAAC,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;gBACrF,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC;gBACpD,IAAI,CAAC,CAAC,QAAQ,EAAE,SAAS,EAAE,WAAW,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;oBAClG,OAAO,IAAI,CAAC;gBACd,CAAC;YACH,CAAC;YAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;gBACtC,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBACzB,IAAI,KAAK,EAAE,CAAC;oBACV,MAAM,MAAM,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;oBACrC,IAAI,MAAM;wBAAE,OAAO,MAAM,CAAC;gBAC5B,CAAC;YACH,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,UAAU,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,UAAU;YAAE,OAAO,UAAU,CAAC;IACpC,CAAC;IAED,IAAI,IAAI,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;QACtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAC5B,IAAI,KAAK,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,CAAC;gBACxG,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;gBAC5D,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;oBACrB,OAAO,IAAI,CAAC;gBACd,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,SAAS,iBAAiB,CAAC,CAAiB;QAC1C,IAAI,CAAC,CAAC,IAAI,KAAK,YAAY,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;YACrF,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC;YACpD,IAAI,CAAC,CAAC,QAAQ,EAAE,SAAS,EAAE,WAAW,EAAE,QAAQ,EAAE,UAAU,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBAClI,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YACzB,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,MAAM,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;gBACxC,IAAI,MAAM;oBAAE,OAAO,MAAM,CAAC;YAC5B,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,aAAa,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;IAC9C,IAAI,aAAa;QAAE,OAAO,aAAa,CAAC;IAExC,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;IAE1D,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,mFAAmF,CAAC,CAAC;IACvH,IAAI,cAAc;QAAE,OAAO,cAAc,CAAC,CAAC,CAAC,CAAC;IAE7C,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,qCAAqC,CAAC,CAAC;IAC1E,IAAI,eAAe;QAAE,OAAO,eAAe,CAAC,CAAC,CAAC,CAAC;IAE/C,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,2CAA2C,CAAC,CAAC;IAC9E,IAAI,aAAa;QAAE,OAAO,aAAa,CAAC,CAAC,CAAC,CAAC;IAE3C,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,kCAAkC,CAAC,CAAC;IAClE,IAAI,UAAU;QAAE,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC;IAErC,OAAO,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;AAC3C,CAAC"}
|