Move endpoint categories into their own packages
This commit is contained in:
21 files changed
+533
-452
No files matched your search
@@ -0,0 +1,82 @@
|
||||
package daily
|
||||
|
||||
import (
|
||||
"crypto/md5"
|
||||
"crypto/rand"
|
||||
"encoding/base64"
|
||||
"encoding/binary"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"github.com/go-co-op/gocron"
|
||||
"github.com/pagefaultgames/pokerogue-server/db"
|
||||
)
|
||||
|
||||
const secondsPerDay = 60 * 60 * 24
|
||||
|
||||
var (
|
||||
dailyRunScheduler = gocron.NewScheduler(time.UTC)
|
||||
dailyRunSecret []byte
|
||||
)
|
||||
|
||||
func Init() error {
|
||||
secret, err := os.ReadFile("secret.key")
|
||||
if err != nil {
|
||||
if !os.IsNotExist(err) {
|
||||
return fmt.Errorf("failed to read daily seed secret: %s", err)
|
||||
}
|
||||
|
||||
newSecret := make([]byte, 32)
|
||||
_, err := rand.Read(newSecret)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to generate daily seed secret: %s", err)
|
||||
}
|
||||
|
||||
err = os.WriteFile("secret.key", newSecret, 0400)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to write daily seed secret: %s", err)
|
||||
}
|
||||
|
||||
secret = newSecret
|
||||
}
|
||||
|
||||
dailyRunSecret = secret
|
||||
|
||||
err = db.TryAddDailyRun(Seed())
|
||||
if err != nil {
|
||||
log.Print(err)
|
||||
}
|
||||
|
||||
log.Printf("Daily Run Seed: %s", Seed())
|
||||
|
||||
scheduleRefresh()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func Seed() string {
|
||||
return base64.StdEncoding.EncodeToString(deriveDailyRunSeed(time.Now().UTC()))
|
||||
}
|
||||
|
||||
func scheduleRefresh() {
|
||||
dailyRunScheduler.Every(1).Day().At("00:00").Do(func() error {
|
||||
err := Init()
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}())
|
||||
dailyRunScheduler.StartAsync()
|
||||
}
|
||||
|
||||
func deriveDailyRunSeed(seedTime time.Time) []byte {
|
||||
day := make([]byte, 8)
|
||||
binary.BigEndian.PutUint64(day, uint64(seedTime.Unix()/secondsPerDay))
|
||||
|
||||
hashedSeed := md5.Sum(append(day, dailyRunSecret...))
|
||||
|
||||
return hashedSeed[:]
|
||||
}
|
||||
@@ -0,0 +1,23 @@
|
||||
package daily
|
||||
|
||||
import (
|
||||
"log"
|
||||
|
||||
"github.com/pagefaultgames/pokerogue-server/db"
|
||||
"github.com/pagefaultgames/pokerogue-server/defs"
|
||||
)
|
||||
|
||||
// /daily/rankings - fetch daily rankings
|
||||
func Rankings(uuid []byte, category, page int) ([]defs.DailyRanking, error) {
|
||||
err := db.UpdateAccountLastActivity(uuid)
|
||||
if err != nil {
|
||||
log.Print("failed to update account last activity")
|
||||
}
|
||||
|
||||
rankings, err := db.FetchRankings(category, page)
|
||||
if err != nil {
|
||||
log.Print("failed to retrieve rankings")
|
||||
}
|
||||
|
||||
return rankings, nil
|
||||
}
|
||||
@@ -0,0 +1,17 @@
|
||||
package daily
|
||||
|
||||
import (
|
||||
"log"
|
||||
|
||||
"github.com/pagefaultgames/pokerogue-server/db"
|
||||
)
|
||||
|
||||
// /daily/rankingpagecount - fetch daily ranking page count
|
||||
func RankingPageCount(category int) (int, error) {
|
||||
pageCount, err := db.FetchRankingPageCount(category)
|
||||
if err != nil {
|
||||
log.Print("failed to retrieve ranking page count")
|
||||
}
|
||||
|
||||
return pageCount, nil
|
||||
}
|
||||
Reference in new issue
Block a user