1

from twisted.words.protocols import irc

     2

from twisted.internet import protocol

     3

from twisted.python import log

     4

import traceback

     5

     6

import buffer

     7

     8

from irc_config import *

     9

    10

class ReplyException (Exception) :

    11

    def __init__ (self, reply) :

    12

        self.reply = reply

    13

    14

class BotProtocol (irc.IRCClient, object) :

    15

    """

    16

        Fixme IRC bot

    17

    """

    18

    19

    def __init__ (self) :

    20

        self.nickname = NICKNAME

    21

        self.username = USERNAME

    22

    23

    # housekeeping

    24

    def connectionMade (self) :

    25

        log.msg("Connected")

    26

        super(BotProtocol, self).connectionMade()

    27

    28

        self.nexus = self.factory.nexus

    29

    30

    def connectionLost (self, reason) :

    31

        log.msg("Connection lost: %s" % reason)

    32

        super(BotProtocol, self).connectionLost(reason)

    33

    34

    def signedOn (self) :

    35

        log.msg("Signed on, joining channel %s" % CHANNEL)

    36

        self.join(CHANNEL)

    37

    38

    def joined (self, channel) :

    39

        log.msg("Joined channel %s" % channel)

    40

    41

        self.factory.connection = self

    42

    43

    # our actual functionality

    44

    def send (self, msg) :

    45

        msg = str(msg)

    46

    47

        if len(msg) > 480 :

    48

            log.msg("truncating: %s" % msg)

    49

            msg = msg[:480] + "..."

    50

    51

        msg = msg.replace("\n", "\\n").replace("\r", "\\r").replace("\0", "\\0")

    52

    53

        self.notice(CHANNEL, msg)

    54

    55

    def sendEvent (self, event) :

    56

        self.send("[%s.%s] %s" % (event.module.name, event.type, event.msg))

    57

    58

    def moduleConnected (self, module, addr) :

    59

        self.send("{modules.%s} connected from %s:%d, version %s" % (module.name, addr.host, addr.port, module.version))

    60

    61

    def moduleDisconnected (self, module, reason) :

    62

        self.send("{modules.%s} disconnected: %s" % (module.name, reason))

    63

    64

    class _noDefault : pass

    65

    66

    def _lookupCommand (self, command, default=_noDefault) :

    67

        if '.' in command :

    68

            raise ReplyException("No support for module commands yet :P")

    69

        else :

    70

            method = getattr(self, "cmd_%s" % command, None)

    71

    72

        if method :

    73

            return method

    74

        elif default is self._noDefault :

    75

            raise ReplyException("No such command '%s'. See `help commands'" % command)

    76

        else :

    77

            return default

    78

    79

    def privmsg (self, user, channel, message) :

    80

        if message.lower().startswith(self.nickname.lower()) :

    81

            me, command = message.split(":", 1)

    82

    83

            args = command.strip().split()

    84

            command = args.pop(0)

    85

    86

            try :

    87

                method = self._lookupCommand(command)

    88

    89

                reply = method(*args)

    90

    91

                if reply :

    92

                    self.send(reply)

    93

    94

            except ReplyException, e :

    95

                self.send(e.reply)

    96

    97

            except Exception, e :

    98

                self.send("Error: %s: %s" % (e.__class__.__name__, e))

    99

                traceback.print_exc()

   100

   101

    def cmd_help (self, cmd="help") :

   102

        """help <command|module> - Display help about the given command or module"""

   103

   104

        method = self._lookupCommand(cmd, None)

   105

   106

        if method :

   107

            return method.__doc__

   108

        else :

   109

            try :

   110

                module, addr = self.nexus.getModuleInfo(cmd)

   111

   112

                return "%s is version %d from %s:%d. Events: %s. See `commands %s' for a list of commands" % (module.name, module.version, addr.host, addr.port, ', '.join(module.event_types), module.name)

   113

   114

            except KeyError :

   115

                raise ReplyException("No command/module called `%s'. See `help commands'" % cmd)

   116

   117

    def cmd_commands (self, module=None) :

   118

        """commands [<module>] - Show primary commands, or commands in the given module (see `help modules')"""

   119

   120

        if module :

   121

            raise ReplyException("No support for module commands yet :P")

   122

        else :

   123

            return "Commands: %s" % ', '.join(

   124

                attr_name.split('_', 1)[1]

   125

                for attr_name in BotProtocol.__dict__.iterkeys()

   126

                if attr_name.startswith("cmd_")

   127

            )

   128

   129

    def cmd_modules (self) :

   130

        """modules - Show a list of connected modules"""

   131

   132

        return "Modules: %s" % ', '.join(

   133

            module.name

   134

            for module in self.nexus.getModules()

   135

        )