#  Shitty plugin made by iloveportalz0r
#  Broken by The Noodle
#  Improved by Lukeroge
#  Mute added by neersighted
from util import hook
import sys
import usertracking
import time

# Added to make the move to a new auth system a lot easier
def isadmin(input):
    if input.nick in input.bot.config["admins"]:
        return True
    else:
        return False
        
@hook.command
def quit(inp, input=None, db=None, notice=None):
    ".quit [reason] -- kills the bot"
    if not isadmin(input):
        notice("Only bot admins can use this command!")
        return
    if inp:
        input.conn.send("QUIT :Kill switch activated by "+input.nick+" (" + inp + ")")
    else:
        input.conn.send("QUIT :Kill switch activated by "+input.nick+" (no reason)")
    time.sleep(3)
    sys.exit()
        

@hook.command
def join(inp, input=None, db=None, notice=None):
    ".join <channel> -- joins a channel"
    if not isadmin(input):
        notice("Only bot admins can use this command!")
        return
    notice("Attempting to join " + inp + "...")
    input.conn.send("JOIN " + inp)


@hook.command
def cycle(inp, input=None, db=None, notice=None):
    ".cycle <channel> -- cycles a channel"
    if not isadmin(input):
        notice("Only bot admins can use this command!")
        return
    notice("Attempting to cycle " + inp + "...")
    input.conn.send("PART " + inp)
    input.conn.send("JOIN " + inp)

@hook.command
def part(inp, input=None, notice=None):
    ".part <channel> -- leaves a channel"
    if not isadmin(input):
        notice("Only bot admins can use this command!")
        return
    notice("Attempting to part from " + inp + "...")
    input.conn.send("PART " + inp)

@hook.command
def nick(inp, input=None, notice=None):
    ".nick <nick> -- change the bots nickname to <nick>"
    if not isadmin(input):
        notice("Only bot admins can use this command!")
        return
    notice("Changing nick to " + inp + ".")
    input.conn.send("NICK " + inp)

@hook.command
def raw(inp, input=None, notice=None):
    ".raw <command> - Send a RAW IRC command!"
    if not isadmin(input):
        notice("Only bot admins can use this command!")
        return
    notice("Raw command sent.")
    input.conn.send(inp)

@hook.command
def kick(inp, input=None, notice=None):
    ".kick [channel] <user> [reason] -- kick a user!"
    if not isadmin(input):
        notice("Only bot admins can use this command!")
        return
    split = inp.split(" ")
    if split[0][0] == "#":
        chan = split[0]
        user = split[1]
        out = "KICK %s %s" % (chan, user)
        if len(split) > 2:
            reason = ""
            for x in split[2:]:
                reason = reason + x + " "
            reason = reason[:-1]
            out = out+" :"+reason
    else:
        chan = input.chan
        user = split[0]
        out = "KICK %s %s" % (input.chan, split[0])
        if len(split) > 1:
            reason = ""
            for x in split[1:]:
                reason = reason + x + " "
            reason = reason[:-1]
            out = out + " :" + reason

    notice("Attempting to kick %s from %s..." % (user, chan))         
    input.conn.send(out)

@hook.command
def say(inp, input=None, notice=None):
    ".say [channel] <message> -- makes the bot say <message> in [channel]. if [channel] is blank the bot will say the <message> in the channel the command was used in."
    if not isadmin(input):
        notice("Only bot admins can use this command!")
        return
    split = inp.split(" ")
    if split[0][0] == "#":
        message = ""
        for x in split[1:]:
            message = message + x + " "
        message = message[:-1]
        out = "PRIVMSG %s :%s" % (split[0], message)
    else:
        message = ""
        for x in split[0:]:
            message = message + x + " "
        message = message[:-1]
        out = "PRIVMSG %s :%s" % (input.chan, message)
    input.conn.send(out)

@hook.command("me")
@hook.command
def act(inp, input=None, notice=None):
    ".act [channel] <action> -- makes the bot act <action> in [channel]. if [channel] is blank the bot will act the <action> in the channel the command was used in."
    if not isadmin(input):
        notice("Only bot admins can use this command!")
        return
    split = inp.split(" ")
    if split[0][0] == "#":
        message = ""
        for x in split[1:]:
            message = message + x + " "
        message = message[:-1]
        out = "PRIVMSG %s :\x01ACTION %s\x01" % (split[0], message)
    else:
        message = ""
        for x in split[0:]:
            message = message + x + " "
        message = message[:-1]
        out = "PRIVMSG %s :\x01ACTION %s\x01" % (input.chan, message)
    input.conn.send(out)

@hook.command
def topic(inp, input=None, notice=None):
    ".topic [channel] <topic> -- change the topic of a channel"
    if not isadmin(input):
        notice("Only bot admins can use this command!")
        return
    split = inp.split(" ")
    if split[0][0] == "#":
        out = "PRIVMSG %s :%s" % (split[0], message)
    else:
        out = "TOPIC %s :%s" % (input.chan, message)
    input.conn.send(out)

@hook.sieve
def mutesieve(bot, input, func, kind, args):
    if kind == "event":
        return input
    if "chan" in input.keys() and input.chan in input.conn.users.channels and hasattr(input.conn.users[input.chan], "mute"):
        if input.command == "PRIVMSG" and input.lastparam[1:] == "unmute":
            return input
        else:
            return None
    return input

@hook.command
def mute(inp, input=None, db=None, bot=None, users=None):
    if inp and inp in input.conn.users.channels.keys():
        input.chan = inp
    ".mute <channel> -- Mutes the bot"
    if usertracking.query(db, bot.config, input.nick, input.chan, "mute") or "o" in users[input.chan].usermodes[input.nick]:
        users[input.chan].mute = "%s %d" % (input.nick, time.time())
        input.notice("Muted")
    else:
        input.notice("Only bot admins can use this command!")

@hook.command
def unmute(inp, input=None, db=None, bot=None, users=None):
    if inp and inp in users.channels.keys():
        input.chan = inp
    ".unmute <channel> -- Unmutes the bot"
    if usertracking.query(db, bot.config, input.nick, input.chan, "mute") or "o" in users[input.chan].usermodes[input.nick]:
        if hasattr(users[input.chan], "mute"):
            input.notice("Unmuted")
            del users[input.chan].mute
        else:
            input.notice("Not Muted")
    else:
        input.notice("Only bot admins can use this command!")