phoenixandphoenix
de171b790d
Closes #34 Reviewed-on: phoenix/textsender-auth#37 Co-authored-by: phoenix <kundeng00@pm.me> Co-committed-by: phoenix <kundeng00@pm.me>
96 lines
2.9 KiB
Go
96 lines
2.9 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"git.kundeng.us/phoenix/textsender-models/tx0/user"
|
|
"github.com/google/uuid"
|
|
"github.com/jackc/pgx/v5"
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
)
|
|
|
|
// TODO: Rename this to ServiceUserStore
|
|
type ServiceStore interface {
|
|
CheckWithUsername(ctx context.Context, username string) (bool, error)
|
|
GetWithUsername(ctx context.Context, username string) (*user.ServiceUser, error)
|
|
GetWithId(ctx context.Context, id uuid.UUID) (*user.ServiceUser, error)
|
|
Create(ctx context.Context, serviceUser *user.ServiceUser) error
|
|
UpdateLastLogin(ctx context.Context, id uuid.UUID, lastLogin time.Time) (int64, error)
|
|
}
|
|
|
|
type PGServiceStore struct {
|
|
db *pgxpool.Pool
|
|
}
|
|
|
|
func NewServiceStore(db *pgxpool.Pool) *PGServiceStore {
|
|
return &PGServiceStore{db: db}
|
|
}
|
|
|
|
func (s *PGServiceStore) CheckWithUsername(ctx context.Context, username string) (bool, error) {
|
|
var exists bool
|
|
query := `SELECT EXISTS(SELECT 1 FROM service_users WHERE Username = $1)`
|
|
|
|
if err := s.db.QueryRow(ctx, query, username).Scan(&exists); err != nil {
|
|
if err == pgx.ErrNoRows {
|
|
return exists, nil
|
|
} else {
|
|
return exists, fmt.Errorf("Error querying row: %v", err)
|
|
}
|
|
} else {
|
|
return exists, nil
|
|
}
|
|
}
|
|
|
|
func (s *PGServiceStore) GetWithUsername(ctx context.Context, username string) (*user.ServiceUser, error) {
|
|
var serviceUser user.ServiceUser
|
|
query := `SELECT id, username, passphrase, created FROM service_users WHERE username = $1`
|
|
|
|
if err := s.db.QueryRow(ctx, query, username).Scan(&serviceUser.Id, &serviceUser.Username, &serviceUser.Passphrase, &serviceUser.Created); err != nil {
|
|
if err == pgx.ErrNoRows {
|
|
return nil, nil
|
|
} else {
|
|
return nil, fmt.Errorf("Error querying row: %v", err)
|
|
}
|
|
} else {
|
|
return &serviceUser, nil
|
|
}
|
|
}
|
|
|
|
func (s *PGServiceStore) GetWithId(ctx context.Context, id uuid.UUID) (*user.ServiceUser, error) {
|
|
var serviceUser user.ServiceUser
|
|
query := `SELECT id, username, passphrase, created FROM service_users WHERE id = $1`
|
|
|
|
if err := s.db.QueryRow(ctx, query, id).Scan(&serviceUser.Id, &serviceUser.Username, &serviceUser.Passphrase, &serviceUser.Created); err != nil {
|
|
if err == pgx.ErrNoRows {
|
|
return nil, nil
|
|
} else {
|
|
return nil, fmt.Errorf("Error querying row: %v", err)
|
|
}
|
|
} else {
|
|
return &serviceUser, nil
|
|
}
|
|
}
|
|
|
|
func (s *PGServiceStore) Create(ctx context.Context, serviceUser *user.ServiceUser) error {
|
|
query := `
|
|
INSERT INTO service_users (username, passphrase)
|
|
VALUES ($1, $2)
|
|
RETURNING id, created
|
|
`
|
|
|
|
return s.db.QueryRow(ctx, query, serviceUser.Username, serviceUser.Passphrase).Scan(
|
|
&serviceUser.Id, &serviceUser.Created,
|
|
)
|
|
}
|
|
|
|
func (s *PGServiceStore) UpdateLastLogin(ctx context.Context, id uuid.UUID, lastLogin time.Time) (int64, error) {
|
|
query := `UPDATE service_users SET last_login = $1 WHERE id = $2`
|
|
if affected, err := s.db.Exec(ctx, query, lastLogin, id); err != nil {
|
|
return 0, err
|
|
} else {
|
|
return affected.RowsAffected(), nil
|
|
}
|
|
}
|