From d60056583839f7bf90555a29aea2ec7c088325b8 Mon Sep 17 00:00:00 2001
From: visoulier <victor.soulier@etu.uca.fr>
Date: Wed, 28 Feb 2024 20:12:44 +0100
Subject: [PATCH] ajout application filter

---
 .../sciencequest/sae/ApplicationFilter.java   | 113 ++++++++++++++++++
 1 file changed, 113 insertions(+)
 create mode 100644 SpringBootProject/src/main/java/fr/iut/sciencequest/sae/ApplicationFilter.java

diff --git a/SpringBootProject/src/main/java/fr/iut/sciencequest/sae/ApplicationFilter.java b/SpringBootProject/src/main/java/fr/iut/sciencequest/sae/ApplicationFilter.java
new file mode 100644
index 0000000..b5836c7
--- /dev/null
+++ b/SpringBootProject/src/main/java/fr/iut/sciencequest/sae/ApplicationFilter.java
@@ -0,0 +1,113 @@
+package fr.iut.sciencequest.sae;
+
+
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ArrayNode;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import jakarta.servlet.FilterChain;
+import jakarta.servlet.ServletException;
+import jakarta.servlet.ServletOutputStream;
+import jakarta.servlet.WriteListener;
+import jakarta.servlet.http.HttpServletRequest;
+import jakarta.servlet.http.HttpServletResponse;
+import jakarta.servlet.http.HttpServletResponseWrapper;
+import org.springframework.core.annotation.Order;
+import org.springframework.http.MediaType;
+import org.springframework.stereotype.Component;
+import org.springframework.web.filter.OncePerRequestFilter;
+
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+
+@Component
+@Order(1)
+public class ApplicationFilter extends OncePerRequestFilter {
+    private final ObjectMapper objectMapper;
+
+    public ApplicationFilter(ObjectMapper objectMapper) {
+        this.objectMapper = objectMapper;
+    }
+
+    @Override
+    protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain)
+            throws ServletException, IOException {
+        // Intercept and modify the JSON response
+        JsonInterceptingResponseWrapper responseWrapper = new JsonInterceptingResponseWrapper(response);
+
+        filterChain.doFilter(request, responseWrapper);
+
+        // Check if the response content type is JSON
+        if (responseWrapper.getContentType() != null && responseWrapper.getContentType().startsWith(MediaType.APPLICATION_JSON_VALUE)) {
+            try {
+                // Parse the JSON response
+                JsonNode root = objectMapper.readTree(responseWrapper.getContent());
+
+                // Check if the _embedded node exists and if it contains the tupleBackedMapList node
+                JsonNode embeddedNode = root.get("_embedded");
+                if (embeddedNode != null && embeddedNode.isObject()) {
+                    JsonNode tupleBackedMapListNode = embeddedNode.get("tupleBackedMapList");
+                    if (tupleBackedMapListNode != null && tupleBackedMapListNode.isArray()) {
+                        ArrayNode tupleBackedMapList = (ArrayNode) tupleBackedMapListNode;
+
+                        // Remove the _embedded node
+                        ((ObjectNode) root).remove("_embedded");
+
+                        // Set the tupleBackedMapList directly to the root
+                        ((ObjectNode) root).set("_embedded", tupleBackedMapList);
+                    }
+                }
+
+                // Convert the modified JSON back to string
+                String modifiedResponseBody = objectMapper.writeValueAsString(root);
+
+                // Write the modified JSON back to the response
+                response.getOutputStream().write(modifiedResponseBody.getBytes());
+            } catch (IOException e) {
+                // Handle exception
+                e.printStackTrace();
+            }
+        }
+    }
+
+    private static class JsonInterceptingResponseWrapper extends HttpServletResponseWrapper {
+        private ByteArrayOutputStream baos = new ByteArrayOutputStream();
+
+        public JsonInterceptingResponseWrapper(HttpServletResponse response) {
+            super(response);
+        }
+
+        @Override
+        public ServletOutputStream getOutputStream() throws IOException {
+            return new ServletOutputStreamWrapper(baos);
+        }
+
+        public String getContent() {
+            return baos.toString();
+        }
+    }
+
+    private static class ServletOutputStreamWrapper extends ServletOutputStream {
+        private ByteArrayOutputStream baos;
+
+        public ServletOutputStreamWrapper(ByteArrayOutputStream baos) {
+            this.baos = baos;
+        }
+
+        @Override
+        public void write(int b) throws IOException {
+            baos.write(b);
+        }
+
+        @Override
+        public boolean isReady() {
+            return true;
+        }
+
+        @Override
+        public void setWriteListener(WriteListener writeListener) {
+            throw new UnsupportedOperationException("Not implemented");
+        }
+    }
+
+}
\ No newline at end of file