import telebot
from TelegramBot import *
from Formatter import *
from persistence import *
from FortniteClient import FortniteClient
from pythonFortniteStatus.FortniteStatus import *

class StartCommand(CommandHandler):

    __telegramBot: TelegramBot
    __userRepository: UserRepository

    def __init__(self, telegramBot: TelegramBot, userRepository: UserRepository):
        self.__telegramBot = telegramBot
        self.__userRepository = userRepository

    async def handle(self, message: telebot.types.Message):
        if message.chat.type == 'private':
            alias = message.chat.username
        else:
            alias = message.chat.title
        self.__userRepository.putUser(message.chat.id, alias)
        await self.__telegramBot.reply(message, 'This chat successfully registered to receive Fortnite updates')

class GetStatusCommand(CommandHandler):

    __telegramBot: TelegramBot
    __fortniteStatus: FortniteStatus

    def __init__(self, telegramBot: TelegramBot):
        self.__telegramBot = telegramBot
        self.__fortniteStatus = FortniteStatus()

    async def handle(self, message: telebot.types.Message):
        await self.__telegramBot.reply(message, formatFortniteStatus(self.__fortniteStatus.getStatus()))

class GetFriendsCommand(CommandHandler):

    __telegramBot: TelegramBot
    __fortniteClient: FortniteClient

    def __init__(self, telegramBot: TelegramBot, fortniteClient: FortniteClient):
        self.__telegramBot = telegramBot
        self.__fortniteClient = fortniteClient

    async def handle(self, message: telebot.types.Message):
        friends = await self.__fortniteClient.get_friends()
        await self.__telegramBot.reply(message, formatUsers(friends))


class GetStatsCommand(CommandHandler):

    __telegramBot: TelegramBot
    __fortniteClient: FortniteClient

    def __init__(self, telegramBot: TelegramBot, fortniteClient: FortniteClient):
        self.__telegramBot = telegramBot
        self.__fortniteClient = fortniteClient

    async def handle(self, message: telebot.types.Message):
        friends = await self.__fortniteClient.get_friends()
        stats = [await friend.fetch_stats() for friend in friends]
        await self.__telegramBot.reply(message, formatUserStatsList(stats))

class GetTodayStatsCommand(CommandHandler):

    __telegramBot: TelegramBot
    __fortniteClient: FortniteClient
    __statsRepository: StatsRepository

    def __init__(self, telegramBot: TelegramBot, fortniteClient: FortniteClient, statsRepository: StatsRepository):
        self.__telegramBot = telegramBot
        self.__fortniteClient = fortniteClient
        self.__statsRepository = statsRepository

    async def handle(self, message: telebot.types.Message):
        persisted_stats = self.__statsRepository.getStats()
        friends = await self.__fortniteClient.get_friends()
        current_stats = [await friend.fetch_stats() for friend in friends]
        await self.__telegramBot.reply(message, formatUserStatsDifference(persisted_stats, current_stats))

class RecordStatsCommand(CommandHandler):

    __telegramBot: TelegramBot
    __fortniteClient: FortniteClient
    __statsRepository: StatsRepository

    def __init__(self, telegramBot: TelegramBot, fortniteClient: FortniteClient, statsRepository: StatsRepository):
        self.__telegramBot = telegramBot
        self.__fortniteClient = fortniteClient
        self.__statsRepository = statsRepository

    async def handle(self, message: telebot.types.Message):
        friends = await self.__fortniteClient.get_friends()
        for friend in friends:
            await self.__statsRepository.putStats(friend)