fix: SerializableCookie.hostOnly

muedsa · Nov 28, 2024 · 23ed8b5 · 23ed8b5
1 parent 0c24a28
commit 23ed8b5
Showing 1 changed file with 17 additions and 11 deletions.
diff --git a/api/src/main/java/com/muedsa/tvbox/tool/SerializableCookie.kt b/api/src/main/java/com/muedsa/tvbox/tool/SerializableCookie.kt
@@ -60,16 +60,22 @@ data class SerializableCookie(
             hostOnly = cookie.hostOnly,
         )
 
-        fun from(cookie: HttpCookie): SerializableCookie = SerializableCookie(
-            name = cookie.name,
-            value = cookie.value,
-            expiresAt = if (cookie.maxAge == -1L) 253402300799999L else System.currentTimeMillis() + cookie.maxAge * 1000,
-            domain = cookie.domain,
-            path = cookie.path,
-            secure = cookie.secure,
-            httpOnly = cookie.isHttpOnly,
-            persistent = true,
-            hostOnly = false,
-        )
+        fun from(cookie: HttpCookie): SerializableCookie {
+            val hostOnly = cookie.domain.startsWith(".")
+            val domain = if (hostOnly) {
+                cookie.domain.removePrefix(".")
+            } else cookie.domain
+            return SerializableCookie(
+                name = cookie.name,
+                value = cookie.value,
+                expiresAt = if (cookie.maxAge == -1L) 253402300799999L else System.currentTimeMillis() + cookie.maxAge * 1000,
+                domain = domain,
+                path = cookie.path,
+                secure = cookie.secure,
+                httpOnly = cookie.isHttpOnly,
+                persistent = true,
+                hostOnly = hostOnly,
+            )
+        }
     }
 }