fix

Author: CodeShellDev

internals/proxy/middlewares/body.go

@@ -20,7 +20,11 @@ func (data BodyMiddleware) Use() http.Handler {
 	next := data.Next
 
 	return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
-		messageAliases := getSettings(req).MESSAGE_ALIASES
+		messageAliases := getSettingsByReq(req).MESSAGE_ALIASES
+
+		if messageAliases == nil {
+			messageAliases = getSettings("*").MESSAGE_ALIASES
+		}
 
 		body, err := request.GetReqBody(w, req)
 

internals/proxy/middlewares/common.go

@@ -23,13 +23,17 @@ type contextKey string
 
 const tokenKey contextKey = "token"
 
-func getSettings(req *http.Request) *config.SETTING_ {
+func getSettingsByReq(req *http.Request) *config.SETTING_ {
     token, ok := req.Context().Value(tokenKey).(string)
 
     if !ok {
         token = "*"
     }
 
+    return getSettings(token)
+}
+
+func getSettings(token string) *config.SETTING_ {
     settings, exists := config.ENV.SETTINGS[token]
 
     if !exists || settings == nil {

internals/proxy/middlewares/endpoints.go

@@ -15,7 +15,11 @@ func (data EndpointsMiddleware) Use() http.Handler {
 	next := data.Next
 
 	return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
-		blockedEndpoints := getSettings(req).BLOCKED_ENDPOINTS
+		blockedEndpoints := getSettingsByReq(req).BLOCKED_ENDPOINTS
+
+		if blockedEndpoints == nil {
+			blockedEndpoints = getSettings("*").BLOCKED_ENDPOINTS
+		}
 
 		reqPath := req.URL.Path
 

internals/proxy/middlewares/template.go

@@ -22,7 +22,11 @@ func (data TemplateMiddleware) Use() http.Handler {
 	next := data.Next
 
 	return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
-		variables := getSettings(req).VARIABLES
+		variables := getSettingsByReq(req).VARIABLES
+
+		if variables == nil {
+			variables = getSettings("*").VARIABLES
+		}
 
 		body, err := request.GetReqBody(w, req)