Play scala web-socket 中的 oauth2 实现

Question

我已经使用 Play2.3 scala 开发了 web-socket，我正在尝试使用 oauth-provider 在 Play scala web-socket 中实现 oauth2。

web-socket的代码sn-p：

object LoginWS {
  def props(out: ActorRef) = Props(new LoginWS(out))
}

class LoginWS(out: ActorRef) extends Actor {
  def receive = {
    case json_req: JsObject =>
        val login_status : String = "Success"
        out ! Json.toJson(JsObject(Seq("login_status" -> JsString(login_status))))
  }
}

object WebSocketController extends Controller {

  def login = WebSocket.acceptWithActor[JsValue, JsValue] { request =>
    out =>
      LoginWS.props(out)
  }

}

REST服务的oauth2的代码sn-p：

我已将“scala-oauth2-provider-slick”示例用于 oauth2 实现。

trait MyOAuth extends OAuth2Provider {
  override val tokenEndpoint: TokenEndpoint = MyTokenEndpoint
}

object OAuth2Controller extends Controller with MyOAuth {

  val cats = TableQuery[CatsTable] 
  implicit val catFormat = Json.format[Cat]

  def accessToken = Action.async { implicit request =>
    issueAccessToken(new Oauth2DataHandler())
  }

  def list = Action.async { implicit request =>
    authorize(new Oauth2DataHandler()) { authInfo =>
      DBAction { implicit rs =>
        Ok(toJson(cats.list))
      }.apply(request)
    }
  }
}

如何在上面的 web-socket 代码中实现 oauth2？

有人知道，如何为上述 web-socket 实现 oauth2？

更新：

我尝试实现 oauth 和 web socket：

代码 sn-p：

  def login = WebSocket.tryAcceptWithActor[JsValue, JsValue] { request =>
      authorize(new Oauth2DataHandler()) { authInfo =>
      val user = authInfo.user // User is defined on your system
      println(user.toString())
    }
  }

遇到以下问题：

1) polymorphic expression cannot be instantiated to expected type; found : [A]scala.concurrent.Future[play.api.mvc.Result] required: Either[play.api.mvc.Result,play.api.mvc.WebSocket.HandlerProps] (which expands to) Either[play.api.mvc.Result,akka.actor.ActorRef ⇒ akka.actor.Props]

2) type mismatch; found : Unit required: scala.concurrent.Future[play.api.mvc.Result]

Answer 1

OAuth2Provider 提供了protectedResource 实例，所以你可以直接调用handleRequest 方法在你的WebSocket 控制器中进行授权。 客户端通过 HTTP 头或查询参数发送访问令牌。

我建议您使用 tryAcceptWithActor 而不是 acceptWithActor。 因为如果用户发送无效的访问令牌，您需要返回错误的请求。

object LoginWS {
  def props(user: User)(out: ActorRef) = Props(new LoginWS(user, out))
}

object WebSocketController extends Controller extends MyOAuth {

  def login = WebSocket.tryAcceptWithActor[JsValue, JsValue] { request =>
    Future.successful {
      (protectedResource.handleRequest(request, new Oauth2DataHandler())).map {
        case Left(e) => e match {
          case e: OAuthError if e.statusCode == 400 => Left(BadRequest.withHeaders(responseOAuthErrorHeader(e)))
          case e: OAuthError if e.statusCode == 401 => Left(Unauthorized.withHeaders(responseOAuthErrorHeader(e)))
        }
        case Right(authInfo) => Right(LoginWS.props(authInfo.user) _)
      }
    }
  }
}

我不确定这段代码是否可以编译。 但是我已经使用这个库编写了类似的代码。