from django.shortcuts import render
from django.urls import reverse
from django.views import View
from mixins.views import ObjectList, ObjectRead

from core.clients import transport
from core.views.manage.permissions import SuperUserRequiredMixin


class WhatsApp(SuperUserRequiredMixin, View):
    template_name = "pages/signal.html"
    service = "whatsapp"
    page_title = "WhatsApp"
    accounts_url_name = "whatsapp_accounts"

    def get(self, request):
        return render(
            request,
            self.template_name,
            {
                "service": self.service,
                "service_label": self.page_title,
                "accounts_url_name": self.accounts_url_name,
            },
        )


class WhatsAppAccounts(SuperUserRequiredMixin, ObjectList):
    list_template = "partials/signal-accounts.html"
    service = "whatsapp"

    context_object_name_singular = "WhatsApp Account"
    context_object_name = "WhatsApp Accounts"
    list_url_name = "whatsapp_accounts"
    list_url_args = ["type"]

    def _normalize_accounts(self, rows):
        out = []
        for item in rows or []:
            if isinstance(item, dict):
                value = (
                    item.get("number")
                    or item.get("id")
                    or item.get("jid")
                    or item.get("account")
                )
                if value:
                    out.append(str(value))
            elif item:
                out.append(str(item))
        return out

    def get_queryset(self, **kwargs):
        self.extra_context = {
            "service": "whatsapp",
            "service_label": "WhatsApp",
            "account_add_url_name": "whatsapp_account_add",
            "show_contact_actions": False,
            "service_warning": transport.get_service_warning("whatsapp"),
        }
        return self._normalize_accounts(transport.list_accounts("whatsapp"))


class WhatsAppAccountAdd(SuperUserRequiredMixin, ObjectRead):
    detail_template = "partials/whatsapp-account-add.html"
    service = "whatsapp"
    context_object_name_singular = "Add Account"
    context_object_name = "Add Account"
    detail_url_name = "whatsapp_account_add"
    detail_url_args = ["type", "device"]

    def _device_name(self) -> str:
        form_args = self.request.POST.dict()
        return form_args.get("device", "GIA Device")

    def _refresh_only(self) -> bool:
        form_args = self.request.POST.dict()
        return str(form_args.get("refresh") or "") == "1"

    def _detail_context(self, kwargs, obj):
        detail_url_args = {
            arg: kwargs[arg]
            for arg in self.detail_url_args
            if arg in kwargs
        }
        return {
            "object": obj,
            "detail_url": reverse(self.detail_url_name, kwargs=detail_url_args),
        }

    def post(self, request, *args, **kwargs):
        self.request = request
        if self._refresh_only() and request.htmx:
            obj = self.get_object(**kwargs)
            return render(
                request,
                self.detail_template,
                self._detail_context(kwargs, obj),
            )
        return super().get(request, *args, **kwargs)

    def get_object(self, **kwargs):
        device_name = self._device_name()
        if not self._refresh_only():
            transport.request_pairing(self.service, device_name)
        try:
            image_bytes = transport.get_link_qr(self.service, device_name)
            return {
                "ok": True,
                "image_b64": transport.image_bytes_to_base64(image_bytes),
                "warning": transport.get_service_warning(self.service),
            }
        except Exception as exc:
            error_text = str(exc)
            return {
                "ok": False,
                "pending": "pairing qr" in error_text.lower(),
                "device": device_name,
                "error": error_text,
                "warning": transport.get_service_warning(self.service),
            }