feat(plugin-nlp): cmd.intend() should return this

packages/plugin-nlp/src/index.ts

@@ -1,11 +1,11 @@
-import { Command, Meta, Context, ParsedLine, ParsedCommandLine } from 'koishi-core'
+import { Command, Meta, Context, ParsedLine, ParsedCommandLine, App } from 'koishi-core'
 import { tag, load } from 'nodejieba'
 import { resolve } from 'path'
 
 declare module 'koishi-core/dist/command' {
   interface Command {
-    intend (keyword: string, callback: IntenderCallback): void
-    intend (keywords: string[], callback: IntenderCallback): void
+    intend (keyword: string, callback: IntenderCallback): this
+    intend (keywords: string[], callback: IntenderCallback): this
   }
 }
 
@@ -48,6 +48,7 @@ const intenders: Intender[] = []
 Command.prototype.intend = function (this: Command, arg0: string | string[], callback: IntenderCallback) {
   const keywords = typeof arg0 === 'string' ? [arg0] : arg0
   intenders.push({ command: this, keywords, callback })
+  return this
 }
 
 export const name = 'nlp'

packages/plugin-nlp/tests/index.spec.ts

@@ -0,0 +1,42 @@
+import { App } from 'koishi-test-utils'
+import { resolve } from 'path'
+import * as nlp from '../src'
+
+test('basic support', async () => {
+  const app = new App()
+  const session = app.createSession('user', 123)
+  app.plugin(nlp)
+
+  app.command('weather <location>')
+    .intend('天气', (meta) => {
+      const tag = meta.$parsed.tags.find(({ tag }) => tag === 'ns')
+      if (tag) return { args: [tag.word] }
+    })
+    .action(({ meta }, location) => meta.$send(location))
+
+  await session.shouldHaveNoReply('今天的天气')
+  await session.shouldHaveNoReply('今天的北京')
+  await session.shouldHaveNoReply('幻想乡今天的天气')
+  await session.shouldHaveReply('北京今天的天气', '北京')
+})
+
+test('user dict', async () => {
+  const app = new App()
+  const session = app.createSession('user', 123)
+  app.plugin(nlp, {
+    userDict: resolve(__dirname, 'user.dict.utf8'),
+  })
+
+  app.command('weather <location>')
+    .intend(['天气'], (meta) => {
+      const tag = meta.$parsed.tags.find(({ tag }) => tag === 'ns')
+      if (tag) return { confidence: 0.9, args: [tag.word] }
+      return { confidence: 0.5 }
+    })
+    .action(({ meta }, location) => meta.$send(location))
+
+  await session.shouldHaveNoReply('今天的天气')
+  await session.shouldHaveNoReply('今天的北京')
+  await session.shouldHaveReply('北京今天的天气', '北京')
+  await session.shouldHaveReply('幻想乡今天的天气', '幻想乡')
+})

packages/plugin-nlp/tests/user.dict.utf8

@@ -0,0 +1 @@
+幻想乡 ns