added discord token check to auth
This commit is contained in:
39
internal/store/tokencheck.go
Normal file
39
internal/store/tokencheck.go
Normal file
@@ -0,0 +1,39 @@
|
||||
package store
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"time"
|
||||
|
||||
"git.haelnorr.com/h/oslstats/internal/db"
|
||||
)
|
||||
|
||||
type TokenCheck struct {
|
||||
Token *db.DiscordToken
|
||||
ExpiresAt time.Time
|
||||
}
|
||||
|
||||
func (s *Store) NewTokenCheck(user *db.User, token *db.DiscordToken) error {
|
||||
if user == nil {
|
||||
return errors.New("user cannot be nil")
|
||||
}
|
||||
if token == nil {
|
||||
return errors.New("token cannot be nil")
|
||||
}
|
||||
s.tokenchecks.Store(user.ID, &TokenCheck{
|
||||
Token: token,
|
||||
ExpiresAt: time.Now().Add(5 * time.Minute),
|
||||
})
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Store) CheckToken(user *db.User) *db.DiscordToken {
|
||||
res, ok := s.tokenchecks.Load(user.ID)
|
||||
if !ok {
|
||||
return nil
|
||||
}
|
||||
check := (res).(*TokenCheck)
|
||||
check.ExpiresAt = time.Now().Add(5 * time.Minute)
|
||||
s.tokenchecks.Delete(user.ID)
|
||||
s.tokenchecks.Store(user.ID, check)
|
||||
return check.Token
|
||||
}
|
||||
Reference in New Issue
Block a user