from irc import IRC
from botbrain import BotBrain

class IRCBot:
    irc = None #IRC wrapper

    nick = ''
    server = ''
    port = 6667

    channels = [] #List of channels to join
    keys = []     #The keys for the channels

    def __init__(self, nick='', server='', port=6667, channels=[], keys=[]):
        """
            Initialises stuff
        """
        self.nick = nick
        self.server = server
        self.port = port
        self.channels = channels
        self.keys = keys

        self.irc = IRC(nick,server,port)

        #callbacks
        self.irc.on_connect = self.on_connect
        self.irc.on_disconnect = self.on_disconnect
        self.irc.on_privmsg = self.on_privmsg

        self.irc.start()

    def on_connect(self):
        """
            We connected to a server
        """
        for i in range(0, len(channels)):
            self.irc.join(channels[i], key[i])

    def on_privmsg(self, usermask, channel, message):
        """
            We received a message from the usermask on channel
        """
        nick = usermask.split('!')[0]

        print(nick + " said: " + message)
        self.irc.privmsg(channel, "Hi " + nick)

    def on_disconnect(self):
        """
            We got disconnected
        """
        print("*** Disconnected.")

if __name__ == "__main__":
    bot = IRCBot("Hayden|Bot",
                 "irc.killanet.net",
                 6667,
                 ["#haydentest"],
                 [""])