diff options
Diffstat (limited to 'src/main')
-rw-r--r-- | src/main/scala/xyz/driver/core/rest.scala | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/src/main/scala/xyz/driver/core/rest.scala b/src/main/scala/xyz/driver/core/rest.scala index aa2a75e..b3a9116 100644 --- a/src/main/scala/xyz/driver/core/rest.scala +++ b/src/main/scala/xyz/driver/core/rest.scala @@ -88,8 +88,16 @@ package rest { } val sanitizeRequestEntity: Directive0 = { + val transformer = Flow.fromFunction(escapeScriptTags) + import HttpEntity._ mapRequest( - request => request.mapEntity(entity => entity.transformDataBytes(Flow.fromFunction(escapeScriptTags)))) + request => request.mapEntity(_ match { + case e: Chunked => e.transformDataBytes(transformer) + case e: Strict => e.transformDataBytes(e.contentLength + ???, transformer) + case e: Default => e.transformDataBytes(e.contentLength + ???, transformer) + } + ) + ) } } |