try session in header

gotson · Aug 27, 2024 · 8fed7ff · 8fed7ff
1 parent 8e5652f
commit 8fed7ff
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 8 deletions.
diff --git a/komga/src/main/kotlin/org/gotson/komga/infrastructure/security/SecurityConfiguration.kt b/komga/src/main/kotlin/org/gotson/komga/infrastructure/security/SecurityConfiguration.kt
@@ -188,13 +188,13 @@ class SecurityConfiguration(
       // despite the response being exactly the same
       // this happens when requesting /v1/user/profile
       // Kobo error: packetdump.warning) Invalid JSON script: QVariant(Invalid) "illegal value"
-//      sessionManagement {
-//        sessionCreationPolicy = SessionCreationPolicy.IF_REQUIRED
-//        sessionConcurrency {
-//          sessionRegistry = theSessionRegistry
-//          maximumSessions = -1
-//        }
-//      }
+      sessionManagement {
+        sessionCreationPolicy = SessionCreationPolicy.IF_REQUIRED
+        sessionConcurrency {
+          sessionRegistry = theSessionRegistry
+          maximumSessions = -1
+        }
+      }
 
       addFilterBefore<AnonymousAuthenticationFilter>(koboAuthenticationFilter())
     }

diff --git a/...ain/kotlin/org/gotson/komga/infrastructure/security/session/SmartHttpSessionIdResolver.kt b/...ain/kotlin/org/gotson/komga/infrastructure/security/session/SmartHttpSessionIdResolver.kt
@@ -32,5 +32,11 @@ class SmartHttpSessionIdResolver(
     request.getResolver().expireSession(request, response)
   }
 
-  private fun HttpServletRequest.getResolver() = if (this.getHeader(sessionHeaderName) != null) header else cookie
+  private fun HttpServletRequest.getResolver() =
+    if (this.getHeader(sessionHeaderName) != null
+      || this.servletPath.startsWith("/kobo/")
+    )
+      header
+    else
+      cookie
 }