package build

import (
	"bytes"
	"encoding/binary"
	"io"
	"time"
)

func GetNowStr(isClient bool) string {
	var msg string
	msg += time.Now().Format("2006-01-02 15:04:05.000")
	if isClient {
		msg += "| cli -> ser |"
	} else {
		msg += "| ser -> cli |"
	}
	return msg
}

func ReadStringFromByte(b []byte) (string, int) {

	var l int
	l = bytes.IndexByte(b, 0x00)
	if l == -1 {
		l = len(b)
	}
	return string(b[0:l]), l
}

func LengthBinary(b []byte) (uint32, int) {

	var first = int(b[0])
	if first > 0 && first <= 250 {
		return uint32(first), 1
	}
	if first == 251 {
		return 0, 1
	}
	if first == 252 {
		return binary.LittleEndian.Uint32(b[1:2]), 1
	}
	if first == 253 {
		return binary.LittleEndian.Uint32(b[1:4]), 3
	}
	if first == 254 {
		return binary.LittleEndian.Uint32(b[1:9]), 8
	}
	return 0, 0
}

func LengthEncodedInt(input []byte) (num uint64, isNull bool, n int) {

	switch input[0] {

	case 0xfb:
		n = 1
		isNull = true
		return
	case 0xfc:
		num = uint64(input[1]) | uint64(input[2])<<8
		n = 3
		return
	case 0xfd:
		num = uint64(input[1]) | uint64(input[2])<<8 | uint64(input[3])<<16
		n = 4
		return
	case 0xfe:
		num = uint64(input[1]) | uint64(input[2])<<8 | uint64(input[3])<<16 |
			uint64(input[4])<<24 | uint64(input[5])<<32 | uint64(input[6])<<40 |
			uint64(input[7])<<48 | uint64(input[8])<<56
		n = 9
		return
	}

	num = uint64(input[0])
	n = 1
	return
}

func LengthEncodedString(b []byte) ([]byte, bool, int, error) {

	num, isNull, n := LengthEncodedInt(b)
	if num < 1 {
		return nil, isNull, n, nil
	}

	n += int(num)

	if len(b) >= n {
		return b[n-int(num) : n], false, n, nil
	}
	return nil, false, n, io.EOF
}