2012-08-14 03:51:55 +00:00
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
|
|
|
# Nemubot is a modulable IRC bot, built around XML configuration files.
|
|
|
|
# Copyright (C) 2012 Mercier Pierre-Olivier
|
|
|
|
#
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
|
|
# (at your option) any later version.
|
|
|
|
#
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
# GNU Affero General Public License for more details.
|
|
|
|
#
|
|
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
2012-08-16 02:32:47 +00:00
|
|
|
from datetime import datetime
|
2012-08-23 16:20:45 +00:00
|
|
|
from queue import Queue
|
2012-08-16 02:32:47 +00:00
|
|
|
import threading
|
|
|
|
|
2012-08-27 22:27:02 +00:00
|
|
|
from botcaps import BotCaps
|
2012-08-23 16:20:45 +00:00
|
|
|
from consumer import Consumer
|
2012-08-14 04:56:28 +00:00
|
|
|
import event
|
2012-08-27 22:27:02 +00:00
|
|
|
from networkbot import NetworkBot
|
2012-08-14 03:51:55 +00:00
|
|
|
from server import Server
|
|
|
|
|
2012-08-30 23:36:23 +00:00
|
|
|
ID_letters = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|
|
|
|
2012-08-27 22:27:02 +00:00
|
|
|
class Bot(BotCaps):
|
2012-08-14 03:51:55 +00:00
|
|
|
def __init__(self, servers=dict(), modules=dict(), mp=list()):
|
2012-08-27 22:27:02 +00:00
|
|
|
BotCaps.__init__(self, 3.2, "3.2-dev")
|
2012-08-14 03:51:55 +00:00
|
|
|
|
2012-08-27 22:27:02 +00:00
|
|
|
# Keep global context: servers and modules
|
2012-08-14 03:51:55 +00:00
|
|
|
self.servers = servers
|
|
|
|
self.modules = modules
|
|
|
|
|
2012-08-27 22:27:02 +00:00
|
|
|
# Context paths
|
2012-08-14 03:51:55 +00:00
|
|
|
self.modules_path = mp
|
2012-08-23 16:20:45 +00:00
|
|
|
self.datas_path = './datas/'
|
2012-08-14 03:51:55 +00:00
|
|
|
|
2012-08-27 22:27:02 +00:00
|
|
|
# Events
|
2012-08-23 16:20:45 +00:00
|
|
|
self.events = list()
|
2012-08-16 02:32:47 +00:00
|
|
|
self.event_timer = None
|
|
|
|
|
2012-08-27 22:27:02 +00:00
|
|
|
# Other known bots, making a bots network
|
|
|
|
self.network = dict()
|
|
|
|
|
|
|
|
# Messages to be treated
|
2012-08-23 16:20:45 +00:00
|
|
|
self.msg_queue = Queue()
|
|
|
|
self.msg_thrd = list()
|
|
|
|
self.msg_thrd_size = -1
|
|
|
|
|
2012-08-16 02:32:47 +00:00
|
|
|
|
|
|
|
def add_event(self, evt):
|
2012-08-30 23:36:23 +00:00
|
|
|
"""Register an event and return its identifiant for futur update"""
|
|
|
|
# Find ID
|
|
|
|
now = datetime.now()
|
|
|
|
evt.id = "%d%c%d%d%c%d%d%c%d" % (now.year, ID_letters[now.microsecond % 52], now.month, now.day, ID_letters[now.microsecond % 42], now.hour, now.minute, ID_letters[now.microsecond % 32], now.second)
|
|
|
|
|
2012-08-16 02:32:47 +00:00
|
|
|
# Add the event in place
|
|
|
|
t = evt.current
|
|
|
|
i = 0
|
|
|
|
for i in range(0, len(self.events)):
|
|
|
|
if self.events[i].current > t:
|
|
|
|
break
|
|
|
|
self.events.insert(i, evt)
|
|
|
|
if i == 0:
|
|
|
|
self.update_timer()
|
|
|
|
|
2012-08-30 23:36:23 +00:00
|
|
|
return evt.id
|
|
|
|
|
|
|
|
def del_event(self, id):
|
|
|
|
"""Find and remove an event from list"""
|
|
|
|
for evt in self.events:
|
|
|
|
if evt.id == id:
|
|
|
|
self.events.remove(evt)
|
|
|
|
return True
|
|
|
|
return False
|
|
|
|
|
2012-08-16 02:32:47 +00:00
|
|
|
def update_timer(self):
|
2012-08-16 02:53:46 +00:00
|
|
|
"""Relaunch the timer to end with the closest event"""
|
2012-08-16 02:32:47 +00:00
|
|
|
# Reset the timer if this is the first item
|
|
|
|
if self.event_timer is not None:
|
|
|
|
self.event_timer.cancel()
|
|
|
|
if len(self.events) > 0:
|
|
|
|
#print ("Update timer, next in", self.events[0].time_left.seconds,
|
|
|
|
# "seconds")
|
|
|
|
if datetime.now() >= self.events[0].current:
|
|
|
|
self.end_timer()
|
|
|
|
else:
|
|
|
|
self.event_timer = threading.Timer(
|
|
|
|
self.events[0].time_left.seconds + 1, self.end_timer)
|
|
|
|
self.event_timer.start()
|
|
|
|
#else:
|
|
|
|
# print ("Update timer: no timer left")
|
|
|
|
|
|
|
|
def end_timer(self):
|
2012-08-16 02:53:46 +00:00
|
|
|
"""Function called at the end of the timer"""
|
2012-08-16 02:32:47 +00:00
|
|
|
#print ("end timer")
|
|
|
|
while len(self.events)>0 and datetime.now() >= self.events[0].current:
|
|
|
|
#print ("end timer: while")
|
|
|
|
evt = self.events.pop(0)
|
|
|
|
evt.launch_check()
|
|
|
|
if evt.next is not None:
|
|
|
|
self.add_event(evt)
|
|
|
|
self.update_timer()
|
2012-08-14 04:56:28 +00:00
|
|
|
|
2012-08-14 03:51:55 +00:00
|
|
|
|
|
|
|
def addServer(self, node, nick, owner, realname):
|
|
|
|
"""Add a new server to the context"""
|
|
|
|
srv = Server(node, nick, owner, realname)
|
|
|
|
if srv.id not in self.servers:
|
|
|
|
self.servers[srv.id] = srv
|
|
|
|
if srv.autoconnect:
|
|
|
|
srv.launch(self)
|
|
|
|
return True
|
|
|
|
else:
|
|
|
|
return False
|
|
|
|
|
|
|
|
|
|
|
|
def add_module(self, module):
|
|
|
|
"""Add a module to the context, if already exists, unload the
|
|
|
|
old one before"""
|
|
|
|
# Check if the module already exists
|
|
|
|
for mod in self.modules.keys():
|
|
|
|
if self.modules[mod].name == module.name:
|
|
|
|
self.unload_module(self.modules[mod].name)
|
|
|
|
break
|
|
|
|
|
|
|
|
self.modules[module.name] = module
|
|
|
|
return True
|
|
|
|
|
|
|
|
|
|
|
|
def add_modules_path(self, path):
|
|
|
|
"""Add a path to the modules_path array, used by module loader"""
|
|
|
|
# The path must end by / char
|
|
|
|
if path[len(path)-1] != "/":
|
|
|
|
path = path + "/"
|
|
|
|
|
|
|
|
if path not in self.modules_path:
|
|
|
|
self.modules_path.append(path)
|
|
|
|
return True
|
|
|
|
|
|
|
|
return False
|
|
|
|
|
|
|
|
|
|
|
|
def unload_module(self, name, verb=False):
|
|
|
|
"""Unload a module"""
|
|
|
|
if name in self.modules:
|
|
|
|
self.modules[name].save()
|
|
|
|
if hasattr(self.modules[name], "unload"):
|
|
|
|
self.modules[name].unload()
|
|
|
|
# Remove from the dict
|
|
|
|
del self.modules[name]
|
|
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
|
|
|
2012-08-23 16:20:45 +00:00
|
|
|
def receive_message(self, srv, raw_msg, private = False):
|
|
|
|
"""Queued the message for treatment"""
|
2012-08-26 16:00:56 +00:00
|
|
|
self.msg_queue.put_nowait((srv, raw_msg, datetime.now(), private))
|
2012-08-23 16:20:45 +00:00
|
|
|
|
|
|
|
# Launch a new thread if necessary
|
|
|
|
if self.msg_queue.qsize() > self.msg_thrd_size:
|
|
|
|
c = Consumer(self)
|
|
|
|
self.msg_thrd.append(c)
|
|
|
|
c.start()
|
|
|
|
self.msg_thrd_size += 2
|
|
|
|
|
|
|
|
|
2012-08-27 22:27:02 +00:00
|
|
|
def add_networkbot(self, srv, dest, dcc=None):
|
|
|
|
"""Append a new bot into the network"""
|
|
|
|
id = srv.id + "/" + dest
|
|
|
|
if id not in self.network:
|
|
|
|
self.network[id] = NetworkBot(self, srv, dest, dcc)
|
|
|
|
return self.network[id]
|
|
|
|
|
|
|
|
|
2012-08-14 03:51:55 +00:00
|
|
|
def quit(self, verb=False):
|
|
|
|
"""Save and unload modules and disconnect servers"""
|
2012-08-16 02:53:46 +00:00
|
|
|
if self.event_timer is not None:
|
|
|
|
if verb: print ("Stop the event timer...")
|
|
|
|
self.event_timer.cancel()
|
|
|
|
|
2012-08-14 03:51:55 +00:00
|
|
|
if verb: print ("Save and unload all modules...")
|
|
|
|
k = list(self.modules.keys())
|
|
|
|
for mod in k:
|
|
|
|
self.unload_module(mod, verb)
|
|
|
|
|
|
|
|
if verb: print ("Close all servers connection...")
|
|
|
|
k = list(self.servers.keys())
|
|
|
|
for srv in k:
|
|
|
|
self.servers[srv].disconnect()
|
|
|
|
|
|
|
|
|
|
|
|
def hotswap(bak):
|
|
|
|
return Bot(bak.servers, bak.modules, bak.modules_path)
|
|
|
|
|
|
|
|
def reload():
|
|
|
|
import imp
|
|
|
|
|
|
|
|
import prompt.builtins
|
|
|
|
imp.reload(prompt.builtins)
|
|
|
|
|
2012-08-14 04:56:28 +00:00
|
|
|
import event
|
|
|
|
imp.reload(event)
|
|
|
|
|
2012-08-14 03:51:55 +00:00
|
|
|
import hooks
|
|
|
|
imp.reload(hooks)
|
|
|
|
|
|
|
|
import xmlparser
|
|
|
|
imp.reload(xmlparser)
|
|
|
|
import xmlparser.node
|
|
|
|
imp.reload(xmlparser.node)
|
|
|
|
|
|
|
|
import importer
|
|
|
|
imp.reload(importer)
|
|
|
|
|
|
|
|
import server
|
|
|
|
imp.reload(server)
|
|
|
|
|
|
|
|
import channel
|
|
|
|
imp.reload(channel)
|
|
|
|
|
|
|
|
import DCC
|
|
|
|
imp.reload(DCC)
|
|
|
|
|
|
|
|
import message
|
|
|
|
imp.reload(message)
|