X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fthymeproxy%2FApplication.java;h=1c33ce2fcaf2730d9ab59e5671ad2a485afba10d;hb=7ef40926db0d3d05443af767fece91ccbc2ddfa1;hp=ff5174a15dd06beac7f6905164af1916cfd7bece;hpb=ca2da8b36b6cf13c5b885ff7e74bc22d3576f6c7;p=maven-thymeleaf-skin

diff --git a/src/main/java/de/juplo/thymeproxy/Application.java b/src/main/java/de/juplo/thymeproxy/Application.java
index ff5174a..1c33ce2 100644
--- a/src/main/java/de/juplo/thymeproxy/Application.java
+++ b/src/main/java/de/juplo/thymeproxy/Application.java
@@ -4,18 +4,28 @@ import de.juplo.thymeleaf.JuploDialect;
 import java.util.HashMap;
 import java.util.Locale;
 import java.util.Map;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import org.apache.http.HttpStatus;
 import org.apache.http.impl.client.CloseableHttpClient;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.boot.autoconfigure.web.BasicErrorController;
+import org.springframework.boot.autoconfigure.web.ErrorAttributes;
+import org.springframework.boot.autoconfigure.web.ErrorController;
+import org.springframework.boot.autoconfigure.web.ErrorProperties;
 import org.springframework.boot.autoconfigure.web.ServerProperties;
 import org.springframework.context.annotation.Bean;
 import org.springframework.core.Ordered;
 import org.springframework.core.env.Environment;
+import org.springframework.http.MediaType;
+import org.springframework.web.servlet.ModelAndView;
 import org.springframework.web.servlet.View;
 import org.springframework.web.servlet.config.annotation.ViewControllerRegistry;
 import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;
 import org.springframework.web.servlet.mvc.UrlFilenameViewController;
+import org.thymeleaf.exceptions.TemplateInputException;
 import org.thymeleaf.resourceresolver.IResourceResolver;
 import org.thymeleaf.resourceresolver.UrlResourceResolver;
 import org.thymeleaf.spring4.view.ThymeleafViewResolver;
@@ -38,7 +48,6 @@ public class Application extends WebMvcConfigurerAdapter
     RegexUrlHandlerMapping mapping = new RegexUrlHandlerMapping();
     mapping.setOrder(Ordered.HIGHEST_PRECEDENCE);
     Map<String, Object> mappings = new HashMap<>();
-    mappings.put("/thymeleaf/.*\\.html", proxy);
     mappings.put("/img/.+", proxy);
     mappings.put("/css/.+", proxy);
     mappings.put("/js/.+", proxy);
@@ -114,7 +123,13 @@ public class Application extends WebMvcConfigurerAdapter
   @Bean
   public View error(ThymeleafViewResolver resolver) throws Exception
   {
-    return resolver.resolveViewName("templates/error", Locale.getDefault());
+    return resolver.resolveViewName("templates/500", Locale.getDefault());
+  }
+
+  @Bean
+  public ErrorController errorController(ErrorAttributes errorAttributes)
+  {
+    return new CustomErrorController(errorAttributes, properties.getError());
   }
 
 
@@ -129,4 +144,51 @@ public class Application extends WebMvcConfigurerAdapter
   {
     SpringApplication.run(Application.class, args);
   }
+
+
+  static class CustomErrorController extends BasicErrorController
+  {
+    public final static String TEMPLATE_INPUT_EXCEPTION =
+        TemplateInputException.class.getCanonicalName();
+
+
+    CustomErrorController(
+        ErrorAttributes errorAttributes,
+        ErrorProperties errorProperties
+        )
+    {
+      super(errorAttributes, errorProperties);
+    }
+
+
+    @Override
+    public ModelAndView errorHtml(
+        HttpServletRequest request,
+        HttpServletResponse response
+        )
+    {
+      Map<String, Object> model =
+          getErrorAttributes(
+              request,
+              isIncludeStackTrace(request, MediaType.TEXT_HTML)
+              );
+
+      String view;
+
+      switch ((String)model.get("exception"))
+      {
+        case "org.thymeleaf.exceptions.TemplateInputException":
+          response.setStatus(HttpStatus.SC_NOT_FOUND);
+          view = "templates/404";
+          break;
+
+        default:
+          response.setStatus(getStatus(request).value());
+          view = "templates/500";
+          break;
+      }
+
+      return new ModelAndView(view, model);
+    }
+  }
 }
\ No newline at end of file