"""
Notification models for KeepItGoing.
"""

import uuid
from django.conf import settings
from django.db import models


class Notification(models.Model):
    """
    Stores notifications for users.
    """
    NOTIFICATION_TYPES = [
        ('reminder', 'Task Reminder'),
        ('due_soon', 'Due Soon'),
        ('overdue', 'Overdue'),
        ('shared', 'Task Shared'),
        ('comment', 'Comment'),
        ('daily_email', 'Daily Email'),
    ]

    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    user = models.ForeignKey(
        settings.AUTH_USER_MODEL,
        on_delete=models.CASCADE,
        related_name='notifications'
    )
    notification_type = models.CharField(max_length=20, choices=NOTIFICATION_TYPES)
    title = models.CharField(max_length=255)
    message = models.TextField()
    task = models.ForeignKey(
        'tasks.Task',
        on_delete=models.CASCADE,
        null=True,
        blank=True,
        related_name='notifications'
    )
    is_read = models.BooleanField(default=False)
    read_at = models.DateTimeField(null=True, blank=True)
    created_at = models.DateTimeField(auto_now_add=True)

    class Meta:
        db_table = 'notifications'
        ordering = ['-created_at']

    def __str__(self):
        return f"{self.notification_type}: {self.title}"


class ScheduledReminder(models.Model):
    """
    Tracks scheduled reminders for tasks.
    """
    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    task = models.ForeignKey(
        'tasks.Task',
        on_delete=models.CASCADE,
        related_name='scheduled_reminders'
    )
    remind_at = models.DateTimeField()
    is_sent = models.BooleanField(default=False)
    sent_at = models.DateTimeField(null=True, blank=True)
    created_at = models.DateTimeField(auto_now_add=True)

    class Meta:
        db_table = 'scheduled_reminders'
        ordering = ['remind_at']

    def __str__(self):
        return f"Reminder for {self.task.title} at {self.remind_at}"