Make MIMEparse more forgiven

Miel Vander Sande · Miel Vander Sande · commit 90908ebd6002 · 2018-04-11T15:00:24.000+02:00
diff --git a/src/main/java/org/linkeddatafragments/servlet/LinkedDataFragmentServlet.java b/src/main/java/org/linkeddatafragments/servlet/LinkedDataFragmentServlet.java
@@ -4,6 +4,7 @@
 import java.io.File;
 import java.io.FileReader;
 import java.io.IOException;
+import java.nio.charset.StandardCharsets;
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.HashMap;
@@ -13,6 +14,8 @@
 import javax.servlet.http.HttpServlet;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
+
+import org.apache.http.HttpHeaders;
 import org.apache.jena.riot.Lang;
 import org.linkeddatafragments.config.ConfigReader;
 import org.linkeddatafragments.datasource.DataSourceFactory;
@@ -94,10 +97,10 @@ public void init(ServletConfig servletConfig) throws ServletException {
 
             // register content types
             MIMEParse.register("text/html");
-            MIMEParse.register(Lang.TTL.getHeaderString());
-            MIMEParse.register(Lang.JSONLD.getHeaderString());
-            MIMEParse.register(Lang.NTRIPLES.getHeaderString());
             MIMEParse.register(Lang.RDFXML.getHeaderString());
+            MIMEParse.register(Lang.NTRIPLES.getHeaderString());
+            MIMEParse.register(Lang.JSONLD.getHeaderString());
+            MIMEParse.register(Lang.TTL.getHeaderString());
         } catch (Exception e) {
             throw new ServletException(e);
         }
@@ -158,12 +161,13 @@ public void doGet(HttpServletRequest request, HttpServletResponse response) thro
         ILinkedDataFragment fragment = null;
         try {
             // do conneg
-            String bestMatch = MIMEParse.bestMatch(request.getHeader("Accept"));
+            String acceptHeader = request.getHeader(HttpHeaders.ACCEPT);
+            String bestMatch = MIMEParse.bestMatch(acceptHeader);
 
             // set additional response headers
-            response.setHeader("Server", "Linked Data Fragments Server");
+            response.setHeader(HttpHeaders.SERVER, "Linked Data Fragments Server");
             response.setContentType(bestMatch);
-            response.setCharacterEncoding("utf-8");
+            response.setCharacterEncoding(StandardCharsets.UTF_8.name());
             
             // create a writer depending on the best matching mimeType
             ILinkedDataFragmentWriter writer = LinkedDataFragmentWriterFactory.create(config.getPrefixes(), dataSources, bestMatch);
@@ -189,11 +193,13 @@ public void doGet(HttpServletRequest request, HttpServletResponse response) thro
                     throw new ServletException(ex1);
                 }
             } catch (Exception e) {
+                e.printStackTrace();
                 response.setStatus(500);
                 writer.writeError(response.getOutputStream(), e);
             }
           
         } catch (Exception e) {
+            e.printStackTrace();
             throw new ServletException(e);
         }
         finally {
diff --git a/src/main/java/org/linkeddatafragments/util/MIMEParse.java b/src/main/java/org/linkeddatafragments/util/MIMEParse.java
@@ -264,8 +264,11 @@ public static String bestMatch(List<String> supported, String header) throws NoR
         
         List<ParseResults> parseResults = new LinkedList<ParseResults>();
         List<FitnessAndQuality> weightedMatches = new LinkedList<FitnessAndQuality>();
-        for (String r : StringUtils.split(header, ','))
-            parseResults.add(parseMediaRange(r));
+        for (String r : StringUtils.split(header, ',')) {
+            try {
+                parseResults.add(parseMediaRange(r));
+            } catch (Exception ex) {} // Fail silently on malformed media type and omit from list.
+        }
 
         for (String s : supported)
         {
@@ -274,10 +277,13 @@ public static String bestMatch(List<String> supported, String header) throws NoR
             fitnessAndQuality.mimeType = s;
             weightedMatches.add(fitnessAndQuality);
         }
-        Collections.sort(weightedMatches);
 
+
+
+        Collections.sort(weightedMatches);
         FitnessAndQuality lastOne = weightedMatches
                 .get(weightedMatches.size() - 1);
+
         return NumberUtils.compare(lastOne.quality, 0) != 0 ? lastOne.mimeType : supported.get(0);
     }
     
