modify /v1/about to include secured-signal-api

Author: CodeShellDev

internals/proxy/middlewares/api.go

@@ -0,0 +1,131 @@
+package middlewares
+
+import (
+	"bytes"
+	"io"
+	"net/http"
+	"os"
+	"regexp"
+	"strings"
+
+	"github.com/codeshelldev/gotl/pkg/logger"
+	"github.com/codeshelldev/gotl/pkg/request"
+	"github.com/codeshelldev/secured-signal-api/internals/config"
+)
+
+var InternalAPI Middleware = Middleware{
+	Name: "_Internal_API",
+	Use: internalAPIHandler,
+}
+
+func internalAPIHandler(next http.Handler) http.Handler {
+	mux := http.NewServeMux()
+
+	const aboutEndpoint = "/v1/about"
+	mux.HandleFunc(aboutEndpoint, func(w http.ResponseWriter, req *http.Request) {
+		res, err := redirectRequestToBackend(req, config.DEFAULT.API.URL + aboutEndpoint)
+		
+		if err != nil {
+			logger.Error("Error requesting Backend: ", err.Error())
+			http.Error(w, "Internal Server Error ", http.StatusBadGateway)
+			return
+		}
+
+		body, err := request.GetResBody(res)
+
+		if err != nil {
+			logger.Error("Could not get Response Body: ", err.Error())
+			http.Error(w, "Internal Server Error", http.StatusInternalServerError)
+			return
+		}
+
+		for key, values := range res.Header {
+			for _, value := range values {
+				w.Header().Add(key, value)
+			}
+		}
+
+		if !body.Empty {
+			var version string
+
+			if isValidSemver(os.Getenv("IMAGE_TAG")) {
+				version, _ = strings.CutPrefix(version, "v")
+			}
+			
+			payload := map[string]any{
+				"version": version,
+				"auth_required": !config.ENV.INSECURE,
+			}
+
+			body.Data["secured-signal-api"] = payload
+
+			err := body.Write(w)
+
+			if err != nil {
+				logger.Error("Could not write to Response Body: ", err.Error())
+				http.Error(w, "Internal Server Error", http.StatusInternalServerError)
+				return
+			}
+		}
+
+		w.WriteHeader(res.StatusCode)
+	})
+
+	mux.HandleFunc("/favicon.ico", func(w http.ResponseWriter, req *http.Request) {
+		http.ServeFile(w, req, config.ENV.FAVICON_PATH)
+	})
+
+	mux.Handle("/", next)
+
+	return mux
+}
+
+func isValidSemver(version string) bool {
+	re, err := regexp.Compile(`^v?([0-9]+)\.([0-9]+)\.([0-9]+)(-[0-9A-Za-z.-]+)?(\+[0-9A-Za-z.-]+)?$`)
+	
+	if err != nil {
+		return false
+	}
+
+	return re.MatchString(version)
+}
+
+func redirectRequestToBackend(req *http.Request, url string) (*http.Response, error) {
+	var body io.Reader
+
+    if req.Body != nil {
+        bodyBytes, err := io.ReadAll(req.Body)
+
+        if err != nil {
+            return nil, err
+        }
+        req.Body.Close()
+
+        body = bytes.NewReader(bodyBytes)
+        req.Body = io.NopCloser(bytes.NewReader(bodyBytes))
+    }
+
+	return requestBackend(req.Method, url, body, req.Header)
+}
+
+func requestBackend(method, url string, body io.Reader, headers map[string][]string) (*http.Response, error) {
+    backendReq, err := http.NewRequest(method, url, body)
+    if err != nil {
+        return nil, err
+    }
+
+    for key, values := range headers {
+        for _, value := range values {
+            backendReq.Header.Add(key, value)
+        }
+    }
+
+    client := &http.Client{}
+
+    res, err := client.Do(backendReq)
+    if err != nil {
+        return nil, err
+    }
+
+    return res, nil
+}

internals/proxy/middlewares/auth.go

@@ -108,8 +108,6 @@ var BodyAuth = AuthMethod{
 			return "", nil
 		}
 
-		body.Write(req)
-
 		if body.Empty {
 			return "", nil
 		}
@@ -129,7 +127,7 @@ var BodyAuth = AuthMethod{
 		if isValidToken(tokens, auth) {
 			delete(body.Data, authField)
 
-			body.Write(req)
+			body.UpdateReq(req)
 
 			return auth, nil
 		}
@@ -149,7 +147,7 @@ var QueryAuth = AuthMethod{
 		const oldAuthQuery = "authorization"
 
 		if req.URL.Query().Has("@" + oldAuthQuery) {
-			fullURL, _ := request.ParseRequestURL(req)
+			fullURL, _ := request.ParseReqURL(req)
 			urlWithNewAuthQuery := strings.Replace(fullURL.String(), "@" + oldAuthQuery, "@{s,fg=bright_red}" + oldAuthQuery + "{/}{b,fg=green}" + authQuery + "{/}", 1)
 
 			deprecation.Error(req.URL.String(), deprecation.DeprecationMessage{

internals/proxy/middlewares/mapping.go

@@ -64,7 +64,7 @@ func mappingHandler(next http.Handler) http.Handler {
 		if modifiedBody {
 			body.Data = bodyData
 
-			err := body.Write(req)
+			err := body.UpdateReq(req)
 
 			if err != nil {
 				logger.Error("Could not write to Request Body: ", err.Error())

internals/proxy/middlewares/message.go

@@ -72,7 +72,7 @@ func messageHandler(next http.Handler) http.Handler {
 		if modifiedBody {
 			body.Data = bodyData
 
-			err := body.Write(req)
+			err := body.UpdateReq(req)
 
 			if err != nil {
 				logger.Error("Could not write to Request Body: ", err.Error())

internals/proxy/middlewares/server.go

@@ -96,7 +96,7 @@ func templateHandler(next http.Handler) http.Handler {
 		if modifiedBody {
 			body.Data = bodyData
 
-			err := body.Write(req)
+			err := body.UpdateReq(req)
 
 			if err != nil {
 				logger.Error("Could not write to Request Body: ", err.Error())

internals/proxy/middlewares/template.go

@@ -67,7 +67,7 @@ func Create(targetUrl string) Proxy {
 
 func (proxy Proxy) Init() http.Handler {
 	handler := m.NewChain().
-		Use(m.Server).
+		Use(m.InternalAPI).
 		Use(m.Auth).
 		Use(m.InternalMiddlewareLogger).
 		Use(m.InternalProxy).