package api

import (
	"gitee.com/h79/goutils/auth/token"
	"gitee.com/h79/goutils/common/app"
	"gitee.com/h79/goutils/common/http"
	"gitee.com/h79/goutils/common/stringutil"
	"strings"
)

type HeaderContext interface {
	Set(key, value string)
	Get(key string) string
}

type CacheContext interface {
	Get(key string) (interface{}, bool)
	Set(key string, value interface{})
}

type HttpContext interface {
	Send(httpCode int, data Response)
	Abort()
	Next()
	Param(key string) string
	Query(key string) string
	ClientIP() string
}

type Context interface {
	HttpContext() HttpContext
	HeaderContext() HeaderContext
	CacheContext() CacheContext
	AuthContext() token.Engine
}

func Get[T any](ctx Context, key string) T {
	var a T
	if v, exist := ctx.CacheContext().Get(key); exist {
		a = v.(T)
	}
	return a
}

func SetUid(ctx Context, uid string) {
	ctx.CacheContext().Set(KUid, uid)
}

func GetUid(ctx Context) string {
	return Get[string](ctx, KUid)
}

func SetUidI64(ctx Context, uid int64) {
	ctx.CacheContext().Set(KUid, stringutil.Int64ToString(uid))
}

func GetUidI64(ctx Context) int64 {
	return stringutil.StringToInt64(Get[string](ctx, KUid))
}

func GetHead(ctx Context) *Header {
	return Get[*Header](ctx, KHead)
}

func GetBase(ctx Context) Base {
	header := GetHead(ctx)
	return header.Base
}

func GetReqHead(ctx Context) ReqHead {
	header := GetHead(ctx)
	return header.ReqHead
}

func GetCustom(ctx Context) interface{} {
	header := GetHead(ctx)
	return header.Custom
}

func GetSystem(ctx Context) app.System {
	return GetBase(ctx).System
}

func GetApp(ctx Context) app.Info {
	return GetBase(ctx).App
}

func GetClientIP(ctx Context) string {
	return GetBase(ctx).ClientIP
}

func QueryInt64(ctx Context, key string) int64 {
	return stringutil.StringToInt64(ctx.HttpContext().Query(key))
}

func HeadInt64(ctx Context, key string) int64 {
	return stringutil.StringToInt64(ctx.HeaderContext().Get(key))
}

func ParamInt64(ctx Context, key string) int64 {
	return stringutil.StringToInt64(ctx.HttpContext().Param(key))
}

const (
	UNK   = "unk"
	XML   = "xml"
	JSON  = "json"
	PLAIN = "plain"
)

func GetContentType(ctx Context) string {
	return ContentTypeTo(ctx.HeaderContext().Get("Content-Type"))
}

func ContentTypeTo(ct string) string {
	ct = strings.ToLower(ct)
	if strings.Contains(ct, http.MimeJSON) {
		return JSON
	}
	if strings.Contains(ct, http.MimeXML2) ||
		strings.Contains(ct, http.MimeXML) {
		return XML
	}
	if strings.Contains(ct, http.MimePlain) {
		return PLAIN
	}
	return UNK
}