package chain

import (
	"context"
	"strings"

	"git.linkiio.cn/linkpay/api/invoke/apapter"
	"github.com/gogf/gf/v2/frame/g"
	"google.golang.org/grpc"
	"google.golang.org/grpc/metadata"
)

/**
 * 超时链
 * @author  dc.To
 * @version 20250409
 */
func ClientTimeoutInvokerChain(ctx context.Context, method string, req, res interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
	return invoker(ctx, method, req, res, cc, opts...)
}

/**
 * 上下文调用链
 * @param  void
 * @author  dc.To
 * @version 20250424
 */
func ClientContextInvokerChain(ctx context.Context, method string, req, res interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
	return invoker(ctx, method, req, nil, cc, opts...)
}

/**
 * Context Unary Interceptor
 * @param  void
 * @author  dc.To
 * @version 20250425
 */
func ClientContextUnaryChain(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
	return handler(ctx, req)
}

/**
 * Metadata Unary Interceptor
 * @author  dc.To
 * @version 20250418
 */
func ClientMetadataUnaryChain(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {

	md, ok := metadata.FromIncomingContext(ctx)
	if ok {
		for k, v := range md {
			if strings.HasPrefix(k, "x-cfg-") && len(v) > 0 {
				g.Cfg(strings.TrimPrefix(k, "x-cfg-")).SetAdapter(apapter.NewContextAdapter(ctx, k))
			}
		}
	} else {
		g.Log("unary").Error(ctx, "No Metadata from incoming cfg context")
	}
	return handler(ctx, req)
}