feat: get sentence from plugin

AwesomeHamster · Dec 11, 2023 · ec6316f · ec6316f
1 parent 98ab560
commit ec6316f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/config.ts b/src/config.ts
@@ -1,6 +1,7 @@
 import { Schema } from 'koishi'
 
 export interface Config {
+  sentences?: boolean
   /**
    * @default "https://v1.hitokoto.cn"
    */
@@ -16,6 +17,7 @@ export interface Config {
 }
 
 export const Config: Schema<Config> = Schema.object({
+  sentences: Schema.boolean().description('是否使用本地一言语料库(需要安装 sentences 服务)').default(false),
   apiUrl: Schema.string().description('获取一言的 API 地址').default('https://v1.hitokoto.cn'),
   minLength: Schema.number().description('一言的最小长度'),
   maxLength: Schema.number().description('一言的最大长度'),

diff --git a/src/index.ts b/src/index.ts
@@ -1,9 +1,14 @@
 import { Context, Quester } from 'koishi'
+import type { SentencesParams } from 'koishi-plugin-hitokoto-sentences'
 
 import { HitokotoApi } from './api'
 import { Config } from './config'
 import i18n from './i18n'
 
+export const inject = {
+  optional: ['sentences'],
+}
+
 export const name = 'hitokoto'
 
 export { Config, HitokotoApi }
@@ -41,7 +46,9 @@ export async function apply(ctx: Context, config: Config = {}): Promise<void> {
       }
 
       try {
-        const resp = await ctx.hitokoto.getHitokoto(params)
+        const resp = config.sentences && ctx.sentences
+          ? ctx.sentences.getSentence(params as SentencesParams)
+          : await ctx.hitokoto.getHitokoto(params)
         return session?.text('.format', resp)
       } catch (error) {
         const err = error as Error