package ru.mcs.diary.lesson;

import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import java.time.LocalDate;
import java.util.List;
import java.util.Optional;

@Repository
public interface LessonRepository extends JpaRepository<Lesson, Long> {

    @Query("SELECT l FROM Lesson l JOIN FETCH l.group g LEFT JOIN FETCH g.subject " +
            "WHERE l.teacher.id = :teacherId AND l.date = :date ORDER BY l.startTime")
    List<Lesson> findByTeacherIdAndDate(@Param("teacherId") Long teacherId, @Param("date") LocalDate date);

    @Query("SELECT l FROM Lesson l JOIN FETCH l.group g LEFT JOIN FETCH g.subject " +
            "WHERE l.teacher.id = :teacherId AND l.date BETWEEN :startDate AND :endDate ORDER BY l.date, l.startTime")
    List<Lesson> findByTeacherIdAndDateBetween(@Param("teacherId") Long teacherId,
                                               @Param("startDate") LocalDate startDate,
                                               @Param("endDate") LocalDate endDate);

    Optional<Lesson> findByIdAndTeacherId(Long id, Long teacherId);

    @Query("SELECT l FROM Lesson l JOIN FETCH l.group g LEFT JOIN FETCH g.subject LEFT JOIN FETCH g.students " +
            "WHERE l.id = :id AND l.teacher.id = :teacherId")
    Optional<Lesson> findByIdWithDetails(@Param("id") Long id, @Param("teacherId") Long teacherId);

    @Query("SELECT COUNT(l) FROM Lesson l WHERE l.teacher.id = :teacherId AND l.date = :date")
    long countByTeacherIdAndDate(@Param("teacherId") Long teacherId, @Param("date") LocalDate date);
}