@monlite/fts 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 +71 -0
- package/dist/index.cjs +95 -0
- package/dist/index.cjs.map +1 -0
- package/dist/index.d.cts +33 -0
- package/dist/index.d.ts +33 -0
- package/dist/index.js +92 -0
- package/dist/index.js.map +1 -0
- package/package.json +65 -0
package/LICENSE
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
MIT License
|
|
2
|
+
|
|
3
|
+
Copyright (c) 2026 Emad Jumaah
|
|
4
|
+
|
|
5
|
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
6
|
+
of this software and associated documentation files (the "Software"), to deal
|
|
7
|
+
in the Software without restriction, including without limitation the rights
|
|
8
|
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
9
|
+
copies of the Software, and to permit persons to whom the Software is
|
|
10
|
+
furnished to do so, subject to the following conditions:
|
|
11
|
+
|
|
12
|
+
The above copyright notice and this permission notice shall be included in all
|
|
13
|
+
copies or substantial portions of the Software.
|
|
14
|
+
|
|
15
|
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
16
|
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
17
|
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
18
|
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
19
|
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
20
|
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
21
|
+
SOFTWARE.
|
package/README.md
ADDED
|
@@ -0,0 +1,71 @@
|
|
|
1
|
+
# 🌙 @monlite/fts
|
|
2
|
+
|
|
3
|
+
> Full-text search for [`@monlite/core`](https://www.npmjs.com/package/@monlite/core),
|
|
4
|
+
> powered by SQLite's built-in **FTS5**. Adds `collection.search()`.
|
|
5
|
+
|
|
6
|
+
A monlite plugin — opt in by passing it to `createDb`, point it at the fields you
|
|
7
|
+
want searchable, and it maintains an FTS5 index automatically (on every write,
|
|
8
|
+
including changes applied by `@monlite/sync`).
|
|
9
|
+
|
|
10
|
+
```ts
|
|
11
|
+
import { createDb } from "@monlite/core";
|
|
12
|
+
import { fts } from "@monlite/fts";
|
|
13
|
+
|
|
14
|
+
const db = createDb("./app.db", {
|
|
15
|
+
plugins: [fts({ posts: ["title", "body"], users: ["name", "profile.bio"] })],
|
|
16
|
+
});
|
|
17
|
+
|
|
18
|
+
await db.collection("posts").create({
|
|
19
|
+
data: { title: "Hello world", body: "the quick brown fox" },
|
|
20
|
+
});
|
|
21
|
+
|
|
22
|
+
const results = await db.collection("posts").search("quick");
|
|
23
|
+
// [ { _id, title, body, _score, … } ] — ranked, full documents
|
|
24
|
+
```
|
|
25
|
+
|
|
26
|
+
## Install
|
|
27
|
+
|
|
28
|
+
```bash
|
|
29
|
+
npm install @monlite/core @monlite/fts
|
|
30
|
+
```
|
|
31
|
+
|
|
32
|
+
No native dependency — FTS5 ships inside SQLite, so this works on both monlite
|
|
33
|
+
backends (`better-sqlite3` and the built-in `node:sqlite`).
|
|
34
|
+
|
|
35
|
+
## API
|
|
36
|
+
|
|
37
|
+
```ts
|
|
38
|
+
fts(spec: Record<string, string[]>): MonlitePlugin
|
|
39
|
+
```
|
|
40
|
+
|
|
41
|
+
`spec` maps a collection name to the field paths to index (dot-notation for
|
|
42
|
+
nested fields, e.g. `"profile.bio"`).
|
|
43
|
+
|
|
44
|
+
```ts
|
|
45
|
+
collection.search(query, {
|
|
46
|
+
limit?: number, // default 50
|
|
47
|
+
where?: WhereInput<T>, // also constrain with a normal monlite filter
|
|
48
|
+
}): Promise<Array<WithId<T> & { _score: number }>>
|
|
49
|
+
```
|
|
50
|
+
|
|
51
|
+
- `query` uses FTS5 [MATCH syntax](https://www.sqlite.org/fts5.html#full_text_query_syntax)
|
|
52
|
+
(bare terms are AND-ed; `"a phrase"`; `term*` prefix; `a OR b`).
|
|
53
|
+
- Results are ordered by relevance; `_score` is higher = better.
|
|
54
|
+
- `where` is applied after matching, so you can combine search with structured filters.
|
|
55
|
+
|
|
56
|
+
```ts
|
|
57
|
+
import { reindex } from "@monlite/fts";
|
|
58
|
+
reindex(db, "posts", ["title", "body"]); // rebuild a collection's index
|
|
59
|
+
```
|
|
60
|
+
|
|
61
|
+
## How it works
|
|
62
|
+
|
|
63
|
+
For each configured collection, the plugin creates an FTS5 virtual table
|
|
64
|
+
(`<collection>_fts`) keyed by the document `_id`. It indexes on `init` (backfilling
|
|
65
|
+
existing documents when the index is empty) and keeps it current via the plugin
|
|
66
|
+
`afterWrite` hook. Search runs `MATCH` against that table, then returns the live
|
|
67
|
+
documents from the collection in rank order.
|
|
68
|
+
|
|
69
|
+
## License
|
|
70
|
+
|
|
71
|
+
MIT 🌙
|
package/dist/index.cjs
ADDED
|
@@ -0,0 +1,95 @@
|
|
|
1
|
+
'use strict';
|
|
2
|
+
|
|
3
|
+
// src/index.ts
|
|
4
|
+
var ftsTable = (coll) => `${coll}_fts`;
|
|
5
|
+
var col = (i) => `f${i}`;
|
|
6
|
+
function extractText(doc, path) {
|
|
7
|
+
let cur = doc;
|
|
8
|
+
for (const seg of path.split(".")) {
|
|
9
|
+
if (cur == null) return "";
|
|
10
|
+
cur = cur[seg];
|
|
11
|
+
}
|
|
12
|
+
if (cur == null) return "";
|
|
13
|
+
if (typeof cur === "string") return cur;
|
|
14
|
+
if (Array.isArray(cur))
|
|
15
|
+
return cur.filter((x) => x != null).map(String).join(" ");
|
|
16
|
+
if (typeof cur === "number" || typeof cur === "boolean") return String(cur);
|
|
17
|
+
return "";
|
|
18
|
+
}
|
|
19
|
+
function indexDoc(db, coll, fields, id) {
|
|
20
|
+
const sqlite = db.sqlite;
|
|
21
|
+
sqlite.prepare(`DELETE FROM "${ftsTable(coll)}" WHERE doc_id = ?`).run(id);
|
|
22
|
+
const doc = db.collection(coll).getRaw(id);
|
|
23
|
+
if (!doc) return;
|
|
24
|
+
const cols = fields.map((_, i) => `"${col(i)}"`).join(", ");
|
|
25
|
+
const placeholders = fields.map(() => "?").join(", ");
|
|
26
|
+
const values = fields.map((f) => extractText(doc, f));
|
|
27
|
+
sqlite.prepare(
|
|
28
|
+
`INSERT INTO "${ftsTable(coll)}"(doc_id, ${cols}) VALUES (?, ${placeholders})`
|
|
29
|
+
).run(id, ...values);
|
|
30
|
+
}
|
|
31
|
+
function reindex(db, coll, fields) {
|
|
32
|
+
const sqlite = db.sqlite;
|
|
33
|
+
sqlite.exec(`DELETE FROM "${ftsTable(coll)}"`);
|
|
34
|
+
for (const doc of db.collection(coll).findManyCore({})) {
|
|
35
|
+
indexDoc(db, coll, fields, doc._id);
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
function search(db, coll, spec, query, opts) {
|
|
39
|
+
const fields = spec[coll.name];
|
|
40
|
+
if (!fields) {
|
|
41
|
+
throw new Error(
|
|
42
|
+
`Collection "${coll.name}" is not configured for full-text search`
|
|
43
|
+
);
|
|
44
|
+
}
|
|
45
|
+
const limit = opts?.limit ?? 50;
|
|
46
|
+
const rows = db.sqlite.prepare(
|
|
47
|
+
`SELECT doc_id, rank FROM "${ftsTable(coll.name)}" WHERE "${ftsTable(coll.name)}" MATCH ? ORDER BY rank LIMIT ?`
|
|
48
|
+
).all(query, limit);
|
|
49
|
+
let allowed = null;
|
|
50
|
+
if (opts?.where) {
|
|
51
|
+
const ids = rows.map((r) => r.doc_id);
|
|
52
|
+
const idIn = { _id: { in: ids } };
|
|
53
|
+
const matching = coll.findManyCore({
|
|
54
|
+
where: { AND: [opts.where, idIn] }
|
|
55
|
+
});
|
|
56
|
+
allowed = new Set(matching.map((d) => d._id));
|
|
57
|
+
}
|
|
58
|
+
const out = [];
|
|
59
|
+
for (const r of rows) {
|
|
60
|
+
if (allowed && !allowed.has(r.doc_id)) continue;
|
|
61
|
+
const doc = coll.getRaw(r.doc_id);
|
|
62
|
+
if (doc) out.push({ ...doc, _score: -r.rank });
|
|
63
|
+
}
|
|
64
|
+
return Promise.resolve(out);
|
|
65
|
+
}
|
|
66
|
+
function fts(spec) {
|
|
67
|
+
let database;
|
|
68
|
+
return {
|
|
69
|
+
name: "fts",
|
|
70
|
+
init(db) {
|
|
71
|
+
database = db;
|
|
72
|
+
for (const [coll, fields] of Object.entries(spec)) {
|
|
73
|
+
const cols = fields.map((_, i) => `"${col(i)}"`).join(", ");
|
|
74
|
+
db.sqlite.exec(
|
|
75
|
+
`CREATE VIRTUAL TABLE IF NOT EXISTS "${ftsTable(coll)}" USING fts5(doc_id UNINDEXED, ${cols})`
|
|
76
|
+
);
|
|
77
|
+
const count = db.sqlite.prepare(`SELECT count(*) AS n FROM "${ftsTable(coll)}"`).get();
|
|
78
|
+
if (count.n === 0) reindex(db, coll, fields);
|
|
79
|
+
}
|
|
80
|
+
},
|
|
81
|
+
afterWrite(db, { collection, ids }) {
|
|
82
|
+
const fields = spec[collection];
|
|
83
|
+
if (!fields) return;
|
|
84
|
+
for (const id of ids) indexDoc(db, collection, fields, id);
|
|
85
|
+
},
|
|
86
|
+
collectionMethods: {
|
|
87
|
+
search: (collection, query, opts) => search(database, collection, spec, query, opts)
|
|
88
|
+
}
|
|
89
|
+
};
|
|
90
|
+
}
|
|
91
|
+
|
|
92
|
+
exports.fts = fts;
|
|
93
|
+
exports.reindex = reindex;
|
|
94
|
+
//# sourceMappingURL=index.cjs.map
|
|
95
|
+
//# sourceMappingURL=index.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../src/index.ts"],"names":[],"mappings":";;;AA4BA,IAAM,QAAA,GAAW,CAAC,IAAA,KAAiB,CAAA,EAAG,IAAI,CAAA,IAAA,CAAA;AAC1C,IAAM,GAAA,GAAM,CAAC,CAAA,KAAc,CAAA,CAAA,EAAI,CAAC,CAAA,CAAA;AAGhC,SAAS,WAAA,CAAY,KAA0B,IAAA,EAAsB;AACnE,EAAA,IAAI,GAAA,GAAW,GAAA;AACf,EAAA,KAAA,MAAW,GAAA,IAAO,IAAA,CAAK,KAAA,CAAM,GAAG,CAAA,EAAG;AACjC,IAAA,IAAI,GAAA,IAAO,MAAM,OAAO,EAAA;AACxB,IAAA,GAAA,GAAM,IAAI,GAAG,CAAA;AAAA,EACf;AACA,EAAA,IAAI,GAAA,IAAO,MAAM,OAAO,EAAA;AACxB,EAAA,IAAI,OAAO,GAAA,KAAQ,QAAA,EAAU,OAAO,GAAA;AACpC,EAAA,IAAI,KAAA,CAAM,QAAQ,GAAG,CAAA;AACnB,IAAA,OAAO,GAAA,CACJ,MAAA,CAAO,CAAC,CAAA,KAAM,CAAA,IAAK,IAAI,CAAA,CACvB,GAAA,CAAI,MAAM,CAAA,CACV,IAAA,CAAK,GAAG,CAAA;AACb,EAAA,IAAI,OAAO,QAAQ,QAAA,IAAY,OAAO,QAAQ,SAAA,EAAW,OAAO,OAAO,GAAG,CAAA;AAC1E,EAAA,OAAO,EAAA;AACT;AAEA,SAAS,QAAA,CACP,EAAA,EACA,IAAA,EACA,MAAA,EACA,EAAA,EACM;AACN,EAAA,MAAM,SAAS,EAAA,CAAG,MAAA;AAClB,EAAA,MAAA,CAAO,OAAA,CAAQ,gBAAgB,QAAA,CAAS,IAAI,CAAC,CAAA,kBAAA,CAAoB,CAAA,CAAE,IAAI,EAAE,CAAA;AACzE,EAAA,MAAM,MAAM,EAAA,CAAG,UAAA,CAAW,IAAI,CAAA,CAAE,OAAO,EAAE,CAAA;AACzC,EAAA,IAAI,CAAC,GAAA,EAAK;AACV,EAAA,MAAM,IAAA,GAAO,MAAA,CAAO,GAAA,CAAI,CAAC,CAAA,EAAG,CAAA,KAAM,CAAA,CAAA,EAAI,GAAA,CAAI,CAAC,CAAC,CAAA,CAAA,CAAG,CAAA,CAAE,KAAK,IAAI,CAAA;AAC1D,EAAA,MAAM,eAAe,MAAA,CAAO,GAAA,CAAI,MAAM,GAAG,CAAA,CAAE,KAAK,IAAI,CAAA;AACpD,EAAA,MAAM,MAAA,GAAS,OAAO,GAAA,CAAI,CAAC,MAAM,WAAA,CAAY,GAAA,EAAK,CAAC,CAAC,CAAA;AACpD,EAAA,MAAA,CACG,OAAA;AAAA,IACC,gBAAgB,QAAA,CAAS,IAAI,CAAC,CAAA,UAAA,EAAa,IAAI,gBAAgB,YAAY,CAAA,CAAA;AAAA,GAC7E,CACC,GAAA,CAAI,EAAA,EAAI,GAAG,MAAM,CAAA;AACtB;AAGO,SAAS,OAAA,CAAQ,EAAA,EAAa,IAAA,EAAc,MAAA,EAAwB;AACzE,EAAA,MAAM,SAAS,EAAA,CAAG,MAAA;AAClB,EAAA,MAAA,CAAO,IAAA,CAAK,CAAA,aAAA,EAAgB,QAAA,CAAS,IAAI,CAAC,CAAA,CAAA,CAAG,CAAA;AAC7C,EAAA,KAAA,MAAW,GAAA,IAAO,GAAG,UAAA,CAAW,IAAI,EAAE,YAAA,CAAa,EAAE,CAAA,EAAG;AACtD,IAAA,QAAA,CAAS,EAAA,EAAI,IAAA,EAAM,MAAA,EAAS,GAAA,CAAoB,GAAG,CAAA;AAAA,EACrD;AACF;AAEA,SAAS,MAAA,CACP,EAAA,EACA,IAAA,EACA,IAAA,EACA,OACA,IAAA,EAC4B;AAC5B,EAAA,MAAM,MAAA,GAAS,IAAA,CAAK,IAAA,CAAK,IAAI,CAAA;AAC7B,EAAA,IAAI,CAAC,MAAA,EAAQ;AACX,IAAA,MAAM,IAAI,KAAA;AAAA,MACR,CAAA,YAAA,EAAe,KAAK,IAAI,CAAA,wCAAA;AAAA,KAC1B;AAAA,EACF;AACA,EAAA,MAAM,KAAA,GAAQ,MAAM,KAAA,IAAS,EAAA;AAC7B,EAAA,MAAM,IAAA,GAAO,GAAG,MAAA,CACb,OAAA;AAAA,IACC,CAAA,0BAAA,EAA6B,SAAS,IAAA,CAAK,IAAI,CAAC,CAAA,SAAA,EACpC,QAAA,CAAS,IAAA,CAAK,IAAI,CAAC,CAAA,+BAAA;AAAA,GACjC,CACC,GAAA,CAAI,KAAA,EAAO,KAAK,CAAA;AAEnB,EAAA,IAAI,OAAA,GAA8B,IAAA;AAClC,EAAA,IAAI,MAAM,KAAA,EAAO;AACf,IAAA,MAAM,MAAM,IAAA,CAAK,GAAA,CAAI,CAAC,CAAA,KAAM,EAAE,MAAM,CAAA;AACpC,IAAA,MAAM,OAAO,EAAE,GAAA,EAAK,EAAE,EAAA,EAAI,KAAI,EAAE;AAChC,IAAA,MAAM,QAAA,GAAW,KAAK,YAAA,CAAa;AAAA,MACjC,OAAO,EAAE,GAAA,EAAK,CAAC,IAAA,CAAK,KAAA,EAAO,IAAI,CAAA;AAAE,KAClC,CAAA;AACD,IAAA,OAAA,GAAU,IAAI,IAAI,QAAA,CAAS,GAAA,CAAI,CAAC,CAAA,KAAM,CAAA,CAAE,GAAG,CAAC,CAAA;AAAA,EAC9C;AAEA,EAAA,MAAM,MAAyB,EAAC;AAChC,EAAA,KAAA,MAAW,KAAK,IAAA,EAAM;AACpB,IAAA,IAAI,WAAW,CAAC,OAAA,CAAQ,GAAA,CAAI,CAAA,CAAE,MAAM,CAAA,EAAG;AACvC,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,MAAA,CAAO,CAAA,CAAE,MAAM,CAAA;AAChC,IAAA,IAAI,GAAA,EAAK,GAAA,CAAI,IAAA,CAAK,EAAE,GAAG,KAAK,MAAA,EAAQ,CAAC,CAAA,CAAE,IAAA,EAAyB,CAAA;AAAA,EAClE;AACA,EAAA,OAAO,OAAA,CAAQ,QAAQ,GAAG,CAAA;AAC5B;AAYO,SAAS,IAAI,IAAA,EAA8B;AAChD,EAAA,IAAI,QAAA;AACJ,EAAA,OAAO;AAAA,IACL,IAAA,EAAM,KAAA;AAAA,IACN,KAAK,EAAA,EAAI;AACP,MAAA,QAAA,GAAW,EAAA;AACX,MAAA,KAAA,MAAW,CAAC,IAAA,EAAM,MAAM,KAAK,MAAA,CAAO,OAAA,CAAQ,IAAI,CAAA,EAAG;AACjD,QAAA,MAAM,IAAA,GAAO,MAAA,CAAO,GAAA,CAAI,CAAC,CAAA,EAAG,CAAA,KAAM,CAAA,CAAA,EAAI,GAAA,CAAI,CAAC,CAAC,CAAA,CAAA,CAAG,CAAA,CAAE,KAAK,IAAI,CAAA;AAC1D,QAAA,EAAA,CAAG,MAAA,CAAO,IAAA;AAAA,UACR,CAAA,oCAAA,EAAuC,QAAA,CAAS,IAAI,CAAC,kCACnB,IAAI,CAAA,CAAA;AAAA,SACxC;AAEA,QAAA,MAAM,KAAA,GAAQ,EAAA,CAAG,MAAA,CACd,OAAA,CAAQ,CAAA,2BAAA,EAA8B,SAAS,IAAI,CAAC,CAAA,CAAA,CAAG,CAAA,CACvD,GAAA,EAAI;AACP,QAAA,IAAI,MAAM,CAAA,KAAM,CAAA,EAAG,OAAA,CAAQ,EAAA,EAAI,MAAM,MAAM,CAAA;AAAA,MAC7C;AAAA,IACF,CAAA;AAAA,IACA,UAAA,CAAW,EAAA,EAAI,EAAE,UAAA,EAAY,KAAI,EAAG;AAClC,MAAA,MAAM,MAAA,GAAS,KAAK,UAAU,CAAA;AAC9B,MAAA,IAAI,CAAC,MAAA,EAAQ;AACb,MAAA,KAAA,MAAW,MAAM,GAAA,EAAK,QAAA,CAAS,EAAA,EAAI,UAAA,EAAY,QAAQ,EAAE,CAAA;AAAA,IAC3D,CAAA;AAAA,IACA,iBAAA,EAAmB;AAAA,MACjB,MAAA,EAAQ,CAAC,UAAA,EAAY,KAAA,EAAe,IAAA,KAClC,OAAO,QAAA,EAAU,UAAA,EAAY,IAAA,EAAM,KAAA,EAAO,IAAI;AAAA;AAClD,GACF;AACF","file":"index.cjs","sourcesContent":["import type {\n Collection,\n Doc,\n Monlite,\n MonlitePlugin,\n WhereInput,\n WithId,\n} from \"@monlite/core\";\n\n/** Map of collection name → searchable field paths (dot-notation allowed). */\nexport type FtsSpec = Record<string, string[]>;\n\nexport interface SearchOptions<T = Doc> {\n /** Max results (default 50). */\n limit?: number;\n /** Additionally constrain matches with a normal monlite where clause. */\n where?: WhereInput<T>;\n}\n\nexport type SearchResult<T = Doc> = WithId<T> & { _score: number };\n\n// Make `collection.search()` typed wherever @monlite/fts is imported.\ndeclare module \"@monlite/core\" {\n interface Collection<T> {\n search(query: string, opts?: SearchOptions<T>): Promise<SearchResult<T>[]>;\n }\n}\n\nconst ftsTable = (coll: string) => `${coll}_fts`;\nconst col = (i: number) => `f${i}`;\n\n/** Extract searchable text for a field path from a document. */\nfunction extractText(doc: Record<string, any>, path: string): string {\n let cur: any = doc;\n for (const seg of path.split(\".\")) {\n if (cur == null) return \"\";\n cur = cur[seg];\n }\n if (cur == null) return \"\";\n if (typeof cur === \"string\") return cur;\n if (Array.isArray(cur))\n return cur\n .filter((x) => x != null)\n .map(String)\n .join(\" \");\n if (typeof cur === \"number\" || typeof cur === \"boolean\") return String(cur);\n return \"\";\n}\n\nfunction indexDoc(\n db: Monlite,\n coll: string,\n fields: string[],\n id: string,\n): void {\n const sqlite = db.sqlite;\n sqlite.prepare(`DELETE FROM \"${ftsTable(coll)}\" WHERE doc_id = ?`).run(id);\n const doc = db.collection(coll).getRaw(id);\n if (!doc) return; // deleted\n const cols = fields.map((_, i) => `\"${col(i)}\"`).join(\", \");\n const placeholders = fields.map(() => \"?\").join(\", \");\n const values = fields.map((f) => extractText(doc, f));\n sqlite\n .prepare(\n `INSERT INTO \"${ftsTable(coll)}\"(doc_id, ${cols}) VALUES (?, ${placeholders})`,\n )\n .run(id, ...values);\n}\n\n/** Rebuild a collection's FTS index from scratch. */\nexport function reindex(db: Monlite, coll: string, fields: string[]): void {\n const sqlite = db.sqlite;\n sqlite.exec(`DELETE FROM \"${ftsTable(coll)}\"`);\n for (const doc of db.collection(coll).findManyCore({})) {\n indexDoc(db, coll, fields, (doc as WithId<Doc>)._id);\n }\n}\n\nfunction search<T = Doc>(\n db: Monlite,\n coll: Collection<T>,\n spec: FtsSpec,\n query: string,\n opts?: SearchOptions<T>,\n): Promise<SearchResult<T>[]> {\n const fields = spec[coll.name];\n if (!fields) {\n throw new Error(\n `Collection \"${coll.name}\" is not configured for full-text search`,\n );\n }\n const limit = opts?.limit ?? 50;\n const rows = db.sqlite\n .prepare(\n `SELECT doc_id, rank FROM \"${ftsTable(coll.name)}\" ` +\n `WHERE \"${ftsTable(coll.name)}\" MATCH ? ORDER BY rank LIMIT ?`,\n )\n .all(query, limit) as Array<{ doc_id: string; rank: number }>;\n\n let allowed: Set<string> | null = null;\n if (opts?.where) {\n const ids = rows.map((r) => r.doc_id);\n const idIn = { _id: { in: ids } } as WhereInput<T>;\n const matching = coll.findManyCore({\n where: { AND: [opts.where, idIn] } as WhereInput<T>,\n });\n allowed = new Set(matching.map((d) => d._id));\n }\n\n const out: SearchResult<T>[] = [];\n for (const r of rows) {\n if (allowed && !allowed.has(r.doc_id)) continue;\n const doc = coll.getRaw(r.doc_id);\n if (doc) out.push({ ...doc, _score: -r.rank } as SearchResult<T>);\n }\n return Promise.resolve(out);\n}\n\n/**\n * Full-text search plugin (SQLite FTS5). Pass it to `createDb({ plugins: [...] })`\n * with a map of collection → searchable fields. Adds `collection.search()`, keeps\n * the index in sync on every write, and backfills existing documents on open.\n *\n * ```ts\n * const db = createDb(\"./app.db\", { plugins: [fts({ posts: [\"title\", \"body\"] })] });\n * await db.collection(\"posts\").search(\"hello world\");\n * ```\n */\nexport function fts(spec: FtsSpec): MonlitePlugin {\n let database: Monlite;\n return {\n name: \"fts\",\n init(db) {\n database = db;\n for (const [coll, fields] of Object.entries(spec)) {\n const cols = fields.map((_, i) => `\"${col(i)}\"`).join(\", \");\n db.sqlite.exec(\n `CREATE VIRTUAL TABLE IF NOT EXISTS \"${ftsTable(coll)}\" ` +\n `USING fts5(doc_id UNINDEXED, ${cols})`,\n );\n // Backfill when the index is empty (e.g. enabling FTS on an existing db).\n const count = db.sqlite\n .prepare(`SELECT count(*) AS n FROM \"${ftsTable(coll)}\"`)\n .get() as { n: number };\n if (count.n === 0) reindex(db, coll, fields);\n }\n },\n afterWrite(db, { collection, ids }) {\n const fields = spec[collection];\n if (!fields) return;\n for (const id of ids) indexDoc(db, collection, fields, id);\n },\n collectionMethods: {\n search: (collection, query: string, opts?: SearchOptions) =>\n search(database, collection, spec, query, opts),\n },\n };\n}\n"]}
|
package/dist/index.d.cts
ADDED
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
import { Doc, WhereInput, WithId, MonlitePlugin, Monlite } from '@monlite/core';
|
|
2
|
+
|
|
3
|
+
/** Map of collection name → searchable field paths (dot-notation allowed). */
|
|
4
|
+
type FtsSpec = Record<string, string[]>;
|
|
5
|
+
interface SearchOptions<T = Doc> {
|
|
6
|
+
/** Max results (default 50). */
|
|
7
|
+
limit?: number;
|
|
8
|
+
/** Additionally constrain matches with a normal monlite where clause. */
|
|
9
|
+
where?: WhereInput<T>;
|
|
10
|
+
}
|
|
11
|
+
type SearchResult<T = Doc> = WithId<T> & {
|
|
12
|
+
_score: number;
|
|
13
|
+
};
|
|
14
|
+
declare module "@monlite/core" {
|
|
15
|
+
interface Collection<T> {
|
|
16
|
+
search(query: string, opts?: SearchOptions<T>): Promise<SearchResult<T>[]>;
|
|
17
|
+
}
|
|
18
|
+
}
|
|
19
|
+
/** Rebuild a collection's FTS index from scratch. */
|
|
20
|
+
declare function reindex(db: Monlite, coll: string, fields: string[]): void;
|
|
21
|
+
/**
|
|
22
|
+
* Full-text search plugin (SQLite FTS5). Pass it to `createDb({ plugins: [...] })`
|
|
23
|
+
* with a map of collection → searchable fields. Adds `collection.search()`, keeps
|
|
24
|
+
* the index in sync on every write, and backfills existing documents on open.
|
|
25
|
+
*
|
|
26
|
+
* ```ts
|
|
27
|
+
* const db = createDb("./app.db", { plugins: [fts({ posts: ["title", "body"] })] });
|
|
28
|
+
* await db.collection("posts").search("hello world");
|
|
29
|
+
* ```
|
|
30
|
+
*/
|
|
31
|
+
declare function fts(spec: FtsSpec): MonlitePlugin;
|
|
32
|
+
|
|
33
|
+
export { type FtsSpec, type SearchOptions, type SearchResult, fts, reindex };
|
package/dist/index.d.ts
ADDED
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
import { Doc, WhereInput, WithId, MonlitePlugin, Monlite } from '@monlite/core';
|
|
2
|
+
|
|
3
|
+
/** Map of collection name → searchable field paths (dot-notation allowed). */
|
|
4
|
+
type FtsSpec = Record<string, string[]>;
|
|
5
|
+
interface SearchOptions<T = Doc> {
|
|
6
|
+
/** Max results (default 50). */
|
|
7
|
+
limit?: number;
|
|
8
|
+
/** Additionally constrain matches with a normal monlite where clause. */
|
|
9
|
+
where?: WhereInput<T>;
|
|
10
|
+
}
|
|
11
|
+
type SearchResult<T = Doc> = WithId<T> & {
|
|
12
|
+
_score: number;
|
|
13
|
+
};
|
|
14
|
+
declare module "@monlite/core" {
|
|
15
|
+
interface Collection<T> {
|
|
16
|
+
search(query: string, opts?: SearchOptions<T>): Promise<SearchResult<T>[]>;
|
|
17
|
+
}
|
|
18
|
+
}
|
|
19
|
+
/** Rebuild a collection's FTS index from scratch. */
|
|
20
|
+
declare function reindex(db: Monlite, coll: string, fields: string[]): void;
|
|
21
|
+
/**
|
|
22
|
+
* Full-text search plugin (SQLite FTS5). Pass it to `createDb({ plugins: [...] })`
|
|
23
|
+
* with a map of collection → searchable fields. Adds `collection.search()`, keeps
|
|
24
|
+
* the index in sync on every write, and backfills existing documents on open.
|
|
25
|
+
*
|
|
26
|
+
* ```ts
|
|
27
|
+
* const db = createDb("./app.db", { plugins: [fts({ posts: ["title", "body"] })] });
|
|
28
|
+
* await db.collection("posts").search("hello world");
|
|
29
|
+
* ```
|
|
30
|
+
*/
|
|
31
|
+
declare function fts(spec: FtsSpec): MonlitePlugin;
|
|
32
|
+
|
|
33
|
+
export { type FtsSpec, type SearchOptions, type SearchResult, fts, reindex };
|
package/dist/index.js
ADDED
|
@@ -0,0 +1,92 @@
|
|
|
1
|
+
// src/index.ts
|
|
2
|
+
var ftsTable = (coll) => `${coll}_fts`;
|
|
3
|
+
var col = (i) => `f${i}`;
|
|
4
|
+
function extractText(doc, path) {
|
|
5
|
+
let cur = doc;
|
|
6
|
+
for (const seg of path.split(".")) {
|
|
7
|
+
if (cur == null) return "";
|
|
8
|
+
cur = cur[seg];
|
|
9
|
+
}
|
|
10
|
+
if (cur == null) return "";
|
|
11
|
+
if (typeof cur === "string") return cur;
|
|
12
|
+
if (Array.isArray(cur))
|
|
13
|
+
return cur.filter((x) => x != null).map(String).join(" ");
|
|
14
|
+
if (typeof cur === "number" || typeof cur === "boolean") return String(cur);
|
|
15
|
+
return "";
|
|
16
|
+
}
|
|
17
|
+
function indexDoc(db, coll, fields, id) {
|
|
18
|
+
const sqlite = db.sqlite;
|
|
19
|
+
sqlite.prepare(`DELETE FROM "${ftsTable(coll)}" WHERE doc_id = ?`).run(id);
|
|
20
|
+
const doc = db.collection(coll).getRaw(id);
|
|
21
|
+
if (!doc) return;
|
|
22
|
+
const cols = fields.map((_, i) => `"${col(i)}"`).join(", ");
|
|
23
|
+
const placeholders = fields.map(() => "?").join(", ");
|
|
24
|
+
const values = fields.map((f) => extractText(doc, f));
|
|
25
|
+
sqlite.prepare(
|
|
26
|
+
`INSERT INTO "${ftsTable(coll)}"(doc_id, ${cols}) VALUES (?, ${placeholders})`
|
|
27
|
+
).run(id, ...values);
|
|
28
|
+
}
|
|
29
|
+
function reindex(db, coll, fields) {
|
|
30
|
+
const sqlite = db.sqlite;
|
|
31
|
+
sqlite.exec(`DELETE FROM "${ftsTable(coll)}"`);
|
|
32
|
+
for (const doc of db.collection(coll).findManyCore({})) {
|
|
33
|
+
indexDoc(db, coll, fields, doc._id);
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
function search(db, coll, spec, query, opts) {
|
|
37
|
+
const fields = spec[coll.name];
|
|
38
|
+
if (!fields) {
|
|
39
|
+
throw new Error(
|
|
40
|
+
`Collection "${coll.name}" is not configured for full-text search`
|
|
41
|
+
);
|
|
42
|
+
}
|
|
43
|
+
const limit = opts?.limit ?? 50;
|
|
44
|
+
const rows = db.sqlite.prepare(
|
|
45
|
+
`SELECT doc_id, rank FROM "${ftsTable(coll.name)}" WHERE "${ftsTable(coll.name)}" MATCH ? ORDER BY rank LIMIT ?`
|
|
46
|
+
).all(query, limit);
|
|
47
|
+
let allowed = null;
|
|
48
|
+
if (opts?.where) {
|
|
49
|
+
const ids = rows.map((r) => r.doc_id);
|
|
50
|
+
const idIn = { _id: { in: ids } };
|
|
51
|
+
const matching = coll.findManyCore({
|
|
52
|
+
where: { AND: [opts.where, idIn] }
|
|
53
|
+
});
|
|
54
|
+
allowed = new Set(matching.map((d) => d._id));
|
|
55
|
+
}
|
|
56
|
+
const out = [];
|
|
57
|
+
for (const r of rows) {
|
|
58
|
+
if (allowed && !allowed.has(r.doc_id)) continue;
|
|
59
|
+
const doc = coll.getRaw(r.doc_id);
|
|
60
|
+
if (doc) out.push({ ...doc, _score: -r.rank });
|
|
61
|
+
}
|
|
62
|
+
return Promise.resolve(out);
|
|
63
|
+
}
|
|
64
|
+
function fts(spec) {
|
|
65
|
+
let database;
|
|
66
|
+
return {
|
|
67
|
+
name: "fts",
|
|
68
|
+
init(db) {
|
|
69
|
+
database = db;
|
|
70
|
+
for (const [coll, fields] of Object.entries(spec)) {
|
|
71
|
+
const cols = fields.map((_, i) => `"${col(i)}"`).join(", ");
|
|
72
|
+
db.sqlite.exec(
|
|
73
|
+
`CREATE VIRTUAL TABLE IF NOT EXISTS "${ftsTable(coll)}" USING fts5(doc_id UNINDEXED, ${cols})`
|
|
74
|
+
);
|
|
75
|
+
const count = db.sqlite.prepare(`SELECT count(*) AS n FROM "${ftsTable(coll)}"`).get();
|
|
76
|
+
if (count.n === 0) reindex(db, coll, fields);
|
|
77
|
+
}
|
|
78
|
+
},
|
|
79
|
+
afterWrite(db, { collection, ids }) {
|
|
80
|
+
const fields = spec[collection];
|
|
81
|
+
if (!fields) return;
|
|
82
|
+
for (const id of ids) indexDoc(db, collection, fields, id);
|
|
83
|
+
},
|
|
84
|
+
collectionMethods: {
|
|
85
|
+
search: (collection, query, opts) => search(database, collection, spec, query, opts)
|
|
86
|
+
}
|
|
87
|
+
};
|
|
88
|
+
}
|
|
89
|
+
|
|
90
|
+
export { fts, reindex };
|
|
91
|
+
//# sourceMappingURL=index.js.map
|
|
92
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../src/index.ts"],"names":[],"mappings":";AA4BA,IAAM,QAAA,GAAW,CAAC,IAAA,KAAiB,CAAA,EAAG,IAAI,CAAA,IAAA,CAAA;AAC1C,IAAM,GAAA,GAAM,CAAC,CAAA,KAAc,CAAA,CAAA,EAAI,CAAC,CAAA,CAAA;AAGhC,SAAS,WAAA,CAAY,KAA0B,IAAA,EAAsB;AACnE,EAAA,IAAI,GAAA,GAAW,GAAA;AACf,EAAA,KAAA,MAAW,GAAA,IAAO,IAAA,CAAK,KAAA,CAAM,GAAG,CAAA,EAAG;AACjC,IAAA,IAAI,GAAA,IAAO,MAAM,OAAO,EAAA;AACxB,IAAA,GAAA,GAAM,IAAI,GAAG,CAAA;AAAA,EACf;AACA,EAAA,IAAI,GAAA,IAAO,MAAM,OAAO,EAAA;AACxB,EAAA,IAAI,OAAO,GAAA,KAAQ,QAAA,EAAU,OAAO,GAAA;AACpC,EAAA,IAAI,KAAA,CAAM,QAAQ,GAAG,CAAA;AACnB,IAAA,OAAO,GAAA,CACJ,MAAA,CAAO,CAAC,CAAA,KAAM,CAAA,IAAK,IAAI,CAAA,CACvB,GAAA,CAAI,MAAM,CAAA,CACV,IAAA,CAAK,GAAG,CAAA;AACb,EAAA,IAAI,OAAO,QAAQ,QAAA,IAAY,OAAO,QAAQ,SAAA,EAAW,OAAO,OAAO,GAAG,CAAA;AAC1E,EAAA,OAAO,EAAA;AACT;AAEA,SAAS,QAAA,CACP,EAAA,EACA,IAAA,EACA,MAAA,EACA,EAAA,EACM;AACN,EAAA,MAAM,SAAS,EAAA,CAAG,MAAA;AAClB,EAAA,MAAA,CAAO,OAAA,CAAQ,gBAAgB,QAAA,CAAS,IAAI,CAAC,CAAA,kBAAA,CAAoB,CAAA,CAAE,IAAI,EAAE,CAAA;AACzE,EAAA,MAAM,MAAM,EAAA,CAAG,UAAA,CAAW,IAAI,CAAA,CAAE,OAAO,EAAE,CAAA;AACzC,EAAA,IAAI,CAAC,GAAA,EAAK;AACV,EAAA,MAAM,IAAA,GAAO,MAAA,CAAO,GAAA,CAAI,CAAC,CAAA,EAAG,CAAA,KAAM,CAAA,CAAA,EAAI,GAAA,CAAI,CAAC,CAAC,CAAA,CAAA,CAAG,CAAA,CAAE,KAAK,IAAI,CAAA;AAC1D,EAAA,MAAM,eAAe,MAAA,CAAO,GAAA,CAAI,MAAM,GAAG,CAAA,CAAE,KAAK,IAAI,CAAA;AACpD,EAAA,MAAM,MAAA,GAAS,OAAO,GAAA,CAAI,CAAC,MAAM,WAAA,CAAY,GAAA,EAAK,CAAC,CAAC,CAAA;AACpD,EAAA,MAAA,CACG,OAAA;AAAA,IACC,gBAAgB,QAAA,CAAS,IAAI,CAAC,CAAA,UAAA,EAAa,IAAI,gBAAgB,YAAY,CAAA,CAAA;AAAA,GAC7E,CACC,GAAA,CAAI,EAAA,EAAI,GAAG,MAAM,CAAA;AACtB;AAGO,SAAS,OAAA,CAAQ,EAAA,EAAa,IAAA,EAAc,MAAA,EAAwB;AACzE,EAAA,MAAM,SAAS,EAAA,CAAG,MAAA;AAClB,EAAA,MAAA,CAAO,IAAA,CAAK,CAAA,aAAA,EAAgB,QAAA,CAAS,IAAI,CAAC,CAAA,CAAA,CAAG,CAAA;AAC7C,EAAA,KAAA,MAAW,GAAA,IAAO,GAAG,UAAA,CAAW,IAAI,EAAE,YAAA,CAAa,EAAE,CAAA,EAAG;AACtD,IAAA,QAAA,CAAS,EAAA,EAAI,IAAA,EAAM,MAAA,EAAS,GAAA,CAAoB,GAAG,CAAA;AAAA,EACrD;AACF;AAEA,SAAS,MAAA,CACP,EAAA,EACA,IAAA,EACA,IAAA,EACA,OACA,IAAA,EAC4B;AAC5B,EAAA,MAAM,MAAA,GAAS,IAAA,CAAK,IAAA,CAAK,IAAI,CAAA;AAC7B,EAAA,IAAI,CAAC,MAAA,EAAQ;AACX,IAAA,MAAM,IAAI,KAAA;AAAA,MACR,CAAA,YAAA,EAAe,KAAK,IAAI,CAAA,wCAAA;AAAA,KAC1B;AAAA,EACF;AACA,EAAA,MAAM,KAAA,GAAQ,MAAM,KAAA,IAAS,EAAA;AAC7B,EAAA,MAAM,IAAA,GAAO,GAAG,MAAA,CACb,OAAA;AAAA,IACC,CAAA,0BAAA,EAA6B,SAAS,IAAA,CAAK,IAAI,CAAC,CAAA,SAAA,EACpC,QAAA,CAAS,IAAA,CAAK,IAAI,CAAC,CAAA,+BAAA;AAAA,GACjC,CACC,GAAA,CAAI,KAAA,EAAO,KAAK,CAAA;AAEnB,EAAA,IAAI,OAAA,GAA8B,IAAA;AAClC,EAAA,IAAI,MAAM,KAAA,EAAO;AACf,IAAA,MAAM,MAAM,IAAA,CAAK,GAAA,CAAI,CAAC,CAAA,KAAM,EAAE,MAAM,CAAA;AACpC,IAAA,MAAM,OAAO,EAAE,GAAA,EAAK,EAAE,EAAA,EAAI,KAAI,EAAE;AAChC,IAAA,MAAM,QAAA,GAAW,KAAK,YAAA,CAAa;AAAA,MACjC,OAAO,EAAE,GAAA,EAAK,CAAC,IAAA,CAAK,KAAA,EAAO,IAAI,CAAA;AAAE,KAClC,CAAA;AACD,IAAA,OAAA,GAAU,IAAI,IAAI,QAAA,CAAS,GAAA,CAAI,CAAC,CAAA,KAAM,CAAA,CAAE,GAAG,CAAC,CAAA;AAAA,EAC9C;AAEA,EAAA,MAAM,MAAyB,EAAC;AAChC,EAAA,KAAA,MAAW,KAAK,IAAA,EAAM;AACpB,IAAA,IAAI,WAAW,CAAC,OAAA,CAAQ,GAAA,CAAI,CAAA,CAAE,MAAM,CAAA,EAAG;AACvC,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,MAAA,CAAO,CAAA,CAAE,MAAM,CAAA;AAChC,IAAA,IAAI,GAAA,EAAK,GAAA,CAAI,IAAA,CAAK,EAAE,GAAG,KAAK,MAAA,EAAQ,CAAC,CAAA,CAAE,IAAA,EAAyB,CAAA;AAAA,EAClE;AACA,EAAA,OAAO,OAAA,CAAQ,QAAQ,GAAG,CAAA;AAC5B;AAYO,SAAS,IAAI,IAAA,EAA8B;AAChD,EAAA,IAAI,QAAA;AACJ,EAAA,OAAO;AAAA,IACL,IAAA,EAAM,KAAA;AAAA,IACN,KAAK,EAAA,EAAI;AACP,MAAA,QAAA,GAAW,EAAA;AACX,MAAA,KAAA,MAAW,CAAC,IAAA,EAAM,MAAM,KAAK,MAAA,CAAO,OAAA,CAAQ,IAAI,CAAA,EAAG;AACjD,QAAA,MAAM,IAAA,GAAO,MAAA,CAAO,GAAA,CAAI,CAAC,CAAA,EAAG,CAAA,KAAM,CAAA,CAAA,EAAI,GAAA,CAAI,CAAC,CAAC,CAAA,CAAA,CAAG,CAAA,CAAE,KAAK,IAAI,CAAA;AAC1D,QAAA,EAAA,CAAG,MAAA,CAAO,IAAA;AAAA,UACR,CAAA,oCAAA,EAAuC,QAAA,CAAS,IAAI,CAAC,kCACnB,IAAI,CAAA,CAAA;AAAA,SACxC;AAEA,QAAA,MAAM,KAAA,GAAQ,EAAA,CAAG,MAAA,CACd,OAAA,CAAQ,CAAA,2BAAA,EAA8B,SAAS,IAAI,CAAC,CAAA,CAAA,CAAG,CAAA,CACvD,GAAA,EAAI;AACP,QAAA,IAAI,MAAM,CAAA,KAAM,CAAA,EAAG,OAAA,CAAQ,EAAA,EAAI,MAAM,MAAM,CAAA;AAAA,MAC7C;AAAA,IACF,CAAA;AAAA,IACA,UAAA,CAAW,EAAA,EAAI,EAAE,UAAA,EAAY,KAAI,EAAG;AAClC,MAAA,MAAM,MAAA,GAAS,KAAK,UAAU,CAAA;AAC9B,MAAA,IAAI,CAAC,MAAA,EAAQ;AACb,MAAA,KAAA,MAAW,MAAM,GAAA,EAAK,QAAA,CAAS,EAAA,EAAI,UAAA,EAAY,QAAQ,EAAE,CAAA;AAAA,IAC3D,CAAA;AAAA,IACA,iBAAA,EAAmB;AAAA,MACjB,MAAA,EAAQ,CAAC,UAAA,EAAY,KAAA,EAAe,IAAA,KAClC,OAAO,QAAA,EAAU,UAAA,EAAY,IAAA,EAAM,KAAA,EAAO,IAAI;AAAA;AAClD,GACF;AACF","file":"index.js","sourcesContent":["import type {\n Collection,\n Doc,\n Monlite,\n MonlitePlugin,\n WhereInput,\n WithId,\n} from \"@monlite/core\";\n\n/** Map of collection name → searchable field paths (dot-notation allowed). */\nexport type FtsSpec = Record<string, string[]>;\n\nexport interface SearchOptions<T = Doc> {\n /** Max results (default 50). */\n limit?: number;\n /** Additionally constrain matches with a normal monlite where clause. */\n where?: WhereInput<T>;\n}\n\nexport type SearchResult<T = Doc> = WithId<T> & { _score: number };\n\n// Make `collection.search()` typed wherever @monlite/fts is imported.\ndeclare module \"@monlite/core\" {\n interface Collection<T> {\n search(query: string, opts?: SearchOptions<T>): Promise<SearchResult<T>[]>;\n }\n}\n\nconst ftsTable = (coll: string) => `${coll}_fts`;\nconst col = (i: number) => `f${i}`;\n\n/** Extract searchable text for a field path from a document. */\nfunction extractText(doc: Record<string, any>, path: string): string {\n let cur: any = doc;\n for (const seg of path.split(\".\")) {\n if (cur == null) return \"\";\n cur = cur[seg];\n }\n if (cur == null) return \"\";\n if (typeof cur === \"string\") return cur;\n if (Array.isArray(cur))\n return cur\n .filter((x) => x != null)\n .map(String)\n .join(\" \");\n if (typeof cur === \"number\" || typeof cur === \"boolean\") return String(cur);\n return \"\";\n}\n\nfunction indexDoc(\n db: Monlite,\n coll: string,\n fields: string[],\n id: string,\n): void {\n const sqlite = db.sqlite;\n sqlite.prepare(`DELETE FROM \"${ftsTable(coll)}\" WHERE doc_id = ?`).run(id);\n const doc = db.collection(coll).getRaw(id);\n if (!doc) return; // deleted\n const cols = fields.map((_, i) => `\"${col(i)}\"`).join(\", \");\n const placeholders = fields.map(() => \"?\").join(\", \");\n const values = fields.map((f) => extractText(doc, f));\n sqlite\n .prepare(\n `INSERT INTO \"${ftsTable(coll)}\"(doc_id, ${cols}) VALUES (?, ${placeholders})`,\n )\n .run(id, ...values);\n}\n\n/** Rebuild a collection's FTS index from scratch. */\nexport function reindex(db: Monlite, coll: string, fields: string[]): void {\n const sqlite = db.sqlite;\n sqlite.exec(`DELETE FROM \"${ftsTable(coll)}\"`);\n for (const doc of db.collection(coll).findManyCore({})) {\n indexDoc(db, coll, fields, (doc as WithId<Doc>)._id);\n }\n}\n\nfunction search<T = Doc>(\n db: Monlite,\n coll: Collection<T>,\n spec: FtsSpec,\n query: string,\n opts?: SearchOptions<T>,\n): Promise<SearchResult<T>[]> {\n const fields = spec[coll.name];\n if (!fields) {\n throw new Error(\n `Collection \"${coll.name}\" is not configured for full-text search`,\n );\n }\n const limit = opts?.limit ?? 50;\n const rows = db.sqlite\n .prepare(\n `SELECT doc_id, rank FROM \"${ftsTable(coll.name)}\" ` +\n `WHERE \"${ftsTable(coll.name)}\" MATCH ? ORDER BY rank LIMIT ?`,\n )\n .all(query, limit) as Array<{ doc_id: string; rank: number }>;\n\n let allowed: Set<string> | null = null;\n if (opts?.where) {\n const ids = rows.map((r) => r.doc_id);\n const idIn = { _id: { in: ids } } as WhereInput<T>;\n const matching = coll.findManyCore({\n where: { AND: [opts.where, idIn] } as WhereInput<T>,\n });\n allowed = new Set(matching.map((d) => d._id));\n }\n\n const out: SearchResult<T>[] = [];\n for (const r of rows) {\n if (allowed && !allowed.has(r.doc_id)) continue;\n const doc = coll.getRaw(r.doc_id);\n if (doc) out.push({ ...doc, _score: -r.rank } as SearchResult<T>);\n }\n return Promise.resolve(out);\n}\n\n/**\n * Full-text search plugin (SQLite FTS5). Pass it to `createDb({ plugins: [...] })`\n * with a map of collection → searchable fields. Adds `collection.search()`, keeps\n * the index in sync on every write, and backfills existing documents on open.\n *\n * ```ts\n * const db = createDb(\"./app.db\", { plugins: [fts({ posts: [\"title\", \"body\"] })] });\n * await db.collection(\"posts\").search(\"hello world\");\n * ```\n */\nexport function fts(spec: FtsSpec): MonlitePlugin {\n let database: Monlite;\n return {\n name: \"fts\",\n init(db) {\n database = db;\n for (const [coll, fields] of Object.entries(spec)) {\n const cols = fields.map((_, i) => `\"${col(i)}\"`).join(\", \");\n db.sqlite.exec(\n `CREATE VIRTUAL TABLE IF NOT EXISTS \"${ftsTable(coll)}\" ` +\n `USING fts5(doc_id UNINDEXED, ${cols})`,\n );\n // Backfill when the index is empty (e.g. enabling FTS on an existing db).\n const count = db.sqlite\n .prepare(`SELECT count(*) AS n FROM \"${ftsTable(coll)}\"`)\n .get() as { n: number };\n if (count.n === 0) reindex(db, coll, fields);\n }\n },\n afterWrite(db, { collection, ids }) {\n const fields = spec[collection];\n if (!fields) return;\n for (const id of ids) indexDoc(db, collection, fields, id);\n },\n collectionMethods: {\n search: (collection, query: string, opts?: SearchOptions) =>\n search(database, collection, spec, query, opts),\n },\n };\n}\n"]}
|
package/package.json
ADDED
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
{
|
|
2
|
+
"name": "@monlite/fts",
|
|
3
|
+
"version": "0.1.0",
|
|
4
|
+
"description": "Full-text search for @monlite/core, powered by SQLite FTS5. Adds collection.search().",
|
|
5
|
+
"type": "module",
|
|
6
|
+
"main": "./dist/index.cjs",
|
|
7
|
+
"module": "./dist/index.js",
|
|
8
|
+
"types": "./dist/index.d.ts",
|
|
9
|
+
"exports": {
|
|
10
|
+
".": {
|
|
11
|
+
"import": {
|
|
12
|
+
"types": "./dist/index.d.ts",
|
|
13
|
+
"default": "./dist/index.js"
|
|
14
|
+
},
|
|
15
|
+
"require": {
|
|
16
|
+
"types": "./dist/index.d.cts",
|
|
17
|
+
"default": "./dist/index.cjs"
|
|
18
|
+
}
|
|
19
|
+
}
|
|
20
|
+
},
|
|
21
|
+
"files": [
|
|
22
|
+
"dist"
|
|
23
|
+
],
|
|
24
|
+
"sideEffects": false,
|
|
25
|
+
"keywords": [
|
|
26
|
+
"monlite",
|
|
27
|
+
"fts",
|
|
28
|
+
"fts5",
|
|
29
|
+
"full-text-search",
|
|
30
|
+
"sqlite",
|
|
31
|
+
"search"
|
|
32
|
+
],
|
|
33
|
+
"license": "MIT",
|
|
34
|
+
"author": "Emad Jumaah <emadjumaah@gmail.com>",
|
|
35
|
+
"repository": {
|
|
36
|
+
"type": "git",
|
|
37
|
+
"url": "git+https://github.com/qataruts/monlite.git",
|
|
38
|
+
"directory": "packages/fts"
|
|
39
|
+
},
|
|
40
|
+
"homepage": "https://github.com/qataruts/monlite/tree/main/packages/fts#readme",
|
|
41
|
+
"bugs": {
|
|
42
|
+
"url": "https://github.com/qataruts/monlite/issues"
|
|
43
|
+
},
|
|
44
|
+
"publishConfig": {
|
|
45
|
+
"access": "public"
|
|
46
|
+
},
|
|
47
|
+
"engines": {
|
|
48
|
+
"node": ">=18"
|
|
49
|
+
},
|
|
50
|
+
"dependencies": {
|
|
51
|
+
"@monlite/core": "^0.9.0"
|
|
52
|
+
},
|
|
53
|
+
"devDependencies": {
|
|
54
|
+
"@types/node": "^22.10.0",
|
|
55
|
+
"better-sqlite3": "^11.8.0",
|
|
56
|
+
"tsup": "^8.3.5",
|
|
57
|
+
"typescript": "^5.7.2",
|
|
58
|
+
"vitest": "^2.1.8"
|
|
59
|
+
},
|
|
60
|
+
"scripts": {
|
|
61
|
+
"build": "tsup",
|
|
62
|
+
"test": "vitest run",
|
|
63
|
+
"typecheck": "tsc --noEmit"
|
|
64
|
+
}
|
|
65
|
+
}
|