Files
Hydro/packages/better-search/model.ts
T
2021-04-02 00:39:21 +08:00

18 lines
413 B
TypeScript

import nodejieba from 'nodejieba';
import * as bus from 'hydrooj/dist/service/bus';
declare module 'hydrooj' {
interface Pdoc {
search?: string
}
}
const jiebaHook = async (pdoc) => {
if (!pdoc.title) return;
const segments = nodejieba.cutForSearch(pdoc.title);
pdoc.search = segments.join(' ');
};
bus.on('problem/before-add', jiebaHook);
bus.on('problem/before-edit', jiebaHook);