Mercurial > hg > LightPIT / file diff

--- a/src/main/java/de/uapcore/lightpit/modules/ErrorModule.java	Wed May 13 21:10:23 2020 +0200
+++ b/src/main/java/de/uapcore/lightpit/modules/ErrorModule.java	Wed May 13 21:46:26 2020 +0200
@@ -32,6 +32,7 @@
 
 import javax.servlet.annotation.WebServlet;
 import javax.servlet.http.HttpServletRequest;
+import java.util.Optional;
 
 /**
  * Entry point for the application.
@@ -49,10 +50,16 @@
 public final class ErrorModule extends AbstractLightPITServlet {
 
     public static final String REQ_ATTR_ERROR_CODE = "errorCode";
+    public static final String REQ_ATTR_RETURN_LINK = "returnLink";
 
     private ResponseType handle(HttpServletRequest req, int sc) {
 
         req.setAttribute(REQ_ATTR_ERROR_CODE, sc);
+
+        Optional.ofNullable(req.getHeader("Referer")).ifPresent(
+                referer -> req.setAttribute(REQ_ATTR_RETURN_LINK, referer)
+        );
+
         setStylesheet(req, "error");
         setDynamicFragment(req, "error");