from django.db import models

from apps.common.models import TeamOwnedModel, TimeStampedModel


class Asset(TeamOwnedModel):
    class Type(models.TextChoices):
        IMAGE = "image", "Image"
        VIDEO = "video", "Video"
        AUDIO = "audio", "Audio"
        SUBTITLE = "subtitle", "Subtitle"
        DOCUMENT = "document", "Document"

    class Source(models.TextChoices):
        UPLOAD = "upload", "Upload"
        AI_GENERATED = "ai_generated", "AI Generated"
        EXPORTED = "exported", "Exported"
        SYSTEM = "system", "System"

    class Category(models.TextChoices):
        PERSON = "person", "Person"
        SCENE = "scene", "Scene"
        PRODUCT_IMAGE = "product_image", "Product Image"
        VIDEO_CLIP = "video_clip", "Video Clip"
        FINAL_VIDEO = "final_video", "Final Video"
        UPLOAD = "upload", "Upload"
        UNCATEGORIZED = "uncategorized", "Uncategorized"

    name = models.CharField(max_length=255)
    asset_type = models.CharField(max_length=24, choices=Type.choices)
    source = models.CharField(max_length=32, choices=Source.choices)
    category = models.CharField(max_length=32, choices=Category.choices, default=Category.UNCATEGORIZED)
    description = models.TextField(blank=True)
    origin_task = models.ForeignKey(
        "ai.AITask",
        on_delete=models.SET_NULL,
        null=True,
        blank=True,
        related_name="generated_assets",
    )
    metadata = models.JSONField(default=dict, blank=True)
    is_deleted = models.BooleanField(default=False)

    def __str__(self) -> str:
        return self.name


class AssetFile(TimeStampedModel):
    asset = models.ForeignKey(Asset, on_delete=models.CASCADE, related_name="files")
    object_key = models.CharField(max_length=512)
    bucket = models.CharField(max_length=128)
    content_type = models.CharField(max_length=128, blank=True)
    size_bytes = models.BigIntegerField(default=0)
    checksum = models.CharField(max_length=128, blank=True)
    width = models.PositiveIntegerField(null=True, blank=True)
    height = models.PositiveIntegerField(null=True, blank=True)
    duration_ms = models.PositiveIntegerField(null=True, blank=True)
    preview_url = models.URLField(blank=True)
    is_primary = models.BooleanField(default=True)

    class Meta:
        indexes = [
            models.Index(fields=["bucket", "object_key"]),
        ]

    def __str__(self) -> str:
        return self.object_key


class AssetTag(TimeStampedModel):
    team = models.ForeignKey("accounts.Team", on_delete=models.CASCADE, related_name="asset_tags")
    name = models.CharField(max_length=64)

    class Meta:
        unique_together = [("team", "name")]

    def __str__(self) -> str:
        return self.name


class AssetTagging(TimeStampedModel):
    asset = models.ForeignKey(Asset, on_delete=models.CASCADE, related_name="taggings")
    tag = models.ForeignKey(AssetTag, on_delete=models.CASCADE, related_name="taggings")

    class Meta:
        unique_together = [("asset", "tag")]


class AssetUsage(TimeStampedModel):
    asset = models.ForeignKey(Asset, on_delete=models.CASCADE, related_name="usages")
    project = models.ForeignKey(
        "projects.Project",
        on_delete=models.CASCADE,
        null=True,
        blank=True,
        related_name="asset_usages",
    )
    usage_type = models.CharField(max_length=64)
    context = models.JSONField(default=dict, blank=True)