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

     9

PORT            = 6667

    10

NICKNAME        = "FixBot"

    11

USERNAME        = "fixme"

    12

CHANNEL         = "#fixme-test"

    13

    14

class ReplyException (Exception) :

    15

    def __init__ (self, reply) :

    16

        self.reply = reply

    17

    18

class BotProtocol (irc.IRCClient, object) :

    19

    """

    20

        Fixme IRC bot

    21

    """

    22

    23

    def __init__ (self) :

    24

        self.nickname = NICKNAME

    25

        self.username = USERNAME

    26

    27

    # housekeeping

    28

    def connectionMade (self) :

    29

        log.msg("Connected")

    30

        super(BotProtocol, self).connectionMade()

    31

    32

        self.nexus = self.factory.nexus

    33

    34

    def connectionLost (self, reason) :

    35

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

    36

        super(BotProtocol, self).connectionLost(reason)

    37

    38

    def signedOn (self) :

    39

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

    40

        self.join(CHANNEL)

    41

    42

    def joined (self, channel) :

    43

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

    44

    45

        self.factory.connection = self

    46

    47

    # our actual functionality

    48

    def send (self, msg) :

    49

        msg = str(msg)

    50

    51

        if len(msg) > 480 :

    52

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

    53

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

    54

    55

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

    56

    57

        self.msg(CHANNEL, msg)

    58

    59

    def sendEvent (self, event) :

    60

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

    61

    62

    def moduleConnected (self, module, addr) :

    63

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

    64

    65

    def moduleDisconnected (self, module, reason) :

    66

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

    67

    68

    class _noDefault : pass

    69

    70

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

    71

        if '.' in command :

    72

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

    73

        else :

    74

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

    75

    76

        if method :

    77

            return method

    78

        elif default is self._noDefault :

    79

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

    80

        else :

    81

            return default

    82

    83

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

    84

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

    85

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

    86

    87

            args = command.strip().split()

    88

            command = args.pop(0)

    89

    90

            try :

    91

                method = self._lookupCommand(command)

    92

    93

                reply = method(*args)

    94

    95

                if reply :

    96

                    self.send(reply)

    97

    98

            except ReplyException, e :

    99

                self.send(e.reply)

   100

   101

            except Exception, e :

   102

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

   103

                traceback.print_exc()

   104

   105

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

   106

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

   107

   108

        method = self._lookupCommand(cmd, None)

   109

   110

        if method :

   111

            return method.__doc__

   112

        else :

   113

            try :

   114

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

   115

   116

                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)

   117

   118

            except KeyError :

   119

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

   120

   121

    def cmd_commands (self, module=None) :

   122

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

   123

   124

        if module :

   125

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

   126

        else :

   127

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

   128

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

   129

                for attr_name in BotProtocol.__dict__.iterkeys()

   130

                if attr_name.startswith("cmd_")

   131

            )

   132

   133

    def cmd_modules (self) :

   134

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

   135

   136

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

   137

            module.name

   138

            for module in self.nexus.getModules()

   139

        )