package models

import (
	"time"

	"gorm.io/gorm"
)

type Assignment struct {
	ID          uint           `gorm:"primarykey" json:"id"`
	UserID      uint           `gorm:"not null;index" json:"user_id"`
	Title       string         `gorm:"not null" json:"title"`
	Description string         `json:"description"`
	Subject     string         `json:"subject"`
	Priority    string         `gorm:"not null;default:medium" json:"priority"` // low, medium, high
	DueDate     time.Time      `gorm:"not null" json:"due_date"`
	IsCompleted bool           `gorm:"default:false" json:"is_completed"`
	CompletedAt *time.Time     `json:"completed_at,omitempty"`
	CreatedAt   time.Time      `json:"created_at"`
	UpdatedAt   time.Time      `json:"updated_at"`
	DeletedAt   gorm.DeletedAt `gorm:"index" json:"-"`

	User *User `gorm:"foreignKey:UserID" json:"user,omitempty"`
}

func (a *Assignment) IsOverdue() bool {
	return !a.IsCompleted && time.Now().After(a.DueDate)
}

func (a *Assignment) IsDueToday() bool {
	now := time.Now()
	return a.DueDate.Year() == now.Year() &&
		a.DueDate.Month() == now.Month() &&
		a.DueDate.Day() == now.Day()
}

func (a *Assignment) IsDueThisWeek() bool {
	now := time.Now()
	weekLater := now.AddDate(0, 0, 7)
	return a.DueDate.After(now) && a.DueDate.Before(weekLater)
}