Add error checking in places, set up automatic relay provisioning and fix starting bots

commands/auto.py

@@ -0,0 +1,29 @@
+import main
+from modules import provision
+
+class AutoCommand:
+    def __init__(self, *args):
+        self.auto(*args)
+
+    def auto(self, addr, authed, data, obj, spl, success, failure, info, incUsage, length):
+        if authed:
+            if length == 3:
+                if not spl[1] in main.network.keys():
+                    failure("No such network: %s" % spl[1])
+                    return
+                if not spl[2].isdigit():
+                    failure("Must be integer, not %s" % spl[2])
+                    return
+
+                id, alias = main.network[spl[1]].add_relay(int(spl[2]))
+                success("Successfully created relay %s on network %s with alias %s" % (str(id), spl[1], alias))
+                main.saveConf("network")
+                rtrn = provision.provisionRelay(int(spl[2]), spl[1])
+                success("Started provisioning network %s on relay %s for alias %s" % (spl[1], spl[2], rtrn))
+                return
+
+            else:
+                incUsage("auto")
+                return
+        else:
+            incUsage(None)

commands/cmd.py

@@ -8,16 +8,12 @@ class CmdCommand:
     def cmd(self, addr, authed, data, obj, spl, success, failure, info, incUsage, length):
         if authed:
             if length > 4:
-                if not spl[1] in main.network.keys():
+                if not spl[1].isdigit():
-                    failure("No such network: %s" % spl[1])
+                    failure("Must be integer, not %s" % spl[1])
                     return
-                if not int(spl[2]) in main.network[spl[1]].relays.keys():
+                commands = {spl[3]: [" ".join(spl[4:])]}
-                    failure("No such relay: %s on network: %s" % (spl[2], spl[1]))
-                    return
-
-                commands = {spl[4]: [" ".join(spl[5:])]}
                 success("Sending commands to relay %s as user %s" % (spl[2], spl[3]))
-                deliverRelayCommands(main.network[spl[1]].relays[spl[2]], commands, user=spl[3]+"/"+spl[1])
+                deliverRelayCommands(int(spl[1]), commands, user=spl[2])
                 return
             else:
                 incUsage("cmd")

commands/enable.py

@@ -11,16 +11,18 @@ class EnableCommand:
                 if not spl[1] in main.network.keys():
                     failure("No such network: %s" % spl[1])
                     return
+                if not spl[2].isdigit():
+                    failure("Must be a number, not %s" % spl[2])
+                    return
                 if not int(spl[2]) in main.network[spl[1]].relays.keys():
-                    failure("No such relay: %s in network %s" % (spl[2], spl[1]))
+                    failure("No such relay on %s: %s" % (spl[2], spl[1]))
                     return
 
                 main.network[spl[1]].relays[int(spl[2])]["enabled"] = True
-                user = main.network[spl[1]].aliases[int(spl[2])]
+                user = main.network[spl[1]].aliases[int(spl[2])]["nick"]
                 network = spl[1]
-                relay = main.network[spl[1]].relays[int(spl[2])]
                 commands = {"status": ["Connect"]}
-                deliverRelayCommands(relay, commands, user=user+"/"+network)
+                deliverRelayCommands(int(spl[2]), commands, user=user+"/"+network)
                 main.saveConf("network")
                 if not spl[1]+spl[2] in main.IRCPool.keys():
                     main.network[spl[1]].start_bot(int(spl[2]))

commands/get.py

@@ -1,22 +0,0 @@
-import main
-
-class GetCommand:
-    def __init__(self, *args):
-        self.get(*args)
-
-    def get(self, addr, authed, data, obj, spl, success, failure, info, incUsage, length):
-        if authed:
-            if length == 3:
-                if not spl[1] in main.pool.keys():
-                    failure("Name does not exist: %s" % spl[1])
-                    return
-                if not spl[1] in main.IRCPool.keys():
-                    failure("Name has no instance: %s" % spl[1])
-                    return
-                info(str(main.IRCPool[spl[1]].get(spl[2])))
-                return
-            else:
-                incUsage("get")
-                return
-        else:
-            incUsage(None)

commands/join.py

@@ -11,7 +11,7 @@ class JoinCommand:
                     failure("Network does not exist: %s" % spl[1])
                     return
                 if not int(spl[2]) in main.network[spl[1]].relays.keys():
-                    failure("Relay % does not exist on network %", (spl[2], spl[1]))
+                    failure("Relay %s does not exist on network %s" % (spl[2], spl[1]))
                     return
                 if not spl[1]+spl[2] in main.IRCPool.keys():
                     failure("Name has no instance: %s" % spl[1])

commands/network.py

@@ -23,7 +23,7 @@ class NetworkCommand:
                         failure("Auth must be sasl, ns or none, not %s" % spl[5])
                         return
                     else:
-                        main.network[spl[2]] = Network(spl[2], spl[3], spl[4], spl[5].lower(), spl[6].lower())
+                        main.network[spl[2]] = Network(spl[2], spl[3], int(spl[4]), spl[5].lower(), spl[6].lower())
                         success("Successfully created network: %s" % spl[2])
                         main.saveConf("network")
                         return

commands/provision.py

@@ -1,60 +0,0 @@
-import main
-from modules import provision
-
-class ProvisionCommand:
-    def __init__(self, *args):
-        self.provision(*args)
-
-    def provision(self, addr, authed, data, obj, spl, success, failure, info, incUsage, length):
-        if authed:
-            if length == 4 or length == 3:
-                if not spl[1] in main.relay.keys():
-                    failure("No such relay: %s" % spl[1])
-                    return
-                if not spl[2] in main.alias.keys():
-                    failure("No such alias: %s" % spl[2])
-                    return
-            if length == 4: # provision for relay, alias and network
-                if not spl[3] in main.network.keys():
-                    failure("No such network: %s" % spl[3])
-                    return
-
-                #if "users" in main.relay[spl[1]]:
-                #    if not spl[2] in main.relay[spl[1]]["users"]:
-                #        failure("Relay %s not provisioned for alias %s" % (spl[1], spl[2]))
-                #        return
-                #else:
-                #        failure("Relay %s not provisioned for alias %s" % (spl[1], spl[2]))
-                #        return
-
-                rtrn = provision.provisionRelayForNetwork(spl[1], spl[2], spl[3])
-                #if rtrn == "PROVISIONED":
-                #    failure("Relay %s already provisioned for alias %s on network %s" % (spl[1], spl[2], spl[3]))
-                #    return
-                #elif rtrn == "DUPLICATE":
-                #    failure("Instance with relay %s and alias %s already exists for network %s" % (spl[1], spl[2], spl[3]))
-                #    return
-                if rtrn:
-                    success("Started provisioning network %s on relay %s for alias %s" % (spl[3], spl[1], spl[2]))
-                    info("Instance name is %s" % rtrn)
-                    return
-                else:
-                    failure("Failure while provisioning relay %s" % spl[1])
-                    return
-            if length == 3: # provision for relay and alias only
-                rtrn = provision.provisionRelayForAlias(spl[1], spl[2])
-                #if rtrn == "PROVISIONED":
-                #    failure("Relay %s already provisioned for alias %s" % (spl[1], spl[2]))
-                #    return
-                if rtrn:
-                    success("Started provisioning relay %s for alias %s" % (spl[1], spl[2]))
-                    return
-                else:
-                    failure("Failure while provisioning relay %s" % spl[1])
-                    return
-
-            else:
-                incUsage("provision")
-                return
-        else:
-            incUsage(None)

commands/relay.py

@@ -7,29 +7,47 @@ class RelayCommand:
 
     def relay(self, addr, authed, data, obj, spl, success, failure, info, incUsage, length):
         if authed:
-            if length == 7:
+            if length == 3:
                 if spl[1] == "add":
+                    if spl[2] in main.network.keys():
+                        id, alias = main.network[spl[2]].add_relay()
+                        success("Successfully created relay %s on network %s with alias %s" % (str(id), spl[2], alias))
+                        main.saveConf("network")
+                        return
+                    else:
+                        failure("No such network: %s" % spl[2])
+                        return
+                elif spl[1] == "list":
                     if spl[2] not in main.network.keys():
                         failure("No such network: %s" % spl[2])
                         return
-                    if not spl[4].isdigit():
+                    info(dump(main.network[spl[2]].relays))
-                        failure("Port must be an integer, not %s" % spl[4])
-                        return
-                    else:
-                        id, alias = main.network[spl[2]].add_relay(spl[3], spl[4], spl[5], spl[6])
-                        success("Successfully created relay %s on network %s with alias %s" % (str(id), spl[2], alias))
-                        main.saveConf("network")
                     return
                 else:
                     incUsage("relay")
                     return
 
             elif length == 4:
-                if spl[1] == "del":
+                if spl[1] == "add":
-                    if spl[2] not in main.network.keys():
+                    if spl[2] in main.network.keys():
+                        if not spl[3].isdigit():
+                            failure("Must be a number, not %s" % spl[3])
+                            return
+                        id, alias = main.network[spl[2]].add_relay(int(spl[3]))
+                        success("Successfully created relay %s on network %s with alias %s" % (str(id), spl[2], alias))
+                        main.saveConf("network")
+                        return
+                    else:
                         failure("No such network: %s" % spl[2])
                         return
-                    if int(spl[3]) not in main.network[spl[2]].relays.keys():
+                elif spl[1] == "del":
+                    if not spl[2] in main.network.keys():
+                        failure("No such network: %s" % spl[2])
+                        return
+                    if not spl[3].isdigit():
+                        failure("Must be a number, not %s" % spl[3])
+                        return
+                    if not int(spl[3]) in main.network[spl[2]].relays.keys():
                         failure("No such relay: %s on network %s" % (spl[3], spl[2]))
                         return
                     main.network[spl[2]].delete_relay(int(spl[3]))
@@ -39,16 +57,6 @@ class RelayCommand:
                 else:
                     incUsage("relay")
                     return
-            elif length == 3:
-                if spl[1] == "list":
-                    if spl[2] not in main.network.keys():
-                        failure("No such network: %s" % spl[2])
-                        return
-                    info(dump(main.network[spl[2]].relays))
-                    return
-                else:
-                    incUsage("relay")
-                    return
             else:
                 incUsage("relay")
                 return

commands/swho.py

@@ -0,0 +1,40 @@
+import main
+
+class SwhoCommand:
+    def __init__(self, *args):
+        self.swho(*args)
+
+    def swho(self, addr, authed, data, obj, spl, success, failure, info, incUsage, length):
+        if authed:
+            if length == 2:
+                if not spl[1] in main.network.keys():
+                    failure("Network does not exist: %s" % spl[1])
+                    return
+                for i in main.IRCPool.keys():
+                    if spl[1] in i:
+                        for x in main.IRCPool[i].channels:
+                            main.IRCPool[i].who(x)
+                success("Sent WHO to all channels on all networks on %s" % spl[1])
+                return
+            elif length == 3:
+                if not spl[1] in main.network.keys():
+                    failure("Network does not exist: %s" % spl[1])
+                    return
+                matches = []
+                for i in main.IRCPool.keys():
+                    if spl[1] in i:
+                        for x in main.IRCPool[i].channels:
+                            if x == spl[2]:
+                                main.IRCPool[i].who(x)
+                                matches.append(i)
+                if matches == []:
+                    failure("No matches found for channel %s" % spl[2])
+                    return
+                success("Sent WHO to %s on: %s" % (spl[2], ", ".join(matches)))
+                return
+
+            else:
+                incUsage("swho")
+                return
+        else:
+            incUsage(None)

conf/example/config.json

@@ -4,7 +4,7 @@
         "Address": "127.0.0.1",
         "UseSSL": true
     },
-    "Relay": {
+    "RelayAPI": {
         "Enabled": true,
         "Port": 13868,
         "Address": "127.0.0.1",
@@ -16,6 +16,12 @@
     "UsePassword": true,
     "ConnectOnCreate": false,
     "Debug": false,
+    "Relay": {
+        "Host": "127.0.0.1",
+        "Port": "201x",
+        "User": "sir",
+        "Password": "sir"
+    },
     "Dist": {
         "Enabled": true,
         "SendOutput": false,

conf/help.json

@@ -1,27 +1,27 @@
 {
     "pass": "pass <password>",
     "logout": "logout",
-    "del": "del <network> <relay>",
+    "del": "del <name>",
-    "mod": "mod <network> <variable> <value>",
+    "mod": "mod <name> [<key>] [<value>]",
-    "get": "get <name> <variable>",
     "who": "who <query>",
-    "join": "join <network> <relay> <channel> [<key>]",
+    "join": "join <name> <num> <channel> [<key>]",
-    "part": "part <network> <relay> <channel>",
+    "part": "part <name> <num> <channel>",
-    "enable": "enable <network> <id>",
+    "enable": "enable <name> <num>",
-    "disable": "disable <network> <id>",
+    "disable": "disable <name> <num>",
+    "list": "list",
     "stats": "stats [<name>]",
     "save": "save <(file)|list|all>",
     "load": "load <(file)|list|all>",
     "dist": "dist",
     "loadmod": "loadmod <module>",
-    "msg": "msg <network> <relay> <target> <message...>",
+    "msg": "msg <name> <target> <message...>",
     "mon": "mon -h",
     "chans": "chans <nick> [<nick> ...]",
     "users": "users <channel> [<channel> ...]",
-    "relay": "relay <add|del|list> [<network> <host> <port> <user> <password> | <network> <id> | <network>]",
+    "relay": "relay <add|del|list> [<network>] [<num>]",
     "network": "network <add|del|list> [<name> <address> <port> <ssl|plain> <sasl|ns|none>]",
-    "provision": "provision <relay> <alias> [<network>]",
+    "auto": "auto <network> <relay>",
-    "cmd": "cmd <network> <relay> <user> <entity> <text ...>",
+    "cmd": "cmd <relay> <user> <entity> <text ...>",
     "token": "token <add|del|list> [<key>] [<relay>]",
     "all": "all <entity> <text ...>",
     "allc": "allc <network|alias> <(network)|(alias)> <entity> <text ...>"

conf/masterbuf.json

@@ -1 +0,0 @@
-[]

core/bot.py

@@ -14,6 +14,7 @@ from modules import monitor
 
 from core.relay import sendRelayNotification
 from utils.dedup import dedup
+from utils.getrelay import getRelay
 
 import main
 from utils.logging.log import *
@@ -21,33 +22,34 @@ from utils.logging.send import *
 
 from twisted.internet.ssl import DefaultOpenSSLContextFactory
 
-def deliverRelayCommands(relay, relayCommands, user=None, stage2=None):
+def deliverRelayCommands(num, relayCommands, user=None, stage2=None):
     # where relay is a dictionary extracted from the Network object
     keyFN = main.certPath+main.config["Key"]
     certFN = main.certPath+main.config["Certificate"]
     contextFactory = DefaultOpenSSLContextFactory(keyFN.encode("utf-8", "replace"),
             certFN.encode("utf-8", "replace"))
-    bot = IRCBotFactory(None, relay, relayCommands, user, stage2)
+    bot = IRCBotFactory(net=None, num=num, relayCommands=relayCommands, user=user, stage2=stage2)
-    rct = reactor.connectSSL(relay["host"],
+    host, port = getRelay(num)
-            int(relay["port"]),
+    rct = reactor.connectSSL(host,
+            port,
             bot, contextFactory)
 
 class IRCRelay(IRCClient):
-    def __init__(self, relay, relayCommands, user, stage2):
+    def __init__(self, num, relayCommands, user, stage2):
         self.connected = False
         self.buffer = ""
         if user == None:
-            self.user = relay["user"]
+            self.user = main.config["Relay"]["User"]
         else:
             self.user = user
-        password = relay["password"]
+        password = main.config["Relay"]["Password"]
-        self.nickname = self.user
+        self.nickname = "relay"
-        self.realname = self.user
+        self.realname = "relay"
         self.username = self.user
         self.password = self.user+":"+password
 
         self.relayCommands = relayCommands
-        self.relay = relay
+        self.num = num
         self.stage2 = stage2
 
     def parsen(self, user):
@@ -67,15 +69,16 @@ class IRCRelay(IRCClient):
         if nick[0] == main.config["Tweaks"]["ZNC"]["Prefix"]:
             nick = nick[1:]
             if nick in self.relayCommands.keys():
-                sendAll("[%s] %s -> %s" % (self.relay, nick, msg))
+                sendAll("[%s] %s -> %s" % (self.num, nick, msg))
 
     def irc_ERR_PASSWDMISMATCH(self, prefix, params):
-        log("%s: relay password mismatch" % self.relay)
+        print(', '.join("%s: %s" % item for item in vars(self).items()))
-        sendAll("%s: relay password mismatch" % self.relay)
+        log("%s: relay password mismatch" % self.num)
+        sendAll("%s: relay password mismatch" % self.num)
 
     def signedOn(self):
         self.connected = True
-        log("signed on as a relay: %s" % self.relay)
+        log("signed on as a relay: %s" % self.num)
         #sendRelayNotification("Relay", {"type": "conn", "status": "connected"})  nobody actually cares
         for i in self.relayCommands.keys():
             for x in self.relayCommands[i]:
@@ -85,25 +88,24 @@ class IRCRelay(IRCClient):
                 user = self.stage2[0].pop(0)
                 commands = self.stage2[0].pop(0)
                 del self.stage2[0]
-                deliverRelayCommands(self.relay, commands, user, self.stage2)
+                deliverRelayCommands(self.num, commands, user, self.stage2)
         deferLater(reactor, 1, self.transport.loseConnection)
         return
 
 class IRCBot(IRCClient):
-    def __init__(self, name, relay):
+    def __init__(self, net, num):
         self.connected = False
         self.channels = []
-        self.net = "".join([x for x in name if not x in digits])
+        self.net = net
-        if self.net == "":
+        self.num = num
-            error("Network with all numbers: %s" % name)
         self.buffer = ""
-        self.name = name
+        self.name = net + str(num)
-        alias = relay["alias"]
+        alias = main.network[self.net].aliases[num]
-
+        relay = main.network[self.net].relays[num]
         self.nickname = alias["nick"]
         self.realname = alias["realname"]
         self.username = alias["nick"]+"/"+relay["net"]
-        self.password = relay["password"]
+        self.password = main.config["Relay"]["Password"]
         self.userinfo = None
         self.fingerReply = None
         self.versionName = None
@@ -445,26 +447,30 @@ class IRCBot(IRCClient):
             self.event(type="mode", muser=user, target=channel, modes=m, status=toset, modeargs=a)
 
 class IRCBotFactory(ReconnectingClientFactory):
-    def __init__(self, name, relay=None, relayCommands=None, user=None, stage2=None):
+    def __init__(self, net, num=None, relayCommands=None, user=None, stage2=None):
-        if not name == None:
+        if net == None:
-            self.name = name
+            self.num = num
-            self.net = "".join([x for x in self.name if not x in digits])
+            self.name = "Relay to %i" % num
+            self.relay = True
         else:
-            self.name = "Relay to "+relay["net"]+relay["id"]
+            self.name = net + str(num)
+            self.num = num
+            self.net = net
+            self.relay = False
         self.client = None
         self.maxDelay = main.config["Tweaks"]["Delays"]["MaxDelay"]
         self.initialDelay = main.config["Tweaks"]["Delays"]["InitialDelay"]
         self.factor = main.config["Tweaks"]["Delays"]["Factor"]
         self.jitter = main.config["Tweaks"]["Delays"]["Jitter"]
 
-        self.relay, self.relayCommands, self.user, self.stage2 = relay, relayCommands, user, stage2
+        self.relayCommands, self.user, self.stage2 = relayCommands, user, stage2
 
     def buildProtocol(self, addr):
-        if self.relay == None:
+        if self.net == None:
-            entry = IRCBot(self.name, self.relay)
+            entry = IRCRelay(self.num, self.relayCommands, self.user, self.stage2)
-            main.IRCPool[self.name] = entry
         else:
-            entry = IRCRelay(self.relay, self.relayCommands, self.user, self.stage2)
+            entry = IRCBot(self.net, self.num)
+            main.IRCPool[self.name] = entry
 
         self.client = entry
         return entry

core/relay.py

@@ -108,7 +108,7 @@ class Relay(Protocol):
             return
 
     def connectionMade(self):
-        log("Connection from %s:%s" % (self.addr.host, self.addr.port))
+        log("Relay connection from %s:%s" % (self.addr.host, self.addr.port))
         #self.send("Greetings.")
 
     def connectionLost(self, reason):

modules/network.py

@@ -1,14 +1,16 @@
+from twisted.internet.ssl import DefaultOpenSSLContextFactory
 import json
+
 import modules.alias as alias
 from twisted.internet import reactor
 from core.bot import IRCBot, IRCBotFactory
-from twisted.internet.ssl import DefaultOpenSSLContextFactory
 import main
 from utils.logging.log import *
+from utils.getrelay import getRelay
 
 class Network:
-    def __init__(self, name, host, port, security, auth):
+    def __init__(self, net, host, port, security, auth):
-        self.name = name
+        self.net = net
         self.host = host
         self.port = port
         self.security = security
@@ -18,40 +20,39 @@ class Network:
         self.relays = {}
         self.aliases = {}
 
-    def add_relay(self, host, port, user, password):
+    def add_relay(self, num=None):
+        if not num:
             self.last += 1
-        self.relays[self.last] = {
+            num = self.last
-            "host": host,
+        self.relays[num] = {
-            "port": port,
-            "user": user,
-            "password": password,
             "enabled": False,
-            "net": self.name,
+            "net": self.net,
-            "id": self.last
+            "id": num
         }
-        self.aliases[self.last] = alias.generate_alias()
+        self.aliases[num] = alias.generate_alias()
-        return self.last, self.aliases[self.last]["nick"]
+        return num, self.aliases[num]["nick"]
 
     def delete_relay(self, id):
         del self.relays[id]
         del self.aliases[id]
 
-    def start_bot(self, relay):
+    def start_bot(self, num):
         # a single name is given to relays in the backend
         # e.g. freenode1 for the first relay on freenode network
-        name = self.name + relay
         keyFN = main.certPath+main.config["Key"]
         certFN = main.certPath+main.config["Certificate"]
         contextFactory = DefaultOpenSSLContextFactory(keyFN.encode("utf-8", "replace"), certFN.encode("utf-8", "replace"))
-        bot = IRCBotFactory(name)
+        bot = IRCBotFactory(self.net, num)
-        rct = reactor.connectSSL(k, port, bot, contextFactory)
+        #host, port = self.relays[num]["host"], self.relays[num]["port"]
-
+        host, port = getRelay(num)
+        rct = reactor.connectSSL(host, port, bot, contextFactory)
+        name = self.net + str(num)
         main.ReactorPool[name] = rct
         main.FactoryPool[name] = bot
 
-        log("Started bot on relay %s on %s", (relay, self.host))
+        log("Started bot on relay %s on %s" % (num, self.host))
 
     def start_bots(self):
-        for relay in self.relays:
+        for num in self.relays.keys():
-            if relay["enabled"]:
+            if self.relays[num]["enabled"]:
-                start_bot(relay)
+                self.start_bot(num)

modules/provision.py

@@ -1,36 +1,35 @@
 import main
 from core.bot import deliverRelayCommands
 from utils.logging.log import *
-from core.helper import startBot
 
-def provisionUserData(relay, alias, nick, altnick, ident, realname, password):
+def provisionUserData(num, nick, altnick, ident, realname, unused): # last field is password, which we don't want to inherit here, but still want to use * expansion, so this is a bit of a hack
     commands = {}
     commands["controlpanel"] = []
-    commands["controlpanel"].append("AddUser %s %s" % (alias, password))
+    commands["controlpanel"].append("AddUser %s %s" % (nick, main.config["Relay"]["Password"]))
-    commands["controlpanel"].append("Set Nick %s %s" % (alias, nick))
+    commands["controlpanel"].append("Set Nick %s %s" % (nick, nick))
-    commands["controlpanel"].append("Set Altnick %s %s" % (alias, altnick))
+    commands["controlpanel"].append("Set Altnick %s %s" % (nick, altnick))
-    commands["controlpanel"].append("Set Ident %s %s" % (alias, ident))
+    commands["controlpanel"].append("Set Ident %s %s" % (nick, ident))
-    commands["controlpanel"].append("Set RealName %s %s" % (alias, realname))
+    commands["controlpanel"].append("Set RealName %s %s" % (nick, realname))
-    deliverRelayCommands(relay, commands)
+    deliverRelayCommands(num, commands)
     return
 
-def provisionNetworkData(relay, alias, network, host, port, security, auth, password):
+def provisionNetworkData(num, nick, network, host, port, security, auth, password):
     commands = {}
     stage2commands = {}
     stage3commands = {}
     commands["controlpanel"] = []
-    commands["controlpanel"].append("AddNetwork %s %s" % (alias, network))
+    commands["controlpanel"].append("AddNetwork %s %s" % (nick, network))
     if security == "ssl":
-        commands["controlpanel"].append("SetNetwork TrustAllCerts %s %s true" % (alias, network)) # Don't judge me
+        commands["controlpanel"].append("SetNetwork TrustAllCerts %s %s true" % (nick, network)) # Don't judge me
-        commands["controlpanel"].append("AddServer %s %s %s +%s" % (alias, network, host, port))
+        commands["controlpanel"].append("AddServer %s %s %s +%s" % (nick, network, host, port))
     elif security == "plain": 
-        commands["controlpanel"].append("AddServer %s %s %s %s" % (alias, network, host, port))
+        commands["controlpanel"].append("AddServer %s %s %s %s" % (nick, network, host, port))
     if auth == "sasl":
         stage2commands["status"] = []
         stage2commands["sasl"] = []
         stage2commands["status"].append("LoadMod sasl")
         stage2commands["sasl"].append("Mechanism plain")
-        stage2commands["sasl"].append("Set %s %s" % (alias, password))
+        stage2commands["sasl"].append("Set %s %s" % (nick, password))
     elif auth == "ns":
         stage2commands["status"] = []
         stage2commands["nickserv"] = []
@@ -43,60 +42,23 @@ def provisionNetworkData(relay, alias, network, host, port, security, auth, pass
         stage2commands["status"] = []
         stage2commands["status"].append("LoadMod disconkick")
     stage2commands["status"].append("LoadMod chansaver")
-    deliverRelayCommands(relay, commands,
+    deliverRelayCommands(num, commands,
-        stage2=[[alias+"/"+network, stage2commands],
+        stage2=[[nick+"/"+network, stage2commands],
-                [alias+"/"+network, stage3commands]])
+                [nick+"/"+network, stage3commands]])
     return
 
-def provisionRelayForAlias(relay, alias):
+def provisionRelayForNetwork(num, alias, network):
-    #if "users" in main.relay[relay].keys():
+    provisionNetworkData(num, alias, network,
-    #    if alias in main.relay[relay]["users"]:
+            main.network[network].host,
-    #        return "PROVISIONED"
+            main.network[network].port,
-    #else:
+            main.network[network].security,
-    #    main.relay[relay]["users"] = []
+            main.network[network].auth,
-    #main.relay[relay]["users"].append(alias)
+            main.network[network].aliases[num]["password"])
-    provisionUserData(relay, alias, main.alias[alias]["nick"],
+    return
-            main.alias[alias]["altnick"],
-            main.alias[alias]["ident"],
-            main.alias[alias]["realname"],
-            main.relay[relay]["password"])
-    #main.saveConf("relay")
-    return True
 
-def provisionRelayForNetwork(relay, alias, network):
+def provisionRelay(num, network):
-    #if set(["users", "networks"]).issubset(main.relay[relay].keys()):
+    aliasObj = main.network[network].aliases[num]
-    #    if network in main.relay[relay]["networks"] and alias in main.relay[relay]["users"]:
+    alias = aliasObj["nick"]
-    #        return "PROVISIONED"
+    provisionUserData(num, *aliasObj.values())
-    #else:
+    provisionRelayForNetwork(num, alias, network)
-    #    main.relay[relay]["networks"] = []
+    return alias
-    #main.relay[relay]["networks"].append(network)
-    provisionNetworkData(relay, alias, network,
-            main.network[network]["host"],
-            main.network[network]["port"],
-            main.network[network]["security"],
-            main.network[network]["auth"],
-            main.alias[alias]["password"])
-    #main.saveConf("relay")
-    storedNetwork = False
-    num = 1
-    while not storedNetwork:
-        i = str(num)
-        if num == 1000:
-            error("Iteration limit exceeded while trying to choose name for r: %s a: %s n: %s" % (relay, alias, network))
-            return False
-
-        if network+i in main.pool.keys():
-            if main.pool[network+i]["alias"] == alias and main.pool[network+i]["relay"] == relay:
-                return "DUPLICATE"
-            num += 1
-        else:
-            main.pool[network+i] = {"relay": relay,
-                                    "alias": alias,
-                                    "network": network,
-                                    "enabled": main.config["ConnectOnCreate"]}
-            main.saveConf("pool")
-            if main.config["ConnectOnCreate"]:
-                startBot(network+i)
-
-            storedNetwork = True
-    return network+i

threshold

@@ -29,14 +29,14 @@ if __name__ == "__main__":
     else:
         reactor.listenTCP(main.config["Listener"]["Port"], listener, interface=main.config["Listener"]["Address"])
         log("Threshold running on %s:%s" % (main.config["Listener"]["Address"], main.config["Listener"]["Port"]))
-    if main.config["Relay"]["Enabled"]:
+    if main.config["RelayAPI"]["Enabled"]:
         relay = RelayFactory()
-        if main.config["Relay"]["UseSSL"] == True:
+        if main.config["RelayAPI"]["UseSSL"] == True:
-            reactor.listenSSL(main.config["Relay"]["Port"], relay, DefaultOpenSSLContextFactory(main.certPath+main.config["Key"], main.certPath+main.config["Certificate"]), interface=main.config["Relay"]["Address"])
+            reactor.listenSSL(main.config["RelayAPI"]["Port"], relay, DefaultOpenSSLContextFactory(main.certPath+main.config["Key"], main.certPath+main.config["Certificate"]), interface=main.config["RelayAPI"]["Address"])
-            log("Threshold relay running with SSL on %s:%s" % (main.config["Relay"]["Address"], main.config["Relay"]["Port"]))
+            log("Threshold relay running with SSL on %s:%s" % (main.config["RelayAPI"]["Address"], main.config["RelayAPI"]["Port"]))
         else:
-            reactor.listenTCP(main.config["Relay"]["Port"], relay, interface=main.config["Relay"]["Address"])
+            reactor.listenTCP(main.config["RelayAPI"]["Port"], relay, interface=main.config["RelayAPI"]["Address"])
-            log("Threshold relay running on %s:%s" % (main.config["Relay"]["Address"], main.config["Relay"]["Port"]))
+            log("Threshold relay running on %s:%s" % (main.config["RelayAPI"]["Address"], main.config["RelayAPI"]["Port"]))
     for net in main.network.keys():
         main.network[net].start_bots()
     modules.counters.setupCounterLoop()

utils/getrelay.py

@@ -0,0 +1,12 @@
+import main
+
+def getRelay(num):
+    host = main.config["Relay"]["Host"].replace("x", str(num))
+    port = int(str(main.config["Relay"]["Port"]).replace("x", str(num)))
+    user = main.config["Relay"]["User"]
+    password = main.config["Relay"]["Password"]
+    try:
+        port = int(port)
+    except ValueError:
+        return False
+    return [host, port]

utils/loaders/command_loader.py

@@ -1,6 +1,7 @@
 from os import listdir
 
 from utils.logging.debug import debug
+from utils.logging.log import *
 import commands
 
 from main import CommandMap