omniauth.lib.TwitterProvider.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2010-2013 Matthew Henderson
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package omniauth.lib
import omniauth.Omniauth
import dispatch.classic._
import oauth._
import oauth.OAuth._
import twitter.Auth
import xml.NodeSeq
import net.liftweb.json.JsonParser
import net.liftweb.http._
import net.liftweb.common._
import omniauth.AuthInfo
import net.liftweb.util.Helpers._

class TwitterProvider(val key:String, val secret:String) extends OmniauthProvider {
  def providerName = TwitterProvider.providerName
  def providerPropertyKey = TwitterProvider.providerPropertyKey
  def providerPropertySecret = TwitterProvider.providerPropertySecret

  def signIn(): NodeSeq = doTwitterSignin
  def callback(): NodeSeq = doTwitterCallback
  implicit val formats = net.liftweb.json.DefaultFormats

  val consumer = Consumer(key, secret)

  def twitterAuthenticateUrl(token: Token) = Omniauth.twitterOauthRequest / "authenticate" with_token token

  def doTwitterSignin () : NodeSeq = {
    logger.debug("doTwitterSignin")
    logger.debug(consumer)
    var callbackUrl = Omniauth.siteAuthBaseUrl+"auth/"+providerName+"/callback"
    logger.debug(callbackUrl)
    var requestToken = Omniauth.http(Auth.request_token(consumer, callbackUrl))
    val auth_uri = twitterAuthenticateUrl(requestToken).to_uri
    logger.debug(auth_uri.toString)
    Omniauth.setRequestToken(requestToken)
    S.redirectTo(auth_uri.toString)
  }

  def doTwitterCallback () : NodeSeq = {
    logger.debug("doTwitterCallback")
    val verifier = S.param("oauth_verifier") openOr S.redirectTo(Omniauth.failureRedirect)
    var requestToken = Omniauth.currentRequestToken openOr S.redirectTo(Omniauth.failureRedirect)
    Omniauth.http(Auth.access_token(consumer, requestToken, verifier)) match {
      case (access_tok, tempUid, screen_name) => {
        val accessToken = AuthToken(access_tok.value, None, None, emptyForBlank(access_tok.secret))
        if(validateToken(accessToken)){
          S.redirectTo(Omniauth.successRedirect)
        }else{
          S.redirectTo(Omniauth.failureRedirect)
        }
      }
      case _ => S.redirectTo(Omniauth.failureRedirect)
    }
  }

  def validateToken(accessToken:AuthToken): Boolean = {
    val tokenParts = accessToken.token
    if(accessToken.secret.isEmpty){
      logger.debug("tokenParts.length != 2: "+accessToken.token +",")
      return false
    }
    val authToken = Token(accessToken.token, accessToken.secret.get)
    logger.debug("authToken "+authToken)
    val verifyCreds = Omniauth.TwitterHost / "1.1/account/verify_credentials.json" <@ (consumer, authToken)
    try{
      val json = Omniauth.http(verifyCreds >- JsonParser.parse)

      val uid = (json \ "id").extract[String]
      val name = (json \ "name").extract[String]
      val nickName = (json \ "screen_name").extract[String]


      val ai = AuthInfo(providerName,uid,name,accessToken,Some(authToken.secret),Some(nickName))
      Omniauth.setAuthInfo(ai)
      logger.debug(ai)
      true
    } catch {
      case e:Exception => logger.debug("Exception= "+e);false;
    }
  }

  def tokenToId(accessToken:AuthToken): Box[String] = {
    val tokenParts = accessToken.token.split(",")
    if(tokenParts.length != 2){
      logger.debug("tokenParts.length != 2: "+accessToken)
      return Empty
    }
    val authToken = Token(tokenParts(0), tokenParts(1))
    logger.debug("authToken "+authToken)
    val verifyCreds = Omniauth.TwitterHost / "1.1/account/verify_credentials.json" <@ (consumer, authToken)
    try{
      val json = Omniauth.http(verifyCreds >- JsonParser.parse)
      (json \ "id").extractOpt[String]
    }catch {
      case e:Exception => logger.debug("Exception= "+e);Empty;
    }
  }
}

object TwitterProvider{
  val providerName:String = "twitter"
  val providerPropertyKey = "omniauth.twitterkey"
  val providerPropertySecret = "omniauth.twittersecret"
}