Clean up daily package

pull/4/head
maru 8 months ago
parent 1f0f38d38e
commit c3a3746639
No known key found for this signature in database
GPG Key ID: 37689350E9CD0F0D

@ -17,12 +17,14 @@ import (
const secondsPerDay = 60 * 60 * 24
var (
dailyRunScheduler = gocron.NewScheduler(time.UTC)
dailyRunSecret []byte
scheduler = gocron.NewScheduler(time.UTC)
secret []byte
)
func Init() error {
secret, err := os.ReadFile("secret.key")
var err 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)
@ -42,41 +44,37 @@ func Init() error {
secret = newSecret
}
dailyRunSecret = secret
err = db.TryAddDailyRun(Seed())
err = recordNewRun()
if err != nil {
log.Print(err)
}
log.Printf("Daily Run Seed: %s", Seed())
scheduleRefresh()
scheduler.Every(1).Day().At("00:00").Do(recordNewRun())
scheduler.StartAsync()
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()
return base64.StdEncoding.EncodeToString(deriveSeed(time.Now().UTC()))
}
func deriveDailyRunSeed(seedTime time.Time) []byte {
func deriveSeed(seedTime time.Time) []byte {
day := make([]byte, 8)
binary.BigEndian.PutUint64(day, uint64(seedTime.Unix()/secondsPerDay))
hashedSeed := md5.Sum(append(day, dailyRunSecret...))
hashedSeed := md5.Sum(append(day, secret...))
return hashedSeed[:]
}
func recordNewRun() error {
err := db.TryAddDailyRun(Seed())
if err != nil {
return err
}
return nil
}

Loading…
Cancel
Save