from django.contrib.postgres.fields import ArrayField
from django.db import models
from django.utils.text import slugify
from apps.core.models import TrackedModel
import pghistory


@pghistory.track()
class Company(TrackedModel):

    # Import managers
    from ..managers import CompanyManager

    objects = CompanyManager()

    class CompanyRole(models.TextChoices):
        OPERATOR = "OPERATOR", "Park Operator"
        PROPERTY_OWNER = "PROPERTY_OWNER", "Property Owner"

    name = models.CharField(max_length=255)
    slug = models.SlugField(max_length=255, unique=True)
    roles = ArrayField(
        models.CharField(max_length=20, choices=CompanyRole.choices),
        default=list,
        blank=True,
    )
    description = models.TextField(blank=True)
    website = models.URLField(blank=True)

    # Operator-specific fields
    founded_year = models.PositiveIntegerField(blank=True, null=True)
    parks_count = models.IntegerField(default=0)
    rides_count = models.IntegerField(default=0)

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

    def __str__(self):
        return self.name

    class Meta:
        app_label = "parks"
        ordering = ["name"]
        verbose_name_plural = "Companies"


class CompanyHeadquarters(models.Model):
    """
    Simple address storage for company headquarters without coordinate tracking.
    Focus on human-readable location information for display purposes.
    """

    # Relationships
    company = models.OneToOneField(
        "Company", on_delete=models.CASCADE, related_name="headquarters"
    )

    # Address Fields (No coordinates needed)
    street_address = models.CharField(
        max_length=255,
        blank=True,
        help_text="Mailing address if publicly available",
    )
    city = models.CharField(
        max_length=100, db_index=True, help_text="Headquarters city"
    )
    state_province = models.CharField(
        max_length=100,
        blank=True,
        db_index=True,
        help_text="State/Province/Region",
    )
    country = models.CharField(
        max_length=100,
        default="USA",
        db_index=True,
        help_text="Country where headquarters is located",
    )
    postal_code = models.CharField(
        max_length=20, blank=True, help_text="ZIP or postal code"
    )

    # Optional mailing address if different or more complete
    mailing_address = models.TextField(
        blank=True,
        help_text="Complete mailing address if different from basic address",
    )

    # Metadata
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

    @property
    def formatted_location(self):
        """Returns a formatted address string for display."""
        components = []
        if self.street_address:
            components.append(self.street_address)
        if self.city:
            components.append(self.city)
        if self.state_province:
            components.append(self.state_province)
        if self.postal_code:
            components.append(self.postal_code)
        if self.country and self.country != "USA":
            components.append(self.country)
        return (
            ", ".join(components)
            if components
            else f"{
                self.city}, {
                self.country}"
        )

    @property
    def location_display(self):
        """Simple city, state/country display for compact views."""
        parts = [self.city]
        if self.state_province:
            parts.append(self.state_province)
        elif self.country != "USA":
            parts.append(self.country)
        return ", ".join(parts) if parts else "Unknown Location"

    def __str__(self):
        return f"{self.company.name} Headquarters - {self.location_display}"

    class Meta:
        verbose_name = "Company Headquarters"
        verbose_name_plural = "Company Headquarters"
        ordering = ["company__name"]
        indexes = [
            models.Index(fields=["city", "country"]),
        ]