Generalise cookie-setting

buerokratt · Dec 22, 2023 · f87e3fb · f87e3fb
1 parent 0ba30c0
commit f87e3fb
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/src/main/java/ee/buerokratt/ruuter/domain/steps/ReturnStep.java b/src/main/java/ee/buerokratt/ruuter/domain/steps/ReturnStep.java
@@ -47,17 +47,18 @@ private Map<String, String> formatHeaders(DslInstance di) {
             .collect(toMap(Entry::getKey, this::entryValueToHeaderString));
     }
 
+    private void addToCookie(LinkedHashMap<String, Object> cookie, String key, Object value) {
+        if (!cookie.containsKey(key))
+            cookie.put(key, value);
+    }
     private Map.Entry<String, Object> addDefaultCookies(Map.Entry<String, Object> entry, DslInstance di) {
         if ("Set-Cookie".equals(entry.getKey())) {
             LinkedHashMap<String, Object> cookie = new LinkedHashMap<>((HashMap<String, Object>) entry.getValue());
-            if (!cookie.containsKey("Path"))
-                cookie.put("Path", "/" );
-
-            if (!cookie.containsKey("HttpOnly"))
-                cookie.put("HttpOnly", true);
 
-            if (!cookie.containsKey("SetSecure"))
-                cookie.put("Secure", true);
+            addToCookie(cookie, "Path", "/");
+            addToCookie(cookie, "HttpOnly", true);
+            addToCookie(cookie, "Secure", true);
+            addToCookie(cookie, "Max-Age", 28800);
 
             entry.setValue(cookie);
         }