Brian S. Stephan
b9ef46606d
this indicates that none of the modules could actually do anything with that text, so whatever happened probably didn't fire any modules (and if it did, they didn't provide return text anyway)
141 lines
5.3 KiB
Python
141 lines
5.3 KiB
Python
"""
|
|
Alias - have internal shortcuts to commands
|
|
Copyright (C) 2010 Brian S. Stephan
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU 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 General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
"""
|
|
|
|
from ConfigParser import NoOptionError, NoSectionError
|
|
import re
|
|
|
|
from extlib import irclib
|
|
|
|
from Module import Module
|
|
|
|
class Alias(Module):
|
|
"""
|
|
Allows for commands to be aliased as !command, circumventing bot addressing stuff.
|
|
"""
|
|
|
|
def on_pubmsg(self, connection, event):
|
|
"""
|
|
Handle pubmsg events. Does some variable setup and initial sanity checking before
|
|
calling Module.do, which should be implemented by subclasses and what can be
|
|
ultimately responsible for the work.
|
|
|
|
Reimplemented so that the recursion stuff doesn't happen (in case we want to
|
|
write an alias that would use recursion).
|
|
"""
|
|
|
|
nick = irclib.nm_to_n(event.source())
|
|
userhost = irclib.nm_to_uh(event.source())
|
|
replypath = event.target()
|
|
what = event.arguments()[0]
|
|
|
|
admin_unlocked = False
|
|
|
|
try:
|
|
if userhost == self.config.get('dr.botzo', 'admin_userhost'):
|
|
admin_unlocked = True
|
|
except NoOptionError: pass
|
|
|
|
# only do commands if the bot has been addressed directly
|
|
addressed_pattern = '^' + connection.get_nickname() + '[:,]?\s+'
|
|
addressed_re = re.compile(addressed_pattern)
|
|
|
|
need_prefix = True
|
|
try:
|
|
need_prefix = self.config.getboolean(self.__class__.__name__, 'meta.pubmsg_needs_bot_prefix')
|
|
except NoOptionError: pass
|
|
except NoSectionError: pass
|
|
|
|
if not addressed_re.match(what) and need_prefix:
|
|
return
|
|
else:
|
|
what = addressed_re.sub('', what)
|
|
|
|
# try reloading
|
|
self.reload(connection, event, nick, userhost, replypath, what, admin_unlocked)
|
|
|
|
self.do(connection, event, nick, userhost, replypath, what, admin_unlocked)
|
|
|
|
def on_privmsg(self, connection, event):
|
|
"""
|
|
Handle privmsg events. Does some variable setup and initial sanity checking before
|
|
calling Module.do, which should be implemented by subclasses and what can be
|
|
ultimately responsible for the work.
|
|
|
|
Reimplemented so that the recursion stuff doesn't happen (in case we want to
|
|
write an alias that would use recursion).
|
|
"""
|
|
|
|
nick = irclib.nm_to_n(event.source())
|
|
userhost = irclib.nm_to_uh(event.source())
|
|
replypath = nick
|
|
what = event.arguments()[0]
|
|
|
|
admin_unlocked = False
|
|
|
|
try:
|
|
if userhost == self.config.get('dr.botzo', 'admin_userhost'):
|
|
admin_unlocked = True
|
|
except NoOptionError: pass
|
|
|
|
self.do(connection, event, nick, userhost, replypath, what, admin_unlocked)
|
|
|
|
def do(self, connection, event, nick, userhost, replypath, what, admin_unlocked):
|
|
"""
|
|
See if there is an alias ("!command") in the text, and if so, translate it into
|
|
an internal bot command and run it.
|
|
"""
|
|
|
|
# first see if the aliases are being directly manipulated via add/remove
|
|
whats = what.split(' ')
|
|
try:
|
|
if whats[0] == '!alias' and whats[1] == 'add' and len(whats) >= 4:
|
|
if not self.config.has_section(self.__class__.__name__):
|
|
self.config.add_section(self.__class__.__name__)
|
|
|
|
self.config.set(self.__class__.__name__, whats[2], ' '.join(whats[3:]))
|
|
replystr = 'added alias ' + whats[2]
|
|
return self.reply(connection, replypath, replystr)
|
|
if whats[0] == '!alias' and whats[1] == 'remove' and len(whats) >= 3:
|
|
if not self.config.has_section(self.__class__.__name__):
|
|
self.config.add_section(self.__class__.__name__)
|
|
|
|
if self.config.remove_option(self.__class__.__name__, whats[2]):
|
|
replystr = 'removed alias ' + whats[2]
|
|
return self.reply(connection, replypath, replystr)
|
|
except NoOptionError: pass
|
|
except NoSectionError: pass
|
|
|
|
# now that that's over, try doing alias work
|
|
try:
|
|
alias_list = self.config.options(self.__class__.__name__)
|
|
self.remove_metaoptions(alias_list)
|
|
|
|
for alias in alias_list:
|
|
alias_re = re.compile(alias)
|
|
if alias_re.match(what):
|
|
command = re.sub(alias, self.config.get(self.__class__.__name__, alias), what)
|
|
reply = self.try_recursion(connection, event, nick, userhost, None, command, admin_unlocked)
|
|
if not reply == command:
|
|
return self.reply(connection, replypath, reply)
|
|
|
|
except NoOptionError: pass
|
|
except NoSectionError: pass
|
|
|
|
# vi:tabstop=4:expandtab:autoindent
|
|
# kate: indent-mode python;indent-width 4;replace-tabs on;
|