96 lines
2.2 KiB
Go
96 lines
2.2 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/dhax/go-base/auth/jwt"
|
|
"github.com/dhax/go-base/auth/pwdless"
|
|
"github.com/uptrace/bun"
|
|
)
|
|
|
|
// AuthStore implements database operations for account pwdlessentication.
|
|
type AuthStore struct {
|
|
db *bun.DB
|
|
}
|
|
|
|
// NewAuthStore return an AuthStore.
|
|
func NewAuthStore(db *bun.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.NewSelect().
|
|
Model(a).
|
|
Where("id = ?", id).
|
|
Scan(context.Background())
|
|
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.NewSelect().
|
|
Model(a).
|
|
Column("id", "active", "email", "name").
|
|
Where("email = ?", e).
|
|
Scan(context.Background())
|
|
return a, err
|
|
}
|
|
|
|
// UpdateAccount upates account data related to pwdlessentication.
|
|
func (s *AuthStore) UpdateAccount(a *pwdless.Account) error {
|
|
_, err := s.db.NewUpdate().
|
|
Model(a).
|
|
Column("last_login").
|
|
WherePK().
|
|
Exec(context.Background())
|
|
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.NewSelect().
|
|
Model(token).
|
|
Where("token = ?", t).
|
|
Scan(context.Background())
|
|
return token, err
|
|
}
|
|
|
|
// CreateOrUpdateToken creates or updates an existing refresh token.
|
|
func (s *AuthStore) CreateOrUpdateToken(t *jwt.Token) error {
|
|
if t.ID == 0 {
|
|
_, err := s.db.NewInsert().
|
|
Model(t).
|
|
Exec(context.Background())
|
|
return err
|
|
}
|
|
_, err := s.db.NewUpdate().
|
|
Model(t).
|
|
WherePK().
|
|
Exec(context.Background())
|
|
return err
|
|
}
|
|
|
|
// DeleteToken deletes a refresh token.
|
|
func (s *AuthStore) DeleteToken(t *jwt.Token) error {
|
|
_, err := s.db.NewDelete().
|
|
Model(t).
|
|
WherePK().
|
|
Exec(context.Background())
|
|
return err
|
|
}
|
|
|
|
// PurgeExpiredToken deletes expired refresh token.
|
|
func (s *AuthStore) PurgeExpiredToken() error {
|
|
_, err := s.db.NewDelete().
|
|
Model((*jwt.Token)(nil)).
|
|
Where("expiry < ?", time.Now()).
|
|
Exec(context.Background())
|
|
return err
|
|
}
|