alfredirc.py

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

import sys
import socket
import string
import os
import MySQLdb
from utils import *
import logging

## DB STUFF FUNCTIONS ###

def connect():
	return MySQLdb.connect(host=DBHOST,
	user=DBUSER,
	passwd=DBPASS,
	db=DBNAME)
def execute_query(db,query):
	cur = db.cursor()
	cur.execute(query)
	return cur.fetchall()

def close_db(db):
	db.close

### IRC FUNCTIONS ###

def send_pong(msg):
	logger.info('[ INFO ] Sending pong %s...' % msg)
	s.send(bytes('PONG %s\r\n' % msg))

def send_notice(user,msg):
	logger.info('[ INFO ] Sending notice %s %s...' % (user,msg))
	s.send(bytes('NOTICE %s :%s\r\n' % (user, msg)))

def send_message(chan, msg):
	logger.info('[ INFO ] Sending message %s %s...' % (chan,msg))
	s.send(bytes('PRIVMSG %s :%s\r\n' % (chan, msg)))

def set_mode(chan,nick,mode):
	logger.info('[ INFO ] Setting mode %s %s %s...' % (chan,nick,mode))
	s.send(bytes('MODE %s %s %s\r\n' % (chan,mode,nick)))

def send_kick(chan,nick):
	logger.info('[ INFO ] Kick user %s  from %s' % (chan,nick))
	s.send(bytes('KICK %s %s\r\n' % (chan,nick)))

def send_nick(nick):
	logger.info('[ INFO ] Sending nick %s ... ' % nick)
	s.send(bytes('NICK %s\r\n' % nick))

def send_pass(password):
	logger.info('[ INFO ] Sending password %s ...' % password)
	s.send(bytes('PASS %s\r\n' % password))

def join_channel(chan):
	logger.info('[ INFO ] Sending join %s ...' % chan)
	s.send(bytes('JOIN %s\r\n' % chan))

def part_channel(chan):
	logger.info('[ INFO ] Sending part %s ...' % chan)
	s.send(bytes('PART %s\r\n' % chan))

def quit():
	logger.info('[ INFO ] Sending quit...')
	s.send(bytes('QUIT %s\r\n'))

### HELPING 
def get_sender(msg):
	result = ""
	for char in msg:
		if char == "!":
			break
		if char != ":":
			result += char
	return result


def get_message(msg):
	result = ""
	i = 3
	length = len(msg)
	while i < length:
		result += msg[i] + " "
		i += 1
	result = result.lstrip(':')
	return result

def is_private(msg):
	return not msg[2].startswith('#')

def is_command(msg):
	return msg.startswith('!')

def get_operators():
	logger.info('[ INFO ] Getting operators...')
	operators=[]
	for user in access:
		if user['level']==OPERATOR_LEVEL
			operator.append(user['nick'])
	return operators


def execute_command(cmd,avaiable_commands):
	if cmd in avaiable_commands:
		logger.info('[ INFO ] Executing command %s' % cmd)
		c=avaiable_commands[cmd]
		s="| "
		if c[2] is not None:
			send_message(CHAN,'%s' % c[2])
		for co in avaiable_commands:
			if avaiable_commands[co][0]==c[1]:
				s+="!%s | " %co
		if s != '| ':
			send_message(CHAN,s)


def parse_private_message(msg,user):
	if len(msg) >= 1:
		print ('Sender %s'  % user)
		operators=get_operators()
		if user in operators:
			msg = msg.split(' ')
			options = {'!die' : die_handler,
			'!testdb':db_handler}
			if msg[0] in options:
				send_message(user,options[msg[0]]())
		else:
			logger.warn('[ WARN ] Private command from unauthorized user %s  %s' % (user,msg))
			send_message(user,'Not autorized!')

###### 

### COMMANDS HANDLER ######

def die_handler():
	logger.warn('[ WARN ] Received die command!')
	send_message(CHAN, 'Bye folks')
	part_channel(CHAN)
	logger.warn('[ WARN ] Shooting down!')
	quit()
	sys.exit(0)


def db_handler():
	db=connect()
	res=execute_query(db,'SELECT * FROM %s' % DBTABLE)
	logger.info('[ INFO ] Testing db Result: %s' % res)
	close_db(db)
	return res

#####


CONF_FILE='alfred.conf'
ACCESS_FILE='access.json'
LOG_FILE='bog.log'


OPERATOR_LEVEL='o'
VOICE_LEVEL='v'

logging.basicConfig(filename=LOG_FILE,format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
	level=logging.INFO)
logger = logging.getLogger(__name__)

logger.info('[ INFO ] Configuration file: "%s"' % CONF_FILE)
logger.info('[ INFO ] Loading configuration...')

conf=load_configuration(CONF_FILE)

logger.info('[ INFO ] Access file: "%s"' % ACCESS_FILE)
logger.info('[ INFO ] Loading access...')
access=load_access_file(ACCESS_FILE)

DBHOST=conf['dbserver']
DBUSER=conf['dbuser']
DBPASS=conf['dbpass']
DBNAME=conf['dbname']
DBTABLE=conf['dbtable']

HOST=conf['server']
PORT=int(conf['port'])
NICK=conf['nick']
IDENT=conf['ident']
REALNAME=conf['real']
CHAN=conf['chan']

logger.info('[ INFO ] Loading commands from db...')

commands=command_db()
commands_map={}
for c in commands:
	print c[0]
	commands_map[c[0]]=(c[1],c[2],c[3])

readbuffer=""

logger.info('[ INFO ] Opening socket...')
s=socket.socket( )
logger.info('[ INFO ] Connection to irc server %s:%s ...' % (HOST,PORT))
s.connect((HOST, PORT))
logger.info('[ INFO ] Sending nick "%s" to server ' % NICK)
send_nick(NICK)
logger.info('[ INFO ] Sendind ident info to server  ident: %s realname: %s'% (IDENT,REALNAME))
s.send("USER %s %s bla :%s\r\n" % (IDENT, HOST, REALNAME))

logger.info('[ INFO ] Joining channel %s'% CHAN)
join_channel(CHAN)


while 1:
	readbuffer=readbuffer+s.recv(1024).decode('UTF-8')
	temp=string.split(readbuffer, "\n")
	readbuffer=temp.pop()


	for line in temp:
		single_line=string.rstrip(line)
		single_line=string.split(single_line)
	if len(single_line)>=1:
		if single_line[0]=="PING" :
					logger.info('[ INFO ] Ping received!')
					send_pong(single_line[1])
					print('PONG!\n')
		if single_line[1] == 'PRIVMSG':
			sender=get_sender(single_line[0])
			message = get_message(single_line)
			if is_private(single_line):
				logger.info('[ INFO ] New private message from %s text:%s' % (sender,message))
				parse_private_message(message,sender)
			else:
				logger.info('[ INFO ] New message in channel from %s text:%s' % (sender,message))
				if is_command(message):
					cmd = message[1:].strip()
					logger.info('[ INFO ] Message is a command: %s' % cmd)
					execute_command(cmd,commands_map)