diff options
Diffstat (limited to 'fortnite_client/fortnite_events.py')
| -rw-r--r-- | fortnite_client/fortnite_events.py | 64 |
1 files changed, 61 insertions, 3 deletions
diff --git a/fortnite_client/fortnite_events.py b/fortnite_client/fortnite_events.py index 918b1e1..1efbc4e 100644 --- a/fortnite_client/fortnite_events.py +++ b/fortnite_client/fortnite_events.py @@ -7,6 +7,9 @@ from telegram_bot import * from persistence import * from formatter import * +__accept_keyword__ = "accept" +__decline_keyword__ = "decline" + class ClientInitObserverImpl(ClientInitObserver): async def on_event(self, fortnite_client: FortniteClient) -> None: print('----------------') @@ -43,10 +46,65 @@ class NewFriendObserverImpl(NewFriendObserver): def __init__(self, telegram_bot: TelegramBot, stats_repository: StatsRepository): self.__telegram_bot = telegram_bot self.__stats_repository = stats_repository + + async def on_new_friend_request(self, request_id: str, display_name: str, fortnite_client: FortniteClient) -> None: + __new_friend_reply_markup = telebot.types.InlineKeyboardMarkup() + __new_friend_reply_markup.add(telebot.types.InlineKeyboardButton('Accept', callback_data="{}:{}:{}".format(__accept_keyword__, request_id, display_name))) + __new_friend_reply_markup.add(telebot.types.InlineKeyboardButton('Decline', callback_data="{}:{}:{}".format(__decline_keyword__, request_id, display_name))) + + self.__telegram_bot.register_callback_query( + NewFriendCallbackQueryHandler(self.__telegram_bot, fortnite_client, self), + lambda call: call.data.startswith(__accept_keyword__) or call.data.startswith(__decline_keyword__)) + + await self.__telegram_bot.send_message_to_all(format_new_friend_request(display_name), reply_markup=__new_friend_reply_markup) - async def on_event(self, friend: User) -> None: + async def on_friend_added(self, fornite_friend: fortnitepy.friend.Friend) -> None: + # Try fetch stats + try: + await self.__try_get_stats(fornite_friend) + except: + print(traceback.format_exc()) + await self.__on_unavailable_public_stats(fornite_friend.display_name) + + friend = User.from_fortnite_friend(fornite_friend) await self.__stats_repository.put_stats(friend, datetime.datetime.now()) await self.__telegram_bot.send_message_to_all(format_new_friend(friend.display_name)) - async def on_unavailable_public_stats(self, display_name) -> None: - await self.__telegram_bot.send_message_to_all(format_friend_disabled_public_stats(display_name))
\ No newline at end of file + async def __try_get_stats(self, friend: fortnitepy.friend.Friend): + await friend.fetch_br_stats() + + async def __on_unavailable_public_stats(self, display_name) -> None: + await self.__telegram_bot.send_message_to_all(format_friend_disabled_public_stats(display_name)) + +class NewFriendCallbackQueryHandler(CallbackQueryHandler): + __telegram_bot: TelegramBot + __fortnite_client: FortniteClient + __new_friend_observer: NewFriendObserver + + def __init__(self, telegram_bot: TelegramBot, fortnite_client: FortniteClient, new_friend_observer: NewFriendObserver): + self.__telegram_bot = telegram_bot + self.__fortnite_client = fortnite_client + self.__new_friend_observer = new_friend_observer + + async def handle(self, call: telebot.types.CallbackQuery): + if self.__fortnite_client.is_initialized(): + answered: bool = False + if call.data.startswith(__accept_keyword__) or call.data.startswith(__decline_keyword__): + splitted = call.data.split(":") + await self.__handle_request(splitted[0], splitted[1], splitted[2]) + + await self.__telegram_bot.answer_callback_query(callback_query_id=call.id) + + async def __handle_request(self, action: str, request_id: str, display_name: str): + pending_friends = self.__fortnite_client.pending_friends + for pending_friend in pending_friends: + if isinstance(pending_friend, fortnitepy.friend.IncomingPendingFriend): + incoming_request = typing.cast(fortnitepy.friend.IncomingPendingFriend, pending_friend) + if incoming_request._id == request_id: + if action == __accept_keyword__: + await incoming_request.accept() + accepted_friend = incoming_request.client.get_friend(incoming_request._id) + await self.__new_friend_observer.on_friend_added(accepted_friend) + elif action == __decline_keyword__: + await incoming_request.decline() + await self.__telegram_bot.send_message_to_all(format_friend_declined(display_name))
\ No newline at end of file |