package repository

import (
	"evanpage-backend/internal/domain"

	"gorm.io/gorm"
)

type BookmarkRepository struct {
	db *gorm.DB
}

func NewBookmarkRepository(db *gorm.DB) *BookmarkRepository {
	return &BookmarkRepository{db: db}
}

func (r *BookmarkRepository) Create(bm *domain.Bookmark) error {
	return r.db.Create(bm).Error
}

func (r *BookmarkRepository) FindByID(id uint) (*domain.Bookmark, error) {
	var bm domain.Bookmark
	err := r.db.First(&bm, id).Error
	return &bm, err
}

func (r *BookmarkRepository) ListByUser(userID uint) ([]domain.Bookmark, error) {
	var bms []domain.Bookmark
	err := r.db.Where("user_id = ?", userID).Order("sort_order asc, created_at desc").Find(&bms).Error
	return bms, err
}

func (r *BookmarkRepository) ListByUserAndCategory(userID uint, category string) ([]domain.Bookmark, error) {
	var bms []domain.Bookmark
	err := r.db.Where("user_id = ? AND category = ?", userID, category).Order("sort_order asc, created_at desc").Find(&bms).Error
	return bms, err
}

func (r *BookmarkRepository) ListCategoriesByUser(userID uint) ([]string, error) {
	var categories []string
	err := r.db.Model(&domain.Bookmark{}).Where("user_id = ?", userID).Distinct().Pluck("category", &categories).Error
	return categories, err
}

func (r *BookmarkRepository) Update(bm *domain.Bookmark) error {
	return r.db.Save(bm).Error
}

func (r *BookmarkRepository) Delete(id uint) error {
	return r.db.Delete(&domain.Bookmark{}, id).Error
}