chore: build

backend/internal/modules/media/repository.go

@@ -13,7 +13,7 @@ var ErrNotFound = errors.New("media not found")
 type Repository interface {
 	Create(ctx context.Context, m *Media) error
 	FindByID(ctx context.Context, id uuid.UUID) (*Media, error)
-	List(ctx context.Context) ([]*Media, error)
+	List(ctx context.Context, productID *uuid.UUID) ([]*Media, error)
 	Update(ctx context.Context, m *Media) error
 	Delete(ctx context.Context, id uuid.UUID) error
 }
@@ -32,21 +32,38 @@ func (r *gormRepository) Create(ctx context.Context, m *Media) error {
 
 func (r *gormRepository) FindByID(ctx context.Context, id uuid.UUID) (*Media, error) {
 	var m Media
-	err := r.db.WithContext(ctx).Where("id = ?", id).First(&m).Error
+
+	err := r.db.WithContext(ctx).
+		Where("id = ?", id).
+		First(&m).Error
+
 	if errors.Is(err, gorm.ErrRecordNotFound) {
 		return nil, ErrNotFound
 	}
+
 	if err != nil {
 		return nil, err
 	}
+
 	return &m, nil
 }
 
-func (r *gormRepository) List(ctx context.Context) ([]*Media, error) {
+func (r *gormRepository) List(ctx context.Context, productID *uuid.UUID) ([]*Media, error) {
 	var list []*Media
-	if err := r.db.WithContext(ctx).Order("created_at desc").Find(&list).Error; err != nil {
+
+	q := r.db.WithContext(ctx).
+		Order("created_at desc")
+
+	if productID != nil {
+		q = q.Where("product_id = ?", *productID)
+	} else {
+		q = q.Where("product_id IS NULL")
+	}
+
+	if err := q.Find(&list).Error; err != nil {
 		return nil, err
 	}
+
 	return list, nil
 }
 
@@ -55,12 +72,16 @@ func (r *gormRepository) Update(ctx context.Context, m *Media) error {
 }
 
 func (r *gormRepository) Delete(ctx context.Context, id uuid.UUID) error {
-	res := r.db.WithContext(ctx).Delete(&Media{}, "id = ?", id)
+	res := r.db.WithContext(ctx).
+		Delete(&Media{}, "id = ?", id)
+
 	if res.Error != nil {
 		return res.Error
 	}
+
 	if res.RowsAffected == 0 {
 		return ErrNotFound
 	}
+
 	return nil
 }