import importlib
from argparse import ArgumentParser
from os import getenv

import requests
from django.core.management.base import BaseCommand, CommandError
from django.db import transaction

from engine.core.management.commands import DEEPL_TARGET_LANGUAGES_MAPPING

DEEPL_API_URL = "https://api.deepl.com/v2/translate"


class Command(BaseCommand):
    help = (
        "Translate a model field into another language via DeepL and store it "
        "in the translated_<lang> field created by django-modeltranslation."
    )

    def add_arguments(self, parser: ArgumentParser) -> None:
        parser.add_argument(
            "-t",
            "--target",
            required=True,
            help="Dotted path to the field to translate, e.g. core.models.Product.description",
        )
        parser.add_argument(
            "-l",
            "--language",
            required=True,
            help="Modeltranslation language code to translate into, e.g. de-de, fr-fr, zh-hans",
        )

    def handle(self, *args, **options) -> None:
        target = options["target"]
        lang = options["language"].lower()

        if lang not in DEEPL_TARGET_LANGUAGES_MAPPING:
            raise CommandError(f"Unknown language '{lang}'.")
        deepl_lang = DEEPL_TARGET_LANGUAGES_MAPPING[lang]
        if deepl_lang == "unsupported":
            raise CommandError(f"DeepL does not support translating into '{lang}'.")

        try:
            module_path, model_name, field_name = target.rsplit(".", 2)
        except ValueError as e:
            raise CommandError(
                "Invalid target format. Use app.module.Model.field, e.g. core.models.Product.description"
            ) from e

        try:
            module = importlib.import_module(module_path)
            model = getattr(module, model_name)
        except (ImportError, AttributeError) as e:
            raise CommandError(
                f"Could not import model '{model_name}' from '{module_path}': {e}"
            ) from e

        dest_suffix = lang.replace("-", "_")
        dest_field = f"{field_name}_{dest_suffix}"

        if not hasattr(model, dest_field):
            raise CommandError(
                f"Model '{model_name}' has no field '{dest_field}'. "
                "Did you run makemigrations/migrate after setting up modeltranslation?"
            )

        auth_key = getenv("DEEPL_AUTH_KEY")
        if not auth_key:
            raise CommandError("Environment variable DEEPL_AUTH_KEY is not set.")

        qs = model.objects.exclude(**{f"{field_name}__isnull": True}).exclude(
            **{f"{field_name}": ""}
        )
        total = qs.count()
        if total == 0:
            self.stdout.write("No instances with non-empty source field found.")
            return

        self.stdout.write(
            f"Translating {total} objects from '{field_name}' into '{dest_field}'."
        )

        for obj in qs.iterator():
            src_text = getattr(obj, field_name)
            existing = getattr(obj, dest_field, None)
            if existing:
                self.stdout.write(f"Skipping {obj.pk}: '{dest_field}' already set.")
                continue

            resp = requests.post(
                DEEPL_API_URL,
                data={
                    "text": src_text,
                    "target_lang": deepl_lang,
                },
                headers={"Authorization": f"DeepL-Auth-Key {auth_key}"},
                timeout=30,
            )
            if resp.status_code != 200:
                self.stderr.write(
                    f"DeepL API error for {obj.pk}: {resp.status_code} {resp.text}"
                )
                continue

            data = resp.json()
            try:
                translated = data["translations"][0]["text"]
            except (KeyError, IndexError):
                self.stderr.write(f"Unexpected DeepL response for {obj.pk}: {data}")
                continue

            setattr(obj, dest_field, translated)
            try:
                with transaction.atomic():
                    obj.save(update_fields=[dest_field])
            except Exception as e:
                self.stderr.write(f"Error saving {obj.pk}: {e}")
            else:
                self.stdout.write(f"✓ {obj.pk}")

        self.stdout.write(self.style.SUCCESS("Done."))