package binary

import (
	"bytes"
	"encoding/binary"
	"encoding/hex"

	"github.com/Mrs4s/MiraiGo/utils"
)

// Writer 写入
type Writer bytes.Buffer

func NewWriterF(f func(writer *Writer)) []byte {
	w := SelectWriter()
	f(w)
	b := append([]byte(nil), w.Bytes()...)
	PutWriter(w)
	return b
}

// OpenWriterF must call func close
func OpenWriterF(f func(writer *Writer)) (b []byte, close func()) {
	w := SelectWriter()
	w.WriteByte(0)
	f(w)
	return w.Bytes(), func() { PutWriter(w) }
}

func (w *Writer) Write(b []byte) {
	(*bytes.Buffer)(w).Write(b)
}

func (w *Writer) WriteHex(h string) {
	b, _ := hex.DecodeString(h)
	w.Write(b)
}

func (w *Writer) WriteByte(b byte) {
	(*bytes.Buffer)(w).WriteByte(b)
}

func (w *Writer) WriteUInt16(v uint16) {
	b := make([]byte, 2)
	binary.BigEndian.PutUint16(b, v)
	w.Write(b)
}

func (w *Writer) WriteUInt32(v uint32) {
	b := make([]byte, 4)
	binary.BigEndian.PutUint32(b, v)
	w.Write(b)
}

func (w *Writer) WriteUInt64(v uint64) {
	b := make([]byte, 8)
	binary.BigEndian.PutUint64(b, v)
	w.Write(b)
}

func (w *Writer) WriteString(v string) {
	payload := utils.S2B(v)
	w.WriteUInt32(uint32(len(payload) + 4))
	w.Write(payload)
}

func (w *Writer) WriteStringShort(v string) {
	w.WriteBytesShort(utils.S2B(v))
}

func (w *Writer) WriteBool(b bool) {
	if b {
		w.WriteByte(0x01)
	} else {
		w.WriteByte(0x00)
	}
}

func (w *Writer) EncryptAndWrite(key []byte, data []byte) {
	w.Write(NewTeaCipher(key).Encrypt(data))
}

func (w *Writer) WriteIntLvPacket(offset int, f func(writer *Writer)) {
	data, close := OpenWriterF(f)
	w.WriteUInt32(uint32(len(data) + offset))
	w.Write(data)
	close()
}

func (w *Writer) WriteUniPacket(commandName string, sessionId, extraData, body []byte) {
	w1 := SelectWriter()
	{ // WriteIntLvPacket
		w1.WriteString(commandName)
		w1.WriteUInt32(8)
		w1.Write(sessionId)
		if len(extraData) == 0 {
			w1.WriteUInt32(0x04)
		} else {
			w1.WriteUInt32(uint32(len(extraData) + 4))
			w1.Write(extraData)
		}
	}
	data := w1.Bytes()
	w.WriteUInt32(uint32(len(data) + 4))
	w.Write(data)
	PutWriter(w1)
	w.WriteUInt32(uint32(len(body) + 4)) // WriteIntLvPacket
	w.Write(body)
}

func (w *Writer) WriteBytesShort(data []byte) {
	w.WriteUInt16(uint16(len(data)))
	w.Write(data)
}

func (w *Writer) WriteTlvLimitedSize(data []byte, limit int) {
	if len(data) <= limit {
		w.WriteBytesShort(data)
		return
	}
	w.WriteBytesShort(data[:limit])
}

func (w *Writer) Bytes() []byte {
	return (*bytes.Buffer)(w).Bytes()
}

func (w *Writer) Cap() int {
	return (*bytes.Buffer)(w).Cap()
}

func (w *Writer) Reset() {
	(*bytes.Buffer)(w).Reset()
}

func (w *Writer) Grow(n int) {
	(*bytes.Buffer)(w).Grow(n)
}