294 lines
10 KiB
Python
294 lines
10 KiB
Python
# -*- Encoding: utf-8 -*-
|
|
###
|
|
# Copyright (c) 2008-2010 Terence Simpson
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of version 2 of the GNU General Public License as
|
|
# published by the Free Software Foundation.
|
|
#
|
|
# 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.
|
|
#
|
|
###
|
|
|
|
import supybot.utils as utils
|
|
from supybot.commands import *
|
|
import supybot.plugins as plugins
|
|
import supybot.ircutils as ircutils
|
|
import supybot.ircmsgs as ircmsgs
|
|
import supybot.callbacks as callbacks
|
|
import supybot.ircutils as ircutils
|
|
import supybot.ircdb as ircdb
|
|
import supybot.conf as conf
|
|
import os
|
|
import re
|
|
import time
|
|
import packages
|
|
reload(packages)
|
|
|
|
def get_user(msg):
|
|
try:
|
|
user = ircdb.users.getUser(msg.prefix)
|
|
except:
|
|
return False
|
|
return user
|
|
|
|
_stripNickChars = """!"#$%&'()*+,./:;<=>?@~"""
|
|
def stripNick(nick):
|
|
while nick and nick[-1] in _stripNickChars:
|
|
nick = nick[:-1]
|
|
return nick
|
|
|
|
## Taken from Encyclopedia ##
|
|
# Repeat filtering message queue
|
|
msgcache = {}
|
|
def queue(irc, to, msg):
|
|
now = time.time()
|
|
for m in msgcache.keys():
|
|
if msgcache[m] < now - 30:
|
|
msgcache.pop(m)
|
|
for m in msgcache:
|
|
if m[0] == irc and m[1] == to:
|
|
oldmsg = m[2]
|
|
if msg == oldmsg or oldmsg.endswith(msg):
|
|
break
|
|
if msg.endswith(oldmsg):
|
|
msg = msg[:-len(oldmsg)] + 'please see above'
|
|
else:
|
|
msgcache[(irc, to, msg)] = now
|
|
irc.queueMsg(ircmsgs.privmsg(to, msg))
|
|
|
|
class PackageInfo(callbacks.Plugin):
|
|
"""Lookup package information via apt-cache/apt-file"""
|
|
threaded = True
|
|
space_re = re.compile(r' *')
|
|
|
|
def __init__(self, irc):
|
|
self.__parent = super(PackageInfo, self)
|
|
self.__parent.__init__(irc)
|
|
self.Apt = packages.Apt(self)
|
|
|
|
def callPrecedence(self, irc):
|
|
before = []
|
|
for cb in irc.callbacks:
|
|
if cb.name() == 'IRCLogin':
|
|
before.append(cb)
|
|
return (before, [])
|
|
|
|
def __getRelease(self, irc, release, channel, doError=True):
|
|
if release:
|
|
release = release.strip()
|
|
defaultRelease = self.registryValue("defaultRelease", channel)
|
|
if not defaultRelease:
|
|
if doError:
|
|
irc.error("'supybot.plugins.PackageInfo.defaultRelease' is not set")
|
|
return (None, None)
|
|
if not release:
|
|
return (defaultRelease, None)
|
|
(release, rest) = (release.split(' ', 1) + [None])[:2]
|
|
if release[0] in ('|', '>'):
|
|
return (defaultRelease, "%s %s" % (release, rest))
|
|
return (release, rest)
|
|
|
|
def __getChannel(self, channel):
|
|
return ircutils.isChannel(channel) and channel or None
|
|
|
|
def __getReplyChars(self, channel):
|
|
prefix_chars = list(self.registryValue("prefixchar", channel))
|
|
address_chars = list(str( conf.supybot.reply.whenAddressedBy.chars() ))
|
|
if channel:
|
|
address_chars = list(str( conf.supybot.reply.whenAddressedBy.chars.get(channel) ))
|
|
return tuple(set(prefix_chars + address_chars))
|
|
|
|
def __getCommand(self, text, channel):
|
|
reply_chars = self.__getReplyChars(channel)
|
|
my_commands = self.listCommands()
|
|
if text[0] in reply_chars:
|
|
text = text[1:]
|
|
return text.strip().lower().split(' ', 1)[0]
|
|
|
|
def real_info(self, irc, msg, args, package, release):
|
|
"""<package> [<release>]
|
|
|
|
Lookup information for <package>, optionally in <release>
|
|
"""
|
|
channel = self.__getChannel(msg.args[0])
|
|
reply_target = ircutils.replyTo(msg)
|
|
(release, rest) = self.__getRelease(irc, release, channel)
|
|
if not release:
|
|
return
|
|
reply = self.Apt.info(package, release)
|
|
if rest:
|
|
if rest[0] == '|':
|
|
try:
|
|
target = rest
|
|
while target[0] == '|':
|
|
target = target[1:].strip()
|
|
if target.lower() == "me":
|
|
target = msg.nick
|
|
queue(irc, reply_target, "%s: %s" % (target, reply))
|
|
return
|
|
except Exception, e:
|
|
self.log.info("PackageInfo: (info) Exception in pipe: %r" % e)
|
|
pass
|
|
elif rest[0] == '>':
|
|
try:
|
|
while rest[0] == '>':
|
|
rest = rest[1:].strip()
|
|
targets = [_ for _ in rest.split() if _] # Split and discard empty parts
|
|
target = stripNick(targets[0]) # Take the first "nick" and strip off bad chars
|
|
if target.lower() == "me":
|
|
target = msg.nick # redirect
|
|
if not target: # Throw error
|
|
raise Exception, 'No target'
|
|
queue(irc, target, "<%s> wants you to know: %s" % (msg.nick, reply))
|
|
return
|
|
except Exception, e:
|
|
self.log.info("PackageInfo: (info) Exception in redirect: %r" % e)
|
|
pass
|
|
|
|
queue(irc, reply_target, reply)
|
|
|
|
info = wrap(real_info, ['anything', optional('text')])
|
|
|
|
def real_find(self, irc, msg, args, package, release):
|
|
"""<package/filename> [<release>]
|
|
|
|
Search for <package> or, of that fails, find <filename>'s package(s).
|
|
Optionally in <release>
|
|
"""
|
|
channel = self.__getChannel(msg.args[0])
|
|
reply_target = ircutils.replyTo(msg)
|
|
(release, rest) = self.__getRelease(irc, release, channel)
|
|
if not release:
|
|
return
|
|
reply = self.Apt.find(package, release)
|
|
if rest:
|
|
if rest[0] == '|':
|
|
try:
|
|
target = rest
|
|
while target[0] == '|':
|
|
target = target[1:].strip()
|
|
if target.lower() == "me":
|
|
target = msg.nick
|
|
queue(irc, reply_target, "%s: %s" % (target, reply))
|
|
return
|
|
except Exception, e:
|
|
self.log.info("PackageInfo: (find) Exception in pipe: %r" % e)
|
|
pass
|
|
elif rest[0] == '>':
|
|
try:
|
|
while rest[0] == '>':
|
|
rest = rest[1:].strip()
|
|
targets = [_ for _ in rest.split() if _] # Split and discard empty parts
|
|
target = stripNick(targets[0]) # Take the first "nick" and strip off bad chars
|
|
if target.lower() == "me":
|
|
target = msg.nick # redirect
|
|
if not target: # Throw error
|
|
raise Exception, 'No target'
|
|
queue(irc, target, "<%s> wants you to know: %s" % (msg.nick, reply))
|
|
return
|
|
except Exception, e:
|
|
self.log.info("PackageInfo: (find) Exception in redirect: %r" % e)
|
|
pass
|
|
|
|
queue(irc, reply_target, reply)
|
|
|
|
find = wrap(real_find, ['anything', optional('text')])
|
|
|
|
def privmsg(self, irc, msg, user):
|
|
channel = self.__getChannel(msg.args[0])
|
|
text = self.space_re.subn(' ', msg.args[1].strip())[0]
|
|
if text[0] == self.registryValue("prefixchar"):
|
|
text = text[1:].strip()
|
|
if user and text[0] in list(conf.supybot.reply.whenAddressedBy.chars()):
|
|
return
|
|
(cmd, rest) = (text.split(' ', 1) + [None])[:2]
|
|
if cmd not in my_commands:
|
|
return
|
|
if not rest:
|
|
return
|
|
(term, rest) = (rest.split(' ', 1) + [None])[:2]
|
|
if cmd == "find":
|
|
self.real_find(irc, msg, [], term, rest)
|
|
else:
|
|
self.real_info(irc, msg, [], term, rest)
|
|
|
|
def chanmsg(self, irc, msg, user):
|
|
channel = self.__getChannel(msg.args[0])
|
|
text = self.space_re.subn(' ', msg.args[1].strip())[0]
|
|
my_commands = self.listCommands()
|
|
if text[0] != self.registryValue("prefixchar", channel):
|
|
return
|
|
text = text[1:]
|
|
(cmd, rest) = (text.split(' ', 1) + [None])[:2]
|
|
if cmd not in my_commands:
|
|
return
|
|
if not rest:
|
|
return
|
|
(term, rest) = (rest.split(' ', 1) + [None])[:2]
|
|
if cmd == "find":
|
|
self.real_find(irc, msg, [], term, rest)
|
|
else:
|
|
self.real_info(irc, msg, [], term, rest)
|
|
|
|
def doPrivmsg(self, irc, msg):
|
|
if chr(1) in msg.args[1]: # CTCP
|
|
return
|
|
if not msg.args[1]:
|
|
return
|
|
channel = self.__getChannel(msg.args[0])
|
|
if not self.registryValue("enabled", channel):
|
|
return
|
|
user = get_user(msg)
|
|
if channel:
|
|
self.chanmsg(irc, msg, user)
|
|
else:
|
|
if user:
|
|
return
|
|
self.privmsg(irc, msg, user)
|
|
|
|
def inFilter(self, irc, msg):
|
|
if msg.command != "PRIVMSG":
|
|
return msg
|
|
if not conf.supybot.defaultIgnore():
|
|
return msg
|
|
text = msg.args[1].strip()
|
|
if len(text) < 6:
|
|
return msg
|
|
user = get_user(msg)
|
|
channel = self.__getChannel(msg.args[0])
|
|
reply_chars = self.__getReplyChars(channel)
|
|
my_commands = self.listCommands()
|
|
cmd = self.__getCommand(text, channel)
|
|
|
|
if cmd not in my_commands:
|
|
return msg
|
|
|
|
if user:
|
|
if not channel and text[0] == self.registryValue("prefixchar"):
|
|
msg.args = (msg.args[0], text[1:])
|
|
return msg
|
|
|
|
if channel:
|
|
if text[0] not in reply_chars:
|
|
return msg
|
|
|
|
# if not hasattr(irc, 'reply'):
|
|
# irc = callbacks.ReplyIrcProxy(irc, msg)
|
|
# self.doPrivmsg(irc, msg)
|
|
else:
|
|
if text[1] in reply_chars:
|
|
msg.args = (msg.args[0], text[1:])
|
|
irc = callbacks.ReplyIrcProxy(irc, msg)
|
|
self.doPrivmsg(irc, msg)
|
|
|
|
return msg
|
|
|
|
Class = PackageInfo
|
|
|
|
|
|
# vim:set shiftwidth=4 tabstop=4 expandtab textwidth=79:
|