Add convenience method.

mnlipp · May 2, 2024 · cb8c255 · cb8c255
1 parent dc4c714
commit cb8c255
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/org.jgrapes.http/src/org/jgrapes/http/events/Response.java b/org.jgrapes.http/src/org/jgrapes/http/events/Response.java
@@ -18,7 +18,13 @@
 
 package org.jgrapes.http.events;
 
+import java.nio.charset.Charset;
+import java.util.Optional;
+import java.util.stream.Stream;
 import org.jdrupes.httpcodec.MessageHeader;
+import org.jdrupes.httpcodec.protocols.http.HttpField;
+import org.jdrupes.httpcodec.protocols.http.HttpResponse;
+import org.jdrupes.httpcodec.types.Converters;
 
 /**
  * Represents the response (header).
@@ -44,4 +50,25 @@ public MessageHeader response() {
         return response;
     }
 
+    /**
+     * Convenience method for retrieving the {@link Charset}
+     * from the response.
+     *
+     * @return the optional
+     */
+    public Optional<Charset> charset() {
+        return ((HttpResponse) response())
+            .findValue(HttpField.CONTENT_TYPE, Converters.MEDIA_TYPE)
+            .map(mt -> mt.parameters().entrySet().stream())
+            .orElse(Stream.empty())
+            .filter(e -> "charset".equalsIgnoreCase(e.getKey()))
+            .findFirst().map(e -> e.getValue()).map(csn -> {
+                try {
+                    return Charset.forName(csn);
+                } catch (Exception e) {
+                    return null;
+                }
+            });
+    }
+
 }