package entity

import "github.com/photoprism/photoprism/pkg/react"

type ReactionMap map[string]Reaction

func (m ReactionMap) Get(name string) Reaction {
	if result, ok := m[name]; ok {
		return result
	}

	return Reaction{}
}

func (m ReactionMap) Pointer(name string) *Reaction {
	if result, ok := m[name]; ok {
		return &result
	}

	return &Reaction{}
}

var ReactionFixtures = ReactionMap{
	"SubjectJohnLike": Reaction{
		EntityUID: SubjectFixtures.Get("john-doe").SubjUID,
		UserUID:   UserFixtures.Get("alice").UserUID,
		Reaction:  react.Like.String(),
	},
	"PhotoAliceLove": Reaction{
		EntityUID: PhotoFixtures.Get("Photo01").PhotoUID,
		UserUID:   UserFixtures.Pointer("alice").UserUID,
		Reaction:  react.Love.String(),
	},
	"PhotoBobLove": Reaction{
		EntityUID: PhotoFixtures.Get("Photo01").PhotoUID,
		UserUID:   UserFixtures.Pointer("bob").UserUID,
		Reaction:  react.Love.String(),
	},
}

// CreateReactionFixtures inserts known entities into the database for testing.
func CreateReactionFixtures() {
	for _, entity := range ReactionFixtures {
		if err := entity.Create(); err != nil {
			log.Error(err)
		}
	}
}