package repository

import (
	"context"

	"with_you/internal/model"

	"gorm.io/gorm"
)

// ConversationVersionLogRepository 会话版本日志仓储接口
type ConversationVersionLogRepository interface {
	Create(ctx context.Context, log *model.ConversationVersionLog) error
	BatchCreate(ctx context.Context, logs []*model.ConversationVersionLog) error
	GetByVersion(ctx context.Context, userID string, sinceVersion int64, limit int) ([]*model.ConversationVersionLog, error)
	GetMaxVersion(ctx context.Context, userID string) (int64, error)
}

// conversationVersionLogRepository GORM 实现
type conversationVersionLogRepository struct {
	db *gorm.DB
}

// NewConversationVersionLogRepository 创建会话版本日志仓储
func NewConversationVersionLogRepository(db *gorm.DB) ConversationVersionLogRepository {
	return &conversationVersionLogRepository{db: db}
}

func (r *conversationVersionLogRepository) Create(ctx context.Context, log *model.ConversationVersionLog) error {
	return r.db.WithContext(ctx).Create(log).Error
}

func (r *conversationVersionLogRepository) BatchCreate(ctx context.Context, logs []*model.ConversationVersionLog) error {
	if len(logs) == 0 {
		return nil
	}
	return r.db.WithContext(ctx).CreateInBatches(logs, 100).Error
}

func (r *conversationVersionLogRepository) GetByVersion(ctx context.Context, userID string, sinceVersion int64, limit int) ([]*model.ConversationVersionLog, error) {
	var logs []*model.ConversationVersionLog
	err := r.db.WithContext(ctx).
		Where("user_id = ? AND version > ?", userID, sinceVersion).
		Order("version ASC").
		Limit(limit).
		Find(&logs).Error
	return logs, err
}

func (r *conversationVersionLogRepository) GetMaxVersion(ctx context.Context, userID string) (int64, error) {
	var maxVersion int64
	err := r.db.WithContext(ctx).
		Model(&model.ConversationVersionLog{}).
		Where("user_id = ?", userID).
		Select("COALESCE(MAX(version), 0)").
		Scan(&maxVersion).Error
	return maxVersion, err
}