#!/usr/bin/env python
# -*- coding: utf-8 -*-

import sys, socket, shelve, time, traceback, random


HOST="irc.freenode.net"
PORT=6667
NICK="PedroBot"
USERNAME="PedroBot"
REALNAME="PedroBot"
CHANNEL="##pedro3005"
#editing beyond this line is NOT necessary
recv=""
bot_is_opped = 0
passwd = open("passwd", "r")
PASSWORD = passwd.read().split("\n")[0]
owners = []
owners_get = open("owners", "r")
for line in owners_get :
    owners.append(line.split('\n')[0])
log_enabled = 0
factoids = shelve.open("factoids", writeback=True) # don't edit this manually

def parsemsg(s) :
    trailing = 0
    prefix = 0
    if s[0] == ":" :
        s = s[1:].split(' ', 1)
        prefix = s[0]
        s = s[1]
    if " :" in s :
        s = s.split(" :")
        trailing = s[1]
        s = s[0]
    args = s.split()
    command = args.pop(0)
    if trailing != 0 :
        args.append(trailing)
    return prefix, command, args

def chanmsg(*msgs) :
    if nick in owners :
        msg = " ".join(msgs)
        socket.send("PRIVMSG %s :%s\r\n" % (CHANNEL, msg))
    else :
        chanmsg("Permission denied.")

def chanact(*acts) :
    if nick in owners :
        act = " ".join(acts)
        socket.send("PRIVMSG %s :\x01ACTION %s\x01\r\n" % (CHANNEL, act))
    else :
        chanmsg("Permission denied.")

def kill(nick) :
    if not nick in owners :
        chanact("kills %s unmercifully!" % (nick))
    else :
        chanmsg("I shall not kill my master.")

def read_factoids() :
    fact_list = []
    for item in factoids :
        fact_list.append(item[1:])
    socket.send("PRIVMSG %s :Available factoids: %s\r\n" % (CHANNEL, ", ".join(fact_list)))

def new_factoid(name, nothing, *definitions) :
    if not name.startswith("!") :
        definition = " ".join(definitions)
        factoids["!" + name] = definition
        if factoids["!" + name] == definition :
            socket.send("PRIVMSG %s :Factoid %s added successfuly\r\n" % (CHANNEL, name))
    else :
        chanmsg("You're not going to break me, asshole")
def del_factoid(name) :
    if nick in owners :
        try :
            del factoids["!" + name]
        except :
            socket.send("PRIVMSG %s :Factoid %s not found\r\n" % (CHANNEL, name))
        else :
            socket.send("PRIVMSG %s :Factoid %s deleted successfuly\r\n" % (CHANNEL, name))

def add_owner(name) :
    if nick in owners :
        owners.append(name)
        write = open("owners", "a")
        write.write(name + '\n')
    if name in owners :
        chanmsg("Owner %s added successfuly." % (name))

def read_owners() :
    rowners = ", ".join(owners)
    chanmsg("Owners: %s" % (rowners))

def kick(name, nothing, *reasons) :
    reason = " ".join(reasons)
    if nick in owners :
        socket.send("KICK %s %s :%s\r\n" % (CHANNEL, name, reason))
        if not bot_is_opped == 1 :
            chanmsg("I'm not an OP.")
    else :
        chanmsg("Permission denied.")

def ismsg(msg) :
    if command == "PRIVMSG" and args[0] == CHANNEL :
        return True

def log_on() :
    global log_enabled, log
    if nick in owners :
        if log_enabled == 0 :
            log = open(CHANNEL, "a")
            log.write(time.strftime("%a, %d %b %Y %H:%M:%S", time.localtime()) + '\n' + '----------------------------' + '\n')
            log_enabled = 1
            chanmsg("Logging started")
            log.flush()
        else :
            chanmsg("I am already logging.")
    else :
        chanmsg("Permission denied.")
def log_off() :
    global log_enabled, log
    if nick in owners :
        if log_enabled == 1 :
            log_enabled = 0
            log.write("-------END OF LOG-----\n")
            log.flush()
            chanmsg("Stopped logging")
        else :
            chanmsg("I am not logging anything.")

def ircroulette() :
    chance = random.choice(range(1, 4))
    if chance == 3 :
        kick(nick, "because", "*BAM*")
    else :
        chanmsg("*tick*")

def topic(*ntpcs) :
    ntpc = " ".join(ntpcs)
    if nick in owners :
        socket.send("TOPIC %s :%s\r\n" % (CHANNEL, ntpc))
        if not bot_is_opped == 1 :
            chanmsg("I'm not an OP.")
    else :
        chanmsg("Permission denied.")

commands = {
    "kill": kill,
    "factoids": read_factoids,
    "add": new_factoid,
    "del": del_factoid,
    "add_owner": add_owner,
    "owners": read_owners,
    "kick": kick,
    "log": log_on,
    "log_off": log_off,
    "roulette": ircroulette,
    "say": chanmsg,
    "act": chanact,
    "topic": topic
}


socket = socket.socket()
socket.connect((HOST, PORT))
if PASSWORD != "(secret)" :
    socket.send("PASS %s\r\n" % (PASSWORD))
socket.send("NICK %s\r\n" % (NICK))
socket.send("USER %s * * :%s\r\n" % (USERNAME, REALNAME))
socket.send("JOIN %s\r\n" % (CHANNEL))

while True :
    recv = recv + socket.recv(4096)
    s = recv.split("\r\n")
    recv = s.pop()
    for msg in s :
        print msg
        prefix, command, args = parsemsg(msg)
        try :
            message = args[-1]
            bot_cmd = message.split()[0][1:]
            bot_cmd_args = message.split()[1:]
            nick = prefix.split("!")[0]
        except :
            pass
        if command == "MODE" and "+o" in args and "PedroBot" in args :
            bot_is_opped = 1
        if log_enabled == 1 and ismsg(msg) :
            log.write("%s: %s\n" % (nick, message))
            log.flush()
        if command == "PING" :
            socket.send("PONG\r\n")
        elif bot_cmd in commands and message.startswith("!") :
            try :
                commands[bot_cmd](*bot_cmd_args)
            except :
                chanmsg("Error encountered. You probably didn't provide enough or provided too much arguments for a function. Try again")
        elif message.startswith("!") and "!" + bot_cmd in factoids.keys() :
            socket.send("PRIVMSG %s :%s\r\n" % (CHANNEL, factoids["!" + bot_cmd]))