com.stytch.java.common.ParseJWTClaims.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sdk Show documentation

Stytch Java SDK

There is a newer version: 6.0.0

package com.stytch.java.common

import org.jose4j.jwk.HttpsJwks
import org.jose4j.jwt.JwtClaims
import org.jose4j.jwt.consumer.JwtConsumerBuilder
import org.jose4j.keys.resolvers.HttpsJwksVerificationKeyResolver
import java.util.Date

internal data class ParsedJWTClaims(
    val payload: JwtClaims,
    val customClaims: Map = emptyMap(),
)

internal val IGNORED_CLAIMS = listOf("aud", "exp", "iat", "iss", "jti", "nbf", "sub")

internal fun parseJWTClaims(
    jwt: String,
    jwtOptions: JwtOptions,
    jwksClient: HttpsJwks,
    options: ParseJWTClaimsOptions,
): ParsedJWTClaims {
    val now = Date().time
    val jwtConsumer =
        JwtConsumerBuilder().apply {
            setRequireExpirationTime()
            setAllowedClockSkewInSeconds(options.leeway)
            setRequireSubject()
            setExpectedAudience(jwtOptions.audience)
            setExpectedIssuer(jwtOptions.issuer)
            setVerificationKeyResolver(HttpsJwksVerificationKeyResolver(jwksClient))
        }.build()
    val jwtClaims = jwtConsumer.processToClaims(jwt)
    if (options.maxTokenAgeSeconds != null) {
        val iat = jwtClaims.issuedAt
        if ((now - iat.valueInMillis) / 1000 >= options.maxTokenAgeSeconds) {
            throw JWTException.JwtTooOld(
                iat = iat.valueInMillis / 1000,
                maxTokenAgeSeconds = options.maxTokenAgeSeconds,
            )
        }
    }
    val customClaims =
        jwtClaims.claimsMap.filterNot {
            IGNORED_CLAIMS.contains(it.key)
        }
    return ParsedJWTClaims(
        payload = jwtClaims,
        customClaims = customClaims,
    )
}