blob: 8c6bc16a426c6d73628b30f5e3390a901d5ec7ad (
plain) (
blame)
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
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)
}
|