145 lines
4.5 KiB
Python
145 lines
4.5 KiB
Python
import time
|
|
import logging
|
|
import sys
|
|
import re
|
|
import os
|
|
import Queue
|
|
import collections
|
|
|
|
from core import config, irc, main, loader
|
|
|
|
|
|
def clean_name(n):
|
|
"""strip all spaces and capitalization"""
|
|
return re.sub('[^A-Za-z0-9_]+', '', n.replace(" ", "_"))
|
|
|
|
|
|
class Bot(object):
|
|
def __init__(self):
|
|
# basic variables
|
|
self.start_time = time.time()
|
|
self.running = True
|
|
|
|
# set up config and logging
|
|
self.setup()
|
|
self.logger.debug("Bot setup completed.")
|
|
|
|
# start IRC connections
|
|
self.connections = {}
|
|
self.connect()
|
|
|
|
# run plugin loader
|
|
self.plugins = collections.defaultdict(list)
|
|
self.threads = {}
|
|
self.loader = loader.PluginLoader(self)
|
|
|
|
|
|
def stop(self, reason=None):
|
|
"""quits all networks and shuts the bot down"""
|
|
self.logger.info("Stopping bot.")
|
|
self.running = False
|
|
|
|
# wait for the bot loop to stop
|
|
time.sleep(1)
|
|
self.config.observer.stop()
|
|
self.logger.debug("Stopping config reloader.")
|
|
|
|
self.loader.stop()
|
|
self.logger.debug("Stopping plugin loader.")
|
|
|
|
for name, connection in self.connections.iteritems():
|
|
# TODO: end connections properly
|
|
self.logger.debug("({}) Closing connection.".format(name))
|
|
|
|
if reason:
|
|
connection.cmd("QUIT", [reason])
|
|
else:
|
|
connection.cmd("QUIT")
|
|
|
|
connection.stop()
|
|
|
|
self.logger.debug("Logging engine stopped")
|
|
logging.shutdown()
|
|
sys.exit()
|
|
|
|
|
|
def loop(self):
|
|
"""recieves input from the IRC engine and processes it"""
|
|
|
|
for conn in self.connections.itervalues():
|
|
try:
|
|
incoming = conn.parsed_queue.get_nowait()
|
|
main.main(self, conn, incoming)
|
|
except Queue.Empty:
|
|
pass
|
|
|
|
# if no messages are in the incoming queue, sleep
|
|
while all(connection.parsed_queue.empty() for connection in self.connections.itervalues()):
|
|
time.sleep(.1)
|
|
|
|
|
|
def connect(self):
|
|
"""connect to all the networks defined in the bot config"""
|
|
for conf in self.config['connections']:
|
|
# strip all spaces and capitalization from the connection name
|
|
name = clean_name(conf['name'])
|
|
nick = conf['nick']
|
|
server = conf['connection']['server']
|
|
port = conf['connection'].get('port', 6667)
|
|
|
|
self.logger.debug("({}) Creating connection to {}.".format(name, server))
|
|
|
|
if conf['connection'].get('ssl'):
|
|
self.connections[name] = irc.SSLIRC(name, server, nick, conf = conf,
|
|
port = port, channels = conf['channels'],
|
|
ignore_certificate_errors=conf['connection'].get('ignore_cert', True))
|
|
self.logger.debug("({}) Created SSL connection.".format(name))
|
|
else:
|
|
self.connections[name] = irc.IRC(name, server, nick, conf = conf,
|
|
port = port, channels = conf['channels'])
|
|
self.logger.debug("({}) Created connection.".format(name))
|
|
|
|
|
|
def setup(self):
|
|
"""create the logger and config objects"""
|
|
# logging
|
|
self.logger = self.new_logger()
|
|
self.logger.debug("Logging engine started.")
|
|
|
|
# data folder
|
|
self.data_dir = os.path.abspath('data')
|
|
if not os.path.exists(self.data_dir):
|
|
self.logger.debug("Data folder not found, creating.")
|
|
os.mkdir(self.data_dir)
|
|
self.logger.debug("Created data folder.")
|
|
|
|
# config
|
|
self.config = config.Config(self.logger)
|
|
self.logger.debug("Config object created.")
|
|
|
|
|
|
def new_logger(self):
|
|
"""create and return a new logger object"""
|
|
# create logger
|
|
logger = logging.getLogger("cloudbot")
|
|
logger.setLevel(logging.DEBUG)
|
|
|
|
# add a file handler
|
|
log_name = "bot.log"
|
|
fh = logging.FileHandler(log_name)
|
|
fh.setLevel(logging.DEBUG)
|
|
|
|
# stdout handler
|
|
sh = logging.StreamHandler()
|
|
sh.setLevel(logging.INFO)
|
|
|
|
# create a formatter and set the formatter for the handler.
|
|
frmt = logging.Formatter('%(asctime)s [%(levelname)s] %(message)s')
|
|
fh.setFormatter(frmt)
|
|
simple_frmt = logging.Formatter('[%(levelname)s] %(message)s')
|
|
sh.setFormatter(simple_frmt)
|
|
|
|
# add the Handlers to the logger
|
|
logger.addHandler(fh)
|
|
logger.addHandler(sh)
|
|
return logger
|