package ormfield

import (
	"fmt"
	"io"
	"strings"

	"google.golang.org/protobuf/reflect/protoreflect"
)

// StringCodec encodes strings as raw bytes.
type StringCodec struct{}

func (s StringCodec) FixedBufferSize() int {
	return -1
}

func (s StringCodec) ComputeBufferSize(value protoreflect.Value) (int, error) {
	return len(value.Interface().(string)), nil
}

func (s StringCodec) IsOrdered() bool {
	return true
}

func (s StringCodec) Compare(v1, v2 protoreflect.Value) int {
	return strings.Compare(v1.Interface().(string), v2.Interface().(string))
}

func (s StringCodec) Decode(r Reader) (protoreflect.Value, error) {
	bz, err := io.ReadAll(r)
	return protoreflect.ValueOfString(string(bz)), err
}

func (s StringCodec) Encode(value protoreflect.Value, w io.Writer) error {
	_, err := w.Write([]byte(value.Interface().(string)))
	return err
}

// NonTerminalStringCodec encodes strings as null-terminated raw bytes. Null
// values within strings will produce an error.
type NonTerminalStringCodec struct{}

func (s NonTerminalStringCodec) FixedBufferSize() int {
	return -1
}

func (s NonTerminalStringCodec) ComputeBufferSize(value protoreflect.Value) (int, error) {
	return len(value.Interface().(string)) + 1, nil
}

func (s NonTerminalStringCodec) IsOrdered() bool {
	return true
}

func (s NonTerminalStringCodec) Compare(v1, v2 protoreflect.Value) int {
	return strings.Compare(v1.Interface().(string), v2.Interface().(string))
}

func (s NonTerminalStringCodec) Decode(r Reader) (protoreflect.Value, error) {
	var bz []byte
	for {
		b, err := r.ReadByte()
		if b == 0 || err == io.EOF {
			return protoreflect.ValueOfString(string(bz)), err
		}
		bz = append(bz, b)
	}
}

func (s NonTerminalStringCodec) Encode(value protoreflect.Value, w io.Writer) error {
	str := value.Interface().(string)
	bz := []byte(str)
	for _, b := range bz {
		if b == 0 {
			return fmt.Errorf("illegal null terminator found in index string: %s", str)
		}
	}
	_, err := w.Write([]byte(str))
	if err != nil {
		return err
	}
	_, err = w.Write(nullTerminator)
	return err
}

var nullTerminator = []byte{0}