#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Copyright (C) 2007  Daniele Tricoli aka Eriol <eriol@mornie.org>
# Copyright (C) 2010 Andrea Milazzo aka mancausoft <me@mancausoft.org>
#
#  Daniele Tricoli, 2010 Reliceinsed to WTFPL
#
#  This Program is free software. It comes without any warranty, to
#  the extent permitted by applicable law. You can redistribute it
#  and/or modify it under the terms of the Do What The Fuck You Want
#  To Public License, Version 2, as published by Sam Hocevar. See
#  http://sam.zoy.org/wtfpl/COPYING for more details.


import os
import re

from twisted.words.protocols import irc
from twisted.internet import reactor, protocol, ssl

from dataslavebot import *

import logging as log
log.basicConfig(level=log.DEBUG,
                format='[%(asctime)s] %(levelname)s %(message)s',
                datefmt='%d/%m/%Y %H:%M:%S %Z',
                filename='slavebot.log',
                filemode='a')
console = log.StreamHandler()
console.setLevel(log.INFO)
log.getLogger().addHandler(console)


def save():
    f = open('dataslavebot.py', 'w')
    f.write('BOT_NICKNAME = "' + BOT_NICKNAME +'"\n')
    f.write('MASTER_USER = ' + str(MASTER_USER))
    f.write('\n\nRE_CONV = ' + str(RE_CONV))
    f.write('\nRE_CONV_ME = ' + str(RE_CONV_ME))
    f.write("\nOP_BOT = '"+ OP_BOT + "'\n")
    f.write('SERVER = '+ str(SERVER))

class SlaveBot(irc.IRCClient):

    nickname = BOT_NICKNAME

    def __init__(self):
        self.op = False
	self.nicks = {} 

    def opOnlyMaster (self, nick, channel):
        if ((nick in MASTER_USER) or (nick[:-1] in MASTER_USER) or (nick[:-2] in MASTER_USER)) and self.op:
            if nick == "tosky":
                self.msg(channel, "Ďekuji vám, pane " + nick + ", protože mi jste ctíl jako operátor.") 
            elif nick == "geenna":
                self.msg(channel, "Grazie mio padrone " + nick + " mettiti a 90 che cominciamo!.") 
            else:
                self.msg(channel, "Grazie mio padrone " + nick + " per avermi dato l'onore di poterti oppare.") 
            self.sendLine('MODE ' + channel + ' +o ' + nick)
            self.nicks[nick] = 1
 
    def signedOn(self):
        log.info('Signed On:')
        for chan in self.factory.channel:
            self.join(chan)

    def joined(self, channel):
        log.info('Joined: %s', channel)
	self.nicks = {}
        self.msg(channel, "Grazie di avermi creato, datemi l'onore del'op e saro' il vostro schiavo per il resto della mia vita")

    def action (self, username, channel, emote): 
        if (emote.find("si inchina") != -1) or (emote.find("s'inchina") != -1):
            self.msg(channel, username.split('!')[0] + ": Stai attento in questo canale e' pericoloso fare cose del genere!")

    def userLeft (self, user, channel):
        del self.nicks[user]

    def userRenamed(self, oldname, newname):
        self.nicks[newname] = self.nicks[oldname]
        del self.nicks[oldname]

    def userJoined (self, user, channel):
        self.me(channel, "si prostra davanti a " + user)
        self.nicks[user] = 0
        self.opOnlyMaster(user, channel)

    def irc_RPL_NAMREPLY(self, prefix, params):
        log.info('irc_RPL_NAMREPLY')
        nicks = params[3].split()
        for nick in nicks:
           if nick[0] == '@':
	       self.nicks[nick[1:]] =  1
           else:
	       self.nicks[nick] = 0
        if OP_BOT in self.nicks.keys():
            self.msg(params[2], "ciao " + OP_BOT)

    def irc_ERR_NICKNAMEINUSE (self, prefix, params):
       self.nickname = self.nickname + "_"
       self.setNick(self.nickname)

    def privmsg(self, user, channel, message):
        user = user.split('!')[0]
	if channel != self.nickname:
           for regexp in RE_CONV:
              rexp = re.sub('(?i)\$bot', self.nickname, regexp)
              try:
              	if re.search(rexp, message, re.IGNORECASE):
                  msg = re.sub('(?i)\$nick', user, RE_CONV[regexp])
                  self.msg(channel, re.sub('(?i)\$bot', self.nickname, msg))
              except:
                  print "ERRORE PORC!"
           for regexp in RE_CONV_ME:
              rexp = re.sub('(?i)\$bot', self.nickname, regexp)
              if re.search(rexp, message, re.IGNORECASE):
                  msg =re.sub('(?i)\$nick', user, RE_CONV_ME[regexp])
                  self.me(channel, re.sub('(?i)\$bot', self.nickname, msg))	  
           if re.search(self.nickname, message, re.IGNORECASE):
               if re.search('aiutami|help', message, re.IGNORECASE):
                   self.msg(channel, self.polygen('if')[:440].rsplit('.',1)[0] + '.')
               elif re.search('spot', message, re.IGNORECASE):
                   self.msg(channel, self.polygen('poster').replace('\n',''))
               elif re.search('consigli', message, re.IGNORECASE):
                   msg = self.polygen('kamasutra')
                   m = re.search('(?i)\<strong\> (.*) \</strong\>', msg)
                   message =  m.groups()[0]
                   m = re.search('\</strong\>.*\n.*\n+(.*)', msg)
                   message += ': ' + m.groups()[0]
                   self.msg(channel, message) 

        else:
           if re.match('(?i) *LIST( |$)', message):
	       self.msg(user, "MESSAGE: ")
               for i in RE_CONV:
                   self.msg(user, i + " -> " + RE_CONV[i])
	       self.msg(user, "/ME: ")
               for i in RE_CONV_ME:
                   self.msg(user, i + " -> " + RE_CONV_ME[i])

           m = re.match('(?i) *ADD +"(.*)".*,.*"(.*)"', message)
           if m:
               RE_CONV[m.groups()[0]] = m.groups()[1]
               for i in RE_CONV:
                   self.msg(user, i + " -> " + RE_CONV[i])
	       log.info(user + ": "+ message)
               save()

           m = re.match('(?i) *DEL +"(.*)".*', message)
           if m:
               del RE_CONV[m.groups()[0]]
               for i in RE_CONV:
                   self.msg(user, i + " -> " + RE_CONV[i])
	       log.info(user + ": "+ message)
               save()

           m = re.match('(?i) *ADDME +"(.*)".*,.*"(.*)"', message)
           if m:
               RE_CONV_ME[m.groups()[0]] = m.groups()[1]
               for i in RE_CONV_ME:
                   self.msg(user, i + " -> " + RE_CONV_ME[i])
	       log.info(user + ": "+ message)
               save()

           m = re.match('(?i) *DELME +"(.*)".*', message)
           if m:
               del RE_CONV_ME[m.groups()[0]]
               for i in RE_CONV_ME:
                   self.msg(user, i + " -> " + RE_CONV_ME[i])
	       log.info(user + ": "+ message)
               save()

           m = re.match('(?i) *NICK +"(.*)".*', message)
           if m:
               BOT_NICKNAME = m.groups()[0]
               self.nickname = BOT_NICKNAME
               self.setNick(self.nickname)
               save()

           m = re.match('(?i) *TALK +(.*)', message)
           if m:
               msg = m.groups()[0].split(' ',1)
               self.msg(msg[0], msg[1])

           m = re.match('(?i) *ADDOP +(.*)', message)
           if m:
               for nick in m.groups()[0].split(' '):
                   MASTER_USER.append(nick)
                   self.msg(user, str(MASTER_USER))
                   save()
 
           if re.match('(?i) *HELP *.*', message):
               self.msg(user.split('!')[0], 'TALK #channel message | LIST | ADD "reg-exp", "message" | DEL "regexp" | ADDME "reg-exp", "message" | DELME "regexp" | NICK "NEW NICK"')
               self.msg(user.split('!')[0], 'In message you can use $nick. $nick will replace with the name to sender.')
               self.msg(user.split('!')[0], 'In message and regexp you can use $bot. $bot will replace with the name to bot.')

    def modeChanged (self, user, channel, set, modes, args):
        toOp = False
	for i in range(len(args)):
            if modes[i] == 'o':
                user = user.split('!')[0]
                self.op = set
                if set:
                    if args[i] == self.nickname:
                        self.msg(channel, "Grazie " + user + " saro' il vostro schiavo per il resto della mia vita.")
                        toOp = True
                    self.nicks[args[i]] = 1
                else:
                    if args[i] == self.nickname:
                        self.msg(channel, "Grazie " + user + " non sono degno di essere il vostro schiavo. La mia vita e' nelle vostre mani!")
                    self.nicks[args[i]] = 0
	if toOp:
            for nick in self.nicks:
                if self.nicks[nick] == 0  and nick != BOT_NICKNAME:
                    self.opOnlyMaster(nick, channel)

    def polygen (self, name):
        filename = '/usr/share/polygen/ita/' + name + '.grm'
        prog = os.popen("polygen " + filename)
	return prog.read()
    
    def ctcpUnknownQuery (self, user, channel, tag, data):
         pass

class SlaveBotFactory(protocol.ClientFactory):

    protocol = SlaveBot

    def __init__(self, channel):
         self.channel = channel

    def clientConnectionLost(self, connector, reason):
        log.critical('Connection lost: %s', reason)
        connector.connect()

    def clientConnectionFailed(self, connector, reason):
        log.critical('Connection failed: %s', reason)
        connector.connect()

if __name__ == '__main__':

    for server in SERVER:
         f = SlaveBotFactory(server['channel'])
	 if server['ssl'] == 1:
             contextFactory = ssl.ClientContextFactory() 
             reactor.connectSSL(server['name'], server['port'], f, contextFactory)
         else:
             reactor.connectTCP(server['name'], server['port'], f)
    reactor.run()