From 199b7c0f26a39fb644ebb69a6d2536a43819e915 Mon Sep 17 00:00:00 2001
From: ue86388 <dmitrii.morozov@sbb.ch>
Date: Tue, 9 Apr 2024 13:44:23 +0200
Subject: Test

---
 FortniteClient.py          |  3 +++
 FortniteStatusFormatter.py | 11 +++++++++++
 tgbot.py                   | 14 ++++++++++++++
 3 files changed, 28 insertions(+)

diff --git a/FortniteClient.py b/FortniteClient.py
index f426fa7..7c5858c 100755
--- a/FortniteClient.py
+++ b/FortniteClient.py
@@ -22,6 +22,9 @@ class FortniteClient(fortnitepy.Client):
     
     def attach(self, observer: any):
         self.observers.append(observer)
+    
+    def get_friends(self):
+        return self.friends
 
     def __auth_authorization_code(self):
         code = input("Enter authorization code (https://www.epicgames.com/id/api/redirect?clientId=3446cd72694c4a4485d81b77adbb2141&responseType=code):") 
diff --git a/FortniteStatusFormatter.py b/FortniteStatusFormatter.py
index 9595304..be87f0c 100644
--- a/FortniteStatusFormatter.py
+++ b/FortniteStatusFormatter.py
@@ -18,4 +18,15 @@ def formatFortniteStatus(fortniteStatus):
         formatting.mbold("Fortnite status"),
         "",
         '\n'.join(statuses),
+        separator='\n')
+
+def __formatFriend(friend):
+    return friend.display_name
+
+def formatFriends(friends): 
+    friends_formatted = [__formatFriend(friend) for friend in friends]
+    return formatting.format_text(
+        formatting.mbold("Registered friends:"),
+        "",
+        '\n'.join(friends_formatted),
         separator='\n')
\ No newline at end of file
diff --git a/tgbot.py b/tgbot.py
index 2df88ff..ad72f00 100755
--- a/tgbot.py
+++ b/tgbot.py
@@ -23,6 +23,20 @@ def startCommand(message):
     userRepository.putUser(message.chat.id)
     bot.reply_to(message, "This chat successfully registered to receive Fortnite updates!")
 
+@bot.message_handler(commands = ['status'])
+def getStatus(message):
+    bot.reply_to(
+        message,
+        formatFortniteStatus(fortniteStatus.getStatus()),
+        parse_mode='MarkdownV2')
+
+@bot.message_handler(commands = ['friends'])
+def getFriends(message):
+    bot.reply_to(
+        message,
+        formatFriends(fortniteClient.get_friends()),
+        parse_mode='MarkdownV2')
+
 class FortniteStatusObserver(Observer):
     def update(self, fortniteStatus) -> None:
         for user in userRepository.getAllUsers():
-- 
cgit v1.2.3