import pghistory
from django.contrib.postgres.fields import ArrayField
from django.db import models
from django.urls import reverse
from django.utils.text import slugify
from django.conf import settings

from apps.core.history import HistoricalSlug
from apps.core.models import TrackedModel
from apps.core.choices.fields import RichChoiceField


@pghistory.track()
class Company(TrackedModel):
    name = models.CharField(max_length=255)
    slug = models.SlugField(max_length=255, unique=True)
    roles = ArrayField(
        RichChoiceField(choice_group="company_roles", domain="rides", max_length=20),
        default=list,
        blank=True,
    )
    description = models.TextField(blank=True)
    website = models.URLField(blank=True)

    # General company info
    founded_date = models.DateField(null=True, blank=True)

    # Manufacturer-specific fields
    rides_count = models.IntegerField(default=0)
    coasters_count = models.IntegerField(default=0)

    # Frontend URL
    url = models.URLField(blank=True, help_text="Frontend URL for this company")

    def __str__(self):
        return self.name

    def save(self, *args, **kwargs):
        if not self.slug:
            self.slug = slugify(self.name)

        # Generate frontend URL based on primary role
        # CRITICAL: Only MANUFACTURER and DESIGNER are for rides domain
        # OPERATOR and PROPERTY_OWNER are for parks domain and handled separately
        if self.roles:
            frontend_domain = getattr(
                settings, "FRONTEND_DOMAIN", "https://thrillwiki.com"
            )
            primary_role = self.roles[0]  # Use first role as primary

            if primary_role == "MANUFACTURER":
                self.url = f"{frontend_domain}/rides/manufacturers/{self.slug}/"
            elif primary_role == "DESIGNER":
                self.url = f"{frontend_domain}/rides/designers/{self.slug}/"
            # OPERATOR and PROPERTY_OWNER URLs are handled by parks domain, not here

        super().save(*args, **kwargs)

    def get_absolute_url(self):
        # This will need to be updated to handle different roles
        return reverse("companies:detail", kwargs={"slug": self.slug})

    @classmethod
    def get_by_slug(cls, slug):
        """Get company by current or historical slug"""
        try:
            return cls.objects.get(slug=slug), False
        except cls.DoesNotExist:
            # Check pghistory first
            try:
                from django.apps import apps
                history_model = apps.get_model('rides', f'{cls.__name__}Event')
                history_entry = (
                    history_model.objects.filter(slug=slug)
                    .order_by("-pgh_created_at")
                    .first()
                )
                if history_entry:
                    return cls.objects.get(id=history_entry.pgh_obj_id), True
            except LookupError:
                # History model doesn't exist, skip pghistory check
                pass

            # Check manual slug history as fallback
            try:
                historical = HistoricalSlug.objects.get(
                    content_type__model="company", slug=slug
                )
                return cls.objects.get(pk=historical.object_id), True
            except (HistoricalSlug.DoesNotExist, cls.DoesNotExist):
                raise cls.DoesNotExist("No company found with this slug")

    class Meta(TrackedModel.Meta):
        app_label = "rides"
        ordering = ["name"]
        verbose_name_plural = "Companies"