import click
import requests

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


def vm_command(command, otp, vm_name, **kwargs):
    data = {**otp.get_json(), "vm_name": vm_name, "action": command, **kwargs}
    r = requests.post(
        join_path(env_vars.get("UCLOUD_API_SERVER"), "vm", "action"), json=data
    )
    return r


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


@vm.command("create")
@click.option("--name", envvar="OTP_NAME", required=True)
@click.option("--realm", envvar="OTP_REALM", required=True)
@click.option("--seed", envvar="OTP_SEED", required=True)
@click.option("--vm-name", required=True)
@click.option("--cpu", required=True, type=int)
@click.option("--ram", required=True)
@click.option("--os-ssd", required=True)
@click.option("--hdd", default=list(), multiple=True)
@click.option("--image", required=True)
@click.option("--network", default=list(), multiple=True)
def create(name, realm, seed, vm_name, cpu, ram, os_ssd, hdd, image, network):
    data = {
        **OTPCredentials(name, realm, seed).get_json(),
        "vm_name": vm_name,
        "specs": {"cpu": cpu, "ram": ram, "os-ssd": os_ssd, "hdd": hdd},
        "network": network,
        "image": image,
    }
    r = requests.post(
        join_path(env_vars.get("UCLOUD_API_SERVER"), "vm", "create"), json=data
    )
    print(load_dump_pretty(r.content))


@vm.command("start")
@click.option("--name", envvar="OTP_NAME", required=True)
@click.option("--realm", envvar="OTP_REALM", required=True)
@click.option("--seed", envvar="OTP_SEED", required=True)
@click.option("--vm-name", required=True)
@click.option("--in_support_of")
def start(name, realm, seed, vm_name, in_support_of):
    r = vm_command(
        "start", OTPCredentials(name, realm, seed), vm_name, in_support_of=in_support_of
    )
    print(load_dump_pretty(r.content))


@vm.command("stop")
@click.option("--name", envvar="OTP_NAME", required=True)
@click.option("--realm", envvar="OTP_REALM", required=True)
@click.option("--seed", envvar="OTP_SEED", required=True)
@click.option("--vm-name", required=True)
@click.option("--in_support_of")
def stop(name, realm, seed, vm_name, in_support_of):
    r = vm_command(
        "stop", OTPCredentials(name, realm, seed), vm_name, in_support_of=in_support_of
    )
    print(load_dump_pretty(r.content))


@vm.command("delete")
@click.option("--name", envvar="OTP_NAME", required=True)
@click.option("--realm", envvar="OTP_REALM", required=True)
@click.option("--seed", envvar="OTP_SEED", required=True)
@click.option("--vm-name", required=True)
@click.option("--in_support_of")
def delete(name, realm, seed, vm_name, in_support_of):
    r = vm_command(
        "delete",
        OTPCredentials(name, realm, seed),
        vm_name,
        in_support_of=in_support_of,
    )
    print(load_dump_pretty(r.content))


@vm.command("status")
@click.option("--name", envvar="OTP_NAME", required=True)
@click.option("--realm", envvar="OTP_REALM", required=True)
@click.option("--seed", envvar="OTP_SEED", required=True)
@click.option("--vm-name", required=True)
@click.option("--in_support_of")
def status(name, realm, seed, vm_name, in_support_of):
    otp = OTPCredentials(name, realm, seed)
    data = {**otp.get_json(), "vm_name": vm_name, "in_support_of": in_support_of}
    r = requests.get(join_path(env_vars.get("UCLOUD_API_SERVER"), "vm", "status"), json=data)
    print(load_dump_pretty(r.content))


@vm.command("migrate")
@click.option("--name", envvar="OTP_NAME", required=True)
@click.option("--realm", envvar="OTP_REALM", required=True)
@click.option("--seed", envvar="OTP_SEED", required=True)
@click.option("--vm-name", required=True)
@click.option("--destination", required=True)
@click.option("--in_support_of")
def vm_migration(name, realm, seed, vm_name, destination, in_support_of):
    otp = OTPCredentials(name, realm, seed)
    data = {
        **otp.get_json(),
        "vm_name": vm_name,
        "destination": destination,
        "in_support_of": in_support_of,
    }
    r = requests.post(
        join_path(env_vars.get("UCLOUD_API_SERVER"), "vm", "migrate"), json=data
    )
    print(load_dump_pretty(r.content))