from secrets.forms import AddSecret

from admin.decorators import superuser_only
from computes.models import Compute
from django.contrib import messages
from django.http import HttpResponseRedirect
from django.shortcuts import get_object_or_404, render
from libvirt import (VIR_SECRET_USAGE_TYPE_CEPH, VIR_SECRET_USAGE_TYPE_ISCSI, VIR_SECRET_USAGE_TYPE_NONE,
                     VIR_SECRET_USAGE_TYPE_TLS, VIR_SECRET_USAGE_TYPE_VOLUME, libvirtError)
from vrtManager.secrets import wvmSecrets


@superuser_only
def secrets(request, compute_id):
    """
    :param request:
    :param compute_id:
    :return:
    """

    secrets_all = []
    compute = get_object_or_404(Compute, pk=compute_id)
    secret_usage_types = {
        VIR_SECRET_USAGE_TYPE_NONE: "none",
        VIR_SECRET_USAGE_TYPE_VOLUME: "volume",
        VIR_SECRET_USAGE_TYPE_CEPH: "ceph",
        VIR_SECRET_USAGE_TYPE_ISCSI: "iscsi",
        VIR_SECRET_USAGE_TYPE_TLS: "tls",
    }

    try:
        conn = wvmSecrets(compute.hostname, compute.login, compute.password, compute.type)
        secrets = conn.get_secrets()

        for uuid in secrets:
            secrt = conn.get_secret(uuid)
            try:
                secrt_value = conn.get_secret_value(uuid)
            except libvirtError as lib_err:
                secrt_value = None
            secrets_all.append({
                'usage': secrt.usageID(),
                'uuid': secrt.UUIDString(),
                'usageType': secret_usage_types[secrt.usageType()],
                'value': secrt_value
            })
        if request.method == 'POST':
            if 'create' in request.POST:
                form = AddSecret(request.POST)
                if form.is_valid():
                    data = form.cleaned_data
                    conn.create_secret(
                        data['ephemeral'],
                        data['private'],
                        data['usage_type'],
                        data['data'],
                    )
                    return HttpResponseRedirect(request.get_full_path())
                else:
                    for msg_err in form.errors.values():
                        messages.error(request, msg_err.as_text())
            if 'delete' in request.POST:
                uuid = request.POST.get('uuid', '')
                conn.delete_secret(uuid)
                return HttpResponseRedirect(request.get_full_path())
            if 'set_value' in request.POST:
                uuid = request.POST.get('uuid', '')
                value = request.POST.get('value', '')
                try:
                    conn.set_secret_value(uuid, value)
                except Exception as err:
                    messages.error(request, err)
                return HttpResponseRedirect(request.get_full_path())
    except libvirtError as err:
        messages.error(request, err)

    return render(request, 'secrets.html', locals())