cask/src/cask/endpoints/WebSocketEndpoint.scala


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

package cask.endpoints

import cask.internal.Router
import cask.model.Request
import io.undertow.websockets.WebSocketConnectionCallback
import collection.JavaConverters._
sealed trait WebsocketResult
object WebsocketResult{
  implicit class Response[T](value0: cask.model.Response[T])
                            (implicit f: T => cask.model.Response.Data) extends WebsocketResult{
    def value = value0.map(f)
  }
  implicit class Listener(val value: WebSocketConnectionCallback) extends WebsocketResult
}

class websocket(val path: String, override val subpath: Boolean = false) extends cask.main.BaseEndpoint{
  type InnerReturned = WebsocketResult
  val methods = Seq("websocket")
  type Input = Seq[String]
  type InputParser[T] = QueryParamReader[T]
  type OuterReturned = Router.Result[WebsocketResult]
  def wrapFunction(ctx: Request, delegate: Delegate): OuterReturned = {
    delegate(WebEndpoint.buildMapFromQueryParams(ctx))
  }

  def wrapPathSegment(s: String): Input = Seq(s)
}