From 38ab16d4ee92820843e59d8afd772e5616c9ffc6 Mon Sep 17 00:00:00 2001 From: blint <rethelyibalint@gmail.com> Date: Thu, 31 Aug 2023 23:43:39 +0200 Subject: [PATCH] fix credits --- db/db.go | 8 ++++++-- homepage/new.go | 2 +- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/db/db.go b/db/db.go index b992d35..174cf55 100644 --- a/db/db.go +++ b/db/db.go @@ -331,8 +331,6 @@ func (u *User) Load() error { return e } -const credit = 5000 - func SaveMultipleSpend(sp []*Spend) error { e := db.RunInTransaction(context.Background(), func(tx *pg.Tx) error { for _, s := range sp { @@ -349,6 +347,12 @@ func SaveMultipleSpend(sp []*Spend) error { } e = tx.Model(u).WherePK().First() + var credit int64 + if u.Credits > 5000 { + credit = u.Credits + } else { + credit = 5000 + } price := p.Price * s.Amount if price > u.Money+credit { return fmt.Errorf("Not enough funds") diff --git a/homepage/new.go b/homepage/new.go index 609169b..ffb62af 100644 --- a/homepage/new.go +++ b/homepage/new.go @@ -95,7 +95,7 @@ func NewHandler(w http.ResponseWriter, r *http.Request) { } var credit int64 - if session.User.Credits > 0 { + if session.User.Credits > 5000 { credit = session.User.Credits } else { credit = 5000 -- GitLab