mirror of
https://github.com/idanoo/autobrr
synced 2025-07-23 08:49:13 +00:00
1a4f3cf55d
* feat: add ability to create an account via the webui without the need for autobrrctl * refactor redundant code block. * fix: early return and 0 value
116 lines
2.5 KiB
Go
116 lines
2.5 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"github.com/rs/zerolog/log"
|
|
|
|
"github.com/autobrr/autobrr/internal/domain"
|
|
)
|
|
|
|
type UserRepo struct {
|
|
db *DB
|
|
}
|
|
|
|
func NewUserRepo(db *DB) domain.UserRepo {
|
|
return &UserRepo{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 {
|
|
log.Error().Stack().Err(err).Msg("user.store: error building query")
|
|
return 0, err
|
|
}
|
|
|
|
row := r.db.handler.QueryRowContext(ctx, query, args...)
|
|
if err := row.Err(); err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
result := 0
|
|
if err := row.Scan(&result); err != nil {
|
|
log.Error().Err(err).Msg("could not query number of users")
|
|
return 0, err
|
|
}
|
|
|
|
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("username = ?", username)
|
|
|
|
query, args, err := queryBuilder.ToSql()
|
|
if err != nil {
|
|
log.Error().Stack().Err(err).Msg("user.store: error building query")
|
|
return nil, err
|
|
}
|
|
|
|
row := r.db.handler.QueryRowContext(ctx, query, args...)
|
|
if err := row.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var user domain.User
|
|
|
|
if err := row.Scan(&user.ID, &user.Username, &user.Password); err != nil {
|
|
log.Error().Err(err).Msg("could not scan user to struct")
|
|
return nil, err
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
func (r *UserRepo) Store(ctx context.Context, user domain.User) error {
|
|
|
|
var err error
|
|
|
|
queryBuilder := r.db.squirrel.
|
|
Insert("users").
|
|
Columns("username", "password").
|
|
Values(user.Username, user.Password)
|
|
|
|
query, args, err := queryBuilder.ToSql()
|
|
if err != nil {
|
|
log.Error().Stack().Err(err).Msg("user.store: error building query")
|
|
return err
|
|
}
|
|
|
|
_, err = r.db.handler.ExecContext(ctx, query, args...)
|
|
if err != nil {
|
|
log.Error().Stack().Err(err).Msg("user.store: error executing query")
|
|
return err
|
|
}
|
|
|
|
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("username = ?", user.Username)
|
|
|
|
query, args, err := queryBuilder.ToSql()
|
|
if err != nil {
|
|
log.Error().Stack().Err(err).Msg("user.store: error building query")
|
|
return err
|
|
}
|
|
|
|
_, err = r.db.handler.ExecContext(ctx, query, args...)
|
|
if err != nil {
|
|
log.Error().Stack().Err(err).Msg("user.store: error executing query")
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|