75 lines
2.0 KiB
Python
75 lines
2.0 KiB
Python
import argparse
|
|
import json
|
|
|
|
from ..cmd import Log, run
|
|
from ..nix import nix_shell
|
|
|
|
|
|
def ssh(
|
|
host: str,
|
|
user: str = "root",
|
|
password: str | None = None,
|
|
ssh_args: list[str] = [],
|
|
) -> None:
|
|
packages = ["nixpkgs#tor", "nixpkgs#openssh"]
|
|
password_args = []
|
|
if password:
|
|
packages.append("nixpkgs#sshpass")
|
|
password_args = [
|
|
"sshpass",
|
|
"-p",
|
|
password,
|
|
]
|
|
_ssh_args = [
|
|
*ssh_args,
|
|
"ssh",
|
|
"-o",
|
|
"UserKnownHostsFile=/dev/null",
|
|
"-o",
|
|
"StrictHostKeyChecking=no",
|
|
f"{user}@{host}",
|
|
]
|
|
cmd = nix_shell(packages, ["torify", *password_args, *_ssh_args])
|
|
run(cmd, log=Log.BOTH)
|
|
|
|
|
|
def qrcode_scan(picture_file: str) -> str:
|
|
return run(
|
|
nix_shell(
|
|
["nixpkgs#zbar"],
|
|
[
|
|
"zbarimg",
|
|
"--quiet",
|
|
"--raw",
|
|
picture_file,
|
|
],
|
|
),
|
|
).stdout.strip()
|
|
|
|
|
|
def main(args: argparse.Namespace) -> None:
|
|
if args.json:
|
|
with open(args.json) as file:
|
|
ssh_data = json.load(file)
|
|
ssh(host=ssh_data["address"], password=ssh_data["password"])
|
|
elif args.png:
|
|
ssh_data = json.loads(qrcode_scan(args.png))
|
|
ssh(host=ssh_data["address"], password=ssh_data["password"])
|
|
|
|
|
|
def register_parser(parser: argparse.ArgumentParser) -> None:
|
|
group = parser.add_mutually_exclusive_group(required=True)
|
|
group.add_argument(
|
|
"-j",
|
|
"--json",
|
|
help="specify the json file for ssh data (generated by starting the clan installer)",
|
|
)
|
|
group.add_argument(
|
|
"-P",
|
|
"--png",
|
|
help="specify the json file for ssh data as the qrcode image (generated by starting the clan installer)",
|
|
)
|
|
# TODO pass all args we don't parse into ssh_args, currently it fails if arg starts with -
|
|
parser.add_argument("ssh_args", nargs="*", default=[])
|
|
parser.set_defaults(func=main)
|