package govmtools

import (
	"bytes"
	"encoding/binary"
	"fmt"
)

var order = binary.BigEndian

func (s *Socket) RpcSend(data []byte) ([]byte, error) {
	// TODO error handling
	buf := bytes.NewBuffer(nil)

	binary.Write(buf, order, FieldInt64)
	binary.Write(buf, order, FieldIDType)
	binary.Write(buf, order, PacketTypeData)

	binary.Write(buf, order, FieldEmpty)

	binary.Write(buf, order, FieldString)
	binary.Write(buf, order, FieldIDPayload)
	binary.Write(buf, order, uint32(len(data)))

	buf.Write(data)

	res := make([]byte, buf.Len()+4)
	order.PutUint32(res, uint32(buf.Len()))
	copy(res[4:], buf.Bytes())

	_, err := s.conn.Write(res)
	if err != nil {
		return nil, fmt.Errorf("write: %w", err)
	}

	responseLength := make([]byte, 4)
	_, err = s.conn.Read(responseLength)
	if err != nil {
		return nil, fmt.Errorf("read len: %w", err)
	}

	response := make([]byte, order.Uint32(responseLength))
	_, err = s.conn.Read(response)
	if err != nil {
		return nil, fmt.Errorf("read: %w", err)
	}

	return response, nil
}