package repository

import (
	"context"

	"with_you/internal/model"

	"gorm.io/gorm"
)

type EmptyClassroomRepository interface {
	ListByUserAndSemester(userID, semester string) ([]*model.EmptyClassroom, error)
	DeleteByUserAndSemester(ctx context.Context, userID, semester string) error
	BatchCreate(ctx context.Context, classrooms []*model.EmptyClassroom) error
}

type emptyClassroomRepository struct {
	db *gorm.DB
}

func NewEmptyClassroomRepository(db *gorm.DB) EmptyClassroomRepository {
	return &emptyClassroomRepository{db: db}
}

func (r *emptyClassroomRepository) ListByUserAndSemester(userID, semester string) ([]*model.EmptyClassroom, error) {
	var classrooms []*model.EmptyClassroom
	err := r.db.Where("user_id = ? AND semester = ?", userID, semester).
		Order("classroom ASC, weekday ASC").
		Find(&classrooms).Error
	return classrooms, err
}

func (r *emptyClassroomRepository) DeleteByUserAndSemester(ctx context.Context, userID, semester string) error {
	return r.db.WithContext(ctx).
		Where("user_id = ? AND semester = ?", userID, semester).
		Delete(&model.EmptyClassroom{}).Error
}

func (r *emptyClassroomRepository) BatchCreate(ctx context.Context, classrooms []*model.EmptyClassroom) error {
	return r.db.WithContext(ctx).CreateInBatches(classrooms, 100).Error
}