#!/usr/bin/env python
"""Contains the TaskBase to be inherited for Tasks creation"""

import asyncio
from importlib import import_module
from os import listdir
from os.path import isfile, join

import discord


class TaskBase:
    """Base class for bot's background tasks"""

    def __init__(self):
        self.endTask = False

    def set_instance_info(self, bot):
        self.bot = bot
        self.knowledge = bot.knowledge
        self.settings = bot.settings
        self.comm = bot.comm

        self.mainChannel = discord.Object(
            id=bot.settings.get_string('DefaultChannelId'))

    async def run(self):
        """Entry point of task. Do not overwrite."""

        # Wait until the bot is ready before running
        await self.comm.wait_until_ready()

        while not self.comm.is_closed or not self.endTask: 
            await asyncio.sleep(self.get_next_timeout())
            await self.do()

    def get_next_timeout(self):
        """Should return the time, in sec, to wait before each do() call."""
        return 10

    def do(self):
        """Proceed with the task's action. To be overwritten when inherited"""
        pass


def load_tasks(bot, tasksParent):
    """Function that loads the tasks from given paren module directory."""
    # Get fs path to module directory
    dpath = './' + tasksParent.replace('.', '/')

    # Retreive every python script in the tasks module directory
    files = [f for f in listdir(dpath)
             if isfile(join(dpath, f))
                and f.endswith('.py')]

    # Build the list of tasks
    tasks = []
    for file in files:
        # Module name should not end with '.py', so remove that part
        module = file[:-3]
        _taskModule = import_module(tasksParent + '.' + module)
        task = _taskModule.Task()
        tasks.append(task)
        # Inject instance info in task object
        task.set_instance_info(bot)
    print('Loaded ' + str(len(tasks)) + ' tasks from drive.')
    return tasks