from uncloud_cli.commands.helper import OTPCredentials, load_dump_pretty
from uncloud_cli.config import env_vars
from os.path import join as join_path

import click
import requests


@click.group()
def user():
    pass


@user.command("files")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
def list_files(name, realm, seed):
    data = OTPCredentials(name, realm, seed).get_json()
    r = requests.get(
        join_path(env_vars.get("UCLOUD_API_SERVER"), "user", "files"), json=data
    )
    print(load_dump_pretty(r.content))


@user.command("vms")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
def list_vms(name, realm, seed):
    data = OTPCredentials(name, realm, seed).get_json()
    r = requests.get(
        join_path(env_vars.get("UCLOUD_API_SERVER"), "user", "vms"), json=data
    )
    print(load_dump_pretty(r.content))


@user.command("networks")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
def list_networks(name, realm, seed):
    data = OTPCredentials(name, realm, seed).get_json()
    r = requests.get(
        join_path(env_vars.get("UCLOUD_API_SERVER"), "user", "networks"), json=data
    )
    print(load_dump_pretty(r.content))


@user.command("add-ssh")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
@click.option("--key-name", required=True)
@click.option("--key", required=True)
def add_ssh(name, realm, seed, key_name, key):
    otp = OTPCredentials(name, realm, seed)
    data = {**otp.get_json(), "key_name": key_name, "key": key}
    r = requests.post(
        join_path(env_vars.get("UCLOUD_API_SERVER"), "user", "add-ssh"), json=data
    )
    print(load_dump_pretty(r.content))


@user.command("remove-ssh")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
@click.option("--key-name", required=True)
def remove_ssh(name, realm, seed, key_name):
    otp = OTPCredentials(name, realm, seed)
    data = {**otp.get_json(), "key_name": key_name}
    r = requests.get(
        join_path(env_vars.get("UCLOUD_API_SERVER"), "user", "remove-ssh"), json=data
    )
    print(load_dump_pretty(r.content))


@user.command("get-ssh")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
@click.option("--key-name", default="")
def get_ssh(name, realm, seed, key_name):
    otp = OTPCredentials(name, realm, seed)
    data = {**otp.get_json(), "key_name": key_name}
    r = requests.get(
        join_path(env_vars.get("UCLOUD_API_SERVER"), "user", "get-ssh"), json=data
    )
    print(load_dump_pretty(r.content))