import main from modules import provision from utils.logging.log import * from utils.logging.debug import * from copy import deepcopy def needToRegister(net): inst = selectInst(net) if "register" in inst.keys(): if inst["register"]: return True else: return False def selectInst(net): if net in main.irc.keys(): inst = deepcopy(main.irc[net]) for i in main.irc["_"].keys(): if not i in inst: inst[i] = main.irc["_"][i] else: inst = main.irc["_"] return inst def registerAccount(net, num): debug("Attempting to register: %s - %i" % (net, num)) alias = main.alias[num] nickname = alias["nick"] username = nickname+"/"+net password = main.network[net].aliases[num]["password"] inst = selectInst(net) if not inst["register"]: error("Cannot register for %s: function disabled" % (net)) return False entity = inst["entity"] email = inst["email"] cmd = inst["register"] email = email.replace("{nickname}", nickname) cmd = cmd.replace("{password}", password) cmd = cmd.replace("{email}", email) name = net+str(num) main.IRCPool[name].msg(entity, cmd) def confirmAccount(net, num, token): inst = selectInst(net) entity = inst["entity"] cmd = inst["confirm"] cmd = cmd.replace("{token}", token) name = net+str(num) main.IRCPool[name].msg(entity, cmd) enableAuthentication(net, num) def confirmRegistration(net, num): obj = main.network[net] name = net+str(num) if name in main.IRCPool.keys(): debug("Relay authenticated: %s - %i" %(net, num)) main.IRCPool[name].authenticated = True main.IRCPool[name].recheckList() if obj.relays[num]["registered"]: return if name in main.IRCPool.keys(): if main.IRCPool[name]._regAttempt: main.IRCPool[name]._regAttempt.cancel() obj.relays[num]["registered"] = True main.saveConf("network") def enableAuthentication(net, num): obj = main.network[net] nick = main.alias[num]["nick"] security = obj.security auth = obj.auth password = obj.aliases[num]["password"] uname = main.alias[num]["nick"]+"/"+net provision.provisionAuthenticationData(num, nick, net, security, auth, password) # Set up for auth main.IRCPool[net+str(num)].msg(main.config["Tweaks"]["ZNC"]["Prefix"]+"status", "Jump") if selectInst(net)["check"] == False: confirmRegistration(net, num) def registerTest(c): inst = selectInst(c["net"]) if inst["check"] == False: return if "msg" in c.keys() and not c["msg"] == None: if inst["checktype"] == "msg": if c["type"] == "query": if inst["checkmsg"] in c["msg"] and c["nick"] == inst["entity"]: confirmRegistration(c["net"], c["num"]) return if inst["ping"]: if inst["checkmsg2"] in c["msg"] and c["nick"] == inst["entity"]: confirmRegistration(c["net"], c["num"]) return elif inst["checktype"] == "mode": if c["type"] == "self": if c["mtype"] == "mode": if inst["checkmode"] in c["mode"] and c["status"] == True: confirmRegistration(c["net"], c["num"]) return