package chain

import (
	"context"
	"strings"

	"github.com/gogf/gf/v2/encoding/gbase64"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/os/gcfg"
	"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 {
				c := strings.Join(v, "")
				if len(c) > 0 {
					adapter, err := gcfg.NewAdapterContent(gbase64.MustDecodeToString(v[0]))
					if err != nil {
						g.Log("unary").Error(ctx, err)
					}
					g.Cfg(strings.TrimPrefix(k, "x-cfg-")).SetAdapter(adapter)
				}
			}
		}
	}
	return handler(ctx, req)
}