kokkoro-plugin-hitokoto 3.0.0 → 3.0.1

package/README.md

@@ -10,15 +10,35 @@ bun add kokkoro-plugin-hitokoto
 
 ## 指令
 
-发送「/一言」，插件默认会从一言接口随机返回一条动画或漫画语句。
+```text
+/一言 [types]...
+```
+
+`types` 是可选的一言类型。不填写时，插件默认会从一言接口随机返回一条动画或漫画语句。
+
+```text
+/一言
+```
+
+填写类型可以限定本次返回的语句，多个类型使用空格分隔。
+
+```text
+/一言 诗词 文学
+```
+
+支持动画、漫画、游戏、文学、原创、来自网络、其他、影视、诗词、网易云、哲学和抖机灵。
 
 ## 快捷方式
 
-也可以发送「来点骚话」触发，群聊需要开启「获取群内全部消息」权限。
+发送「来点骚话」也会随机返回一条语句，群聊需要开启「获取群内全部消息」权限。
+
+```text
+来点骚话
+```
 
 ## 环境变量
 
-如需修改语句类型，请在项目根目录创建 `.env` 文件，并通过 `HITOKOTO_TYPES` 设置一言接口的 `c` 参数。多个类型使用逗号分隔，默认值为 `a,b`。
+如需修改默认语句类型，请在项目根目录创建 `.env` 文件，并通过 `HITOKOTO_TYPES` 设置一言接口的 `c` 参数。多个类型使用逗号分隔，默认值为 `a,b`。
 
 ```ini
 HITOKOTO_TYPES=c,d

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kokkoro-plugin-hitokoto",
-  "version": "3.0.0",
+  "version": "3.0.1",
   "description": "ヒトコト",
   "keywords": [
     "bot",

package/src/index.ts

@@ -43,20 +43,52 @@ interface Sentence {
 
 const { HITOKOTO_TYPES = 'a,b' } = import.meta.env;
 const HITOKOTO_API = new URL('https://v1.hitokoto.cn');
+const TYPE_CODES = {
+  动画: 'a',
+  漫画: 'b',
+  游戏: 'c',
+  文学: 'd',
+  原创: 'e',
+  来自网络: 'f',
+  其他: 'g',
+  影视: 'h',
+  诗词: 'i',
+  网易云: 'j',
+  哲学: 'k',
+  抖机灵: 'l',
+} as const satisfies Record<string, SentenceType>;
+const TYPE_NAMES = Object.keys(TYPE_CODES).join('、');
 
-for (const type of HITOKOTO_TYPES.split(',')) {
-  HITOKOTO_API.searchParams.append('c', type);
+function resolveTypes(names: string[]) {
+  if (names.length === 0) {
+    return HITOKOTO_TYPES.split(',');
+  }
+  return names.map(name => {
+    const type = TYPE_CODES[<keyof typeof TYPE_CODES>name];
+
+    if (!type) {
+      throw new Error(`不支持的一言类型「${name}」，可用类型为${TYPE_NAMES}`);
+    }
+    return type;
+  });
 }
 
-export default () => {
-  useCommand('/一言', async () => {
-    const response = await fetch(HITOKOTO_API);
+async function getHitokoto(names: string[]) {
+  const url = new URL(HITOKOTO_API);
 
-    if (!response.ok) {
-      throw new Error(`接口请求失败，状态码 ${response.status}`);
-    }
-    const { hitokoto, from } = <Sentence>await response.json();
+  for (const type of resolveTypes(names)) {
+    url.searchParams.append('c', type);
+  }
+  const response = await fetch(url);
 
-    return `『${hitokoto}』——「${from}」`;
-  }).shortcut('来点骚话');
+  if (!response.ok) {
+    throw new Error(`接口请求失败，状态码 ${response.status}`);
+  }
+  const { hitokoto, from } = <Sentence>await response.json();
+
+  return `『${hitokoto}』——「${from}」`;
+}
+
+export default () => {
+  useCommand('/一言 [types]...', context => getHitokoto(context.args.types)).shortcut('来点骚话');
 };