1
0
mirror of https://github.com/osmarks/random-stuff synced 2024-10-31 19:06:15 +00:00
random-stuff/computercraft/keyctl.py
2023-11-13 19:37:05 +00:00

49 lines
2.2 KiB
Python

#!/usr/bin/env python3
import requests
import argparse
import sys
parser = argparse.ArgumentParser(description="Manage keys in SPUDNET")
parser.add_argument("--spudnet_url", "-s", required=True, help="URL of the SPUDNET instance to access")
parser.add_argument("--key", "-K", required=True)
subparsers = parser.add_subparsers(required=True)
subparsers.dest = "command"
subparsers.add_parser("info", help="Get information about this key.")
subparsers.add_parser("dependent_keys", help="List keys issued by this key.")
subparsers.add_parser("disable_key", help="Disable this key.")
issue_key = subparsers.add_parser("issue_key", help="Issue a new key.")
issue_key.add_argument("--channel", "-c", action="append", help="Allow new key access to this channel. Can be repeated. If none specified, defaults to issuing key's channels.")
issue_key.add_argument("--bearer", "-b", help="Specifies bearer of new key")
issue_key.add_argument("--use", "-u", help="Specifies intended use of new key")
issue_key.add_argument("--permission_level", "-p", help="Specifies permission level of new key. Not currently used. Must be <= issuing key's permission level", type=int)
args = parser.parse_args()
def query(subpath, data):
for key, value in list(data.items()):
if value == None: del data[key]
result = requests.post(args.spudnet_url + "/hki/" + subpath, json=data)
if not result.ok:
print(result.text)
sys.exit(1)
else:
return result.json()
if args.command == "info":
print(query("key-info", { "key": args.key }))
elif args.command == "dependent_keys":
print(query("dependent-keys", { "key": args.key }))
elif args.command == "issue_key":
print(query("issue-key", {
"key": args.key,
"bearer": args.bearer,
"use": args.use,
"permission_level": args.permission_level,
"allowed_channels": args.channel
}))
elif args.command == "disable_key":
yn = input("Are you sure? All keys issued by this key will also be disabled. This action is irreversible. (y/n) ")
if yn == "y":
print(query("disable-key", { "key": args.key }))
else:
print("Action cancelled.")