mirror of
https://github.com/idanoo/autobrr
synced 2025-07-22 08:19:12 +00:00
124 lines
2.8 KiB
Go
124 lines
2.8 KiB
Go
// Copyright (c) 2021 - 2023, Ludvig Lundgren and the autobrr contributors.
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
package database
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
"github.com/autobrr/autobrr/internal/domain"
|
|
"github.com/autobrr/autobrr/internal/logger"
|
|
"github.com/autobrr/autobrr/pkg/errors"
|
|
|
|
sq "github.com/Masterminds/squirrel"
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
type UserRepo struct {
|
|
log zerolog.Logger
|
|
db *DB
|
|
}
|
|
|
|
func NewUserRepo(log logger.Logger, db *DB) domain.UserRepo {
|
|
return &UserRepo{
|
|
log: log.With().Str("repo", "user").Logger(),
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func (r *UserRepo) GetUserCount(ctx context.Context) (int, error) {
|
|
queryBuilder := r.db.squirrel.Select("count(*)").From("users")
|
|
|
|
query, args, err := queryBuilder.ToSql()
|
|
if err != nil {
|
|
return 0, errors.Wrap(err, "error building query")
|
|
}
|
|
|
|
row := r.db.handler.QueryRowContext(ctx, query, args...)
|
|
if err := row.Err(); err != nil {
|
|
return 0, errors.Wrap(err, "error executing query")
|
|
}
|
|
|
|
result := 0
|
|
if err := row.Scan(&result); err != nil {
|
|
return 0, errors.Wrap(err, "error scanning row")
|
|
}
|
|
|
|
return result, nil
|
|
}
|
|
|
|
func (r *UserRepo) FindByUsername(ctx context.Context, username string) (*domain.User, error) {
|
|
|
|
queryBuilder := r.db.squirrel.
|
|
Select("id", "username", "password").
|
|
From("users").
|
|
Where(sq.Eq{"username": username})
|
|
|
|
query, args, err := queryBuilder.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "error building query")
|
|
}
|
|
|
|
row := r.db.handler.QueryRowContext(ctx, query, args...)
|
|
if err := row.Err(); err != nil {
|
|
return nil, errors.Wrap(err, "error executing query")
|
|
}
|
|
|
|
var user domain.User
|
|
|
|
if err := row.Scan(&user.ID, &user.Username, &user.Password); err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, nil
|
|
}
|
|
|
|
return nil, errors.Wrap(err, "error scanning row")
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
func (r *UserRepo) Store(ctx context.Context, req domain.CreateUserRequest) error {
|
|
|
|
var err error
|
|
|
|
queryBuilder := r.db.squirrel.
|
|
Insert("users").
|
|
Columns("username", "password").
|
|
Values(req.Username, req.Password)
|
|
|
|
query, args, err := queryBuilder.ToSql()
|
|
if err != nil {
|
|
return errors.Wrap(err, "error building query")
|
|
}
|
|
|
|
_, err = r.db.handler.ExecContext(ctx, query, args...)
|
|
if err != nil {
|
|
return errors.Wrap(err, "error executing query")
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
func (r *UserRepo) Update(ctx context.Context, user domain.User) error {
|
|
|
|
var err error
|
|
|
|
queryBuilder := r.db.squirrel.
|
|
Update("users").
|
|
Set("username", user.Username).
|
|
Set("password", user.Password).
|
|
Where(sq.Eq{"username": user.Username})
|
|
|
|
query, args, err := queryBuilder.ToSql()
|
|
if err != nil {
|
|
return errors.Wrap(err, "error building query")
|
|
}
|
|
|
|
_, err = r.db.handler.ExecContext(ctx, query, args...)
|
|
if err != nil {
|
|
return errors.Wrap(err, "error executing query")
|
|
}
|
|
|
|
return err
|
|
}
|