OpenDiablo2/d2networking/d2server/d2udpclientconnection/udp_client_connection.go

package d2udpclientconnection

import (
	"bytes"
	"compress/gzip"
	"encoding/json"
	"github.com/OpenDiablo2/OpenDiablo2/d2networking/d2client/d2clientconnectiontype"
	"net"

	"github.com/OpenDiablo2/OpenDiablo2/d2game/d2player"

	"github.com/OpenDiablo2/OpenDiablo2/d2networking/d2netpacket"
)

type UDPClientConnection struct {
	id            string
	address       *net.UDPAddr
	udpConnection *net.UDPConn
	playerState   *d2player.PlayerState
}

func CreateUDPClientConnection(udpConnection *net.UDPConn, id string, address *net.UDPAddr) *UDPClientConnection {
	result := &UDPClientConnection{
		id:            id,
		address:       address,
		udpConnection: udpConnection,
	}

	return result
}

func (u UDPClientConnection) GetUniqueId() string {
	return u.id
}

func (u UDPClientConnection) GetConnectionType() d2clientconnectiontype.ClientConnectionType {
	return d2clientconnectiontype.LANClient
}

func (u *UDPClientConnection) SendPacketToClient(packet d2netpacket.NetPacket) error {
	data, err := json.Marshal(packet.PacketData)
	if err != nil {
		return err
	}
	var buff bytes.Buffer
	buff.WriteByte(byte(packet.PacketType))
	writer, _ := gzip.NewWriterLevel(&buff, gzip.BestCompression)
	writer.Write(data)
	writer.Close()
	_, err = u.udpConnection.WriteToUDP(buff.Bytes(), u.address)
	if err != nil {
		return err
	}

	return nil
}

func (u *UDPClientConnection) SetPlayerState(playerState *d2player.PlayerState) {
	u.playerState = playerState
}

func (u *UDPClientConnection) GetPlayerState() *d2player.PlayerState {
	return u.playerState
}