import main from core.bot import deliverRelayCommands from utils.logging.log import * from twisted.internet import reactor def provisionUserNetworkData(num, nick, altnick, ident, realname, network, host, port, security, auth, password): commands = {} stage2commands = {} stage2commands["status"] = [] commands["controlpanel"] = [] commands["controlpanel"].append("AddUser %s %s" % (nick, main.config["Relay"]["Password"])) commands["controlpanel"].append("AddNetwork %s %s" % (nick, network)) commands["controlpanel"].append("Set Nick %s %s" % (nick, nick)) commands["controlpanel"].append("Set Altnick %s %s" % (nick, altnick)) commands["controlpanel"].append("Set Ident %s %s" % (nick, ident)) commands["controlpanel"].append("Set RealName %s %s" % (nick, realname)) if security == "ssl": commands["controlpanel"].append("SetNetwork TrustAllCerts %s %s true" % (nick, network)) # Don't judge me commands["controlpanel"].append("AddServer %s %s %s +%s" % (nick, network, host, port)) elif security == "plain": commands["controlpanel"].append("AddServer %s %s %s %s" % (nick, network, host, port)) if not main.config["ConnectOnCreate"]: stage2commands["status"].append("Disconnect") if main.config["Toggles"]["CycleChans"]: stage2commands["status"].append("LoadMod disconkick") stage2commands["status"].append("LoadMod chansaver") deliverRelayCommands(num, commands, stage2=[[nick+"/"+network, stage2commands]]) def provisionAuthenticationData(num, nick, network, security, auth, password): commands = {} commands["status"] = [] if auth == "sasl": commands["sasl"] = [] commands["status"].append("LoadMod sasl") commands["sasl"].append("Mechanism plain") commands["sasl"].append("Set %s %s" % (nick, password)) elif auth == "ns": commands["nickserv"] = [] commands["status"].append("LoadMod nickserv") commands["nickserv"].append("Set %s" % password) deliverRelayCommands(num, commands, user=nick+"/"+network) def provisionRelay(num, network): # provision user and network data aliasObj = main.alias[num] alias = aliasObj["nick"] provisionUserNetworkData(num, *aliasObj.values(), network, main.network[network].host, main.network[network].port, main.network[network].security, main.network[network].auth, main.network[network].aliases[num]["password"]) if main.config["ConnectOnCreate"]: reactor.callLater(10, main.network[network].start_bot, num) def provisionMultipleRelays(net, relaysNeeded): numsProvisioned = [] for i in range(relaysNeeded): num, alias = main.network[net].add_relay() numsProvisioned.append(num) provisionRelay(num, net) main.saveConf("network") return numsProvisioned