request_context.go

package context

// #include "../../API.h"
import "C"
import (
	. "main/aikido_types"
	"main/log"
)

type CallbackFunction func(int) string

/* Request level context cache (changes on each PHP request) */
type RequestContextData struct {
	Callback                      CallbackFunction // callback to access data from the PHP layer (C++ extension) about the current request and current event
	Method                        *string
	Route                         *string
	RouteParsed                   *string
	URL                           *string
	StatusCode                    *int
	IP                            *string
	RateLimitGroup                *string
	EndpointConfig                **EndpointData
	WildcardEndpointsConfigs      *[]EndpointData
	IsIpBypassed                  *bool
	IsEndpointConfigured          *bool
	IsEndpointRateLimitingEnabled *bool
	IsEndpointProtectionTurnedOff *bool
	IsEndpointIpAllowed           *bool
	IsEndpointRateLimited         bool
	UserAgent                     *string
	UserId                        *string
	UserName                      *string
	BodyRaw                       *string
	BodyParsed                    *map[string]interface{}
	BodyParsedFlattened           *map[string]string
	QueryParsed                   *map[string]interface{}
	QueryParsedFlattened          *map[string]string
	CookiesParsed                 *map[string]interface{}
	CookiesParsedFlattened        *map[string]string
	HeadersParsed                 *map[string]interface{}
	HeadersParsedFlattened        *map[string]string
	RouteParamsRaw                *string
	RouteParamsParsed             *map[string]interface{}
	RouteParamsParsedFlattened    *map[string]string
	GraphQLParsedFlattened        *map[string]string
}

var Context RequestContextData

func Init(callback CallbackFunction) bool {
	Context = RequestContextData{
		Callback: callback,
	}
	return true
}

func Clear() bool {
	Context = RequestContextData{
		Callback: Context.Callback,
	}
	ResetEventContext()
	return true
}

func GetFromCache[T any](fetchDataFn func(), s **T) T {
	if fetchDataFn != nil {
		fetchDataFn()
	}
	if *s == nil {
		var t T
		log.Warnf("Error getting from cache. Returning default value %v...", t)
		return t
	}
	return **s
}

func GetIp() string {
	return GetFromCache(ContextSetIp, &Context.IP)
}

func GetMethod() string {
	return GetFromCache(ContextSetMethod, &Context.Method)
}

func GetRoute() string {
	return GetFromCache(ContextSetRoute, &Context.Route)
}

func GetParsedRoute() string {
	return GetFromCache(ContextSetParsedRoute, &Context.RouteParsed)
}

func GetUrl() string {
	return GetFromCache(ContextSetUrl, &Context.URL)
}

func GetStatusCode() int {
	return GetFromCache(ContextSetStatusCode, &Context.StatusCode)
}

func IsIpBypassed() bool {
	return GetFromCache(ContextSetIsIpBypassed, &Context.IsIpBypassed)
}

func GetBodyRaw() string {
	return GetFromCache(ContextSetBody, &Context.BodyRaw)
}

func GetBodyParsed() map[string]interface{} {
	return GetFromCache(ContextSetBody, &Context.BodyParsed)
}

func GetQueryParsed() map[string]interface{} {
	return GetFromCache(ContextSetQuery, &Context.QueryParsed)
}

func GetCookiesParsed() map[string]interface{} {
	return GetFromCache(ContextSetCookies, &Context.CookiesParsed)
}

func GetHeadersParsed() map[string]interface{} {
	return GetFromCache(ContextSetHeaders, &Context.HeadersParsed)
}

func GetBodyParsedFlattened() map[string]string {
	return GetFromCache(ContextSetBody, &Context.BodyParsedFlattened)
}

func GetQueryParsedFlattened() map[string]string {
	return GetFromCache(ContextSetQuery, &Context.QueryParsedFlattened)
}

func GetCookiesParsedFlattened() map[string]string {
	return GetFromCache(ContextSetCookies, &Context.CookiesParsedFlattened)
}

func GetRouteParamsParsedFlattened() map[string]string {
	return GetFromCache(ContextSetRouteParams, &Context.RouteParamsParsedFlattened)
}

func GetHeadersParsedFlattened() map[string]string {
	return GetFromCache(ContextSetHeaders, &Context.HeadersParsedFlattened)
}

func GetGraphQLParsedFlattened() map[string]string {
	return GetFromCache(ContextSetGraphQL, &Context.GraphQLParsedFlattened)
}

func GetUserAgent() string {
	return GetFromCache(ContextSetUserAgent, &Context.UserAgent)
}

func GetUserId() string {
	return GetFromCache(ContextSetUserId, &Context.UserId)
}

func GetUserName() string {
	return GetFromCache(ContextSetUserName, &Context.UserName)
}

func GetRateLimitGroup() string {
	return GetFromCache(ContextSetRateLimitGroup, &Context.RateLimitGroup)
}

func GetEndpointConfig() *EndpointData {
	return GetFromCache(ContextSetEndpointConfig, &Context.EndpointConfig)
}

func GetWildcardEndpointsConfig() []EndpointData {
	return GetFromCache(ContextSetWildcardEndpointsConfigs, &Context.WildcardEndpointsConfigs)
}

func IsEndpointConfigured() bool {
	return GetFromCache(ContextSetIsEndpointConfigured, &Context.IsEndpointConfigured)
}

func IsEndpointRateLimitingEnabled() bool {
	return GetFromCache(ContextSetIsEndpointRateLimitingEnabled, &Context.IsEndpointRateLimitingEnabled)
}

func IsEndpointIpAllowed() bool {
	return GetFromCache(ContextSetIsEndpointIpAllowed, &Context.IsEndpointIpAllowed)
}

func IsEndpointProtectionTurnedOff() bool {
	return GetFromCache(ContextSetIsEndpointProtectionTurnedOff, &Context.IsEndpointProtectionTurnedOff)
}

func IsEndpointRateLimited() bool {
	return Context.IsEndpointRateLimited
}