io.javalin.http.util.CookieStore.kt Maven / Gradle / Ivy
/*
* Javalin - https://javalin.io
* Copyright 2017 David Åse
* Licensed under Apache 2.0: https://github.com/tipsy/javalin/blob/master/LICENSE
*/
package io.javalin.http.util
import io.javalin.plugin.json.JavalinJson
import java.util.*
import javax.servlet.http.Cookie
@Suppress("UNCHECKED_CAST")
class CookieStore(cookie: String?) {
companion object {
var COOKIE_NAME = "javalin-cookie-store"
}
private val cookieMap = deserialize(cookie)
fun serializeToCookie() = Cookie(COOKIE_NAME, serialize(cookieMap)).apply { path = "/" }
operator fun get(key: String) = cookieMap[key] as T
operator fun set(key: String, value: Any) = cookieMap.put(key, value)
fun clear() = cookieMap.clear()
private fun deserialize(cookie: String?) = if (!cookie.isNullOrEmpty()) {
JavalinJson.fromJson(String(Base64.getDecoder().decode(cookie)), Map::class.java) as MutableMap
} else mutableMapOf()
private fun serialize(map: MutableMap) = Base64.getEncoder().encodeToString(JavalinJson.toJson(map).toByteArray())
}