优化中文搜索

packages/better-search/model.ts

@@ -0,0 +1,23 @@
+import nodejieba from 'nodejieba';
+import { Pdoc } from 'hydrooj';
+import * as document from 'hydrooj/dist/model/document';
+import * as bus from 'hydrooj/dist/service/bus';
+
+declare module 'hydrooj' {
+    interface Pdoc {
+        search?: string
+    }
+}
+
+bus.on('document/add', async (doc) => {
+    if (doc.type !== document.TYPE_PROBLEM) return;
+    const pdoc = doc as Pdoc;
+    const segments = nodejieba.cut(pdoc.title);
+    pdoc.search = segments.join(' ');
+});
+
+bus.on('document/set', async (domainId, docType, docId, $set) => {
+    if (docType !== document.TYPE_PROBLEM || !$set.title) return;
+    const segments = nodejieba.cut($set.title);
+    $set.search = segments.join(' ');
+});