ContentMiddleware.swift 676 B

12345678910111213141516171819
  1. import HTTP
  2. /// Middleware that adds `Server` HTTP header to response.
  3. public final class ContentMiddleware: Middleware {
  4. public init() { }
  5. public func respond(to request: Request, chainingTo next: Responder) throws -> Response {
  6. let response = try next.respond(to: request)
  7. if response.headers[HeaderKey.contentType] == "text/plain; charset=utf-8" {
  8. response.headers[HeaderKey.contentType] = "text/plain"
  9. } else if response.headers[HeaderKey.contentType] == "application/json; charset=utf-8" {
  10. response.headers[HeaderKey.contentType] = "application/json"
  11. }
  12. return response
  13. }
  14. }