package request

import (
	"encoding/json"
	"fmt"

	protocolpkg "github.com/noahlann/nnet/pkg/protocol"
	requestpkg "github.com/noahlann/nnet/pkg/request"
)

// requestImpl 请求实现
// 实现了Request和RequestSetter接口
type requestImpl struct {
	raw       []byte
	data      interface{}
	dataBytes []byte
	header    requestpkg.FrameHeader
	protocol  protocolpkg.Protocol
}

// 确保requestImpl实现了RequestSetter接口
var _ RequestSetter = (*requestImpl)(nil)

// New 创建新的请求对象
func New(raw []byte, protocol protocolpkg.Protocol) requestpkg.Request {
	return &requestImpl{
		raw:      raw,
		protocol: protocol,
	}
}

// Raw 获取原始数据
func (r *requestImpl) Raw() []byte {
	return r.raw
}

// Data 获取数据部分（已自动解码为Go对象）
func (r *requestImpl) Data() interface{} {
	return r.data
}

// SetData 设置数据部分（内部使用，导出以供服务器使用）
func (r *requestImpl) SetData(data interface{}) {
	r.data = data
}

// Bind 绑定到指定的结构体
func (r *requestImpl) Bind(v interface{}) error {
	// 如果Data已经存在，直接使用Data
	if r.data != nil {
		// 使用反射将data的值复制到v
		// 这里简化处理，实际可以使用json或其他序列化方式
		return bindDataToStruct(r.data, v)
	}

	// 如果Data不存在，但有DataBytes，使用DataBytes进行解码
	if len(r.dataBytes) > 0 {
		// 这里需要知道使用什么codec，但Request没有codec信息
		// 所以Bind方法主要用于将已解码的Data绑定到指定的struct
		// 如果需要从DataBytes解码，应该直接使用Data()获取结果
		return fmt.Errorf("data bytes available but no codec specified, use Data() method instead")
	}

	return fmt.Errorf("no data available to bind")
}

// bindDataToStruct 将data绑定到struct
func bindDataToStruct(data interface{}, v interface{}) error {
	// 使用JSON作为中间格式进行绑定
	// 这适用于大部分场景
	dataBytes, err := json.Marshal(data)
	if err != nil {
		return fmt.Errorf("failed to marshal data: %w", err)
	}

	return json.Unmarshal(dataBytes, v)
}

// Header 获取协议帧头
func (r *requestImpl) Header() requestpkg.FrameHeader {
	return r.header
}

// SetHeader 设置协议帧头（内部使用，导出以供服务器使用）
func (r *requestImpl) SetHeader(header requestpkg.FrameHeader) {
	r.header = header
}

// DataBytes 获取数据部分的原始字节
func (r *requestImpl) DataBytes() []byte {
	return r.dataBytes
}

// SetDataBytes 设置数据部分的原始字节（内部使用，导出以供服务器使用）
func (r *requestImpl) SetDataBytes(dataBytes []byte) {
	r.dataBytes = dataBytes
}

// Protocol 获取协议信息
func (r *requestImpl) Protocol() protocolpkg.Protocol {
	return r.protocol
}

// ProtocolVersion 获取协议版本
func (r *requestImpl) ProtocolVersion() string {
	if r.protocol != nil {
		return r.protocol.Version()
	}
	return ""
}

// Reset 重置Request状态（用于对象池）
func (r *requestImpl) Reset() {
	r.raw = nil
	r.data = nil
	r.dataBytes = nil
	r.header = nil
	r.protocol = nil
}