initial commit

database/authStore.go

@@ -0,0 +1,84 @@
+package database
+
+import (
+	"time"
+
+	"github.com/dhax/go-base/models"
+	"github.com/go-pg/pg"
+)
+
+type AuthStore struct {
+	db *pg.DB
+}
+
+func NewAuthStore(db *pg.DB) *AuthStore {
+	return &AuthStore{
+		db: db,
+	}
+}
+
+func (s *AuthStore) GetByID(id int) (*models.Account, error) {
+	a := models.Account{ID: id}
+	err := s.db.Model(&a).
+		Column("account.*").
+		Where("id = ?id").
+		First()
+	return &a, err
+}
+
+func (s *AuthStore) GetByEmail(e string) (*models.Account, error) {
+	a := models.Account{Email: e}
+	err := s.db.Model(&a).
+		Column("id", "active", "email", "name").
+		Where("email = ?email").
+		First()
+	return &a, err
+}
+
+func (s *AuthStore) GetByRefreshToken(t string) (*models.Account, *models.Token, error) {
+	token := models.Token{Token: t}
+	err := s.db.Model(&token).
+		Where("token = ?token").
+		First()
+	if err != nil {
+		return nil, nil, err
+	}
+
+	a := models.Account{ID: token.AccountID}
+	err = s.db.Model(&a).
+		Column("account.*").
+		Where("id = ?id").
+		First()
+
+	return &a, &token, err
+}
+
+func (s *AuthStore) UpdateAccount(a *models.Account) error {
+	_, err := s.db.Model(a).
+		Column("last_login").
+		Update()
+	return err
+}
+
+func (s *AuthStore) SaveRefreshToken(t *models.Token) error {
+	var err error
+	if t.ID == 0 {
+		err = s.db.Insert(t)
+	} else {
+		err = s.db.Update(t)
+	}
+	return err
+}
+
+func (s *AuthStore) DeleteRefreshToken(t *models.Token) error {
+	err := s.db.Delete(t)
+	return err
+}
+
+func (s *AuthStore) PurgeExpiredToken() error {
+	_, err := s.db.Model(&models.Token{}).
+		Where("expiry < ?", time.Now()).
+		Delete()
+
+	return err
+}