added compatibility middleware

added Notification Payload Aliases to improve Compatibility with multiple Apps

Author: CodeShellDev

internals/proxy/middlewares/body.go

@@ -0,0 +1,88 @@
+package middlewares
+
+import (
+	"bytes"
+	"encoding/json"
+	"io"
+	"net/http"
+	"strconv"
+
+	log "github.com/codeshelldev/secured-signal-api/utils/logger"
+)
+
+type MessageAlias struct {
+	Alias    string
+	Priority int
+}
+
+type BodyMiddleware struct {
+	Next           http.Handler
+	MessageAliases []MessageAlias
+}
+
+func (data BodyMiddleware) Use() http.Handler {
+	next := data.Next
+	messageAliases := data.MessageAliases
+
+	return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
+		if req.Body != nil {
+			bodyBytes, err := io.ReadAll(req.Body)
+
+			if err != nil {
+				log.Error("Could not read Body: ", err.Error())
+				http.Error(w, "Internal Error", http.StatusInternalServerError)
+				return
+			}
+
+			req.Body.Close()
+
+			var modifiedBodyData map[string]interface{}
+
+			err = json.Unmarshal(bodyBytes, &modifiedBodyData)
+
+			if err != nil {
+				log.Error("Could not decode Body: ", err.Error())
+				http.Error(w, "Internal Error", http.StatusInternalServerError)
+				return
+			}
+
+			content, ok := modifiedBodyData["message"]
+
+			if !ok || content == "" {
+				best := 0
+
+				for _, alias := range messageAliases {
+					aliasKey := alias.Alias
+					priority := alias.Priority
+
+					value, ok := modifiedBodyData[aliasKey]
+
+					if ok && value != "" && priority > best {
+						content = modifiedBodyData[aliasKey]
+					}
+
+					modifiedBodyData[aliasKey] = nil
+				}
+
+				modifiedBodyData["message"] = content
+
+				bodyBytes, err = json.Marshal(modifiedBodyData)
+
+				if err != nil {
+					log.Error("Could not encode Body: ", err.Error())
+					http.Error(w, "Internal Error", http.StatusInternalServerError)
+					return
+				}
+			}
+
+			modifiedBody := string(bodyBytes)
+
+			req.Body = io.NopCloser(bytes.NewReader(bodyBytes))
+
+			req.ContentLength = int64(len(modifiedBody))
+			req.Header.Set("Content-Length", strconv.Itoa(len(modifiedBody)))
+		}
+
+		next.ServeHTTP(w, req)
+	})
+}

main.go

@@ -26,8 +26,50 @@ func main() {
 
 	initHandler = proxy.Create(ENV.API_URL)
 
+	body_m4 := BodyMiddleware{
+		Next: initHandler,
+		MessageAliases: []MessageAlias{
+			{
+				Alias:    "msg",
+				Priority: 100,
+			},
+			{
+				Alias:    "content",
+				Priority: 99,
+			},
+			{
+				Alias:    "description",
+				Priority: 98,
+			},
+			{
+				Alias:    "text",
+				Priority: 20,
+			},
+			{
+				Alias:    "body",
+				Priority: 15,
+			},
+			{
+				Alias:    "summary",
+				Priority: 10,
+			},
+			{
+				Alias:    "details",
+				Priority: 9,
+			},
+			{
+				Alias:    "payload",
+				Priority: 2,
+			},
+			{
+				Alias:    "data",
+				Priority: 1,
+			},
+		},
+	}
+
 	temp_m3 := TemplateMiddleware{
-		Next:      initHandler,
+		Next:      body_m4.Use(),
 		Variables: ENV.VARIABLES,
 	}