Add HeXi bot codebase: custom plugins, web frontends, tests

- hexi core: message handling, rate limiting, cooldown, plugin manager - Custom plugins: BF stats, daily check-in, quotes, persona cards, etc. - Community plugins vendored under hexi/plugins with local fixes - Web admin frontends (learning-chat, persona-admin), unified hexi/web - Tests for rate_limit/cooldown/memes/persona; poetry.lock Co-Authored-By: Claude <noreply@anthropic.com>
2026-09-01 13:13:40 +08:00
parent 1783c60afa
commit b61d09f09f
3201 changed files with 160436 additions and 171 deletions
@@ -0,0 +1,66 @@
+import os
+import random
+from nonebot import logger
+from .equipment_renderer import render_equipment_card
+from .equipment_store import list_stratagems
+
+
+basic_path = os.path.dirname(__file__)
+save_path = os.path.join(basic_path, "temp")
+img_path = os.path.join(basic_path, "img")
+data_path = os.path.join(basic_path, "data")
+
+
+async def get_random_equipment(count):
+    data = list_stratagems()
+    indices = select_random_equipment(len(data), count)
+    selected_equipment = [data[i] for i in indices]
+
+    return await render_equipment_card(selected_equipment)
+
+
+async def get_equipment_by_combination(type_combination):
+    data = list_stratagems()
+
+    equipment_by_type = categorize_equipment_by_type(data)
+    selected_equipment = select_equipment_by_type(equipment_by_type, type_combination)
+
+    logger.debug(f"根据组合选择的装备: {selected_equipment}")
+    return await render_equipment_card(selected_equipment)
+
+
+def categorize_equipment_by_type(data):
+    equipment_by_type = {}
+    for item in data:
+        equip_type = item['type']
+        if equip_type not in equipment_by_type:
+            equipment_by_type[equip_type] = []
+        equipment_by_type[equip_type].append(item)
+    return equipment_by_type
+
+
+def select_equipment_by_type(equipment_by_type, type_combination):
+    selected_equipment = []
+    backpack_count = 0
+
+    for equip_type, count in type_combination.items():
+        if equip_type in equipment_by_type:
+            available_items = equipment_by_type[equip_type]
+            random.shuffle(available_items)
+            selected = 0
+            for item in available_items:
+                if selected < count:
+                    if item['backpack'] == 1 and backpack_count >= 1:
+                        continue
+                    selected_equipment.append(item)
+                    selected += 1
+                    if item['backpack'] == 1:
+                        backpack_count += 1
+                else:
+                    break
+
+    return selected_equipment
+
+
+def select_random_equipment(max_equipment, count):
+    return random.sample(range(max_equipment), count)