78 lines
2.6 KiB
Go
78 lines
2.6 KiB
Go
package repos
|
|
|
|
import (
|
|
"context"
|
|
"gralias/models"
|
|
)
|
|
|
|
type PlayersRepo interface {
|
|
PlayerGetByName(username string) (*models.Player, error)
|
|
PlayerAdd(player *models.Player) error
|
|
PlayerUpdate(player *models.Player) error
|
|
PlayerDelete(roomID, username string) error
|
|
PlayerSetRoomID(username, roomID string) error
|
|
PlayerExitRoom(username string) error
|
|
PlayerListNames() ([]string, error)
|
|
PlayerList(isBot bool) ([]models.Player, error)
|
|
}
|
|
|
|
func (p *RepoProvider) PlayerListNames() ([]string, error) {
|
|
var names []string
|
|
err := p.DB.SelectContext(context.Background(), &names, "SELECT username FROM players;")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return names, nil
|
|
}
|
|
|
|
func (p *RepoProvider) PlayerGetByName(username string) (*models.Player, error) {
|
|
var player models.Player
|
|
err := p.DB.GetContext(context.Background(), &player, "SELECT id, room_id, username, team, role, is_bot FROM players WHERE username = ?", username)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &player, nil
|
|
}
|
|
|
|
func (p *RepoProvider) PlayerAdd(player *models.Player) error {
|
|
_, err := p.DB.ExecContext(context.Background(), "INSERT INTO players (room_id, username, team, role, is_bot) VALUES (?, ?, ?, ?, ?)",
|
|
player.RoomID, player.Username, player.Team, player.Role, player.IsBot)
|
|
return err
|
|
}
|
|
|
|
func (p *RepoProvider) PlayerUpdate(player *models.Player) error {
|
|
_, err := p.DB.ExecContext(context.Background(), "UPDATE players SET room_id = ?, username = ?, team = ?, role = ?, is_bot = ? WHERE id = ?",
|
|
player.RoomID, player.Username, player.Team, player.Role, player.IsBot, player.ID)
|
|
return err
|
|
}
|
|
|
|
func (p *RepoProvider) PlayerDelete(roomID, username string) error {
|
|
_, err := p.DB.ExecContext(context.Background(), "DELETE FROM players WHERE room_id = ? AND username = ?", roomID, username)
|
|
return err
|
|
}
|
|
|
|
func (p *RepoProvider) PlayerSetRoomID(username, roomID string) error {
|
|
_, err := p.DB.ExecContext(context.Background(), "UPDATE players SET room_id = ? WHERE username = ?", roomID, username)
|
|
return err
|
|
}
|
|
|
|
func (p *RepoProvider) PlayerExitRoom(username string) error {
|
|
_, err := p.DB.ExecContext(context.Background(), "UPDATE players SET room_id = null WHERE username = ?", username)
|
|
return err
|
|
}
|
|
|
|
func (p *RepoProvider) PlayerList(isBot bool) ([]models.Player, error) {
|
|
var players []models.Player
|
|
var query string
|
|
if isBot {
|
|
query = "SELECT id, room_id, username, team, role, is_bot FROM players WHERE is_bot = true;"
|
|
} else {
|
|
query = "SELECT id, room_id, username, team, role, is_bot FROM players WHERE is_bot = false;"
|
|
}
|
|
err := p.DB.SelectContext(context.Background(), &players, query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return players, nil
|
|
}
|