import urllib.request
from datetime import datetime, timedelta

class MatrixModule:
    def matrix_start(self, bot):
        self.starttime = datetime.now()

    async def matrix_message(self, bot, room, event):
        args = event.body.split()
        if len(args) == 2:
            if args[1]=='quit':
                bot.must_be_admin(room, event)
                await bot.send_text(room, f'Quitting, as requested')
                print(f'{event.sender} commanded bot to quit, so quitting..')
                bot.bot_task.cancel()
            elif args[1]=='version':
                uptme = datetime.now() - self.starttime
                await bot.send_text(room, f'Hemppa version {bot.version} - Uptime {uptme} - https://github.com/vranki/hemppa')
            elif args[1]=='reload':
                bot.must_be_admin(room, event)
                await bot.send_text(room, f'Reloading modules..')
                bot.stop()
                for modulename in bot.modules:
                    bot.reload_module(modulename)
                bot.start()
            elif args[1]=='stats':
                roomcount = len(bot.client.rooms)
                usercount = 0
                homeservers = dict()

                for croomid in bot.client.rooms:
                    room = bot.client.rooms[croomid]
                    usercount = usercount + len(room.users)
                    for user in room.users:
                        hs = user.split(':')[1]
                        if homeservers.get(hs):
                            homeservers[hs] = homeservers[hs] + 1
                        else:
                            homeservers[hs] = 1

                homeservers = sorted(homeservers.items(), key=lambda kv: (kv[1], kv[0]), reverse=True)

                if len(homeservers) > 5:
                    homeservers = homeservers[0:5]

                await bot.send_text(room, f'I\'m seeing {usercount} users in {roomcount} rooms. Top 5 homeservers: {homeservers}')
        else:
            await bot.send_text(room, 'Unknown command, sorry.')

    def help(self):
        return('Bot management commands')