62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
package progress
|
|
|
|
import (
|
|
"fmt"
|
|
"sort"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
func (s *Store) ResearchObjects(pack int) ([]int, error) {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
store, err := s.fieldRewardEntries()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
entries, err := store.ListEntries("progress", "field_research")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var ids []int
|
|
prefix := fmt.Sprintf("%d:", pack)
|
|
for key, raw := range entries {
|
|
if string(raw) != "once" {
|
|
return nil, fmt.Errorf("progress: invalid research entry")
|
|
}
|
|
if suffix, found := strings.CutPrefix(key, prefix); found {
|
|
id, e := strconv.Atoi(suffix)
|
|
if e != nil || id <= 0 {
|
|
return nil, fmt.Errorf("progress: invalid research identity")
|
|
}
|
|
ids = append(ids, id)
|
|
}
|
|
}
|
|
sort.Ints(ids)
|
|
return ids, nil
|
|
}
|
|
func (s *Store) ResearchObjectReply(pack, id int) ([]byte, bool, error) {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
store, err := s.fieldRewardEntries()
|
|
if err != nil {
|
|
return nil, false, err
|
|
}
|
|
return store.LoadEntry("progress", "field_research_rewards", fieldRewardKey(pack, id))
|
|
}
|
|
func (s *Store) MarkResearchObject(pack, id int, rewardItems []byte) error {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
if pack <= 0 || id <= 0 {
|
|
return fmt.Errorf("progress: invalid research identity")
|
|
}
|
|
store, err := s.fieldRewardEntries()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err = store.PutEntry("progress", "field_research_rewards", fieldRewardKey(pack, id), rewardItems); err != nil {
|
|
return err
|
|
}
|
|
return store.PutEntry("progress", "field_research", fieldRewardKey(pack, id), []byte("once"))
|
|
}
|