✨ Support Get Birthday Card

plugins/genshin/birthday.py

@@ -1,18 +1,35 @@
 import re
 from datetime import datetime
+from typing import List
 
-from telegram import Update
+from genshin import Client, GenshinException
-from telegram.ext import CommandHandler, CallbackContext
+from genshin.client.routes import Route
+from genshin.utility import recognize_genshin_server
+from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
+from telegram.constants import ParseMode
+from telegram.ext import CommandHandler, CallbackContext, MessageHandler
 from telegram.ext import filters
+from telegram.helpers import create_deep_linked_url
 
 from core.baseplugin import BasePlugin
+from core.cookies import CookiesService
+from core.cookies.error import CookiesNotFoundError
 from core.plugin import Plugin, handler
+from core.user import UserService
+from core.user.error import UserNotFoundError
 from metadata.genshin import AVATAR_DATA
 from metadata.shortname import roleToId, roleToName
 from utils.bot import get_args
 from utils.decorators.error import error_callable
 from utils.decorators.restricts import restricts
+from utils.genshin import fetch_hk4e_token_by_cookie, recognize_genshin_game_biz
+from utils.helpers import get_genshin_client
 from utils.log import logger
+from utils.models.base import RegionEnum
+
+BIRTHDAY_URL = Route(
+    "https://hk4e-api.mihoyo.com/event/birthdaystar/account/post_my_draw",
+)
 
 
 def rm_starting_str(string, starting):
@@ -25,7 +42,11 @@ def rm_starting_str(string, starting):
 class BirthdayPlugin(Plugin, BasePlugin):
     """Birthday."""
 
-    def __init__(self):
+    def __init__(
+        self,
+        user_service: UserService = None,
+        cookie_service: CookiesService = None,
+    ):
         """Load Data."""
         self.birthday_list = {}
         for value in AVATAR_DATA.values():
@@ -33,6 +54,16 @@ class BirthdayPlugin(Plugin, BasePlugin):
             data = self.birthday_list.get(key, [])
             data.append(value["name"])
             self.birthday_list.update({key: data})
+        self.user_service = user_service
+        self.cookie_service = cookie_service
+
+    def get_today_birthday(self) -> List[str]:
+        key = (
+            rm_starting_str(datetime.now().strftime("%m"), "0")
+            + "_"
+            + rm_starting_str(datetime.now().strftime("%d"), "0")
+        )
+        return self.birthday_list.get(key, [])
 
     @handler(CommandHandler, command="birthday", block=False)
     @restricts()
@@ -87,7 +118,7 @@ class BirthdayPlugin(Plugin, BasePlugin):
                         self._add_delete_message_job(context, reply_message.chat_id, reply_message.message_id)
         else:
             logger.info(f"用户 {user.full_name}[{user.id}] 查询今日角色生日列表")
-            today_list = self.birthday_list.get(key, [])
+            today_list = self.get_today_birthday()
             if key == "6_1":
                 text = f"今天是 派蒙、{'、'.join(today_list)} 的生日哦~"
             else:
@@ -96,3 +127,72 @@ class BirthdayPlugin(Plugin, BasePlugin):
             if filters.ChatType.GROUPS.filter(reply_message):
                 self._add_delete_message_job(context, message.chat_id, message.message_id)
                 self._add_delete_message_job(context, reply_message.chat_id, reply_message.message_id)
+
+    @staticmethod
+    async def get_card(client: Client, role_id: int) -> None:
+        """领取画片"""
+        url = BIRTHDAY_URL.get_url()
+        params = {
+            "game_biz": recognize_genshin_game_biz(client.uid),
+            "lang": "zh-cn",
+            "badge_uid": client.uid,
+            "badge_region": recognize_genshin_server(client.uid),
+            "activity_id": "20220301153521",
+        }
+        json = {
+            "role_id": role_id,
+        }
+        await client.cookie_manager.request(url, method="POST", params=params, json=json)
+
+    @handler(CommandHandler, command="birthday_card", block=False)
+    @handler(MessageHandler, filters=filters.Regex("^领取角色生日画片$"), block=False)
+    @restricts()
+    @error_callable
+    async def command_birthday_card_start(self, update: Update, context: CallbackContext) -> None:
+        message = update.effective_message
+        user = update.effective_user
+        logger.info("用户 %s[%s] 领取生日画片命令请求", user.full_name, user.id)
+        today_list = self.get_today_birthday()
+        if not today_list:
+            reply_message = await message.reply_text("今天没有角色过生日哦~")
+            if filters.ChatType.GROUPS.filter(reply_message):
+                self._add_delete_message_job(context, message.chat_id, message.message_id)
+                self._add_delete_message_job(context, reply_message.chat_id, reply_message.message_id)
+            return
+        try:
+            client = await get_genshin_client(user.id)
+            if client.region == RegionEnum.HOYOLAB:
+                text = "此功能当前只支持国服账号哦~"
+            else:
+                await fetch_hk4e_token_by_cookie(client)
+                for name in today_list.copy():
+                    if role_id := roleToId(name):
+                        try:
+                            await self.get_card(client, role_id)
+                        except GenshinException as e:
+                            if e.retcode in {-512008, -512009}:  # 未过生日、已领取过
+                                today_list.remove(name)
+                if today_list:
+                    text = f"成功领取了 {'、'.join(today_list)} 的生日画片~"
+                else:
+                    text = "没有领取到生日画片哦 ~ 可能是已经领取过了"
+            reply_message = await message.reply_text(text)
+            if filters.ChatType.GROUPS.filter(reply_message):
+                self._add_delete_message_job(context, message.chat_id, message.message_id)
+                self._add_delete_message_job(context, reply_message.chat_id, reply_message.message_id)
+        except (UserNotFoundError, CookiesNotFoundError):
+            buttons = [[InlineKeyboardButton("点我绑定账号", url=create_deep_linked_url(context.bot.username, "set_cookie"))]]
+            if filters.ChatType.GROUPS.filter(message):
+                reply_msg = await message.reply_text(
+                    "此功能需要绑定<code>cookie</code>后使用，请先私聊派蒙绑定账号",
+                    reply_markup=InlineKeyboardMarkup(buttons),
+                    parse_mode=ParseMode.HTML,
+                )
+                self._add_delete_message_job(context, reply_msg.chat_id, reply_msg.message_id, 30)
+                self._add_delete_message_job(context, message.chat_id, message.message_id, 30)
+            else:
+                await message.reply_text(
+                    "此功能需要绑定<code>cookie</code>后使用，请先私聊派蒙进行绑定",
+                    parse_mode=ParseMode.HTML,
+                    reply_markup=InlineKeyboardMarkup(buttons),
+                )

resources/bot/help/help.html

@@ -145,6 +145,10 @@
                     <div class="command-name">/birthday</div>
                     <div class="command-description">角色生日</div>
                 </div>
+                <div class="command">
+                    <div class="command-name">/birthday_card</div>
+                    <div class="command-description">领取角色生日画片</div>
+                </div>
              </div>
         </div>