add db get queries

internal/foundry/models/db/user.go

@@ -2,11 +2,13 @@ package db
 
 import (
 	"context"
+	"database/sql"
+	"errors"
 	"time"
 )
 
 type User struct {
-	Id        int
+	Id        int64
 	Name      string
 	Role      int
 	Character string
@@ -18,7 +20,7 @@ type User struct {
 }
 
 type UserStats struct {
-	Id             int
+	Id             int64
 	CoreVersion    string
 	SystemId       string
 	SystemVersion  string
@@ -33,7 +35,7 @@ func (users Users) GetById(id int) *User {
 	return &users[id]
 }
 
-func (m FoundryStateModel) InsertUser(user *User, stateId int) error {
+func (m FoundryStateModel) InsertUser(user *User, stateId int64) error {
 	query := `
 		INSERT INTO users (state_id, name, role, character, color, pronouns)
 		VALUES ($1, $2, $3, $4, $5, $6)
@@ -59,7 +61,7 @@ func (m FoundryStateModel) InsertUser(user *User, stateId int) error {
 	return m.InsertUserStats(&user.Stats, user.Id)
 }
 
-func (m FoundryStateModel) InsertUserHotbar(key string, value string, userId int) error {
+func (m FoundryStateModel) InsertUserHotbar(key string, value string, userId int64) error {
 	query := `
 		INSERT INTO users_hotbar (user_id, key, value)
 		VALUES ($1, $2, $3)`
@@ -76,7 +78,7 @@ func (m FoundryStateModel) InsertUserHotbar(key string, value string, userId int
 	return nil
 }
 
-func (m FoundryStateModel) InsertUserStats(stats *UserStats, userId int) error {
+func (m FoundryStateModel) InsertUserStats(stats *UserStats, userId int64) error {
 	query := `
 		INSERT INTO users_stats (user_id, core_version, system_id, system_version, created_time, modified_time, last_modified_by)
 		VALUES ($1, $2, $3, $4, $5, $6, $7)
@@ -89,3 +91,150 @@ func (m FoundryStateModel) InsertUserStats(stats *UserStats, userId int) error {
 
 	return m.DB.QueryRowContext(ctx, query, args...).Scan(&stats.Id)
 }
+
+func (m FoundryStateModel) GetUsers(idState int64) (Users, error) {
+	if idState < 1 {
+		return nil, ErrorRecordNotFound
+	}
+
+	query := `
+		SELECT id, created_at, name, role, character, color, pronouns
+		FROM users
+		WHERE state_id = $1`
+
+	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
+	defer cancel()
+
+	rows, err := m.DB.QueryContext(ctx, query, idState)
+	if err != nil {
+		switch {
+		case errors.Is(err, sql.ErrNoRows):
+			return nil, ErrorRecordNotFound
+		default:
+			return nil, err
+		}
+	}
+
+	users := make(Users, 0, 1)
+
+	for rows.Next() {
+		var user User
+		err := rows.Scan(
+			&user.Id,
+			&user.CreatedAt,
+			&user.Name,
+			&user.Role,
+			&user.Character,
+			&user.Color,
+			&user.Pronouns,
+		)
+		if err != nil {
+			return nil, err
+		}
+		user.Hotbar, err = m.GetUserHotbar(user.Id)
+		if err != nil {
+			return nil, err
+		}
+		userState, err := m.GetUserStats(user.Id)
+		if err != nil {
+			return nil, err
+		}
+
+		user.Stats = *userState
+		users = append(users, user)
+	}
+
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+
+	return users, nil
+}
+
+func (m FoundryStateModel) GetUserHotbar(idUser int64) (map[string]string, error) {
+	if idUser < 1 {
+		return nil, ErrorRecordNotFound
+	}
+
+	query := `
+		SELECT key, value
+		FROM users_hotbar
+		WHERE user_id = $1`
+
+	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
+	defer cancel()
+
+	rows, err := m.DB.QueryContext(ctx, query, idUser)
+	if err != nil {
+		switch {
+		case errors.Is(err, sql.ErrNoRows):
+			return nil, ErrorRecordNotFound
+		default:
+			return nil, err
+		}
+	}
+
+	hotbar := make(map[string]string)
+	for rows.Next() {
+		var key string
+		var val string
+
+		err := rows.Scan(
+			&key,
+			&val,
+		)
+		if err != nil {
+			return nil, err
+		}
+
+		hotbar[key] = val
+	}
+
+	if err != nil {
+		switch {
+		case errors.Is(err, sql.ErrNoRows):
+			return nil, ErrorRecordNotFound
+		default:
+			return nil, err
+		}
+	}
+
+	return hotbar, nil
+}
+
+func (m FoundryStateModel) GetUserStats(idSystem int64) (*UserStats, error) {
+	if idSystem < 1 {
+		return nil, ErrorRecordNotFound
+	}
+
+	query := `
+		SELECT id, core_version, system_id, system_version, created_time, modified_time, last_modified_by
+		FROM users_stats
+		WHERE user_id = $1`
+
+	var userStats UserStats
+
+	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
+	defer cancel()
+
+	err := m.DB.QueryRowContext(ctx, query, idSystem).Scan(
+		&userStats.Id,
+		&userStats.CoreVersion,
+		&userStats.SystemId,
+		&userStats.SystemVersion,
+		&userStats.CreatedTime,
+		&userStats.ModifiedTime,
+		&userStats.LastModifiedBy,
+	)
+
+	if err != nil {
+		switch {
+		case errors.Is(err, sql.ErrNoRows):
+			return nil, ErrorRecordNotFound
+		default:
+			return nil, err
+		}
+	}
+
+	return &userStats, nil
+}