path: root/cask/src/cask/endpoints/WebSocketEndpoint.scala

package cask.endpoints

import cask.internal.Router
import cask.model.Request
import io.undertow.server.HttpServerExchange
import io.undertow.websockets.WebSocketConnectionCallback
trait WebsocketParam[T] extends Router.ArgReader[Seq[String], T, cask.model.Request]

object WebsocketParam{
  class NilParam[T](f: (Request, String) => T) extends WebsocketParam[T]{
    def arity = 0
    def read(ctx: Request, label: String, v: Seq[String]): T = f(ctx, label)
  }
  implicit object HttpExchangeParam extends NilParam[HttpServerExchange](
    (ctx, label) => ctx.exchange
  )
  class SimpleParam[T](f: String => T) extends WebsocketParam[T]{
    def arity = 1
    def read(ctx: cask.model.Request, label: String, v: Seq[String]): T = f(v.head)
  }

  implicit object StringParam extends SimpleParam[String](x => x)
  implicit object BooleanParam extends SimpleParam[Boolean](_.toBoolean)
  implicit object ByteParam extends SimpleParam[Byte](_.toByte)
  implicit object ShortParam extends SimpleParam[Short](_.toShort)
  implicit object IntParam extends SimpleParam[Int](_.toInt)
  implicit object LongParam extends SimpleParam[Long](_.toLong)
  implicit object DoubleParam extends SimpleParam[Double](_.toDouble)
  implicit object FloatParam extends SimpleParam[Float](_.toFloat)
}

sealed trait WebsocketResult
object WebsocketResult{
  implicit class Response(val value: cask.model.Response) extends WebsocketResult
  implicit class Listener(val value: WebSocketConnectionCallback) extends WebsocketResult
}

class websocket(val path: String, override val subpath: Boolean = false) extends cask.main.BaseEndpoint{
  type Output = WebsocketResult
  val methods = Seq("websocket")
  type Input = Seq[String]
  type InputParser[T] = WebsocketParam[T]
  type Returned = Router.Result[WebsocketResult]
  def wrapFunction(ctx: Request, delegate: Delegate): Returned = delegate(Map())

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


}