package memory

import (
	"context"
	"sync"

	"hyapp/pkg/xerr"
	activitydomain "hyapp/services/activity-service/internal/domain/activity"
)

// Repository 是 activity-service 的测试用内存存储。
type Repository struct {
	mu         sync.RWMutex
	activities map[string]activitydomain.Activity
}

// NewRepository 创建空活动仓库。
func NewRepository() *Repository {
	return &Repository{activities: make(map[string]activitydomain.Activity)}
}

// PutActivity 写入活动状态，测试用它准备数据。
func (r *Repository) PutActivity(activity activitydomain.Activity) {
	r.mu.Lock()
	defer r.mu.Unlock()

	r.activities[activity.ActivityID] = activity
}

// GetActivity 查询活动状态。
func (r *Repository) GetActivity(_ context.Context, activityID string) (activitydomain.Activity, error) {
	r.mu.RLock()
	defer r.mu.RUnlock()

	activity, ok := r.activities[activityID]
	if !ok {
		return activitydomain.Activity{}, xerr.New(xerr.NotFound, "activity not found")
	}

	return activity, nil
}