From 7d1e97f66a795660fc9bf22361e1395d652287fe Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andr=C3=A9=20Hahn?= <andre.hahn@tfgco.com>
Date: Mon, 28 Oct 2019 11:36:02 -0300
Subject: [PATCH] Allow client to customize handshake data

---
 client/client.go | 50 ++++++++++++++++++++++++++++++++++--------------
 1 file changed, 36 insertions(+), 14 deletions(-)

diff --git a/client/client.go b/client/client.go
index a382fa8e..e9c0565c 100644
--- a/client/client.go
+++ b/client/client.go
@@ -37,6 +37,7 @@ import (
 	"github.com/topfreegames/pitaya/conn/message"
 	"github.com/topfreegames/pitaya/conn/packet"
 	"github.com/topfreegames/pitaya/logger"
+	"github.com/topfreegames/pitaya/session"
 	"github.com/topfreegames/pitaya/util/compression"
 )
 
@@ -76,19 +77,20 @@ type pendingRequest struct {
 
 // Client struct
 type Client struct {
-	conn            net.Conn
-	Connected       bool
-	packetEncoder   codec.PacketEncoder
-	packetDecoder   codec.PacketDecoder
-	packetChan      chan *packet.Packet
-	IncomingMsgChan chan *message.Message
-	pendingChan     chan bool
-	pendingRequests map[uint]*pendingRequest
-	pendingReqMutex sync.Mutex
-	requestTimeout  time.Duration
-	closeChan       chan struct{}
-	nextID          uint32
-	messageEncoder  message.Encoder
+	conn                net.Conn
+	Connected           bool
+	packetEncoder       codec.PacketEncoder
+	packetDecoder       codec.PacketDecoder
+	packetChan          chan *packet.Packet
+	IncomingMsgChan     chan *message.Message
+	pendingChan         chan bool
+	pendingRequests     map[uint]*pendingRequest
+	pendingReqMutex     sync.Mutex
+	requestTimeout      time.Duration
+	closeChan           chan struct{}
+	nextID              uint32
+	messageEncoder      message.Encoder
+	clientHandshakeData *session.HandshakeData
 }
 
 // MsgChannel return the incoming message channel
@@ -125,11 +127,31 @@ func New(logLevel logrus.Level, requestTimeout ...time.Duration) *Client {
 		// TODO this should probably be configurable
 		pendingChan:    make(chan bool, 30),
 		messageEncoder: message.NewMessagesEncoder(true),
+		clientHandshakeData: &session.HandshakeData{
+			Sys: session.HandshakeClientData{
+				Platform:    "mac",
+				LibVersion:  "0.3.5-release",
+				BuildNumber: "20",
+				Version:     "2.1",
+			},
+			User: map[string]interface{}{
+				"age": 30,
+			},
+		},
 	}
 }
 
+func (c *Client) SetClientHandshakeData(data *session.HandshakeData) {
+	c.clientHandshakeData = data
+}
+
 func (c *Client) sendHandshakeRequest() error {
-	p, err := c.packetEncoder.Encode(packet.Handshake, []byte(handshakeBuffer))
+	enc, err := json.Marshal(c.clientHandshakeData)
+	if err != nil {
+		return err
+	}
+
+	p, err := c.packetEncoder.Encode(packet.Handshake, enc)
 	if err != nil {
 		return err
 	}