1 # Copyright (c) 2010 Google Inc. All rights reserved. 2 # 3 # Redistribution and use in source and binary forms, with or without 4 # modification, are permitted provided that the following conditions are 5 # met: 6 # 7 # * Redistributions of source code must retain the above copyright 8 # notice, this list of conditions and the following disclaimer. 9 # * Redistributions in binary form must reproduce the above 10 # copyright notice, this list of conditions and the following disclaimer 11 # in the documentation and/or other materials provided with the 12 # distribution. 13 # * Neither the name of Google Inc. nor the names of its 14 # contributors may be used to endorse or promote products derived from 15 # this software without specific prior written permission. 16 # 17 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 18 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 19 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 20 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 21 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 22 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 23 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 24 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 25 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 26 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 27 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28 29 from webkitpy.common.config import irc as config_irc 30 31 from webkitpy.common.thread.messagepump import MessagePump, MessagePumpDelegate 32 from webkitpy.thirdparty.autoinstalled.irc import ircbot 33 from webkitpy.thirdparty.autoinstalled.irc import irclib 34 35 36 class IRCBotDelegate(object): 37 def irc_message_received(self, nick, message): 38 raise NotImplementedError, "subclasses must implement" 39 40 def irc_nickname(self): 41 raise NotImplementedError, "subclasses must implement" 42 43 def irc_password(self): 44 raise NotImplementedError, "subclasses must implement" 45 46 47 class IRCBot(ircbot.SingleServerIRCBot, MessagePumpDelegate): 48 # FIXME: We should get this information from a config file. 49 def __init__(self, 50 message_queue, 51 delegate): 52 self._message_queue = message_queue 53 self._delegate = delegate 54 ircbot.SingleServerIRCBot.__init__( 55 self, 56 [( 57 config_irc.server, 58 config_irc.port, 59 self._delegate.irc_password() 60 )], 61 self._delegate.irc_nickname(), 62 self._delegate.irc_nickname()) 63 self._channel = config_irc.channel 64 65 # ircbot.SingleServerIRCBot methods 66 67 def on_nicknameinuse(self, connection, event): 68 connection.nick(connection.get_nickname() + "_") 69 70 def on_welcome(self, connection, event): 71 connection.join(self._channel) 72 self._message_pump = MessagePump(self, self._message_queue) 73 74 def on_pubmsg(self, connection, event): 75 nick = irclib.nm_to_n(event.source()) 76 request = event.arguments()[0].split(":", 1) 77 if len(request) > 1 and irclib.irc_lower(request[0]) == irclib.irc_lower(self.connection.get_nickname()): 78 response = self._delegate.irc_message_received(nick, request[1]) 79 if response: 80 connection.privmsg(self._channel, response) 81 82 # MessagePumpDelegate methods 83 84 def schedule(self, interval, callback): 85 self.connection.execute_delayed(interval, callback) 86 87 def message_available(self, message): 88 self.connection.privmsg(self._channel, message) 89 90 def final_message_delivered(self): 91 self.die() 92
