116 lines
2.5 KiB
Go
116 lines
2.5 KiB
Go
package store
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
bolt "go.etcd.io/bbolt"
|
|
)
|
|
|
|
var bucketTitles = []byte("titles")
|
|
|
|
type TitleDef struct {
|
|
ID string
|
|
Name string
|
|
Description string
|
|
}
|
|
|
|
var TitleDefs = []TitleDef{
|
|
{ID: "novice", Name: "Novice", Description: "Default title for new players"},
|
|
{ID: "explorer", Name: "Explorer", Description: "Reach floor 5"},
|
|
{ID: "veteran", Name: "Veteran", Description: "Reach floor 10"},
|
|
{ID: "champion", Name: "Champion", Description: "Reach floor 20"},
|
|
{ID: "gold_king", Name: "Gold King", Description: "Accumulate 500+ gold in one run"},
|
|
{ID: "team_player", Name: "Team Player", Description: "Complete 5 multiplayer runs"},
|
|
{ID: "survivor", Name: "Survivor", Description: "Complete a run without dying"},
|
|
}
|
|
|
|
type PlayerTitleData struct {
|
|
ActiveTitle string `json:"active_title"`
|
|
Earned []string `json:"earned"`
|
|
}
|
|
|
|
func (d *DB) EarnTitle(fingerprint, titleID string) (bool, error) {
|
|
newlyEarned := false
|
|
err := d.db.Update(func(tx *bolt.Tx) error {
|
|
b := tx.Bucket(bucketTitles)
|
|
key := []byte(fingerprint)
|
|
var data PlayerTitleData
|
|
|
|
v := b.Get(key)
|
|
if v != nil {
|
|
if err := json.Unmarshal(v, &data); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// Check if already earned
|
|
for _, e := range data.Earned {
|
|
if e == titleID {
|
|
return nil
|
|
}
|
|
}
|
|
|
|
newlyEarned = true
|
|
data.Earned = append(data.Earned, titleID)
|
|
|
|
// Auto-set first earned title as active
|
|
if data.ActiveTitle == "" {
|
|
data.ActiveTitle = titleID
|
|
}
|
|
|
|
encoded, err := json.Marshal(data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return b.Put(key, encoded)
|
|
})
|
|
return newlyEarned, err
|
|
}
|
|
|
|
func (d *DB) SetActiveTitle(fingerprint, titleID string) error {
|
|
return d.db.Update(func(tx *bolt.Tx) error {
|
|
b := tx.Bucket(bucketTitles)
|
|
key := []byte(fingerprint)
|
|
var data PlayerTitleData
|
|
|
|
v := b.Get(key)
|
|
if v != nil {
|
|
if err := json.Unmarshal(v, &data); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// Verify title is earned
|
|
found := false
|
|
for _, e := range data.Earned {
|
|
if e == titleID {
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
if !found {
|
|
return nil
|
|
}
|
|
|
|
data.ActiveTitle = titleID
|
|
encoded, err := json.Marshal(data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return b.Put(key, encoded)
|
|
})
|
|
}
|
|
|
|
func (d *DB) GetTitleData(fingerprint string) (PlayerTitleData, error) {
|
|
var data PlayerTitleData
|
|
err := d.db.View(func(tx *bolt.Tx) error {
|
|
b := tx.Bucket(bucketTitles)
|
|
v := b.Get([]byte(fingerprint))
|
|
if v == nil {
|
|
return nil
|
|
}
|
|
return json.Unmarshal(v, &data)
|
|
})
|
|
return data, err
|
|
}
|