54 lines
1.5 KiB
TypeScript
54 lines
1.5 KiB
TypeScript
import { SQLite } from "../../server/deps.ts";
|
|
import { SysCallMapping } from "../system.ts";
|
|
import { asyncExecute, asyncQuery } from "./store.deno.ts";
|
|
|
|
export function ensureFTSTable(
|
|
db: SQLite,
|
|
tableName: string,
|
|
) {
|
|
const result = db.query(
|
|
`SELECT name FROM sqlite_master WHERE type='table' AND name=?`,
|
|
[tableName],
|
|
);
|
|
if (result.length === 0) {
|
|
asyncExecute(
|
|
db,
|
|
`CREATE VIRTUAL TABLE ${tableName} USING fts5(key, value);`,
|
|
);
|
|
|
|
console.log(`Created fts5 table ${tableName}`);
|
|
}
|
|
return Promise.resolve();
|
|
}
|
|
|
|
export function fullTextSearchSyscalls(
|
|
db: SQLite,
|
|
tableName: string,
|
|
): SysCallMapping {
|
|
return {
|
|
"fulltext.index": async (_ctx, key: string, value: string) => {
|
|
await asyncExecute(db, `DELETE FROM ${tableName} WHERE key = ?`, key);
|
|
await asyncExecute(
|
|
db,
|
|
`INSERT INTO ${tableName} (key, value) VALUES (?, ?)`,
|
|
key,
|
|
value,
|
|
);
|
|
},
|
|
"fulltext.delete": async (_ctx, key: string) => {
|
|
await asyncExecute(db, `DELETE FROM ${tableName} WHERE key = ?`, key);
|
|
},
|
|
"fulltext.search": async (_ctx, phrase: string, limit: number) => {
|
|
console.log("Got search query", phrase);
|
|
return (
|
|
await asyncQuery<any>(
|
|
db,
|
|
`SELECT key, rank FROM ${tableName} WHERE value MATCH ? ORDER BY key, rank LIMIT ?`,
|
|
phrase,
|
|
limit,
|
|
)
|
|
).map((item) => ({ name: item.key, rank: item.rank }));
|
|
},
|
|
};
|
|
}
|