#!/usr/bin/env python
# coding=utf-8
from __future__ import unicode_literals, absolute_import, print_function, division

# sopel imports
import sopel.module
from sopel.config.types import StaticSection, ValidatedAttribute, ListAttribute

from sopel_modules.SpiceBot.Logs import botlogs
from sopel_modules.SpiceBot.Events import botevents
from sopel_modules.SpiceBot_SBTools import join_all_channels, chanadmin_all_channels, channel_list_current

import spicemanip

import time


class SpiceBot_Channels_MainSection(StaticSection):
    announcenew = ValidatedAttribute('announcenew', default=False)
    joinall = ValidatedAttribute('joinall', default=False)
    operadmin = ValidatedAttribute('operadmin', default=False)
    chanignore = ListAttribute('chanignore')


def configure(config):
    config.define_section("SpiceBot_Channels", SpiceBot_Channels_MainSection, validate=False)
    config.SpiceBot_Channels.configure_setting('announcenew', 'SpiceBot_Channels Announce New Channels')
    config.SpiceBot_Channels.configure_setting('joinall', 'SpiceBot_Channels JOIN New Channels')
    config.SpiceBot_Channels.configure_setting('operadmin', 'SpiceBot_Channels OPER ADMIN MODE')
    config.SpiceBot_Channels.configure_setting('chanignore', 'SpiceBot_Channels Ignore JOIN for channels')


def setup(bot):
    botlogs.log('SpiceBot_Channels', "Starting setup procedure")
    botevents.startup_add([botevents.BOT_CHANNELS])

    bot.config.define_section("SpiceBot_Channels", SpiceBot_Channels_MainSection, validate=False)

    if "SpiceBot_Channels" not in bot.memory:
        bot.memory["SpiceBot_Channels"] = {"channels": {}, "InitialProcess": False, "ProcessLock": False}


def shutdown(bot):
    if "SpiceBot_Channels" in bot.memory:
        del bot.memory["SpiceBot_Channels"]


@sopel.module.event(botevents.BOT_CONNECTED)
@sopel.module.rule('.*')
def unkickable_bot(bot, trigger):
    if bot.config.SpiceBot_Channels.operadmin:
        bot.write(('SAMODE', bot.nick, '+q'))


@sopel.module.event(botevents.BOT_CONNECTED)
@sopel.module.rule('.*')
def trigger_channel_list_initial(bot, trigger):

    # Unkickable
    bot.write(('SAMODE', bot.nick, '+q'))

    bot_part_empty(bot)

    bot.write(['LIST'])
    bot.memory['SpiceBot_Channels']['ProcessLock'] = True

    botlogs.log('SpiceBot_Channels', "Initial Channel list populating")
    starttime = time.time()

    while not bot.memory['SpiceBot_Channels']['InitialProcess']:
        timesince = time.time() - starttime
        if timesince < 60:
            pass
        else:
            botlogs.log('SpiceBot_Channels', "Initial Channel list populating Timed Out")
            bot.memory['SpiceBot_Channels']['InitialProcess'] = True

    channel_list_current(bot)
    foundchannelcount = len(bot.memory['SpiceBot_Channels']['channels'].keys())
    botlogs.log('SpiceBot_Channels', "Channel listing finished! " + str(foundchannelcount) + " channel(s) found.")

    join_all_channels(bot)
    chanadmin_all_channels(bot)

    if "*" in bot.memory['SpiceBot_Channels']['channels']:
        del bot.memory['SpiceBot_Channels']['channels']["*"]

    bot_part_empty(bot)
    botevents.trigger(bot, botevents.BOT_CHANNELS, "SpiceBot_Channels")


@sopel.module.event(botevents.BOT_CHANNELS)
@sopel.module.rule('.*')
def trigger_channel_list_recurring(bot, trigger):
    while True:
        try:
            time.sleep(1800)
            bot_part_empty(bot)

            oldlist = list(bot.memory['SpiceBot_Channels']['channels'].keys())
            bot.write(['LIST'])
            bot.memory['SpiceBot_Channels']['ProcessLock'] = True

            while bot.memory['SpiceBot_Channels']['ProcessLock']:
                pass

            newlist = [item.lower() for item in oldlist if item.lower() not in list(bot.memory['SpiceBot_Channels']['channels'.keys()])]
            if "*" in newlist:
                newlist.remove("*")
            if len(newlist) and bot.config.SpiceBot_Channels.announcenew:
                bot.osd(["The Following channel(s) are new:", spicemanip.main(newlist, 'andlist')], bot.channels.keys())

            join_all_channels(bot)

            chanadmin_all_channels(bot)

            if "*" in bot.memory['SpiceBot_Channels']['channels']:
                del bot.memory['SpiceBot_Channels']['channels']["*"]
            bot_part_empty(bot)

        except KeyError:
            return


def bot_part_empty(bot):
    """Don't stay in empty channels"""
    ignorepartlist = []
    if bot.config.core.logging_channel:
        ignorepartlist.append(bot.config.core.logging_channel)

    for channel in bot.channels.keys():
        if len(bot.channels[channel].privileges.keys()) == 1 and channel not in ignorepartlist and channel.startswith("#"):
            bot.part(channel, "Leaving Empty Channel")
            if channel.lower() in bot.memory['SpiceBot_Channels']['channels']:
                del bot.memory['SpiceBot_Channels']['channels'][channel.lower()]