1 files changed, 139 insertions, 0 deletions
diff --git a/ATRI/plugins/wife/__init__.py b/ATRI/plugins/wife/__init__.py
new file mode 100644
index 0000000..a113b8a
--- /dev/null
+++ b/ATRI/plugins/wife/__init__.py
@@ -0,0 +1,139 @@
+import asyncio
+from random import choice
+from pydantic import BaseModel
+
+from nonebot.permission import USER
+from nonebot.rule import Rule
+from nonebot.permission import SUPERUSER
+from nonebot.adapters.cqhttp import Bot, MessageEvent, GroupMessageEvent, Message
+
+from ATRI.utils.limit import FreqLimiter
+from .data_source import Wife
+
+
+_tietie_flmt = FreqLimiter(600)
+
+
+tietie_superuser = Wife().on_message("只与维护者贴贴w", rule=Rule(), permission=SUPERUSER, block=False)
+
+@tietie_superuser.handle()
+async def _tietie_superuser(bot: Bot, event: MessageEvent):
+    user_id = event.get_user_id()
+    if not _tietie_flmt.check(user_id):
+        await tietie_superuser.finish()
+    
+    result = Wife().to_superuser(user_id)
+    _tietie_flmt.start_cd(user_id)
+    await tietie_superuser.finish(Message(result))
+
+
+_wife_flmt = FreqLimiter(10)
+
+class MarryInfo(BaseModel):
+    name: str
+    sex: str
+    wife: str
+
+
+get_wife = Wife().on_command("抽老婆", "随机选择一位幸运裙友成为老婆！")
+
+@get_wife.handle()
+async def _get_wife(bot: Bot, event: GroupMessageEvent):
+    user_id = event.get_user_id()
+    if not _wife_flmt.check(user_id):
+        await get_wife.finish()
+    
+    group_id = event.group_id
+    req_user_info: dict = await bot.get_group_member_info(group_id=group_id, user_id=int(user_id))
+    req_user_card = req_user_info["card"]
+    req_user_sex = req_user_info["sex"]
+    is_nick = "老公" if req_user_sex == "male" else "老婆"
+    
+    repo_0 = (
+        "现在咱将随机抽取一位幸运裙友\n"
+        f"成为{req_user_card}的{is_nick}！"
+    )
+    await bot.send(event, repo_0)
+    await asyncio.sleep(10)
+    
+    prep_list = await bot.get_group_member_list(group_id=group_id)
+    prep_list = [prep.get("user_id", 114514) for prep in prep_list]
+    
+    lucky_user = choice(prep_list)
+    lucky_user_info: dict = await bot.get_group_member_info(group_id=group_id, user_id=lucky_user)
+    lucky_user_card = lucky_user_info["card"]
+    lucky_user_sex = lucky_user_info["sex"]
+    
+    data = Wife().load_marry_list()
+    data[lucky_user] = MarryInfo(
+        name=req_user_card,
+        sex=is_nick,
+        wife=user_id
+    ).dict()
+    data[user_id] = MarryInfo(
+        name=lucky_user_card,
+        sex=lucky_user_sex,
+        wife=lucky_user
+    ).dict()
+    Wife().save_marry_list(data)
+    
+    repo_1 = f"好欸！{lucky_user_card}成为了{req_user_card}的{is_nick}"
+    _wife_flmt.start_cd(user_id)
+    await get_wife.finish(repo_1)
+
+
+_call_wife_flmt = FreqLimiter(60)
+
+
+call_wife = Wife().on_command("老婆", "呼唤老婆/老公！", aliases={"老公", "老婆！", "老公！"}, permission=USER("114514"))
+
+@call_wife.handle()
+async def _call_wife(bot: Bot, event: MessageEvent):
+    user_id = event.get_user_id()
+    if not _wife_flmt.check(user_id):
+        await call_wife.finish()
+    
+    data = Wife().load_marry_list()
+    if user_id not in data:
+        return
+    
+    wife = data[user_id].get("wife", "ignore")
+    sex = data[user_id].get("sex", "male")
+    is_nick = "老公" if sex == "male" else "老婆"
+    repo = f"你已经有{is_nick}啦！是{wife}噢~"
+    _call_wife_flmt.start_cd(user_id)
+    await call_wife.finish(repo)
+
+
+discard_wife = Wife().on_command("我要离婚", "离婚！")
+
+@discard_wife.handle()
+async def _discard_wife(bot: Bot, event: GroupMessageEvent):
+    user_id = event.get_user_id()
+    if not _wife_flmt.check(user_id):
+        await discard_wife.finish()
+    
+    await bot.send(event, "真的吗...（y/是）")
+    msg = str(event.message).strip()
+    rd_list = ["y", "Y", "是", "确认", "对"]
+    if msg not in rd_list:
+        await discard_wife.finish("")
+    
+    group_id = event.group_id
+    group_info: dict = await bot.get_group_member_info(group_id=group_id, user_id=int(user_id))
+    user_card = group_info.get("card", "老婆")
+    
+    data = Wife().load_marry_list()
+    if user_id not in data:
+        await discard_wife.finish("你还没对象呐...")
+    
+    discard_user_info = data[user_id]
+    discard_user_card = discard_user_info["name"]
+    discard_user_id = discard_user_info["wife"]
+    
+    data.pop(user_id)
+    data.pop(discard_user_id)
+    Wife().save_marry_list(data)
+    repo = f"悲，{user_card}抛弃了{discard_user_card}..."
+    _wife_flmt.start_cd(user_id)
+    await discard_wife.finish(repo)