import datetime
import enum
from typing import List, Optional
from sqlalchemy.orm import Mapped, mapped_column, relationship
from app.extensions import db
from app.helpers import get_jakarta_now

enrollments = db.Table('enrollments',
    db.Column('user_id', db.Integer, db.ForeignKey('users.id'), primary_key=True),
    db.Column('course_id', db.Integer, db.ForeignKey('courses.id'), primary_key=True)
)


class KbmActivityType(enum.Enum):
    TEORI = 'teori'
    PRAKTIKUM = 'praktikum'
    UJIAN = 'ujian'
    TUGAS = 'tugas'
    REVIEW = 'review'
    LAINNYA = 'lainnya'


class KbmNote(db.Model):
    """Catatan Kegiatan Belajar Mengajar (KBM)"""
    __tablename__ = 'kbm_notes'

    id: Mapped[int] = mapped_column(db.Integer, primary_key=True)
    course_id: Mapped[int] = mapped_column(db.Integer, db.ForeignKey('courses.id'), nullable=False, index=True)
    teacher_id: Mapped[int] = mapped_column(db.Integer, db.ForeignKey('users.id'), nullable=False, index=True)
    
    # Activity details
    activity_date: Mapped[datetime.datetime] = mapped_column(db.DateTime, nullable=False)
    start_time: Mapped[Optional[datetime.time]] = mapped_column(db.Time, nullable=True)
    end_time: Mapped[Optional[datetime.time]] = mapped_column(db.Time, nullable=True)
    activity_type: Mapped[KbmActivityType] = mapped_column(db.Enum(KbmActivityType), nullable=False, default=KbmActivityType.TEORI)
    topic: Mapped[str] = mapped_column(db.String(200), nullable=False)  # Materi pokok
    description: Mapped[Optional[str]] = mapped_column(db.Text, nullable=True)  # Detail KBM
    notes: Mapped[Optional[str]] = mapped_column(db.Text, nullable=True)  # Catatan tambahan (refleksi, dll)
    
    # Metadata
    created_at: Mapped[datetime] = mapped_column(db.DateTime, default=get_jakarta_now)
    updated_at: Mapped[datetime] = mapped_column(db.DateTime, default=get_jakarta_now, onupdate=get_jakarta_now)

    # Relationships
    course = relationship('Course', back_populates='kbm_notes')
    teacher = relationship('User', foreign_keys=[teacher_id])

    def to_dict(self):
        return {
            'id': self.id,
            'course_id': self.course_id,
            'teacher_id': self.teacher_id,
            'activity_date': self.activity_date.strftime('%Y-%m-%d') if self.activity_date else None,
            'start_time': self.start_time.strftime('%H:%M') if self.start_time else None,
            'end_time': self.end_time.strftime('%H:%M') if self.end_time else None,
            'activity_type': self.activity_type.value,
            'topic': self.topic,
            'description': self.description,
            'notes': self.notes,
            'created_at': self.created_at.strftime('%Y-%m-%d %H:%M:%S') if self.created_at else None,
            'updated_at': self.updated_at.strftime('%Y-%m-%d %H:%M:%S') if self.updated_at else None,
        }

    def __repr__(self):
        return f'<KbmNote {self.topic} on {self.activity_date}>'


class AcademicYear(db.Model):
    __tablename__ = 'academic_years'

    id: Mapped[int] = mapped_column(db.Integer, primary_key=True)
    year: Mapped[str] = mapped_column(db.String(20), nullable=False)
    is_active: Mapped[bool] = mapped_column(db.Boolean, default=False, nullable=False)
    school_id: Mapped[int] = mapped_column(
        db.Integer, db.ForeignKey('schools.id'), nullable=True, index=True
    )

    __table_args__ = (
        db.UniqueConstraint('year', 'school_id', name='uq_academic_year_school'),
    )

    courses: Mapped[List['Course']] = relationship('Course', backref='academic_year', lazy='dynamic')
    school = relationship('School', back_populates='academic_years')

    def __repr__(self) -> str:
        return f'<AcademicYear {self.year}>'


class Course(db.Model):
    __tablename__ = 'courses'

    id: Mapped[int] = mapped_column(db.Integer, primary_key=True)
    name: Mapped[str] = mapped_column(db.String(150), nullable=False)
    class_code: Mapped[str] = mapped_column(db.String(8), nullable=False, index=True)
    color: Mapped[str] = mapped_column(db.String(7), nullable=False, default='#0282c6')

    academic_year_id: Mapped[int] = mapped_column(db.Integer, db.ForeignKey('academic_years.id'), nullable=False)
    teacher_id: Mapped[int] = mapped_column(db.Integer, db.ForeignKey('users.id'), nullable=False)

    teacher: Mapped['User'] = relationship('User', back_populates='courses_taught')
    students: Mapped[List['User']] = relationship('User', secondary=enrollments, lazy='subquery', back_populates='courses_enrolled')

    quizzes: Mapped[List['Quiz']] = relationship('Quiz', back_populates='course', lazy='dynamic', cascade='all, delete-orphan')
    assignments: Mapped[List['Assignment']] = relationship('Assignment', back_populates='course', lazy='dynamic', cascade='all, delete-orphan')
    links: Mapped[List['Link']] = relationship('Link', back_populates='course', lazy='dynamic', cascade='all, delete-orphan')
    files: Mapped[List['File']] = relationship('File', back_populates='course', lazy='dynamic', cascade='all, delete-orphan')
    discussions: Mapped[List['Discussion']] = relationship('Discussion', back_populates='course', lazy='dynamic', cascade='all, delete-orphan')
    kbm_notes: Mapped[List['KbmNote']] = relationship('KbmNote', back_populates='course', lazy='dynamic', cascade='all, delete-orphan')
    content_folders: Mapped[List['ContentFolder']] = relationship('ContentFolder', back_populates='course', lazy='dynamic', cascade='all, delete-orphan')

    # Gradebook relationships
    grade_categories: Mapped[List['GradeCategory']] = relationship('GradeCategory', back_populates='course', lazy='dynamic', cascade='all, delete-orphan')
    learning_objectives: Mapped[List['LearningObjective']] = relationship('LearningObjective', back_populates='course', lazy='dynamic', cascade='all, delete-orphan')
    grade_items: Mapped[List['GradeItem']] = relationship('GradeItem', back_populates='course', lazy='dynamic', cascade='all, delete-orphan')
    
    # Rasch Model relationship
    rasch_analyses: Mapped[List['RaschAnalysis']] = relationship('RaschAnalysis', back_populates='course', lazy='dynamic', cascade='all, delete-orphan')

    def __repr__(self) -> str:
        return f'<Course {self.name}>'


class UserCourseOrder(db.Model):
    __tablename__ = 'user_course_orders'

    user_id: Mapped[int] = mapped_column(db.Integer, db.ForeignKey('users.id'), primary_key=True)
    course_id: Mapped[int] = mapped_column(db.Integer, db.ForeignKey('courses.id'), primary_key=True)
    manual_order: Mapped[int] = mapped_column(db.Integer, default=0, nullable=False)

    def __repr__(self) -> str:
        return f'<UserCourseOrder User:{self.user_id} Course:{self.course_id} Order:{self.manual_order}>'


class Link(db.Model):
    __tablename__ = 'links'

    id: Mapped[int] = mapped_column(db.Integer, primary_key=True)
    name: Mapped[str] = mapped_column(db.String(200), nullable=False)
    url: Mapped[str] = mapped_column(db.String(500), nullable=False)
    course_id: Mapped[int] = mapped_column(db.Integer, db.ForeignKey('courses.id'), nullable=False, index=True)

    # Folder organization
    folder_id: Mapped[Optional[int]] = mapped_column(db.Integer, db.ForeignKey('content_folders.id'), nullable=True, index=True)
    order: Mapped[int] = mapped_column(db.Integer, default=0, nullable=False)

    # Archive status
    is_archived: Mapped[bool] = mapped_column(db.Boolean, default=False, nullable=False, index=True)

    created_at: Mapped[datetime.datetime] = mapped_column(db.DateTime, default=get_jakarta_now)

    course: Mapped[Course] = relationship('Course', back_populates='links')
    folder = relationship('ContentFolder', back_populates='links', foreign_keys=[folder_id])

    def __repr__(self) -> str:
        return f'<Link {self.name}>'


class File(db.Model):
    __tablename__ = 'files'

    id: Mapped[int] = mapped_column(db.Integer, primary_key=True)
    name: Mapped[str] = mapped_column(db.String(200), nullable=False)
    description: Mapped[Optional[str]] = mapped_column(db.Text, nullable=True)
    filename: Mapped[str] = mapped_column(db.String(200), nullable=False)
    course_id: Mapped[int] = mapped_column(db.Integer, db.ForeignKey('courses.id'), nullable=False, index=True)
    start_date: Mapped[Optional[datetime.datetime]] = mapped_column(db.DateTime, nullable=True)
    end_date: Mapped[Optional[datetime.datetime]] = mapped_column(db.DateTime, nullable=True)

    # Archive status
    is_archived: Mapped[bool] = mapped_column(db.Boolean, default=False, nullable=False, index=True)

    # Folder organization
    folder_id: Mapped[Optional[int]] = mapped_column(db.Integer, db.ForeignKey('content_folders.id'), nullable=True, index=True)
    order: Mapped[int] = mapped_column(db.Integer, default=0, nullable=False)

    created_at: Mapped[datetime.datetime] = mapped_column(db.DateTime, default=get_jakarta_now)

    course: Mapped['Course'] = relationship('Course', back_populates='files')
    folder = relationship('ContentFolder', back_populates='files', foreign_keys=[folder_id])

    def __repr__(self) -> str:
        return f'<File {self.name}>'