snow/person/quest.go

package person

import (
	"github.com/ectrc/snow/aid"
	"github.com/ectrc/snow/storage"
	"github.com/google/uuid"
)

type Quest struct {
	ID string
	ProfileID string
	TemplateID string
	State string
	Objectives []string
	ObjectiveCounts []int64
	BundleID string
	ScheduleID string
}

func NewQuest(templateID string, bundleID string, scheduleID string) *Quest {
	return &Quest{
		ID: uuid.New().String(),
		TemplateID: templateID,
		State: "Active",
		Objectives: []string{},
		ObjectiveCounts: []int64{},
		BundleID: bundleID,
		ScheduleID: scheduleID,
	}
}

func NewDailyQuest(templateID string) *Quest {
	return &Quest{
		ID: uuid.New().String(),
		TemplateID: templateID,
		State: "Active",
		Objectives: []string{},
		ObjectiveCounts: []int64{},
	}
}

func FromDatabaseQuest(quest *storage.DB_Quest) *Quest {
	return &Quest{
		ID: quest.ID,
		ProfileID: quest.ProfileID,
		TemplateID: quest.TemplateID,
		State: quest.State,
		Objectives: quest.Objectives,
		ObjectiveCounts: quest.ObjectiveCounts,
		BundleID: quest.BundleID,
		ScheduleID: quest.ScheduleID,
	}
}

func (q *Quest) GenerateFortniteQuestEntry() aid.JSON {
	json := aid.JSON{
		"templateId": q.TemplateID,
		"attributes": aid.JSON{
			"quest_state": q.State,
			"challenge_bundle_id": q.BundleID,
		},
		"quantity": 1,
	}

	for i, objective := range q.Objectives {
		json["attributes"].(aid.JSON)["completion_" + objective] = q.ObjectiveCounts[i]
	}

	return json
}

func (q *Quest) Delete() {
	storage.Repo.DeleteQuest(q.ID)
}

func (q *Quest) AddObjective(objective string, count int64) {
	q.Objectives = append(q.Objectives, objective)
	q.ObjectiveCounts = append(q.ObjectiveCounts, count)
}

func (q *Quest) SetObjectiveCount(objective string, count int64) {
	for i, obj := range q.Objectives {
		if obj == objective {
			q.ObjectiveCounts[i] = count
			return
		}
	}
}

func (q *Quest) UpdateObjectiveCount(objective string, delta int64) {
	for i, obj := range q.Objectives {
		if obj == objective {
			q.ObjectiveCounts[i] += delta
			return
		}
	}
}

func (q *Quest) GetObjectiveCount(objective string) int64 {
	for i, obj := range q.Objectives {
		if obj == objective {
			return q.ObjectiveCounts[i]
		}
	}

	return 0
}

func (q *Quest) GetObjectiveIndex(objective string) int {
	for i, obj := range q.Objectives {
		if obj == objective {
			return i
		}
	}

	return -1
}

func (q *Quest) RemoveObjective(objective string) {
	for i, obj := range q.Objectives {
		if obj == objective {
			q.Objectives = append(q.Objectives[:i], q.Objectives[i+1:]...)
			q.ObjectiveCounts = append(q.ObjectiveCounts[:i], q.ObjectiveCounts[i+1:]...)
			return
		}
	}
}

func (q *Quest) ToDatabase(profileId string) *storage.DB_Quest {
	return &storage.DB_Quest{
		ID: q.ID,
		ProfileID: profileId,
		TemplateID: q.TemplateID,
		State: q.State,
		Objectives: q.Objectives,
		ObjectiveCounts: q.ObjectiveCounts,
		BundleID: q.BundleID,
		ScheduleID: q.ScheduleID,
	}
}

func (q *Quest) Save() {
	storage.Repo.SaveQuest(q.ToDatabase(q.ProfileID))
}