Add shouldGzipResponse

Sources/GzipMiddleware/GzipMiddleware.swift

@@ -3,7 +3,7 @@ import Vapor
 import Foundation
 import ZIPFoundation
 
-/// Server Gzip middlere:
+/// Server Gzip middleware:
 /// 1. checks if the "Accept-Encoding" header contains "gzip"
 /// 2. if so, compresses the body and sets the response header "Content-Encoding" to "gzip",
 public struct GzipServerMiddleware: Middleware, ServiceType {
@@ -11,22 +11,37 @@ public struct GzipServerMiddleware: Middleware, ServiceType {
         return .init()
     }
 
-    private let shouldGzip: (_ request: Request) -> Bool
+    private let shouldGzipRequest: (_ request: Request) -> Bool
+    private let shouldGzipResponse: (_ response: Response) -> Bool
 
     /// The `shouldGzip` closure is asked for every request whether that request
     /// should allow response gzipping. Returns `true` always by default.
     public init(shouldGzip: @escaping (_ request: Request) -> Bool = { _ in true }) {
-        self.shouldGzip = shouldGzip
+        self.shouldGzipRequest = shouldGzip
+        self.shouldGzipResponse = { _ in true }
+    }
+
+    /// The `shouldGzipRequest` closure is asked for every request whether that request
+    /// should allow response gzipping.  `shouldGzipResponse` asks the same for the response.
+    /// Both return`true` always by default.
+    public init(shouldGzipRequest: @escaping (_ request: Request) -> Bool = { _ in true },
+                shouldGzipResponse: @escaping (_ response: Response) -> Bool = { _ in true }
+    ) {
+        self.shouldGzipRequest = shouldGzipRequest
+        self.shouldGzipResponse = shouldGzipResponse
     }
 
     public func respond(to request: Request, chainingTo next: Responder) throws -> EventLoopFuture<Response> {
         let acceptsGzip = request.http.headers[.acceptEncoding].first?.contains("gzip") ?? false
-        guard acceptsGzip && shouldGzip(request) else {
-            return try next.respond(to: request)
+        let response = try next.respond(to: request)
+        guard acceptsGzip && shouldGzipRequest(request) else {
+            return response
         }
 
-        let response = try next.respond(to: request)
         return response.flatMap { response in
+            guard self.shouldGzipResponse(response) else {
+                return request.future(response)
+            }
             var headers = response.http.headers
             headers.replaceOrAdd(name: .contentEncoding, value: "gzip")
             return response.http.body.consumeData(on: request).map { data in