mirror-nourybot/internal/data/user.go

package data

import (
	"database/sql"
	"errors"
	"time"
)

type User struct {
	ID             int       `json:"id"`
	AddedAt        time.Time `json:"-"`
	Login          string    `json:"login"`
	TwitchID       string    `json:"twitchid"`
	Level          int       `json:"level"`
	Location       string    `json:"location,omitempty"`
	LastFMUsername string    `json:"lastfm_username,omitempty"`
}

type UserModel struct {
	DB *sql.DB
}

// Insert inserts a user model into the database.
func (u UserModel) Insert(login, twitchId string) error {
	query := `
	INSERT INTO users(login, twitchid, level, location, lastfm_username)
	VALUES ($1, $2, $3, $4, $5)
	ON CONFLICT (login)
	DO NOTHING
	RETURNING id, added_at;
	`

	args := []interface{}{login, twitchId, "0", "", ""}

	// Execute the query returning the number of affected rows.
	result, err := u.DB.Exec(query, args...)
	if err != nil {
		return err
	}

	// Check how many rows were affected.
	rowsAffected, err := result.RowsAffected()
	if err != nil {
		return err
	}

	if rowsAffected == 0 {
		return ErrUserAlreadyExists
	}

	return nil
}

// SetLocation queries the database for a record with the provided login value
// and if that exists sets the location to the supplied
func (u UserModel) SetLocation(twitchId, location string) error {
	query := `
	UPDATE users
	SET location = $2
	WHERE twitchId = $1`

	result, err := u.DB.Exec(query, twitchId, location)
	if err != nil {
		return err
	}

	// Check how many rows were affected.
	rowsAffected, err := result.RowsAffected()
	if err != nil {
		return err
	}

	// We want atleast 1, if it is 0 the entry did not exist.
	if rowsAffected == 0 {
		return ErrRecordNotFound
	}

	return nil
}

// GetLocation queries the database for a record with the provided login value
// and if that exists returns the location value.
func (u UserModel) GetLocation(twitchId string) (string, error) {
	query := `
	SELECT location
	FROM users
	WHERE twitchid = $1`

	var user User

	err := u.DB.QueryRow(query, twitchId).Scan(
		&user.Location,
	)

	if err != nil {
		switch {
		case errors.Is(err, sql.ErrNoRows):
			return "", ErrRecordNotFound
		default:
			return "", err
		}
	}

	return user.Location, nil
}

// SetLastFM queries the database for a record with the provided login value
// and if that exists sets the lastfm value to the supplied value.
func (u UserModel) SetLastFM(login, lastfmUser string) error {
	query := `
	UPDATE users
	SET lastfm_username = $2
	WHERE login = $1`

	result, err := u.DB.Exec(query, login, lastfmUser)
	if err != nil {
		return err
	}

	// Check how many rows were affected.
	rowsAffected, err := result.RowsAffected()
	if err != nil {
		return err
	}

	// We want atleast 1, if it is 0 the entry did not exist.
	if rowsAffected == 0 {
		return ErrRecordNotFound
	}

	return nil
}

// GetLastFM queries the database for a record with the provided login value
// and if that exists returns the lastfm username.
func (u UserModel) GetLastFM(login string) (string, error) {
	query := `
	SELECT lastfm_username
	FROM users
	WHERE login = $1`

	var user User

	err := u.DB.QueryRow(query, login).Scan(
		&user.LastFMUsername,
	)

	if err != nil {
		switch {
		case errors.Is(err, sql.ErrNoRows):
			return "", ErrRecordNotFound
		default:
			return "", err
		}
	}

	return user.LastFMUsername, nil
}

// SetLocation queries the database for a record with the provided twitch id
// and if that exists returns the level specified in the database for the user.
func (u UserModel) GetLevel(twitchId string) (int, error) {
	query := `
	SELECT level
	FROM users
	WHERE twitchid = $1`

	var user User

	err := u.DB.QueryRow(query, twitchId).Scan(
		&user.Level,
	)

	if err != nil {
		switch {
		case errors.Is(err, sql.ErrNoRows):
			return -1, ErrRecordNotFound
		default:
			return -1, err
		}
	}

	return user.Level, nil
}

// Setlevel searches the database for a record with the provided login value
// and if that exists sets the level to the newly supplied level value.
func (u UserModel) SetLevel(login string, level int) error {
	query := `
	UPDATE users
	SET level = $2
	WHERE login = $1`

	result, err := u.DB.Exec(query, login, level)
	if err != nil {
		return err
	}

	// Check how many rows were affected.
	rowsAffected, err := result.RowsAffected()
	if err != nil {
		return err
	}

	// We want atleast 1, if it is 0 the entry did not exist.
	if rowsAffected == 0 {
		return ErrRecordNotFound
	}

	return nil
}

// Get searches the database for a login name and returns the user struct on success.
func (u UserModel) Get(login string) (*User, error) {
	query := `
	SELECT id, added_at, login, twitchid, level, location, lastfm_username
	FROM users
	WHERE login = $1`

	var user User

	err := u.DB.QueryRow(query, login).Scan(
		&user.ID,
		&user.AddedAt,
		&user.Login,
		&user.TwitchID,
		&user.Level,
		&user.Location,
		&user.LastFMUsername,
	)

	if err != nil {
		switch {
		case errors.Is(err, sql.ErrNoRows):
			return nil, ErrRecordNotFound
		default:
			return nil, err
		}
	}

	return &user, nil
}

// Check checks for a record with the given login name.
func (u UserModel) Check(twitchId string) (*User, error) {
	query := `
	SELECT id, login
	FROM users
	WHERE twitchid = $1`

	var user User

	err := u.DB.QueryRow(query, twitchId).Scan(
		&user.ID,
		&user.Login,
	)

	if err != nil {
		switch {
		case errors.Is(err, sql.ErrNoRows):
			return nil, ErrRecordNotFound
		default:
			return nil, err
		}
	}

	return &user, nil
}

// Delete searches the database for a value with the supplied login name and if
// one exists deletes the record, returning any errors that might occur.
func (u UserModel) Delete(login string) error {
	// Prepare the statement.
	query := `
	DELETE FROM users
	WHERE login = $1`

	// Execute the query returning the number of affected rows.
	result, err := u.DB.Exec(query, login)
	if err != nil {
		return err
	}

	// Check how many rows were affected.
	rowsAffected, err := result.RowsAffected()
	if err != nil {
		return err
	}

	// We want atleast 1, if it is 0 the entry did not exist.
	if rowsAffected == 0 {
		return ErrRecordNotFound
	}

	return nil
}