package sql

import (
	"context"
	"errors"
	"tercul/internal/domain"
	"tercul/internal/platform/config"
	"gorm.io/gorm"
)

type authorRepository struct {
	*BaseRepositoryImpl[domain.Author]
	db *gorm.DB
}

// NewAuthorRepository creates a new AuthorRepository.
func NewAuthorRepository(db *gorm.DB, cfg *config.Config) domain.AuthorRepository {
	baseRepo := NewBaseRepositoryImpl[domain.Author](db, cfg)
	return &authorRepository{
		BaseRepositoryImpl: baseRepo,
		db:                 db,
	}
}

func (r *authorRepository) FindByName(ctx context.Context, name string) (*domain.Author, error) {
	var author domain.Author
	err := r.db.WithContext(ctx).Where("name = ?", name).First(&author).Error
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, domain.ErrEntityNotFound
		}
		return nil, err
	}
	return &author, nil
}

func (r *authorRepository) ListByWorkID(ctx context.Context, workID uint) ([]domain.Author, error) {
	var authors []domain.Author
	err := r.db.WithContext(ctx).Joins("JOIN work_authors ON work_authors.author_id = authors.id").
		Where("work_authors.work_id = ?", workID).Find(&authors).Error
	return authors, err
}

func (r *authorRepository) ListByBookID(ctx context.Context, bookID uint) ([]domain.Author, error) {
	var authors []domain.Author
	err := r.db.WithContext(ctx).Joins("JOIN book_authors ON book_authors.author_id = authors.id").
		Where("book_authors.book_id = ?", bookID).Find(&authors).Error
	return authors, err
}

func (r *authorRepository) ListByCountryID(ctx context.Context, countryID uint) ([]domain.Author, error) {
	var authors []domain.Author
	err := r.db.WithContext(ctx).Where("country_id = ?", countryID).Find(&authors).Error
	return authors, err
}

func (r *authorRepository) GetWithTranslations(ctx context.Context, id uint) (*domain.Author, error) {
	var author domain.Author
	err := r.db.WithContext(ctx).Preload("Translations").First(&author, id).Error
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, domain.ErrEntityNotFound
		}
		return nil, err
	}
	return &author, nil
}