7d9786606f
* fix: 🐛 Fix CodebaseIndexer Bugs * fix: 🎨 Prettier * fix: ✅ Fixed by reoganizing dependent tests * build: 💚 New CI Build
77 lines
2.3 KiB
TypeScript
77 lines
2.3 KiB
TypeScript
import { DatabaseConnection, SqliteDb } from "./refreshIndex.js";
|
|
import {
|
|
IndexResultType,
|
|
MarkCompleteCallback,
|
|
RefreshIndexResults,
|
|
type CodebaseIndex,
|
|
} from "./types.js";
|
|
|
|
import type { IndexTag, IndexingProgressUpdate } from "../index.js";
|
|
|
|
/**
|
|
* This is a CodebaseIndex used for testing which files get indexed.
|
|
* It maintains a SQLite database of all file/tag pairs
|
|
*/
|
|
export class TestCodebaseIndex implements CodebaseIndex {
|
|
relativeExpectedTime: number = 0.1;
|
|
artifactId = "__test__";
|
|
|
|
private static async _createTables(db: DatabaseConnection) {
|
|
await db.exec(`CREATE TABLE IF NOT EXISTS test_index (
|
|
id INTEGER PRIMARY KEY,
|
|
path TEXT NOT NULL,
|
|
branch TEXT NOT NULL,
|
|
directory TEXT NOT NULL
|
|
)`);
|
|
}
|
|
|
|
async *update(
|
|
tag: IndexTag,
|
|
results: RefreshIndexResults,
|
|
markComplete: MarkCompleteCallback,
|
|
repoName: string | undefined,
|
|
): AsyncGenerator<IndexingProgressUpdate, any, unknown> {
|
|
const db = await SqliteDb.get();
|
|
await TestCodebaseIndex._createTables(db);
|
|
|
|
for (const item of [...results.compute, ...results.addTag]) {
|
|
await db.run(
|
|
"INSERT INTO test_index (path, branch, directory) VALUES (?, ?, ?)",
|
|
[item.path, tag.branch, tag.directory],
|
|
);
|
|
}
|
|
|
|
for (const item of [...results.del, ...results.removeTag]) {
|
|
await db.run(
|
|
"DELETE FROM test_index WHERE path = ? AND branch = ? AND directory = ?",
|
|
[item.path, tag.branch, tag.directory],
|
|
);
|
|
}
|
|
|
|
await markComplete(results.compute, IndexResultType.Compute);
|
|
await markComplete(results.addTag, IndexResultType.AddTag);
|
|
await markComplete(results.del, IndexResultType.Delete);
|
|
await markComplete(results.removeTag, IndexResultType.RemoveTag);
|
|
}
|
|
|
|
async getIndexedFilesForTags(tags: IndexTag[]): Promise<string[]> {
|
|
const db = await SqliteDb.get();
|
|
await TestCodebaseIndex._createTables(db);
|
|
|
|
const rows = await db.all(
|
|
`SELECT path FROM test_index WHERE (branch, directory) IN (VALUES ${tags
|
|
.map(() => "(?, ?)")
|
|
.join(", ")})`,
|
|
tags.flatMap((tag) => [tag.branch, tag.directory]),
|
|
);
|
|
|
|
return rows.map((row: any) => row.path);
|
|
}
|
|
|
|
async clearDatabase(): Promise<void> {
|
|
const db = await SqliteDb.get();
|
|
await TestCodebaseIndex._createTables(db);
|
|
await db.run("DELETE FROM test_index");
|
|
}
|
|
}
|