dcedfb5268
Add src/db.ts that dynamically imports bun:sqlite under Bun and better-sqlite3 under Node.js. Exports openDatabase(), loadSqliteVec(), and a shared Database interface. - sqlite-vec loading is now optional — FTS works without it, vector ops throw a clear error if unavailable - CI tests both runtimes: Node 22/23 via vitest, Bun via bun test - All 104 unit tests pass on both Node and Bun
53 lines
1.4 KiB
TypeScript
53 lines
1.4 KiB
TypeScript
/**
|
|
* db.ts - Cross-runtime SQLite compatibility layer
|
|
*
|
|
* Provides a unified Database export that works under both Bun (bun:sqlite)
|
|
* and Node.js (better-sqlite3). The APIs are nearly identical — the main
|
|
* difference is the import path.
|
|
*/
|
|
|
|
export const isBun = typeof globalThis.Bun !== "undefined";
|
|
|
|
let _Database: any;
|
|
let _sqliteVecLoad: (db: any) => void;
|
|
|
|
if (isBun) {
|
|
_Database = (await import("bun:sqlite")).Database;
|
|
const { getLoadablePath } = await import("sqlite-vec");
|
|
_sqliteVecLoad = (db: any) => db.loadExtension(getLoadablePath());
|
|
} else {
|
|
_Database = (await import("better-sqlite3")).default;
|
|
const sqliteVec = await import("sqlite-vec");
|
|
_sqliteVecLoad = (db: any) => sqliteVec.load(db);
|
|
}
|
|
|
|
/**
|
|
* Open a SQLite database. Works with both bun:sqlite and better-sqlite3.
|
|
*/
|
|
export function openDatabase(path: string): Database {
|
|
return new _Database(path) as Database;
|
|
}
|
|
|
|
/**
|
|
* Common subset of the Database interface used throughout QMD.
|
|
*/
|
|
export interface Database {
|
|
exec(sql: string): void;
|
|
prepare(sql: string): Statement;
|
|
loadExtension(path: string): void;
|
|
close(): void;
|
|
}
|
|
|
|
export interface Statement {
|
|
run(...params: any[]): { changes: number; lastInsertRowid: number | bigint };
|
|
get(...params: any[]): any;
|
|
all(...params: any[]): any[];
|
|
}
|
|
|
|
/**
|
|
* Load the sqlite-vec extension into a database.
|
|
*/
|
|
export function loadSqliteVec(db: Database): void {
|
|
_sqliteVecLoad(db);
|
|
}
|