from django.db.models import ForeignKey, Model, IntegerField, BooleanField, CASCADE, PROTECT, OneToOneField
from boxofficefantasy.models import Season, User, Movie
from boxofficefantasy_project.utils import encode_id, decode_id

# Create your models here.
class DraftSession(Model):
    season = ForeignKey(Season, on_delete=CASCADE)
    is_active = BooleanField()
    current_nomination_index = IntegerField()

    @property
    def hashed_id(self):
        if not self.pk: return ""
        return f"{encode_id(self.pk)}"
    
    @classmethod
    def decode_id(cls, hashed_id:str) -> id:
        return decode_id(hashed_id)
    
    def save(self, *args, **kwargs):
        is_new = self.pk is None
        super().save(*args, **kwargs)
        if is_new and not hasattr(self, 'settings'):
            DraftSessionSettings.objects.create(draft_session=self)


class DraftParticipant(Model):
    draft = ForeignKey(DraftSession, on_delete=CASCADE)
    user = ForeignKey(User, on_delete=CASCADE)
    budget = IntegerField()

class DraftMoviePool(Model):
    draft = ForeignKey(DraftSession, on_delete=CASCADE)
    movie = ForeignKey(Movie, on_delete=CASCADE)
    nominated = BooleanField()

class DraftPick(Model):
    draft = ForeignKey(DraftSession, on_delete=CASCADE)
    movie = ForeignKey(Movie, on_delete=CASCADE)
    winner = ForeignKey(User, on_delete=CASCADE)
    bid_amount = IntegerField()
    nomination_order = IntegerField()

class DraftSessionSettings(Model):
    starting_budget = IntegerField(default=100)
    draft_session = OneToOneField(
        DraftSession,
        on_delete=CASCADE,
        related_name="settings"
    )

    def __str__(self):
        return f"Settings for {self.draft_session}"
    
    class Meta:
        verbose_name_plural = "Draft session settings"