commonMain.com.supertokens.sdk.recipes.sessions.Cookies.kt Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of supertokens-sdk-frontend-jvm Show documentation

SuperTokens frontend SDK

The newest version!

package com.supertokens.sdk.recipes.sessions

import com.supertokens.sdk.common.COOKIE_ACCESS_TOKEN
import com.supertokens.sdk.common.COOKIE_REFRESH_TOKEN
import com.supertokens.sdk.common.Routes
import io.ktor.client.plugins.cookies.CookiesStorage
import io.ktor.http.Cookie
import io.ktor.http.Url
import io.ktor.http.fullPath

internal fun defaultCookieStorage(
    sessionRecipe: SessionRecipe,
) =
    object : CookiesStorage {
      override suspend fun addCookie(requestUrl: Url, cookie: Cookie) {
        when (cookie.name) {
          COOKIE_ACCESS_TOKEN -> {
            val token = cookie.value
            if (token.isNotBlank()) {
              sessionRecipe.updateAccessTokenUseCase.updateAccessToken(token)
            }
          }
          COOKIE_REFRESH_TOKEN -> {
            val token = cookie.value
            if (token.isBlank() && !requestUrl.fullPath.endsWith(Routes.Session.SIGN_OUT)) {
              sessionRecipe.signOut(clearServerSession = false)
            } else {
              sessionRecipe.updateRefreshTokenUseCase.updateRefreshToken(token)
            }
          }
        }
      }

      override suspend fun get(requestUrl: Url): List = buildList {
        sessionRecipe.tokensRepository.getAccessToken()?.let {
          add(Cookie(COOKIE_ACCESS_TOKEN, it))
        }
        if (requestUrl.fullPath.endsWith(Routes.Session.REFRESH)) {
          sessionRecipe.tokensRepository.getRefreshToken()?.let {
            add(Cookie(COOKIE_REFRESH_TOKEN, it))
          }
        }
      }

      override fun close() = Unit
    }