mirror of
https://github.com/SamyRai/tercul-backend.git
synced 2025-12-27 00:31:35 +00:00
0395df3ff0
This commit includes a major refactoring of the GORM many-to-many relationships to use explicit join tables, improving stability and compatibility with GORM's features. It also implements a large number of previously unimplemented GraphQL mutations for core entities like Collections, Comments, Likes, and Bookmarks. Key changes: - Refactored polymorphic many-to-many relationships for Copyright and Monetization to use standard many-to-many with explicit join tables. - Implemented GraphQL mutations for Collection, Comment, Like, and Bookmark entities, including input validation and authorization checks. - Added comprehensive integration tests for all new features and refactored code. - Refactored the GraphQL integration test suite to be type-safe, using generics for response handling as requested. - Updated repository interfaces and implementations to support the new data model. - Updated the TODO.md file to reflect the completed work.
69 lines
2.2 KiB
Go
69 lines
2.2 KiB
Go
package sql
|
|
|
|
import (
|
|
"context"
|
|
"tercul/internal/domain"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type collectionRepository struct {
|
|
domain.BaseRepository[domain.Collection]
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewCollectionRepository creates a new CollectionRepository.
|
|
func NewCollectionRepository(db *gorm.DB) domain.CollectionRepository {
|
|
return &collectionRepository{
|
|
BaseRepository: NewBaseRepositoryImpl[domain.Collection](db),
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
// ListByUserID finds collections by user ID
|
|
func (r *collectionRepository) ListByUserID(ctx context.Context, userID uint) ([]domain.Collection, error) {
|
|
var collections []domain.Collection
|
|
if err := r.db.WithContext(ctx).Where("user_id = ?", userID).Find(&collections).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return collections, nil
|
|
}
|
|
|
|
// AddWorkToCollection adds a work to a collection
|
|
func (r *collectionRepository) AddWorkToCollection(ctx context.Context, collectionID uint, workID uint) error {
|
|
collection := &domain.Collection{}
|
|
collection.ID = collectionID
|
|
work := &domain.Work{}
|
|
work.ID = workID
|
|
return r.db.WithContext(ctx).Model(collection).Association("Works").Append(work)
|
|
}
|
|
|
|
// RemoveWorkFromCollection removes a work from a collection
|
|
func (r *collectionRepository) RemoveWorkFromCollection(ctx context.Context, collectionID uint, workID uint) error {
|
|
collection := &domain.Collection{}
|
|
collection.ID = collectionID
|
|
work := &domain.Work{}
|
|
work.ID = workID
|
|
return r.db.WithContext(ctx).Model(collection).Association("Works").Delete(work)
|
|
}
|
|
|
|
// ListPublic finds public collections
|
|
func (r *collectionRepository) ListPublic(ctx context.Context) ([]domain.Collection, error) {
|
|
var collections []domain.Collection
|
|
if err := r.db.WithContext(ctx).Where("is_public = ?", true).Find(&collections).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return collections, nil
|
|
}
|
|
|
|
// ListByWorkID finds collections by work ID
|
|
func (r *collectionRepository) ListByWorkID(ctx context.Context, workID uint) ([]domain.Collection, error) {
|
|
var collections []domain.Collection
|
|
if err := r.db.WithContext(ctx).Joins("JOIN collection_works ON collection_works.collection_id = collections.id").
|
|
Where("collection_works.work_id = ?", workID).
|
|
Find(&collections).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return collections, nil
|
|
}
|