com.wordnik.swagger.auth.service.TokenCache.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of swagger-oauth2-server_2.10 Show documentation
The newest version!
package com.wordnik.swagger.auth.service

import com.wordnik.swagger.auth.model.TokenWrapper

import org.apache.oltu.oauth2.as.issuer.{ MD5Generator, OAuthIssuerImpl }

import scala.collection.mutable.{ HashSet, HashMap }

trait TokenCache {
  def hasAccessCode(accessCode: String): Boolean
  def exchangeRequestIdForCode(requestId: String): String
  def getTokenForAccessCode(accessCode: String): TokenWrapper
  def addAccessCode(accessCode: String, token: TokenWrapper)
  def removeAccessCode(accessCode: String)

  def hasRequestId(requestId: String): Boolean
  def getRequestId(requestId: String): Map[String, Option[String]]
  def addRequestId(requestId: String, requestMap: Map[String, Option[String]])
  def removeRequestId(requestId: String)

  def allowAnonymousTokens(): Boolean
}

object TokenFactory {
  var tokenCache: Option[TokenCache] = None

  def apply(): TokenCache = {
    if(tokenCache == None)
      tokenCache = Some(new DefaultTokenCache())
    tokenCache.get
  }
}

trait TokenStore {
  def hasAccessCode(accessCode: String): Boolean = TokenFactory().hasAccessCode(accessCode)
  def exchangeRequestIdForCode(requestId: String): String = TokenFactory().exchangeRequestIdForCode(requestId)
  def getTokenForAccessCode(accessCode: String): TokenWrapper = TokenFactory().getTokenForAccessCode(accessCode)
  def addAccessCode(accessCode: String, token: TokenWrapper) = TokenFactory().addAccessCode(accessCode, token)
  def removeAccessCode(accessCode: String) = TokenFactory().removeAccessCode(accessCode)

  def hasRequestId(requestId: String): Boolean = TokenFactory().hasRequestId(requestId)
  def getRequestId(requestId: String): Map[String, Option[String]] = TokenFactory().getRequestId(requestId)
  def addRequestId(requestId: String, requestMap: Map[String, Option[String]]) = TokenFactory().addRequestId(requestId, requestMap)
  def removeRequestId(requestId: String) = TokenFactory().removeRequestId(requestId)

  def generateRequestId(clientId: String) = TokenGenerator.generateRandomCode("requestId", clientId)

  def generateCode(clientId: String) = TokenGenerator.generateRandomCode("code", clientId)

  def generateAccessToken() = TokenGenerator.generateAccessToken()

  def allowAnonymousTokens(): Boolean = TokenFactory().allowAnonymousTokens()
}

object TokenGenerator {
  def generateAccessToken() = {
    val oauthIssuerImpl = new OAuthIssuerImpl(new MD5Generator())
    oauthIssuerImpl.accessToken()
  }

  def generateRandomCode(scope: String, seed: String) = {
    val keySource = scope + seed + System.currentTimeMillis.toString + new scala.util.Random(1).nextInt(1000)
    val md = java.security.MessageDigest.getInstance("SHA-1")
    new sun.misc.BASE64Encoder().encode(md.digest(keySource.getBytes))
  }
}