package codec

import (
	ctxpkg "github.com/noahlann/nnet/pkg/context"
	protocolpkg "github.com/noahlann/nnet/pkg/protocol"
)

// Resolver Codec解析器接口
// 每个Resolver负责一种Codec选择策略，支持双向解析（解码和编码）
type Resolver interface {
	// ResolveForDecode 解析用于解码的Codec（数据流入）
	// registry: 编解码器注册表（可以动态注册codec）
	// ctx: 请求上下文
	// raw: 原始数据
	// header: 协议帧头（如果有）
	// 返回: 解析到的Codec，如果无法解析返回nil和nil错误（继续下一个resolver）
	// 注意：resolver可以将解析到的codec注册到registry中，供后续使用
	ResolveForDecode(registry Registry, ctx ctxpkg.Context, raw []byte, header protocolpkg.FrameHeader) (Codec, error)

	// ResolveForEncode 解析用于编码的Codec（数据流出）
	// registry: 编解码器注册表（可以动态注册codec）
	// ctx: 请求上下文
	// data: 要编码的数据（Go对象）
	// header: 协议帧头（如果有）
	// 返回: 解析到的Codec，如果无法解析返回nil和nil错误（继续下一个resolver）
	// 注意：通常流入时已经将codec注册到registry，这里可以直接获取
	ResolveForEncode(registry Registry, ctx ctxpkg.Context, data interface{}, header protocolpkg.FrameHeader) (Codec, error)
}

// ResolverChain Codec解析器链
// 按顺序尝试每个Resolver，直到找到可用的Codec
type ResolverChain struct {
	resolvers   []Resolver
	registry    Registry
	defaultName string
}

// NewResolverChain 创建解析器链
func NewResolverChain(registry Registry, defaultName string) *ResolverChain {
	return &ResolverChain{
		resolvers:    make([]Resolver, 0),
		registry:     registry,
		defaultName:  defaultName,
	}
}

// Add 添加解析器到链中
func (c *ResolverChain) Add(resolver Resolver) *ResolverChain {
	c.resolvers = append(c.resolvers, resolver)
	return c
}

// ResolveForDecode 执行解析器链用于解码，返回第一个成功解析的Codec
// 如果所有resolver都无法解析，返回默认Codec
func (c *ResolverChain) ResolveForDecode(ctx ctxpkg.Context, raw []byte, header protocolpkg.FrameHeader) (Codec, error) {
	// 按顺序尝试每个resolver
	for _, resolver := range c.resolvers {
		codec, err := resolver.ResolveForDecode(c.registry, ctx, raw, header)
		if err != nil {
			// 如果resolver返回错误，继续下一个
			continue
		}
		if codec != nil {
			return codec, nil
		}
	}

	// 所有resolver都无法解析，使用默认
	if c.defaultName != "" {
		if codec, err := c.registry.Get(c.defaultName); err == nil {
			return codec, nil
		}
	}
	return c.registry.Default(), nil
}

// ResolveForEncode 执行解析器链用于编码，返回第一个成功解析的Codec
// 如果所有resolver都无法解析，返回默认Codec
func (c *ResolverChain) ResolveForEncode(ctx ctxpkg.Context, data interface{}, header protocolpkg.FrameHeader) (Codec, error) {
	// 按顺序尝试每个resolver
	for _, resolver := range c.resolvers {
		codec, err := resolver.ResolveForEncode(c.registry, ctx, data, header)
		if err != nil {
			// 如果resolver返回错误，继续下一个
			continue
		}
		if codec != nil {
			return codec, nil
		}
	}

	// 所有resolver都无法解析，使用默认
	if c.defaultName != "" {
		if codec, err := c.registry.Get(c.defaultName); err == nil {
			return codec, nil
		}
	}
	return c.registry.Default(), nil
}