OneHourGameJam-irc-bot/paib.py at master · OneHourGameJam/OneHourGameJam-irc-bot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
import json
import sys, os, imp

import irc.bot
import irc.client
import irc.strings

import time

sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.realpath(__file__))))


class Paib(irc.bot.SingleServerIRCBot):
    def __init__(self, config):
        self.shutupuntil = 0.0
        self.config = config
        irc.bot.SingleServerIRCBot.__init__(self, [(config["connection"]["server"], config["connection"]["port"])],
                                            config["connection"]["nick"], config["connection"]["nick"])
        self.channels_to_join = config["connection"]["channels"]
        self.command_prefix = config["botsettings"]["command_prefix"]

        irc.client.ServerConnection.buffer_class.errors = 'replace'

        self.plugins = []

        self.load_plugins(config["botsettings"]["plugins_folder"])

    def on_nicknameinuse(self, c, e):
        c.nick(c.get_nickname() + "_")
        for plugin in self.plugins:
            plugin.on_nicknameinuse(c.get_nickname())

    def on_welcome(self, c, e):
        for channel in self.channels_to_join:
            c.join(channel)
        for plugin in self.plugins:
            plugin.on_welcome(c.get_nickname())

    def on_privmsg(self, c, e):
        a = e.arguments[0].split(self.command_prefix, 1)
        if len(a) > 1:
            self.do_command(e, a[1].strip(), e.target)

    def on_join(self, c, e):
        for plugin in self.plugins:
            plugin.on_userjoin(e.source.nick, e.target)

    def on_quit(self, c, e):
        for plugin in self.plugins:
            plugin.on_userquit(e.source.nick, e.target)

    def on_pubmsg(self, c, e):
        if not (e.source.nick in self.config["botsettings"]["ignored"]):
            a = e.arguments[0].split(self.command_prefix, 1)
            if time.time() > self.shutupuntil:
                if len(a) > 1:
                    if (a[1].strip() == 'quiet'):
                        self.shutupuntil = time.time() + 60
                        c.privmsg(e.target, "I will STFU for a while.")
                    self.do_command(e, a[1].strip(), e.target)

            for plugin in self.plugins:
                plugin.on_message(e.arguments[0], e.source.nick, e.target)

    def do_command(self, e, cmd, chan):
        nick = e.source.nick
        c = self.connection

        for plugin in self.plugins:
            plugin.on_command(cmd, nick, chan)

    def load_plugins(self, plugins):
        print("Loading plugins")

        self.plugins = []
        for path in os.listdir(plugins):
            self.load_plugin(path)

    def load_plugin(self, path):
        print("Loading %s" % path)

        if os.path.isdir(os.path.join(self.config["botsettings"]["plugins_folder"], path)):
            plugin_file_path = os.path.join(self.config["botsettings"]["plugins_folder"], path, "%s.py" % path)

            if os.path.isfile(plugin_file_path):
                try:
                    print("Trying to import %s" % path)
                    plugin_module = imp.load_source(path, plugin_file_path)
                    print("Loaded %s" % path)
                except:
                    print("Error importing plugin %s!" % path)
                    return False

                if hasattr(plugin_module, "Plugin"):
                    try:
                        print("Creating plugin")
                        plugin = plugin_module.Plugin(self, self.config)
                        print("Loaded plugin %s" % plugin.name)
                    except:
                        print("Failed to load %s" % plugin_file_path)
                        return False

                    for p in self.plugins:
                        if p.name == plugin.name:
                            self.plugins.remove(p)
                    self.plugins.append(plugin)
                    return True


def main():
    configFile = 'config.json'
    try:
        configFile = open(configFile)
        settings = json.loads(configFile.read())
    except:
        print("Could not open the config. Quitting.")
        sys.exit(1)

    paib = Paib(settings)

    paib.start()

if __name__ == "__main__":
    main()