io.toolsplus.atlassian.jwt.JwtBuilder.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of atlassian-jwt-core_2.12 Show documentation

Atlassian Jwt Core

The newest version!

package io.toolsplus.atlassian.jwt

import java.time.Duration

import com.nimbusds.jose.JWSAlgorithm
import com.nimbusds.jose.crypto.MACSigner
import io.toolsplus.atlassian.jwt.api.Predef.RawJwt

/**
  * A builder of JSON Web Tokens.
  */
class JwtBuilder(expireAfter: Duration) {

  private val jwtJsonBuilder = new JwtJsonBuilder(expireAfter)

  def withIssuer(iss: String): JwtBuilder = {
    jwtJsonBuilder.withIssuer(iss)
    this
  }

  def withSubject(sub: String): JwtBuilder = {
    jwtJsonBuilder.withSubject(sub)
    this
  }

  def withAudience(aud: Seq[String]): JwtBuilder = {
    jwtJsonBuilder.withAudience(aud)
    this
  }

  def withExpirationTime(exp: Long): JwtBuilder = {
    jwtJsonBuilder.withExpirationTime(exp)
    this
  }

  def withNotBefore(nbf: Long): JwtBuilder = {
    jwtJsonBuilder.withNotBefore(nbf)
    this
  }

  def withIssuedAt(iat: Long): JwtBuilder = {
    jwtJsonBuilder.withIssuedAt(iat)
    this
  }

  def withQueryHash(queryHash: String): JwtBuilder = {
    jwtJsonBuilder.withQueryHash(queryHash)
    this
  }

  def withClaim(name: String, value: AnyRef): JwtBuilder = {
    jwtJsonBuilder.withClaim(name, value)
    this
  }

  def build(sharedSecret: String, algorithm: JWSAlgorithm = JWSAlgorithm.HS256)
    : Either[JwtSigningError, RawJwt] = {
    val jwtPayload: String = jwtJsonBuilder.build
    createJwtWriter(sharedSecret, algorithm).jsonToJwt(jwtPayload)
  }

  private def createJwtWriter(sharedSecret: String,
                              algorithm: JWSAlgorithm): JwtWriter = {
    new JwtWriter(algorithm, new MACSigner(sharedSecret))
  }
}