new

konabot/plugins/hanzi/__init__.py

@@ -0,0 +1,217 @@
+import random
+from typing import Optional
+import opencc
+
+from nonebot import on_message
+from nonebot.adapters import Event as BaseEvent
+from nonebot.adapters.discord.event import MessageEvent as DiscordMessageEvent
+from nonebot_plugin_alconna import (
+    Alconna,
+    Args,
+    UniMessage,
+    UniMsg,
+    on_alconna,
+)
+
+convert_type = ["简","簡","繁","正","港","日"]
+
+compiled_str = "|".join([f"{a}{mid}{b}" for mid in ["转","轉","転"] for a in convert_type for b in convert_type if a != b])
+
+def hanzi_to_abbr(hanzi: str) -> str:
+    mapping = {
+        "简": "s",
+        "簡": "s",
+        "繁": "t",
+        "正": "t",
+        "港": "hk",
+        "日": "jp",
+    }
+    return mapping.get(hanzi, "")
+
+def check_valid_convert_type(convert_type: str) -> bool:
+    avaliable_set = ["s2t","t2s","s2tw","tw2s","s2hk","hk2s","s2twp","tw2sp","t2tw","hk2t","t2hk","t2jp","jp2t","tw2t"]
+    if convert_type in avaliable_set:
+        return True
+    return False
+
+def convert(source, src_abbr, dst_abbr):
+    convert_type_key = f"{src_abbr}2{dst_abbr}"
+    if not check_valid_convert_type(convert_type_key):
+        # 先转为繁体，再转为目标
+        converter = opencc.OpenCC(f"{src_abbr}2t.json")
+        source = converter.convert(source)
+        src_abbr = "t"
+    converter = opencc.OpenCC(f"{src_abbr}2{dst_abbr}.json")
+    converted = converter.convert(source)
+    return converted
+
+evt = on_alconna(
+    Alconna(
+        f"re:({compiled_str})",
+        Args["source?", str],
+    ),
+    use_cmd_start=True,
+    use_cmd_sep=False,
+    skip_for_unmatch=True,
+)
+
+@evt.handle()
+async def _(msg: UniMsg, event: BaseEvent, source: Optional[str] = None):
+    if isinstance(event, DiscordMessageEvent):
+        content = event.get_message().extract_plain_text()
+    else:
+        content = event.get_message().extract_plain_text()
+
+    prefix = content.split()[0]
+    to_convert = ""
+    # 如果回复了消息，则转换回复的内容
+    if(source is None):
+        if event.reply:
+            to_convert = event.reply.message.extract_plain_text()
+            if not to_convert:
+                return
+        else:
+            return
+    else:
+        to_convert = source
+    parts = []
+    if "转" in prefix:
+        parts = prefix.split("转")
+    elif "轉" in prefix:
+        parts = prefix.split("轉")
+    elif "転" in prefix:
+        parts = prefix.split("転")
+    if len(parts) != 2:
+        notice = "转换格式错误，请使用“简转繁”、“繁转简”等格式。"
+        await evt.send(await UniMessage().text(notice).export())
+        return
+    src, dst = parts
+    src_abbr = hanzi_to_abbr(src)
+    dst_abbr = hanzi_to_abbr(dst)
+    if not src_abbr or not dst_abbr:
+        notice = "不支持的转换类型，请使用“简”、“繁”、“正”、“港”、“日”等。"
+        if src_abbr:
+            notice = convert(notice, "s", src_abbr)
+        await evt.send(await UniMessage().text(notice).export())
+        return
+    
+    converted = convert(to_convert, src_abbr, dst_abbr)
+
+    converted_prefix = convert("转换结果", "s", dst_abbr)
+
+    await evt.send(await UniMessage().text(f"{converted_prefix}：{converted}").export())
+
+shuo = ["说","說"]
+
+full_name_type = ["简体","簡體","繁體","繁体","正體","正体","港話","港话","日文"]
+
+combined_list = [f"{a}{b}" for a in shuo for b in full_name_type]
+
+compiled_str_2 = "|".join(combined_list)
+
+evt = on_alconna(
+    Alconna(
+        f"re:({compiled_str_2})",
+        Args["source?", str]
+    ),
+    use_cmd_start=True,
+    use_cmd_sep=False,
+    skip_for_unmatch=True,
+)
+
+@evt.handle()
+async def _(msg: UniMsg, event: BaseEvent, source: Optional[str] = None):
+    if isinstance(event, DiscordMessageEvent):
+        content = event.get_message().extract_plain_text()
+    else:
+        content = event.get_message().extract_plain_text()
+
+    prefix = content.split()[0]
+    to_convert = ""
+    # 如果回复了消息，则转换回复的内容
+    if(source is None):
+        if event.reply:
+            to_convert = event.reply.message.extract_plain_text()
+            if not to_convert:
+                return
+        else:
+            return
+    else:
+        to_convert = source
+    # 获取目标转换类型
+    dst = ""
+    match prefix:
+        case "说简体" | "說簡體" | "说簡體" | "說简体":
+            dst = "简"
+        case "說繁體" | "说繁体" | "說繁体" | "说繁體":
+            dst = "繁"
+        case "說正體" | "说正体" | "說正体" | "说正體":
+            dst = "正"
+        case "說港話" | "说港话" | "說港话" | "说港話":
+            dst = "港"
+        case "說日文" | "说日文":
+            dst = "日"
+    dst_abbr = hanzi_to_abbr(dst)
+    if not dst_abbr:
+        notice = "不支持的转换类型，请使用“简体”、“繁體”、“正體”、“港話”、“日文”等。"
+        await evt.send(await UniMessage().text(notice).export())
+        return
+    # 循环，将源语言一次次转换为目标语言
+    current_text = to_convert
+    for src_abbr in ["s","hk","jp","tw","t"]:
+        if src_abbr != dst_abbr:
+            current_text = convert(current_text, src_abbr, dst_abbr)
+
+    converted_prefix = convert("转换结果", "s", dst_abbr)
+
+    await evt.send(await UniMessage().text(f"{converted_prefix}：{current_text}").export())
+
+def random_char(char: str) -> str:
+    dst_abbr = random.choice(["s","t","hk","jp","tw"])
+    for src_abbr in ["s","hk","jp","tw","t"]:
+        if src_abbr != dst_abbr:
+            char = convert(char, src_abbr, dst_abbr)
+    return char
+
+def random_string(text: str) -> str:
+    final_text = ""
+    for char in text:
+        final_text += random_char(char)
+    return final_text
+
+random_match = ["混乱字形","混亂字形","乱数字形","亂數字形","ランダム字形"]
+
+evt = on_alconna(
+    Alconna(
+        f"re:({'|'.join(random_match)})",
+        Args["source?", str]
+    ),
+    use_cmd_start=True,
+    use_cmd_sep=False,
+    skip_for_unmatch=True,
+)
+@evt.handle()
+async def _(msg: UniMsg, event: BaseEvent, source: Optional[str] = None):
+    if isinstance(event, DiscordMessageEvent):
+        content = event.get_message().extract_plain_text()
+    else:
+        content = event.get_message().extract_plain_text()
+
+    prefix = content.split()[0]
+    to_convert = ""
+    # 如果回复了消息，则转换回复的内容
+    if(source is None):
+        if event.reply:
+            to_convert = event.reply.message.extract_plain_text()
+            if not to_convert:
+                return
+        else:
+            return
+    else:
+        to_convert = source
+
+    final_text = ""
+    final_text = random_string(to_convert)
+    converted_prefix = convert(random_string("转换结果"), "s", "s")
+
+    await evt.send(await UniMessage().text(f"{converted_prefix}：{final_text}").export())

konabot/plugins/idiomgame/__init__.py

@@ -71,8 +71,8 @@ class TryVerifyState(Enum):
     VERIFIED_AND_REAL = 1
     NOT_IDIOM = 2
     WRONG_FIRST_CHAR = 3
-    VERIFIED_BUT_NO_NEXT = 4
-    VERIFIED_GAME_END = 5
+    BUT_NO_NEXT = 4
+    GAME_END = 5
 
 
 class IdiomGame:
@@ -185,29 +185,33 @@ class IdiomGame:
         """
         return last_char in IdiomGame.AVALIABLE_IDIOM_FIRST_CHAR
 
-    def _verify_idiom(self, idiom: str, user_id: str) -> TryVerifyState:
+    def _verify_idiom(self, idiom: str, user_id: str) -> list[TryVerifyState]:
+        state = []
         # 新成语的首字应与上一条成语的尾字相同
         if idiom[0] != self.last_char:
-            return TryVerifyState.WRONG_FIRST_CHAR
+            state.append(TryVerifyState.WRONG_FIRST_CHAR)
+            return state
         if idiom not in IdiomGame.ALL_IDIOMS and idiom not in IdiomGame.ALL_WORDS:
             self.add_score(user_id, -0.1)
-            return TryVerifyState.NOT_IDIOM
+            state.append(TryVerifyState.NOT_IDIOM)
+            return state
+        # 成语合法，更新状态
+        state.append(TryVerifyState.VERIFIED)
         self.last_idiom = idiom
         self.last_char = idiom[-1]
         self.add_score(user_id, 1)
         if idiom in IdiomGame.ALL_IDIOMS:
+            state.append(TryVerifyState.VERIFIED_AND_REAL)
             self.add_score(user_id, 4) # 再加 4 分
         self.remain_rounds -= 1
         if self.remain_rounds <= 0:
             self.now_playing = False
-            return TryVerifyState.VERIFIED_GAME_END
+            state.append(TryVerifyState.GAME_END)
         if not self.is_nextable(self.last_char):
             # 没有成语可以接了，自动跳过
             self._skip_idiom_async()
-            return TryVerifyState.VERIFIED_BUT_NO_NEXT
-        if idiom in IdiomGame.ALL_IDIOMS:
-            return TryVerifyState.VERIFIED_AND_REAL # 真实成语
-        return TryVerifyState.VERIFIED
+            state.append(TryVerifyState.BUT_NO_NEXT)
+        return state
 
     def get_user_score(self, user_id: str) -> float:
         if user_id not in self.score_board:
@@ -233,9 +237,9 @@ class IdiomGame:
     @classmethod
     def random_idiom_starting_with(cls, first_char: str) -> Optional[str]:
         cls.init_lexicon()
-        if first_char not in cls.IDIOM_FIRST_CHAR:
+        if first_char not in cls.AVALIABLE_IDIOM_FIRST_CHAR:
             return None
-        return secrets.choice(cls.IDIOM_FIRST_CHAR[first_char])
+        return secrets.choice(cls.AVALIABLE_IDIOM_FIRST_CHAR[first_char])
 
     @classmethod
     def init_lexicon(cls):
@@ -249,7 +253,10 @@ class IdiomGame:
 
         # 词语大表
         with open(ASSETS_PATH / "lexicon" / "ci.json", "r", encoding="utf-8") as f:
-            cls.ALL_WORDS = json.load(f)
+            jsonData = json.load(f)
+            cls.ALL_WORDS = [item["ci"] for item in jsonData]
+            logger.debug(f"Loaded {len(cls.ALL_WORDS)} words from ci.json")
+            logger.debug(f"Sample words: {cls.ALL_WORDS[:5]}")
 
         COMMON_WORDS = []
         # 读取 COMMON 词语大表
@@ -258,6 +265,8 @@ class IdiomGame:
                 word = line.strip()
                 if len(word) == 4:
                     COMMON_WORDS.append(word)
+        logger.debug(f"Loaded {len(COMMON_WORDS)} common words from common.txt")
+        logger.debug(f"Sample common words: {COMMON_WORDS[:5]}")
 
         # 读取 THUOCL 成语库
         with open(
@@ -265,7 +274,9 @@ class IdiomGame:
             "r",
             encoding="utf-8",
         ) as f:
-            THUOCL_IDIOMS = [line.split(" ")[0].strip() for line in f]
+            THUOCL_IDIOMS = [line.split(" ")[0].split("\t")[0].strip() for line in f]
+            logger.debug(f"Loaded {len(THUOCL_IDIOMS)} idioms from THUOCL_chengyu.txt")
+            logger.debug(f"Sample idioms: {THUOCL_IDIOMS[:5]}")
 
         # 读取 THUOCL 剩下的所有 txt 文件，只保留四字词
         THUOCL_WORDS = []
@@ -279,9 +290,11 @@ class IdiomGame:
                     encoding="utf-8",
                 ) as f:
                     for line in f:
-                        word = line.lstrip().split(" ")[0].strip()
+                        word = line.lstrip().split(" ")[0].split("\t")[0].strip()
                         if len(word) == 4:
                             THUOCL_WORDS.append(word)
+        logger.debug(f"Loaded {len(THUOCL_WORDS)} words from THUOCL txt files")
+        logger.debug(f"Sample words: {THUOCL_WORDS[:5]}")
 
         # 只有成语的大表
         cls.ALL_IDIOMS = [idiom["word"] for idiom in ALL_IDIOMS_INFOS] + THUOCL_IDIOMS
@@ -441,6 +454,10 @@ async def _(target: DepLongTaskTarget):
     if not instance or not instance.get_playing_state():
         return
     avaliable_idiom = IdiomGame.random_idiom_starting_with(instance.get_last_char())
+    # 发送哈哈狗图片
+    with open(ASSETS_PATH / "img" / "dog" / "haha_dog.jpg", "rb") as f:
+        img_data = f.read()
+    await evt.send(await UniMessage().image(raw=img_data).export())
     await evt.send(await UniMessage().text(f"你们太菜了，全部扣100分！明明还可以接「{avaliable_idiom}」的！").export())
     idiom = await instance.skip_idiom(-100)
     await evt.send(
@@ -472,9 +489,9 @@ async def _(event: BaseEvent, msg: UniMsg, target: DepLongTaskTarget):
     user_idiom = msg.extract_plain_text().strip()
     user_id, user_name = get_user_info(event)
     state = await instance.try_verify_idiom(user_idiom, user_id)
-    if state == TryVerifyState.WRONG_FIRST_CHAR:
+    if TryVerifyState.WRONG_FIRST_CHAR in state:
         return
-    if state == TryVerifyState.NOT_IDIOM:
+    if TryVerifyState.NOT_IDIOM in state:
         await evt.send(
             await UniMessage()
             .at(user_id)
@@ -482,25 +499,25 @@ async def _(event: BaseEvent, msg: UniMsg, target: DepLongTaskTarget):
             .export()
         )
         return
-    if state == TryVerifyState.VERIFIED:
-        await evt.send(
-            await UniMessage()
-            .at(user_id)
-            .text(f" 接上了，喜提 1 分！你有 {instance.get_user_score(user_id)} 分！")
-            .export()
-        )
-    elif state == TryVerifyState.VERIFIED_AND_REAL:
+    if TryVerifyState.VERIFIED_AND_REAL in state:
         await evt.send(
             await UniMessage()
             .at(user_id)
             .text(f" 接上了，这是个真实成语，喜提 5 分！你有 {instance.get_user_score(user_id)} 分！")
             .export()
         )
-    if state == TryVerifyState.VERIFIED_GAME_END:
+    elif TryVerifyState.VERIFIED in state:
+        await evt.send(
+            await UniMessage()
+            .at(user_id)
+            .text(f" 接上了，喜提 1 分！你有 {instance.get_user_score(user_id)} 分！")
+            .export()
+        )
+    if TryVerifyState.GAME_END in state:
         await evt.send(await UniMessage().text("全部回合结束！").export())
         await end_game(event, group_id)
         return
-    if state == TryVerifyState.VERIFIED_BUT_NO_NEXT:
+    if TryVerifyState.BUT_NO_NEXT in state:
         await evt.send(
             await UniMessage()
             .text("但是，这是条死路！你们全部都要扣 100 分！")

konabot/plugins/xibao_generate/__init__.py

@@ -0,0 +1,80 @@
+from typing import Optional
+from nonebot_plugin_alconna import Alconna, Args, UniMessage, UniMsg, on_alconna
+from konabot.common.web_render import WebRenderer
+from nonebot.adapters import Event as BaseEvent
+from playwright.async_api import Page
+
+async def continue_handle(page: Page, content: str) -> None:
+    # 这里可以添加一些预处理逻辑
+    # 找到 id 为 input 的 textarea 元素
+    textarea = await page.query_selector("#input")
+    if textarea:
+        # 在 textarea 中输入内容
+        await textarea.fill(content)
+    # 找到 id 为 submit-btn 的按钮元素
+    submit_button = await page.query_selector("#submit-btn")
+    if submit_button:
+        # 点击按钮提交
+        await submit_button.click()
+
+evt = on_alconna(
+    Alconna(
+        f"生成喜报",
+        Args["content?", str]
+    ),
+    use_cmd_start=True,
+    use_cmd_sep=False,
+    skip_for_unmatch=True,
+)
+@evt.handle()
+async def _(msg: UniMsg, event: BaseEvent, content: Optional[str] = ""):
+
+    screenshot = await WebRenderer.render(
+        url="https://witnessbot.mxowl.com/services/congratulations/",
+        target="#main-canvas",
+        other_function=lambda page: continue_handle(page, content),
+        timeout=30
+    )
+    await evt.send(
+        await UniMessage().image(raw=screenshot).export()
+    )
+
+async def beibao_continue_handle(page: Page, content: str) -> None:
+    # 这里可以添加一些预处理逻辑
+    # 找到 id 为 input 的 textarea 元素
+    textarea = await page.query_selector("#input")
+    if textarea:
+        # 在 textarea 中输入内容
+        await textarea.fill(content)
+    # 找到 class 为 btn btn-outline-primary，for属性为 mode-2 的标签元素
+    mode_radio = await page.query_selector("label.btn.btn-outline-primary[for='mode-2']")
+    if mode_radio:
+        # 点击选择悲报模式
+        await mode_radio.click()
+    # 找到 id 为 submit-btn 的按钮元素
+    submit_button = await page.query_selector("#submit-btn")
+    if submit_button:
+        # 点击按钮提交
+        await submit_button.click()
+
+evt = on_alconna(
+    Alconna(
+        f"生成悲报",
+        Args["content?", str]
+    ),
+    use_cmd_start=True,
+    use_cmd_sep=False,
+    skip_for_unmatch=True,
+)
+@evt.handle()
+async def _(msg: UniMsg, event: BaseEvent, content: Optional[str] = ""):
+
+    screenshot = await WebRenderer.render(
+        url="https://witnessbot.mxowl.com/services/congratulations/",
+        target="#main-canvas",
+        other_function=lambda page: beibao_continue_handle(page, content),
+        timeout=30
+    )
+    await evt.send(
+        await UniMessage().image(raw=screenshot).export()
+    )