from django.db import models
from django.contrib.auth.models import User


class SavedStation(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='saved_stations')
    name = models.CharField(max_length=200)
    url = models.URLField(max_length=500)
    bitrate = models.CharField(max_length=20, blank=True)
    country = models.CharField(max_length=100, blank=True)
    tags = models.CharField(max_length=200, blank=True)
    favicon_url = models.URLField(max_length=500, blank=True)
    is_favorite = models.BooleanField(default=False)
    notes = models.TextField(blank=True)
    created_at = models.DateTimeField(auto_now_add=True)

    class Meta:
        unique_together = ('user', 'url')
        ordering = ['-is_favorite', 'name']

    def __str__(self):
        return f"{self.name} ({self.user.username})"


class StationPlay(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='station_plays')
    station_name = models.CharField(max_length=200)
    station_url = models.URLField(max_length=500)
    started_at = models.DateTimeField(auto_now_add=True)
    ended_at = models.DateTimeField(null=True, blank=True)
    hour_of_day = models.IntegerField()        # 0–23, set on save
    day_of_week = models.IntegerField()        # 0=Mon … 6=Sun, set on save
    is_weekend = models.BooleanField()         # True if day_of_week >= 5

    class Meta:
        ordering = ['-started_at']

    def save(self, *args, **kwargs):
        from django.utils import timezone
        now = timezone.now()
        self.hour_of_day = now.hour
        self.day_of_week = now.weekday()
        self.is_weekend = self.day_of_week >= 5
        super().save(*args, **kwargs)

    def __str__(self):
        return f"{self.station_name} by {self.user.username} at {self.started_at}"


class FocusSession(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='focus_sessions')
    station_name = models.CharField(max_length=200, blank=True)
    completed_at = models.DateTimeField(auto_now_add=True)
    duration_minutes = models.IntegerField(default=25)

    class Meta:
        ordering = ['-completed_at']


class TrackHistory(models.Model):
    user = models.ForeignKey(
        User, null=True, blank=True, on_delete=models.SET_NULL, related_name='track_history'
    )
    session_key = models.CharField(max_length=40, blank=True)  # for anonymous users
    station_name = models.CharField(max_length=200)
    track = models.CharField(max_length=500)
    played_at = models.DateTimeField(auto_now_add=True)
    scrobbled = models.BooleanField(default=False)

    class Meta:
        ordering = ['-played_at']

    def __str__(self):
        return f"{self.track} @ {self.station_name}"