From 1015991e71679a6eab7249fce02727fb8245000f Mon Sep 17 00:00:00 2001 From: Yannik Schmidt Date: Fri, 21 Jul 2023 16:51:10 +0200 Subject: [PATCH] feat: support authentication --- signal-query-dispatch.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/signal-query-dispatch.py b/signal-query-dispatch.py index d120fac..dd602e2 100755 --- a/signal-query-dispatch.py +++ b/signal-query-dispatch.py @@ -18,7 +18,8 @@ def signal_send(user, message): def confirm_dispatch(target, uid): '''Confirm to server that message has been dispatched and can be removed''' - response = requests.post(target + "/confirm-dispatch", json=[{ "uid" : uid }]) + response = requests.post(target + "/confirm-dispatch", json=[{ "uid" : uid }], + auth=(args.user, args.password)) if response.status_code not in [200, 204]: print("Failed to confirm disptach with server for {} ({})".format(uid, response.text), file=sys.stderr) @@ -28,17 +29,22 @@ if __name__ == "__main__": parser = argparse.ArgumentParser(description='Query Atlantis Dispatch for Signal', formatter_class=argparse.ArgumentDefaultsHelpFormatter) + parser.add_argument('--target', required=True) parser.add_argument('--method', default="signal") parser.add_argument('--no-confirm', action="store_true") parser.add_argument('--signal-cli-bin') + parser.add_argument('--user') + parser.add_argument('--password') + args = parser.parse_args() if args.signal_cli_bin: signal_cli_bin = args.signal_cli_bin - response = requests.get(args.target + "/get-dispatch?method={}".format(args.method)) + response = requests.get(args.target + "/get-dispatch?method={}".format(args.method), + auth=(args.user, args.password)) # check status # if response.status_code == HTTP_NOT_FOUND: