From aac224b3bf46db3df21a895bfb6857a4610f25fa Mon Sep 17 00:00:00 2001 From: xhaggi Date: Fri, 25 Oct 2024 13:56:48 +0200 Subject: [PATCH] Handle flash attributes on htmx redirects --- .../boot/mvc/HtmxHandlerInterceptor.java | 19 ++++---- ...sponseHandlerMethodReturnValueHandler.java | 44 ++++++++++++++++--- ...lerMethodReturnValueHandlerController.java | 13 ++++++ ...seHandlerMethodReturnValueHandlerTest.java | 19 +++++++- 4 files changed, 78 insertions(+), 17 deletions(-) diff --git a/htmx-spring-boot/src/main/java/io/github/wimdeblauwe/htmx/spring/boot/mvc/HtmxHandlerInterceptor.java b/htmx-spring-boot/src/main/java/io/github/wimdeblauwe/htmx/spring/boot/mvc/HtmxHandlerInterceptor.java index 0d98cb7..b7b0b27 100644 --- a/htmx-spring-boot/src/main/java/io/github/wimdeblauwe/htmx/spring/boot/mvc/HtmxHandlerInterceptor.java +++ b/htmx-spring-boot/src/main/java/io/github/wimdeblauwe/htmx/spring/boot/mvc/HtmxHandlerInterceptor.java @@ -10,7 +10,6 @@ import org.springframework.web.servlet.HandlerInterceptor; import org.springframework.web.servlet.ModelAndView; import org.springframework.web.servlet.View; -import org.springframework.web.servlet.support.ServletUriComponentsBuilder; import java.lang.reflect.Method; import java.time.Duration; @@ -30,14 +29,13 @@ public HtmxHandlerInterceptor(ObjectMapper objectMapper, HtmxResponseHandlerMeth @Override public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception { if (modelAndView != null) { - modelAndView.getModel().values().forEach( - value -> { - if (value instanceof HtmxResponse) { - buildAndRender((HtmxResponse) value, modelAndView, request, response); - } else if (value instanceof HtmxResponse.Builder) { - buildAndRender(((HtmxResponse.Builder) value).build(), modelAndView, request, response); - } - }); + for (Object value : modelAndView.getModel().values()) { + if (value instanceof HtmxResponse) { + buildAndRender((HtmxResponse) value, modelAndView, request, response); + } else if (value instanceof HtmxResponse.Builder) { + buildAndRender(((HtmxResponse.Builder) value).build(), modelAndView, request, response); + } + } } } @@ -45,7 +43,8 @@ private void buildAndRender(HtmxResponse htmxResponse, ModelAndView mav, HttpSer View v = htmxResponseHandlerMethodReturnValueHandler.toView(htmxResponse); try { v.render(mav.getModel(), request, response); - htmxResponseHandlerMethodReturnValueHandler.addHxHeaders(htmxResponse, response); + // ModelAndViewContainer is not available here, so flash attributes won't work + htmxResponseHandlerMethodReturnValueHandler.addHxHeaders(htmxResponse, request, response, null); } catch (Exception e) { throw new RuntimeException(e); } diff --git a/htmx-spring-boot/src/main/java/io/github/wimdeblauwe/htmx/spring/boot/mvc/HtmxResponseHandlerMethodReturnValueHandler.java b/htmx-spring-boot/src/main/java/io/github/wimdeblauwe/htmx/spring/boot/mvc/HtmxResponseHandlerMethodReturnValueHandler.java index fbd4f47..554e025 100644 --- a/htmx-spring-boot/src/main/java/io/github/wimdeblauwe/htmx/spring/boot/mvc/HtmxResponseHandlerMethodReturnValueHandler.java +++ b/htmx-spring-boot/src/main/java/io/github/wimdeblauwe/htmx/spring/boot/mvc/HtmxResponseHandlerMethodReturnValueHandler.java @@ -2,10 +2,14 @@ import com.fasterxml.jackson.core.JsonProcessingException; import com.fasterxml.jackson.databind.ObjectMapper; +import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletResponse; import org.springframework.beans.factory.ObjectFactory; import org.springframework.core.MethodParameter; +import org.springframework.lang.Nullable; +import org.springframework.ui.ModelMap; import org.springframework.util.Assert; +import org.springframework.util.CollectionUtils; import org.springframework.web.context.request.NativeWebRequest; import org.springframework.web.method.support.HandlerMethodReturnValueHandler; import org.springframework.web.method.support.ModelAndViewContainer; @@ -13,11 +17,14 @@ import org.springframework.web.servlet.ModelAndView; import org.springframework.web.servlet.View; import org.springframework.web.servlet.ViewResolver; +import org.springframework.web.servlet.mvc.support.RedirectAttributes; +import org.springframework.web.servlet.support.RequestContextUtils; import org.springframework.web.util.ContentCachingResponseWrapper; import java.util.Collection; import java.util.HashMap; import java.util.Locale; +import java.util.Map; import java.util.stream.Collectors; public class HtmxResponseHandlerMethodReturnValueHandler implements HandlerMethodReturnValueHandler { @@ -47,7 +54,10 @@ public void handleReturnValue(Object returnValue, HtmxResponse htmxResponse = (HtmxResponse) returnValue; mavContainer.setView(toView(htmxResponse)); - addHxHeaders(htmxResponse, webRequest.getNativeResponse(HttpServletResponse.class)); + HttpServletRequest request = webRequest.getNativeRequest(HttpServletRequest.class); + HttpServletResponse response = webRequest.getNativeResponse(HttpServletResponse.class); + + addHxHeaders(htmxResponse, request, response, mavContainer); } View toView(HtmxResponse htmxResponse) { @@ -74,16 +84,20 @@ View toView(HtmxResponse htmxResponse) { }; } - void addHxHeaders(HtmxResponse htmxResponse, HttpServletResponse response) { + void addHxHeaders(HtmxResponse htmxResponse, HttpServletRequest request, HttpServletResponse response, @Nullable ModelAndViewContainer mavContainer) { addHxTriggerHeaders(response, HtmxResponseHeader.HX_TRIGGER, htmxResponse.getTriggersInternal()); addHxTriggerHeaders(response, HtmxResponseHeader.HX_TRIGGER_AFTER_SETTLE, htmxResponse.getTriggersAfterSettleInternal()); addHxTriggerHeaders(response, HtmxResponseHeader.HX_TRIGGER_AFTER_SWAP, htmxResponse.getTriggersAfterSwapInternal()); if (htmxResponse.getLocation() != null) { - if (htmxResponse.getLocation().hasContextData()) { - setHeaderJsonValue(response, HtmxResponseHeader.HX_LOCATION.getValue(), htmxResponse.getLocation()); + HtmxLocation location = htmxResponse.getLocation(); + if (mavContainer != null) { + saveFlashAttributes(mavContainer, request, response, location.getPath()); + } + if (location.hasContextData()) { + setHeaderJsonValue(response, HtmxResponseHeader.HX_LOCATION.getValue(), location); } else { - response.setHeader(HtmxResponseHeader.HX_LOCATION.getValue(), htmxResponse.getLocation().getPath()); + response.setHeader(HtmxResponseHeader.HX_LOCATION.getValue(), location.getPath()); } } if (htmxResponse.getReplaceUrl() != null) { @@ -93,6 +107,9 @@ void addHxHeaders(HtmxResponse htmxResponse, HttpServletResponse response) { response.setHeader(HtmxResponseHeader.HX_PUSH_URL.getValue(), htmxResponse.getPushUrl()); } if (htmxResponse.getRedirect() != null) { + if (mavContainer != null) { + saveFlashAttributes(mavContainer, request, response, htmxResponse.getRedirect()); + } response.setHeader(HtmxResponseHeader.HX_REDIRECT.getValue(), htmxResponse.getRedirect()); } if (htmxResponse.isRefresh()) { @@ -139,4 +156,21 @@ private void setHeaderJsonValue(HttpServletResponse response, String name, Objec throw new IllegalArgumentException("Unable to set header " + name + " to " + value, e); } } + + private void saveFlashAttributes(ModelAndViewContainer mavContainer, HttpServletRequest request, HttpServletResponse response, String location) { + mavContainer.setRedirectModelScenario(true); + ModelMap model = mavContainer.getModel(); + + if (model instanceof RedirectAttributes redirectAttributes) { + Map flashAttributes = redirectAttributes.getFlashAttributes(); + if (!CollectionUtils.isEmpty(flashAttributes)) { + if (request != null) { + RequestContextUtils.getOutputFlashMap(request).putAll(flashAttributes); + if (response != null) { + RequestContextUtils.saveOutputFlashMap(location, request, response); + } + } + } + } + } } diff --git a/htmx-spring-boot/src/test/java/io/github/wimdeblauwe/htmx/spring/boot/mvc/HtmxResponseHandlerMethodReturnValueHandlerController.java b/htmx-spring-boot/src/test/java/io/github/wimdeblauwe/htmx/spring/boot/mvc/HtmxResponseHandlerMethodReturnValueHandlerController.java index 308a675..98ace6d 100644 --- a/htmx-spring-boot/src/test/java/io/github/wimdeblauwe/htmx/spring/boot/mvc/HtmxResponseHandlerMethodReturnValueHandlerController.java +++ b/htmx-spring-boot/src/test/java/io/github/wimdeblauwe/htmx/spring/boot/mvc/HtmxResponseHandlerMethodReturnValueHandlerController.java @@ -5,6 +5,7 @@ import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.servlet.ModelAndView; +import org.springframework.web.servlet.mvc.support.RedirectAttributes; import java.time.Duration; import java.util.Map; @@ -34,6 +35,12 @@ public HtmxResponse hxLocationWithoutContextData() { return HtmxResponse.builder().location("/path").build(); } + @GetMapping("/hx-location-with-flash-attributes") + public HtmxResponse hxLocationWithoutContextData(RedirectAttributes redirectAttributes) { + redirectAttributes.addFlashAttribute("flash", "test"); + return HtmxResponse.builder().location("/path").build(); + } + @GetMapping("/hx-push-url") public HtmxResponse hxPushUrl() { return HtmxResponse.builder().pushUrl("/path").build(); @@ -44,6 +51,12 @@ public HtmxResponse hxRedirect() { return HtmxResponse.builder().redirect("/path").build(); } + @GetMapping("/hx-redirect-with-flash-attributes") + public HtmxResponse hxRedirectWithFlashAttributes(RedirectAttributes redirectAttributes) { + redirectAttributes.addFlashAttribute("flash", "test"); + return HtmxResponse.builder().redirect("/path").build(); + } + @GetMapping("/hx-refresh") public HtmxResponse hxRefresh() { return HtmxResponse.builder().refresh().build(); diff --git a/htmx-spring-boot/src/test/java/io/github/wimdeblauwe/htmx/spring/boot/mvc/HtmxResponseHandlerMethodReturnValueHandlerTest.java b/htmx-spring-boot/src/test/java/io/github/wimdeblauwe/htmx/spring/boot/mvc/HtmxResponseHandlerMethodReturnValueHandlerTest.java index de6452a..abdabae 100644 --- a/htmx-spring-boot/src/test/java/io/github/wimdeblauwe/htmx/spring/boot/mvc/HtmxResponseHandlerMethodReturnValueHandlerTest.java +++ b/htmx-spring-boot/src/test/java/io/github/wimdeblauwe/htmx/spring/boot/mvc/HtmxResponseHandlerMethodReturnValueHandlerTest.java @@ -9,8 +9,7 @@ import static org.assertj.core.api.Assertions.assertThat; import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get; -import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.header; -import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status; +import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*; @WebMvcTest(HtmxResponseHandlerMethodReturnValueHandlerController.class) @WithMockUser @@ -33,6 +32,14 @@ public void testHxLocationWithoutContextData() throws Exception { .andExpect(header().string("HX-Location", "/path")); } + @Test + public void testHxLocationWithFlashAttributes() throws Exception { + mockMvc.perform(get("/hvhi/hx-location-with-flash-attributes")) + .andExpect(status().isOk()) + .andExpect(header().string("HX-Location", "/path")) + .andExpect(flash().attribute("flash", "test")); + } + @Test public void testHxPushUrl() throws Exception { mockMvc.perform(get("/hvhi/hx-push-url")) @@ -47,6 +54,14 @@ public void testHxRedirect() throws Exception { .andExpect(header().string("HX-Redirect", "/path")); } + @Test + public void testHxRedirectWithFlashAttributes() throws Exception { + mockMvc.perform(get("/hvhi/hx-redirect-with-flash-attributes")) + .andExpect(status().isOk()) + .andExpect(header().string("HX-Redirect", "/path")) + .andExpect(flash().attribute("flash", "test")); + } + @Test public void testHxRefresh() throws Exception { mockMvc.perform(get("/hvhi/hx-refresh"))