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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
package cask.endpoints
import cask.internal.{Router, Util}
import cask.main.Endpoint
import cask.model._
import io.undertow.server.handlers.form.FormParserFactory
import collection.JavaConverters._
sealed trait FormReader[T] extends Router.ArgReader[Seq[FormEntry], T, Request]
object FormReader{
implicit def paramFormReader[T: QueryParamReader] = new FormReader[T]{
def arity = implicitly[QueryParamReader[T]].arity
def read(ctx: Request, label: String, input: Seq[FormEntry]) = {
implicitly[QueryParamReader[T]].read(ctx, label, if (input == null) null else input.map(_.valueOrFileName))
}
}
implicit def formEntryReader = new FormReader[FormEntry]{
def arity = 1
def read(ctx: Request, label: String, input: Seq[FormEntry]) = input.head
}
implicit def formEntriesReader = new FormReader[Seq[FormEntry]]{
def arity = 1
def read(ctx: Request, label: String, input: Seq[FormEntry]) = input
}
implicit def formValueReader = new FormReader[FormValue]{
def arity = 1
def read(ctx: Request, label: String, input: Seq[FormEntry]) = input.head.asInstanceOf[FormValue]
}
implicit def formValuesReader = new FormReader[Seq[FormValue]]{
def arity = 1
def read(ctx: Request, label: String, input: Seq[FormEntry]) = input.map(_.asInstanceOf[FormValue])
}
implicit def formFileReader = new FormReader[FormFile]{
def arity = 1
def read(ctx: Request, label: String, input: Seq[FormEntry]) = input.head.asInstanceOf[FormFile]
}
implicit def formFilesReader = new FormReader[Seq[FormFile]]{
def arity = 1
def read(ctx: Request, label: String, input: Seq[FormEntry]) = input.map(_.asInstanceOf[FormFile])
}
}
class postForm(val path: String, override val subpath: Boolean = false)
extends Endpoint[Response.Raw, Seq[FormEntry]] {
val methods = Seq("post")
type InputParser[T] = FormReader[T]
def wrapFunction(ctx: Request,
delegate: Delegate): Router.Result[Response.Raw] = {
try {
val formData = FormParserFactory.builder().build().createParser(ctx.exchange).parseBlocking()
delegate(
formData
.iterator()
.asScala
.map(k => (k, formData.get(k).asScala.map(FormEntry.fromUndertow).toSeq))
.toMap
)
} catch{case e: Exception =>
Router.Result.Success(cask.model.Response(
"Unable to parse form data: " + e + "\n" + Util.stackTraceString(e),
statusCode = 400
))
}
}
def wrapPathSegment(s: String): Seq[FormEntry] = Seq(FormValue(s, new io.undertow.util.HeaderMap))
}
|