package middleware

import (
	"context"
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/nacos-group/nacos-sdk-go/inner/uuid"
)

const (
	GIN_CONTEXT_KEY = "GinContextKey"
)

// https://gqlgen.com/recipes/gin/#accessing-gincontext
func GinContextToContextMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		ctx := context.WithValue(c.Request.Context(), GIN_CONTEXT_KEY, c)
		c.Request = c.Request.WithContext(ctx)
		c.Next()
	}
}

func GinContextFromContext(ctx context.Context) (*gin.Context, error) {
	ginContext := ctx.Value(GIN_CONTEXT_KEY)
	if ginContext == nil {
		err := fmt.Errorf("could not retrieve gin.Context")
		return nil, err
	}

	gc, ok := ginContext.(*gin.Context)
	if !ok {
		err := fmt.Errorf("gin.Context has wrong type")
		return nil, err
	}
	return gc, nil
}

// cli context
func GetCliCtx() *gin.Context {

	rid := ""
	if uuidGen != uuid.Nil {
		rid = uuidGen.String()
	}

	ctx := &gin.Context{}
	ctx.Set(REQUEST_ID, rid)
	return ctx
}