48 lines
837 B
Go
48 lines
837 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/gbrlsnchs/jwt/v3"
|
||
|
|
cache2 "github.com/patrickmn/go-cache"
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type JWTokens struct {
|
||
|
|
cache *cache2.Cache
|
||
|
|
mutex *sync.Mutex
|
||
|
|
}
|
||
|
|
|
||
|
|
var _tokens *JWTokens
|
||
|
|
|
||
|
|
type DownloadBackupPayload struct {
|
||
|
|
jwt.Payload
|
||
|
|
ServerUuid string `json:"server_uuid"`
|
||
|
|
BackupUuid string `json:"backup_uuid"`
|
||
|
|
UniqueId string `json:"unique_id"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func getTokenStore() *JWTokens {
|
||
|
|
if _tokens == nil {
|
||
|
|
_tokens = &JWTokens{
|
||
|
|
cache: cache2.New(time.Minute*60, time.Minute*5),
|
||
|
|
mutex: &sync.Mutex{},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return _tokens
|
||
|
|
}
|
||
|
|
|
||
|
|
// Determines if a given JWT unique token is valid.
|
||
|
|
func (tokens *JWTokens) IsValidToken(token string) bool {
|
||
|
|
tokens.mutex.Lock()
|
||
|
|
defer tokens.mutex.Unlock()
|
||
|
|
|
||
|
|
_, exists := tokens.cache.Get(token)
|
||
|
|
|
||
|
|
if !exists {
|
||
|
|
_tokens.cache.Add(token, "", time.Minute*60)
|
||
|
|
}
|
||
|
|
|
||
|
|
return !exists
|
||
|
|
}
|