tercul-backend/internal/repositories/author_repository.go

package repositories

import (
	"context"
	"gorm.io/gorm"
	"tercul/internal/models"
)

// AuthorRepository defines CRUD methods specific to Author.
type AuthorRepository interface {
	BaseRepository[models.Author]
	ListByWorkID(ctx context.Context, workID uint) ([]models.Author, error)
	ListByBookID(ctx context.Context, bookID uint) ([]models.Author, error)
	ListByCountryID(ctx context.Context, countryID uint) ([]models.Author, error)
}

type authorRepository struct {
	BaseRepository[models.Author]
	db *gorm.DB
}

// NewAuthorRepository creates a new AuthorRepository.
func NewAuthorRepository(db *gorm.DB) AuthorRepository {
	return &authorRepository{
		BaseRepository: NewBaseRepositoryImpl[models.Author](db),
		db:             db,
	}
}

// ListByWorkID finds authors by work ID
func (r *authorRepository) ListByWorkID(ctx context.Context, workID uint) ([]models.Author, error) {
	var authors []models.Author
	if 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; err != nil {
		return nil, err
	}
	return authors, nil
}

// ListByBookID finds authors by book ID
func (r *authorRepository) ListByBookID(ctx context.Context, bookID uint) ([]models.Author, error) {
	var authors []models.Author
	if 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; err != nil {
		return nil, err
	}
	return authors, nil
}

// ListByCountryID finds authors by country ID
func (r *authorRepository) ListByCountryID(ctx context.Context, countryID uint) ([]models.Author, error) {
	var authors []models.Author
	if err := r.db.WithContext(ctx).Where("country_id = ?", countryID).Find(&authors).Error; err != nil {
		return nil, err
	}
	return authors, nil
}