package services import ( "errors" "fmt" "git.solsynth.dev/hydrogen/passport/pkg/internal/database" "git.solsynth.dev/hydrogen/passport/pkg/internal/models" "gorm.io/gorm" "math/rand" "time" ) func CheckDailyCanSign(user models.Account) error { probe := time.Now().Format("YYYY-MM-DD") var record models.SignRecord if err := database.C.Where("account_id = ? AND created_at::date = ?", user.ID, probe).First(&record).Error; err != nil { if errors.Is(err, gorm.ErrRecordNotFound) { return nil } return fmt.Errorf("unable check daliy sign record: %v", err) } return fmt.Errorf("daliy sign record exists") } func GetTodayDailySign(user models.Account) (models.SignRecord, error) { probe := time.Now().Format("YYYY-MM-DD") var record models.SignRecord if err := database.C.Where("account_id = ? AND created_at::date = ?", user.ID, probe).First(&record).Error; err != nil { return record, fmt.Errorf("unable get daliy sign record: %v", err) } return record, nil } func DailySign(user models.Account) (models.SignRecord, error) { tier := rand.Intn(5) record := models.SignRecord{ ResultTier: tier, ResultExperience: rand.Intn(tier*100) + 100, AccountID: user.ID, } if err := CheckDailyCanSign(user); err != nil { return record, fmt.Errorf("today already signed") } var profile models.AccountProfile if err := database.C.Where("account_id = ?", user.ID).First(&profile).Error; err != nil { return record, fmt.Errorf("unable get account profile: %v", err) } else { profile.Experience += uint64(record.ResultExperience) database.C.Save(&profile) } if err := database.C.Save(&record).Error; err != nil { return record, fmt.Errorf("unable do daliy sign: %v", err) } return record, nil }