56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
import main
|
|
from modules import alias
|
|
from yaml import dump
|
|
|
|
|
|
class AliasCommand:
|
|
def __init__(self, *args):
|
|
self.alias(*args)
|
|
|
|
def alias(
|
|
self, addr, authed, data, obj, spl, success, failure, info, incUsage, length
|
|
):
|
|
if authed:
|
|
if length == 1:
|
|
info(dump(main.alias))
|
|
return
|
|
elif length == 3:
|
|
if spl[1] == "add":
|
|
if not spl[2].isdigit():
|
|
failure("Must be a number, not %s" % spl[2])
|
|
return
|
|
num = int(spl[2])
|
|
for i in range(num):
|
|
if len(main.alias.keys()) == 0:
|
|
nextNum = 1
|
|
else:
|
|
nextNum = max(main.alias.keys()) + 1
|
|
main.alias[nextNum] = alias.generate_alias()
|
|
success("Generated new alias: %i" % nextNum)
|
|
main.saveConf("alias")
|
|
return
|
|
elif spl[1] == "del":
|
|
if not spl[2].isdigit():
|
|
failure("Must be a number, not %s" % spl[2])
|
|
return
|
|
num = int(spl[2])
|
|
if num not in main.alias.keys():
|
|
failure("No such alias: %i" % num)
|
|
return
|
|
failed = False
|
|
for i in main.network.keys():
|
|
if num in main.network[i].aliases.keys():
|
|
failure("Alias in use by %s" % i)
|
|
failed = True
|
|
if failed:
|
|
return
|
|
del main.alias[num]
|
|
success("Removed alias: %i" % num)
|
|
main.saveConf("alias")
|
|
return
|
|
else:
|
|
incUsage("alias")
|
|
return
|
|
else:
|
|
incUsage(None)
|