from django.db import models
from django.core.validators import MaxValueValidator, MinValueValidator
from django.utils import timezone
from django.contrib.auth import get_user_model

import os.path

from mood.models import Mood
from common.helpers import get_upload_path

from colorfield.fields import ColorField

class Theme(models.Model):
    user = models.ForeignKey(get_user_model(), models.CASCADE)
    name = models.CharField(max_length=64)
    icon = models.CharField(default="fas fa-bed", max_length=64)
    color = ColorField(default="#000000")

    def __str__(self):
        return self.name

class Dream(models.Model):
    class DreamTypes(models.IntegerChoices):
        NIGHT = 0, 'Night (main) sleep'
        DAY = 1, 'Daydream'
        NAP = 2, 'Napping'

    user = models.ForeignKey(get_user_model(), models.CASCADE)
    timestamp = models.DateTimeField(default=timezone.now)
    title = models.CharField(max_length=64)
    content = models.TextField()
    type = models.IntegerField(choices=DreamTypes.choices)
    mood = models.ForeignKey(Mood, models.SET_NULL, null=True)
    lucid = models.BooleanField(default=False)
    wet = models.BooleanField(default=False)

    @property
    def short_text(self):
        return self.title or self.content[:64]

    @property
    def theme_set(self):
        return [theme.theme for theme in self.dreamtheme_set.all()]

class DreamTheme(models.Model):
    dream = models.ForeignKey(Dream, models.CASCADE)
    theme = models.ForeignKey(Theme, models.CASCADE)

class DreamMedia(models.Model):
    dream = models.ForeignKey(Dream, models.CASCADE)
    media = models.FileField(get_upload_path)

    @property
    def basename(self):
        return os.path.basename(self.media.name)

class ThemeRating(models.Model):
    theme = models.ForeignKey(Theme, models.CASCADE)
    name = models.CharField(max_length=64)
    icon = models.CharField(default="fas fa-star", max_length=64)
    color = ColorField(default="#000000")
    value = models.IntegerField(validators=[MinValueValidator(0), MaxValueValidator(255)])

class DreamThemeRating(models.Model):
    dream = models.ForeignKey(Dream, models.CASCADE)
    theme_rating = models.ForeignKey(ThemeRating, models.SET_NULL, null=True)
    comment = models.TextField(null=True, blank=True)