func TestUgcHighlander(t *testing.T) {
res, err := models.LobbyGetPlayerSlot(models.LobbyTypeSixes, "red", "scout1")
assert.Equal(t, 0, res)
assert.Nil(t, err)
res, err = models.LobbyGetPlayerSlot(models.LobbyTypeHighlander, "blu", "heavy")
assert.Equal(t, 13, res)
assert.Nil(t, err)
res, err = models.LobbyGetPlayerSlot(models.LobbyTypeHighlander, "blu", "garbageman")
assert.NotNil(t, err)
res, err = models.LobbyGetPlayerSlot(models.LobbyTypeSixes, "ylw", "demoman")
assert.NotNil(t, err)
}
func DebugPlayerSub(server *wsevent.Server, so *wsevent.Client, data []byte) []byte {
reqerr := chelpers.FilterRequest(so, 0, true)
if reqerr != nil {
return reqerr.Encode()
}
var args struct {
Id *uint `json:"id"`
Team *string `json:"team"`
Class *string `json:"class"`
}
err := chelpers.GetParams(data, &args)
if err != nil {
return helpers.NewTPErrorFromError(err).Encode()
}
lob, tperr := models.GetLobbyById(*args.Id)
if tperr != nil {
return tperr.Encode()
}
s, tperr := models.LobbyGetPlayerSlot(lob.Type, *args.Team, *args.Class)
if tperr != nil {
return tperr.Encode()
}
slot := models.LobbySlot{}
err = db.DB.Where("slot = ?", s).First(&slot).Error
if err != nil {
helpers.Logger.Debug("", slot, s)
return helpers.NewTPErrorFromError(err).Encode()
}
player := models.Player{}
err = db.DB.First(&player, slot.PlayerId).Error
if err != nil {
helpers.Logger.Debug("", player)
return helpers.NewTPErrorFromError(err).Encode()
}
sub, _ := models.NewSub(*args.Id, player.SteamId)
db.DB.Save(sub)
models.BroadcastSubList()
return chelpers.EmptySuccessJS
}
func newRequirement(team, class string, requirement Requirement, lobby *models.Lobby) *helpers.TPError {
slot, err := models.LobbyGetPlayerSlot(lobby.Type, team, class)
if err != nil {
return err
}
slotReq := &models.Requirement{
LobbyID: lobby.ID,
Slot: slot,
Hours: requirement.Hours,
Lobbies: requirement.Lobbies,
}
slotReq.Save()
return nil
}
func (Helen) GetSteamIDFromSlot(args Args, steamID *string) error {
slot, tperr := models.LobbyGetPlayerSlot(args.Type, args.Team, args.Class)
var playerID uint
if tperr != nil {
return errors.New(tperr.Error())
}
err := db.DB.DB().QueryRow("SELECT player_id FROM lobby_slots WHERE slot = $1 AND lobby_id = $2", slot, args.LobbyID).Scan(&playerID)
if err != nil {
return err
}
err = db.DB.DB().QueryRow("SELECT steam_id FROM players WHERE id = $1", playerID).Scan(steamID)
if err != nil {
return err
}
return nil
}
func (Lobby) LobbyJoin(server *wsevent.Server, so *wsevent.Client, data []byte) []byte {
reqerr := chelpers.FilterRequest(so, authority.AuthAction(0), true)
if reqerr != nil {
return reqerr.Encode()
}
var args struct {
Id *uint `json:"id"`
Class *string `json:"class"`
Team *string `json:"team" valid:"red,blu"`
}
if err := chelpers.GetParams(data, &args); err != nil {
return helpers.NewTPErrorFromError(err).Encode()
}
//helpers.Logger.Debug("id %d class %s team %s", *args.Id, *args.Class, *args.Team)
player, tperr := models.GetPlayerBySteamId(chelpers.GetSteamId(so.Id()))
if tperr != nil {
return tperr.Encode()
}
lob, tperr := models.GetLobbyById(*args.Id)
if tperr != nil {
return tperr.Encode()
}
if lob.State == models.LobbyStateEnded {
return helpers.NewTPError("Cannot join a closed lobby.", -1).Encode()
}
//Check if player is in the same lobby
var sameLobby bool
if id, err := player.GetLobbyId(); err == nil && id == *args.Id {
sameLobby = true
}
slot, tperr := models.LobbyGetPlayerSlot(lob.Type, *args.Team, *args.Class)
if tperr != nil {
return tperr.Encode()
}
if prevId, err := player.GetLobbyId(); err != nil {
server.RemoveClient(so.Id(), fmt.Sprintf("%d_public", prevId))
server.RemoveClient(so.Id(), fmt.Sprintf("%d_private", prevId))
}
tperr = lob.AddPlayer(player, slot, *args.Team, *args.Class)
if tperr != nil {
return tperr.Encode()
}
if !sameLobby {
chelpers.AfterLobbyJoin(server, so, lob, player)
}
if lob.IsFull() {
lob.State = models.LobbyStateReadyingUp
lob.ReadyUpTimestamp = time.Now().Unix() + 30
lob.Save()
tick := time.After(time.Second * 30)
id := lob.ID
stop := make(chan struct{})
go func() {
select {
case <-tick:
lobby := &models.Lobby{}
db.DB.First(lobby, id)
if lobby.State != models.LobbyStateInProgress {
err := lobby.RemoveUnreadyPlayers()
if err != nil {
helpers.Logger.Error("RemoveUnreadyPlayers: ", err.Error())
err = nil
}
err = lobby.UnreadyAllPlayers()
if err != nil {
helpers.Logger.Error("UnreadyAllPlayers: ", err.Error())
}
lobby.State = models.LobbyStateWaiting
lobby.Save()
}
case <-stop:
return
}
}()
room := fmt.Sprintf("%s_private",
chelpers.GetLobbyRoom(lob.ID))
broadcaster.SendMessageToRoom(room, "lobbyReadyUp",
`{"timeout":30}`)
models.BroadcastLobbyList()
}
err := models.AllowPlayer(*args.Id, player.SteamId, *args.Team+*args.Class)
if err != nil {
helpers.Logger.Error(err.Error())
}
if lob.State == models.LobbyStateInProgress {
bytes, _ := json.Marshal(models.DecorateLobbyConnect(lob, player.Name, *args.Class))
broadcaster.SendMessage(player.SteamId, "lobbyStart", string(bytes))
}
return chelpers.EmptySuccessJS
}
func (Lobby) LobbyJoin(so *wsevent.Client, args struct {
Id *uint `json:"id"`
Class *string `json:"class"`
Team *string `json:"team" valid:"red,blu"`
Password *string `json:"password" empty:"-"`
}) interface{} {
player := chelpers.GetPlayerFromSocket(so.ID)
if banned, until := player.IsBannedWithTime(models.PlayerBanJoin); banned {
str := fmt.Sprintf("You have been banned from joining lobbies till %s", until.Format(time.RFC822))
return helpers.NewTPError(str, -1)
}
//logrus.Debug("id %d class %s team %s", *args.Id, *args.Class, *args.Team)
lob, tperr := models.GetLobbyByID(*args.Id)
if tperr != nil {
return tperr
}
if lob.State == models.LobbyStateEnded {
return helpers.NewTPError("Cannot join a closed lobby.", -1)
}
//Check if player is in the same lobby
var sameLobby bool
if id, err := player.GetLobbyID(false); err == nil && id == *args.Id {
sameLobby = true
}
slot, tperr := models.LobbyGetPlayerSlot(lob.Type, *args.Team, *args.Class)
if tperr != nil {
return tperr
}
if prevId, _ := player.GetLobbyID(false); prevId != 0 && !sameLobby {
lobby, _ := models.GetLobbyByID(prevId)
hooks.AfterLobbyLeave(lobby, player)
}
tperr = lob.AddPlayer(player, slot, *args.Password)
if tperr != nil {
return tperr
}
if !sameLobby {
hooks.AfterLobbyJoin(so, lob, player)
}
//check if lobby isn't already in progress (which happens when the player is subbing)
if lob.IsFull() && lob.State != models.LobbyStateInProgress {
lob.State = models.LobbyStateReadyingUp
lob.ReadyUpTimestamp = time.Now().Unix() + 30
lob.Save()
time.AfterFunc(time.Second*30, func() {
lobby := &models.Lobby{}
db.DB.First(lobby, lob.ID)
//if all player's haven't readied up,
//remove unreadied players and unready the
//rest.
if lobby.State != models.LobbyStateInProgress && lobby.State != models.LobbyStateEnded {
removeUnreadyPlayers(lobby)
lobby.State = models.LobbyStateWaiting
lobby.Save()
}
})
room := fmt.Sprintf("%s_private",
hooks.GetLobbyRoom(lob.ID))
broadcaster.SendMessageToRoom(room, "lobbyReadyUp",
struct {
Timeout int `json:"timeout"`
}{30})
models.BroadcastLobbyList()
}
if lob.State == models.LobbyStateInProgress { //this happens when the player is a substitute
db.DB.Preload("ServerInfo").First(lob, lob.ID)
so.EmitJSON(helpers.NewRequest("lobbyStart", models.DecorateLobbyConnect(lob, player.Name, slot)))
}
return chelpers.EmptySuccessJS
}
func LobbyJoin(so socketio.Socket) func(string) string {
return chelpers.FilterRequest(so, lobbyJoinFilters,
func(params map[string]interface{}) string {
player, tperr := models.GetPlayerBySteamId(chelpers.GetSteamId(so.Id()))
if tperr != nil {
bytes, _ := tperr.ErrorJSON().Encode()
return string(bytes)
}
lobbyid := params["id"].(uint)
classString := params["class"].(string)
teamString := params["team"].(string)
lob, tperr := models.GetLobbyById(uint(lobbyid))
if tperr != nil {
bytes, _ := tperr.ErrorJSON().Encode()
return string(bytes)
}
//Check if player is in the same lobby
var sameLobby bool
if id, err := player.GetLobbyId(); err == nil && id == lobbyid {
sameLobby = true
}
slot, tperr := models.LobbyGetPlayerSlot(lob.Type, teamString, classString)
if tperr != nil {
bytes, _ := tperr.ErrorJSON().Encode()
return string(bytes)
}
helpers.LockRecord(lob.ID, lob)
defer helpers.UnlockRecord(lob.ID, lob)
tperr = lob.AddPlayer(player, slot)
if tperr != nil {
bytes, _ := tperr.ErrorJSON().Encode()
return string(bytes)
}
if !sameLobby {
chelpers.AfterLobbyJoin(so, lob, player)
}
if lob.IsFull() {
lob.State = models.LobbyStateReadyingUp
lob.Save()
lob.ReadyUpTimeoutCheck()
room := fmt.Sprintf("%s_private",
chelpers.GetLobbyRoom(lob.ID))
broadcaster.SendMessageToRoom(room, "lobbyReadyUp",
`{"timeout":30}`)
models.BroadcastLobbyList()
}
models.BroadcastLobbyToUser(lob, player.SteamId)
bytes, _ := chelpers.BuildSuccessJSON(simplejson.New()).Encode()
return string(bytes)
})
}