package messages7

import (
	"slices"

	"github.com/teeworlds-go/go-teeworlds-protocol/chunk7"
	"github.com/teeworlds-go/go-teeworlds-protocol/network7"
	"github.com/teeworlds-go/go-teeworlds-protocol/packer"
	"github.com/teeworlds-go/go-teeworlds-protocol/snapshot7"
)

type SnapSingle struct {
	ChunkHeader *chunk7.ChunkHeader

	GameTick  int
	DeltaTick int
	Crc       int
	PartSize  int

	// TODO: remove data when snapshot packing works
	//       as of right now Data and Snapshot are the same thing
	Data     []byte
	Snapshot snapshot7.Snapshot
}

func (msg *SnapSingle) MsgId() int {
	return network7.MsgSysSnapSingle
}

func (msg *SnapSingle) MsgType() network7.MsgType {
	return network7.TypeNet
}

func (msg *SnapSingle) System() bool {
	return true
}

func (msg *SnapSingle) Vital() bool {
	return false
}

func (msg *SnapSingle) Pack() []byte {
	return slices.Concat(
		packer.PackInt(msg.GameTick),
		packer.PackInt(msg.DeltaTick),
		packer.PackInt(msg.Crc),
		packer.PackInt(msg.PartSize),
		msg.Data,
	)
}

func (msg *SnapSingle) Unpack(u *packer.Unpacker) error {
	msg.GameTick = u.GetInt()
	msg.DeltaTick = u.GetInt()
	msg.Crc = u.GetInt()
	msg.PartSize = u.GetInt()
	msg.Data = u.Rest()

	// genius
	u.Reset(msg.Data)
	err := msg.Snapshot.Unpack(u)
	if err != nil {
		return err
	}

	return nil
}

func (msg *SnapSingle) Header() *chunk7.ChunkHeader {
	return msg.ChunkHeader
}

func (msg *SnapSingle) SetHeader(header *chunk7.ChunkHeader) {
	msg.ChunkHeader = header
}