86 lines
1.9 KiB
Go
86 lines
1.9 KiB
Go
package database
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/dhax/go-base/auth/jwt"
|
|
"github.com/dhax/go-base/auth/pwdless"
|
|
"github.com/go-pg/pg"
|
|
)
|
|
|
|
// AuthStore implements database operations for account pwdlessentication.
|
|
type AuthStore struct {
|
|
db *pg.DB
|
|
}
|
|
|
|
// NewAuthStore return an AuthStore.
|
|
func NewAuthStore(db *pg.DB) *AuthStore {
|
|
return &AuthStore{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
// GetAccount returns an account by ID.
|
|
func (s *AuthStore) GetAccount(id int) (*pwdless.Account, error) {
|
|
a := pwdless.Account{ID: id}
|
|
err := s.db.Model(&a).
|
|
Column("account.*").
|
|
Where("id = ?id").
|
|
First()
|
|
return &a, err
|
|
}
|
|
|
|
// GetAccountByEmail returns an account by email.
|
|
func (s *AuthStore) GetAccountByEmail(e string) (*pwdless.Account, error) {
|
|
a := pwdless.Account{Email: e}
|
|
err := s.db.Model(&a).
|
|
Column("id", "active", "email", "name").
|
|
Where("email = ?email").
|
|
First()
|
|
return &a, err
|
|
}
|
|
|
|
// UpdateAccount upates account data related to pwdlessentication.
|
|
func (s *AuthStore) UpdateAccount(a *pwdless.Account) error {
|
|
_, err := s.db.Model(a).
|
|
Column("last_login").
|
|
WherePK().
|
|
Update()
|
|
return err
|
|
}
|
|
|
|
// GetToken returns refresh token by token identifier.
|
|
func (s *AuthStore) GetToken(t string) (*jwt.Token, error) {
|
|
token := jwt.Token{Token: t}
|
|
err := s.db.Model(&token).
|
|
Where("token = ?token").
|
|
First()
|
|
|
|
return &token, err
|
|
}
|
|
|
|
// CreateOrUpdateToken creates or updates an existing refresh token.
|
|
func (s *AuthStore) CreateOrUpdateToken(t *jwt.Token) error {
|
|
var err error
|
|
if t.ID == 0 {
|
|
err = s.db.Insert(t)
|
|
} else {
|
|
err = s.db.Update(t)
|
|
}
|
|
return err
|
|
}
|
|
|
|
// DeleteToken deletes a refresh token.
|
|
func (s *AuthStore) DeleteToken(t *jwt.Token) error {
|
|
err := s.db.Delete(t)
|
|
return err
|
|
}
|
|
|
|
// PurgeExpiredToken deletes expired refresh token.
|
|
func (s *AuthStore) PurgeExpiredToken() error {
|
|
_, err := s.db.Model(&jwt.Token{}).
|
|
Where("expiry < ?", time.Now()).
|
|
Delete()
|
|
|
|
return err
|
|
}
|