user_repository.go

package repositories

import (
	"context"
	"database/sql"
	"github.com/hiroyky/famiphoto/drivers/mysql"
	"github.com/hiroyky/famiphoto/entities"
	"github.com/hiroyky/famiphoto/errors"
	"github.com/hiroyky/famiphoto/infrastructures/dbmodels"
	"github.com/hiroyky/famiphoto/infrastructures/filters"
	"github.com/volatiletech/sqlboiler/v4/boil"
	"github.com/volatiletech/sqlboiler/v4/queries/qm"
)

type UserRepository interface {
	GetUser(ctx context.Context, userID string) (*dbmodels.User, error)
	GetUsers(ctx context.Context, filter *filters.UserFilter, limit, offset int) ([]*dbmodels.User, error)
	CountUsers(ctx context.Context, filter *filters.UserFilter) (int, error)
	ExistUser(ctx context.Context, userID string) (bool, error)
	CreateUser(ctx context.Context, user *dbmodels.User, password *dbmodels.UserPassword) (*dbmodels.User, error)
	UpdateUserProfile(ctx context.Context, user *dbmodels.User) (*dbmodels.User, error)
}

func NewUserRepository(db mysql.SQLExecutor) UserRepository {
	return &userRepository{
		db: db,
	}
}

type userRepository struct {
	db mysql.SQLExecutor
}

func (r *userRepository) GetUser(ctx context.Context, userID string) (*dbmodels.User, error) {
	user, err := dbmodels.FindUser(ctx, r.db, userID)
	if err == sql.ErrNoRows {
		return nil, errors.New(errors.UserNotFoundError, err)
	} else if err != nil {
		return nil, err
	}
	return user, nil
}

func (r *userRepository) GetUsers(ctx context.Context, filter *filters.UserFilter, limit, offset int) ([]*dbmodels.User, error) {
	mods := filter.WhereMods()
	mods = append(mods, qm.Limit(limit), qm.Offset(offset))

	users, err := dbmodels.Users(mods...).All(ctx, r.db)
	if err != nil {
		return nil, err
	}
	return users, nil
}

func (r *userRepository) CountUsers(ctx context.Context, filter *filters.UserFilter) (int, error) {
	mods := filter.WhereMods()
	total, err := dbmodels.Users(mods...).Count(ctx, r.db)
	if err != nil {
		return 0, err
	}
	return int(total), nil
}

func (r *userRepository) ExistUser(ctx context.Context, userID string) (bool, error) {
	return dbmodels.UserExists(ctx, r.db, userID)
}

func (r *userRepository) CreateUser(ctx context.Context, user *dbmodels.User, password *dbmodels.UserPassword) (*dbmodels.User, error) {
	tx, err := r.db.BeginTx(ctx, nil)
	if err != nil {
		return nil, errors.New(errors.TxnBeginFatal, err)
	}

	if err := user.Insert(ctx, tx, boil.Infer()); err != nil {
		return nil, err
	}
	if err := password.Insert(ctx, tx, boil.Infer()); err != nil {
		return nil, err
	}

	if err := tx.Commit(); err != nil {
		if err := tx.Rollback(); err != nil {
			return nil, errors.New(errors.TxnRollbackFatal, err)
		}
		return nil, errors.New(errors.UserCreateFatal, err)
	}
	return user, nil
}

func (r *userRepository) UpdateUserProfile(ctx context.Context, user *dbmodels.User) (*dbmodels.User, error) {
	if _, err := user.Update(ctx, r.db, boil.Blacklist(dbmodels.UserColumns.Status)); err != nil {
		return nil, err
	}
	return user, nil
}

func (r *userRepository) toDBUserStatus(s entities.UserStatus) int {
	return int(s)
}