evan/ewconfig
1
Fork 0
Code

Add reconnect option

Browse Source
This commit is contained in:
Evan Pratten 2023-10-23 09:52:43 -04:00
parent d036999c0c
commit 17c846897e
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
configs/scripts/guru-vpn
View File

@ -108,17 +108,23 @@ def main() -> int:
prog="guru-vpn", description="Utility for connecting to the Guru VPN"
)
ap.add_argument(
"operation", choices=["connect", "disconnect"], help="Operation to perform"
"operation",
choices=["connect", "disconnect", "reconnect"],
help="Operation to perform",
)
args = ap.parse_args()
# Ensure we can actually get credentials from the Yubikey
if not has_ykman():
print("Could not execute `ykman`. Is it installed?", file=sys.stderr);
print("Could not execute `ykman`. Is it installed?", file=sys.stderr)
return 1
# Handle subcommands
cmd_fns = {"connect": handle_connect, "disconnect": handle_disconnect}
cmd_fns = {
"connect": handle_connect,
"disconnect": handle_disconnect,
"reconnect": lambda args: handle_disconnect(args) or handle_connect(args),
}
return cmd_fns[args.operation](args)