feat: match and datagram handling

Author: paralin

client.go

@@ -5,9 +5,9 @@ import (
 	"errors"
 	"sync"
 
+	"github.com/Sirupsen/logrus"
 	"github.com/faceit/go-steam"
 	"github.com/faceit/go-steam/protocol/gamecoordinator"
-	"github.com/Sirupsen/logrus"
 	"github.com/golang/protobuf/proto"
 	devents "github.com/paralin/go-dota2/events"
 	// gcmm "github.com/paralin/go-dota2/protocol/dota_gcmessages_common_match_management"
@@ -93,6 +93,9 @@ func (d *Dota2) buildHandlerMap() {
 		uint32(gcm.EDOTAGCMsg_k_EMsgGCChatMessage):                    d.handleChatMessage,
 		uint32(gcm.EDOTAGCMsg_k_EMsgGCOtherJoinedChannel):             d.handleJoinedChannel,
 		uint32(gcm.EDOTAGCMsg_k_EMsgGCOtherLeftChannel):               d.handleLeftChannel,
+		uint32(gcm.EDOTAGCMsg_k_EMsgGCToClientSteamDatagramTicket):    d.handleSteamDatagramTicket,
+		uint32(gcsm.EGCBaseClientMsg_k_EMsgGCPingRequest):             d.handlePingRequest,
+		uint32(gcm.EDOTAGCMsg_k_EMsgGCToClientMatchSignedOut):         d.handleMatchSignedOut,
 	}
 }
 
@@ -157,7 +160,8 @@ func (d *Dota2) HandleGCPacket(packet *gamecoordinator.GCPacket) {
 	}
 }
 
-// Pong responds to a Ping.
-func (d *Dota2) Pong() {
+// handlePingRequest handles an incoming ping request from the gc.
+func (d *Dota2) handlePingRequest(packet *gamecoordinator.GCPacket) error {
 	d.write(uint32(gcsm.EGCBaseClientMsg_k_EMsgGCPingResponse), &gcsdkm.CMsgGCClientPing{})
+	return nil
 }

client_datagram.go

@@ -0,0 +1,28 @@
+package dota2
+
+import (
+	"github.com/faceit/go-steam/protocol/gamecoordinator"
+	"github.com/faceit/go-steam/steamid"
+	"github.com/paralin/go-dota2/events"
+	gcccm "github.com/paralin/go-dota2/protocol/dota_gcmessages_client_match_management"
+	gcm "github.com/paralin/go-dota2/protocol/dota_gcmessages_msgid"
+)
+
+// RequestSteamDatagramTicket requests a steam datagram ticket.
+func (d *Dota2) RequestSteamDatagramTicket(serverID steamid.SteamId) {
+	steamID := serverID.ToUint64()
+	d.write(uint32(gcm.EDOTAGCMsg_k_EMsgClientToGCRequestSteamDatagramTicket), &gcccm.CMsgClientToGCRequestSteamDatagramTicket{
+		ServerSteamId: &steamID,
+	})
+}
+
+// handleSteamDatagramTicket handles an incoming steam datagram ticket.
+func (d *Dota2) handleSteamDatagramTicket(packet *gamecoordinator.GCPacket) error {
+	ev := &events.SteamDatagramTicket{}
+	if err := d.unmarshalBody(packet, &ev.CMsgGCToClientSteamDatagramTicket); err != nil {
+		return err
+	}
+
+	d.emit(ev)
+	return nil
+}

client_lobby.go

@@ -33,7 +33,7 @@ func (d *Dota2) LaunchLobby() {
 }
 
 // LeaveCreateLobby attempts to leave any current lobby and creates a new one.
-func (d *Dota2) LeaveCreateLobby(ctx context.Context, details *gcccm.CMsgPracticeLobbySetDetails) error {
+func (d *Dota2) LeaveCreateLobby(ctx context.Context, details *gcccm.CMsgPracticeLobbySetDetails, destroyOldLobby bool) error {
 	cacheCtr, err := d.cache.GetContainerForTypeID(uint32(cso.Lobby))
 	if err != nil {
 		return err
@@ -57,6 +57,12 @@ func (d *Dota2) LeaveCreateLobby(ctx context.Context, details *gcccm.CMsgPractic
 			}
 
 			le.Debug("attempting to leave lobby")
+			if destroyOldLobby && lob.GetLeaderId() == d.client.SteamId().ToUint64() {
+				d.DestroyLobby()
+			}
+			if lob.GetState() != gcmm.CSODOTALobby_UI {
+				d.AbandonLobby()
+			}
 			d.LeaveLobby()
 		} else {
 			wasInNoLobby = true
@@ -78,6 +84,11 @@ func (d *Dota2) LeaveLobby() {
 	d.write(uint32(gcm.EDOTAGCMsg_k_EMsgGCPracticeLobbyLeave), &gcccm.CMsgPracticeLobbyLeave{})
 }
 
+// AbandonLobby abandons the current lobby.
+func (d *Dota2) AbandonLobby() {
+	d.write(uint32(gcm.EDOTAGCMsg_k_EMsgGCAbandonCurrentGame), &gcccm.CMsgAbandonCurrentGame{})
+}
+
 // DestroyLobby attempts to destroy the lobby.
 func (d *Dota2) DestroyLobby() {
 	d.write(uint32(gcm.EDOTAGCMsg_k_EMsgDestroyLobbyRequest), &gccm.CMsgDOTADestroyLobbyRequest{})

client_lobby_members.go

@@ -0,0 +1,24 @@
+package dota2
+
+import (
+	"github.com/faceit/go-steam/steamid"
+	bgcm "github.com/paralin/go-dota2/protocol/base_gcmessages"
+	gcccm "github.com/paralin/go-dota2/protocol/dota_gcmessages_client_match_management"
+	gcm "github.com/paralin/go-dota2/protocol/dota_gcmessages_msgid"
+)
+
+// LobbyKickPlayer kicks a player from the lobby by account ID.
+func (d *Dota2) LobbyKickPlayer(playerID steamid.SteamId) {
+	accountID := playerID.GetAccountId()
+	d.write(uint32(gcm.EDOTAGCMsg_k_EMsgGCPracticeLobbyKick), &gcccm.CMsgPracticeLobbyKick{
+		AccountId: &accountID,
+	})
+}
+
+// LobbyInvitePlayer attempts to invite a player to the current lobby.
+func (d *Dota2) LobbyInvitePlayer(playerID steamid.SteamId) {
+	steamID := playerID.ToUint64()
+	d.write(uint32(bgcm.EGCBaseMsg_k_EMsgGCInviteToLobby), &bgcm.CMsgInviteToLobby{
+		SteamId: &steamID,
+	})
+}

client_lobby_slot.go

@@ -0,0 +1,33 @@
+package dota2
+
+import (
+	"github.com/faceit/go-steam/steamid"
+	gcccm "github.com/paralin/go-dota2/protocol/dota_gcmessages_client_match_management"
+	gcm "github.com/paralin/go-dota2/protocol/dota_gcmessages_msgid"
+	"github.com/paralin/go-dota2/protocol/dota_shared_enums"
+)
+
+// LobbyJoinTeam switches team in a lobby.
+func (d *Dota2) LobbyJoinTeam(team dota_shared_enums.DOTA_GC_TEAM, slot uint32) {
+	d.write(uint32(gcm.EDOTAGCMsg_k_EMsgGCPracticeLobbySetTeamSlot), &gcccm.CMsgPracticeLobbySetTeamSlot{
+		Team: &team,
+		Slot: &slot,
+	})
+}
+
+// LobbySetSlotBotDifficulty sets the difficulty of a slot to a given bot difficulty.
+func (d *Dota2) LobbySetSlotBotDifficulty(team dota_shared_enums.DOTA_GC_TEAM, slot uint32, botDifficulty dota_shared_enums.DOTABotDifficulty) {
+	d.write(uint32(gcm.EDOTAGCMsg_k_EMsgGCPracticeLobbySetTeamSlot), &gcccm.CMsgPracticeLobbySetTeamSlot{
+		Team:          &team,
+		Slot:          &slot,
+		BotDifficulty: &botDifficulty,
+	})
+}
+
+// LobbyKickPlayerFromTeam kicks a player from whatever team they are on.
+func (d *Dota2) LobbyKickPlayerFromTeam(player steamid.SteamId) {
+	accountID := player.GetAccountId()
+	d.write(uint32(gcm.EDOTAGCMsg_k_EMsgGCPracticeLobbyKickFromTeam), &gcccm.CMsgPracticeLobbyKickFromTeam{
+		AccountId: &accountID,
+	})
+}

client_match.go

@@ -0,0 +1,17 @@
+package dota2
+
+import (
+	"github.com/faceit/go-steam/protocol/gamecoordinator"
+	"github.com/paralin/go-dota2/events"
+)
+
+// handleMatchSignedOut handles an incoming steam datagram ticket.
+func (d *Dota2) handleMatchSignedOut(packet *gamecoordinator.GCPacket) error {
+	ev := &events.MatchSignedOut{}
+	if err := d.unmarshalBody(packet, &ev.CMsgGCToClientMatchSignedOut); err != nil {
+		return err
+	}
+
+	d.emit(ev)
+	return nil
+}

events/datagram.go

@@ -0,0 +1,10 @@
+package events
+
+import (
+	gcccm "github.com/paralin/go-dota2/protocol/dota_gcmessages_client_match_management"
+)
+
+// SteamDatagramTicket contains an incoming steam datagram ticket.
+type SteamDatagramTicket struct {
+	gcccm.CMsgGCToClientSteamDatagramTicket
+}

events/match.go

@@ -0,0 +1,10 @@
+package events
+
+import (
+	gccm "github.com/paralin/go-dota2/protocol/dota_gcmessages_client"
+)
+
+// MatchSignedOut is emitted when the match signout event arrives from the GC.
+type MatchSignedOut struct {
+	gccm.CMsgGCToClientMatchSignedOut
+}