2019-01-26 01:57:24 +00:00
|
|
|
import main
|
|
|
|
|
from yaml import dump
|
|
|
|
|
|
2019-08-09 22:06:34 +00:00
|
|
|
class NetworkCommand:
|
2019-01-26 18:58:21 +00:00
|
|
|
def __init__(self, *args):
|
|
|
|
|
self.network(*args)
|
2019-01-26 01:57:24 +00:00
|
|
|
|
|
|
|
|
def network(self, addr, authed, data, obj, spl, success, failure, info, incUsage, length):
|
|
|
|
|
if authed:
|
|
|
|
|
if length == 7:
|
|
|
|
|
if spl[1] == "add":
|
|
|
|
|
if spl[2] in main.network.keys():
|
|
|
|
|
failure("Network already exists: %s" % spl[2])
|
|
|
|
|
return
|
|
|
|
|
if not spl[4].isdigit():
|
|
|
|
|
failure("Port must be an integer, not %s" % spl[4])
|
|
|
|
|
return
|
|
|
|
|
if not spl[5].lower() in ["ssl", "plain"]:
|
|
|
|
|
failure("Security must be ssl or plain, not %s" % spl[5])
|
|
|
|
|
return
|
|
|
|
|
if not spl[6].lower() in ["sasl", "ns", "none"]:
|
|
|
|
|
failure("Auth must be sasl, ns or none, not %s" % spl[5])
|
|
|
|
|
return
|
|
|
|
|
else:
|
|
|
|
|
main.network[spl[2]] = {"host": spl[3],
|
|
|
|
|
"port": spl[4],
|
|
|
|
|
"security": spl[5].lower(),
|
|
|
|
|
"auth": spl[6].lower()}
|
|
|
|
|
success("Successfully created network: %s" % spl[2])
|
|
|
|
|
main.saveConf("network")
|
|
|
|
|
return
|
|
|
|
|
else:
|
|
|
|
|
incUsage("network")
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
elif length == 3:
|
|
|
|
|
if spl[1] == "del":
|
|
|
|
|
if spl[2] in main.network.keys():
|
|
|
|
|
del main.network[spl[2]]
|
|
|
|
|
success("Successfully removed network: %s" % spl[2])
|
|
|
|
|
main.saveConf("network")
|
|
|
|
|
return
|
|
|
|
|
else:
|
|
|
|
|
failure("No such network: %s" % spl[2])
|
|
|
|
|
return
|
|
|
|
|
else:
|
|
|
|
|
incUsage("network")
|
|
|
|
|
return
|
|
|
|
|
elif length == 2:
|
|
|
|
|
if spl[1] == "list":
|
|
|
|
|
info(dump(main.network))
|
|
|
|
|
return
|
|
|
|
|
else:
|
|
|
|
|
incUsage("network")
|
|
|
|
|
return
|
|
|
|
|
else:
|
|
|
|
|
incUsage("network")
|
|
|
|
|
return
|
|
|
|
|
else:
|
|
|
|
|
incUsage(None)
|
