mirror of
https://github.com/PaiGramTeam/MibooGram.git
synced 2024-12-12 15:40:52 +00:00
134 lines
6.6 KiB
Python
134 lines
6.6 KiB
Python
import time
|
|
from typing import TYPE_CHECKING
|
|
|
|
from simnet import Region
|
|
from telegram.constants import ChatAction
|
|
from telegram.ext import filters
|
|
|
|
from core.config import config
|
|
from core.plugin import Plugin, handler
|
|
from core.services.task.models import Task as SignUser, TaskStatusEnum
|
|
from core.services.users.services import UserAdminService
|
|
from gram_core.services.task.services import TaskCardServices
|
|
from plugins.tools.cloud_game import CloudGameHelper
|
|
from plugins.tools.genshin import PlayerNotFoundError, CookiesNotFoundError
|
|
from utils.log import logger
|
|
|
|
if TYPE_CHECKING:
|
|
from telegram import Update
|
|
from telegram.ext import ContextTypes
|
|
|
|
|
|
class CloudGameSign(Plugin):
|
|
"""云游戏每日签到"""
|
|
|
|
CHECK_SERVER, COMMAND_RESULT = range(10400, 10402)
|
|
|
|
def __init__(
|
|
self,
|
|
cloud_game_helper: CloudGameHelper,
|
|
sign_service: TaskCardServices,
|
|
user_admin_service: UserAdminService,
|
|
):
|
|
self.user_admin_service = user_admin_service
|
|
self.sign_service = sign_service
|
|
self.cloud_game_helper = cloud_game_helper
|
|
|
|
async def _process_auto_sign(self, user_id: int, player_id: int, offset: int, chat_id: int, method: str) -> str:
|
|
try:
|
|
async with self.cloud_game_helper.client(user_id, player_id=player_id, offset=offset) as c:
|
|
if c.region is Region.OVERSEAS:
|
|
return "云游戏相关功能仅支持国服"
|
|
await c.get_cloud_game_notifications()
|
|
player_id = c.player_id
|
|
except (PlayerNotFoundError, CookiesNotFoundError):
|
|
return config.notice.user_not_found
|
|
user: SignUser = await self.sign_service.get_by_user_id(user_id, player_id)
|
|
if user:
|
|
if method == "关闭":
|
|
await self.sign_service.remove(user)
|
|
return f"UID {player_id} 关闭云游戏自动签到成功"
|
|
if method == "开启":
|
|
if user.chat_id == chat_id:
|
|
return f"UID {player_id} 云游戏自动签到已经开启过了"
|
|
user.chat_id = chat_id
|
|
user.status = TaskStatusEnum.STATUS_SUCCESS
|
|
await self.sign_service.update(user)
|
|
return f"UID {player_id} 修改云游戏自动签到通知对话成功"
|
|
elif method == "关闭":
|
|
return f"UID {player_id} 您还没有开启云游戏自动签到"
|
|
elif method == "开启":
|
|
user = self.sign_service.create(user_id, player_id, chat_id, TaskStatusEnum.STATUS_SUCCESS)
|
|
await self.sign_service.add(user)
|
|
return f"UID {player_id} 开启云游戏自动签到成功"
|
|
|
|
@handler.command(command="cloud_game_sign", cookie=True, block=False)
|
|
@handler.message(filters=filters.Regex("^云游戏每日签到(.*)"), cookie=True, block=False)
|
|
@handler.command(command="start", filters=filters.Regex("cloud_game_sign$"), block=False)
|
|
async def command_start(self, update: "Update", context: "ContextTypes.DEFAULT_TYPE") -> None:
|
|
user_id = await self.get_real_user_id(update)
|
|
uid, offset = self.get_real_uid_or_offset(update)
|
|
message = update.effective_message
|
|
args = self.get_args(context)
|
|
if len(args) >= 1:
|
|
msg = None
|
|
if args[0] == "开启自动签到":
|
|
if await self.user_admin_service.is_admin(user_id):
|
|
msg = await self._process_auto_sign(user_id, uid, offset, message.chat_id, "开启")
|
|
else:
|
|
msg = await self._process_auto_sign(user_id, uid, offset, user_id, "开启")
|
|
elif args[0] == "关闭自动签到":
|
|
msg = await self._process_auto_sign(user_id, uid, offset, message.chat_id, "关闭")
|
|
if msg:
|
|
self.log_user(update, logger.info, "云游戏自动签到命令请求 || 参数 %s", args[0])
|
|
reply_message = await message.reply_text(msg)
|
|
if filters.ChatType.GROUPS.filter(message):
|
|
self.add_delete_message_job(reply_message, delay=30)
|
|
self.add_delete_message_job(message, delay=30)
|
|
return
|
|
self.log_user(update, logger.info, "云游戏每日签到命令请求")
|
|
if filters.ChatType.GROUPS.filter(message):
|
|
self.add_delete_message_job(message)
|
|
async with self.cloud_game_helper.client(user_id, player_id=uid, offset=offset) as client:
|
|
if client.region is Region.OVERSEAS:
|
|
reply_message = await message.reply_text("云游戏相关功能仅支持国服")
|
|
if filters.ChatType.GROUPS.filter(reply_message):
|
|
self.add_delete_message_job(message)
|
|
self.add_delete_message_job(reply_message)
|
|
return
|
|
await message.reply_chat_action(ChatAction.TYPING)
|
|
sign_text = await self.cloud_game_helper.start_sign(client)
|
|
reply_message = await message.reply_text(sign_text)
|
|
if filters.ChatType.GROUPS.filter(reply_message):
|
|
self.add_delete_message_job(reply_message)
|
|
|
|
@handler.command(command="cloud_game_wallet", cookie=True, block=False)
|
|
async def command_start_wallet(self, update: "Update", _: "ContextTypes.DEFAULT_TYPE") -> None:
|
|
user_id = await self.get_real_user_id(update)
|
|
uid, offset = self.get_real_uid_or_offset(update)
|
|
message = update.effective_message
|
|
self.log_user(update, logger.info, "云游戏查询钱包命令请求")
|
|
async with self.cloud_game_helper.client(user_id, player_id=uid, offset=offset) as client:
|
|
if client.region is Region.OVERSEAS:
|
|
reply_message = await message.reply_text("云游戏相关功能仅支持国服")
|
|
if filters.ChatType.GROUPS.filter(reply_message):
|
|
self.add_delete_message_job(message)
|
|
self.add_delete_message_job(reply_message)
|
|
return
|
|
await message.reply_chat_action(ChatAction.TYPING)
|
|
wallet = await self.cloud_game_helper.get_wallet(client)
|
|
today = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
|
|
free_time = wallet.free_time.free_time
|
|
pay_time = int(wallet.coin.coin_num / wallet.coin.exchange)
|
|
total_time = wallet.total_time
|
|
card = wallet.play_card.msg
|
|
text = (
|
|
"#### 云游戏钱包 ####\n"
|
|
f"时间：{today} (UTC+8)\n"
|
|
f"免费时长：{free_time} 分钟\n"
|
|
f"付费时长：{pay_time} 分钟\n"
|
|
f"畅玩卡：{card}\n"
|
|
f"总时长：{total_time} 分钟"
|
|
)
|
|
await message.reply_text(text)
|