package rest

import (
	"context"
	"net/http"

	"github.com/google/uuid"
)

type contextKey struct {
	name string
}

var (
	reqCtxKey = &contextKey{"HTTPRequest"}
)

type ctxKey int

const (
	requestIDCtxKey    = ctxKey(1)
	requestIDHeaderKey = "X-Request-Id"
)

// HTTPRequest return svr's request
func HTTPRequest(ctx context.Context) *http.Request { // nolint
	val, ok := ctx.Value(reqCtxKey).(*http.Request)
	if !ok {
		panic("missing request in context")
	}
	return val
}

// GenRequestID 生产 request_id
func GenRequestID() string {
	id, _ := uuid.NewRandom()
	return id.String()
}

// RequestIDValue 获取 RequestId 值
func RequestIDValue(ctx context.Context) string {
	v, ok := ctx.Value(requestIDCtxKey).(string)
	if !ok {
		return ""
	}

	return v
}

// WithRequestID 设置 request_id
func WithRequestID(ctx context.Context, id string) context.Context {
	newCtx := context.WithValue(ctx, requestIDCtxKey, id)
	return newCtx
}