from discord.ext import commands
from cogs.utils.dataIO import dataIO
import discord
import os
import asyncio


class Seen:
    '''Check when someone was last seen.'''
    def __init__(self, bot):
        self.bot = bot
        self.seen = dataIO.load_json("data/seen/seen.json")
        self.new_data = False

    async def _get_channel(self, id):
        return discord.utils.get(self.bot.get_all_channels(), id=id)

    async def data_writer(self):
        while self == self.bot.get_cog("Seen"):
            if self.new_data:
                await asyncio.sleep(60)
                dataIO.save_json("data/seen/seen.json", self.seen)
                self.new_data = False
            else:
                await asyncio.sleep(30)

    @commands.command(pass_context=True, no_pm=True, name='seen')
    async def _seen(self, context, username: discord.Member):
        '''seen <@username>'''
        server = context.message.server
        author = username
        print(True if author.id in self.seen[server.id] else False if server.id in self.seen else False)
        if True if author.id in self.seen[server.id] else False if server.id in self.seen else False:
            data = self.seen[server.id][author.id]
            ts = data['TIMESTAMP']
            channel = await self._get_channel(data['CHANNEL'])
            em = discord.Embed(color=discord.Color.green())
            avatar = author.avatar_url if author.avatar else author.default_avatar_url
            em.set_author(name='{} was last seen on {} UTC in #{}'.format(author.display_name, ts, channel.name), icon_url=avatar)
            await self.bot.say(embed=em)
        else:
            message = 'I haven\'t seen {} yet.'.format(author.display_name)
            await self.bot.say('{}'.format(message))

    async def listener(self, message):
        if not message.channel.is_private and self.bot.user.id != message.author.id:
            server = message.server
            channel = message.channel
            author = message.author
            ts = message.timestamp
            data = {}
            data['TIMESTAMP'] = '{} {}:{}:{}'.format(ts.date(), ts.hour, ts.minute, ts.second)
            data['CHANNEL'] = channel.id
            if server.id not in self.seen:
                self.seen[server.id] = {}
            self.seen[server.id][author.id] = data
            self.new_data = True


def check_folder():
    if not os.path.exists('data/seen'):
        print('Creating data/seen folder...')
        os.makedirs('data/seen')


def check_file():
    if not dataIO.is_valid_json("data/seen/seen.json"):
        print("Creating seen.json...")
        dataIO.save_json("data/seen/seen.json", {})


def setup(bot):
    check_folder()
    check_file()
    n = Seen(bot)
    bot.add_listener(n.listener, 'on_message')
    loop = asyncio.get_event_loop()
    loop.create_task(n.data_writer())
    bot.add_cog(n)