Merge pull request #7 from Hexaville/noppoMan-patch-2

take over response headers from context
Hexaville · Jul 13, 2017 · 8684bb6 · 8684bb6
2 parents 843990d + d752c5b
commit 8684bb6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Sources/HexavilleAuth/HexavilleAuth+Router.swift b/Sources/HexavilleAuth/HexavilleAuth+Router.swift
@@ -21,7 +21,9 @@ extension HexavilleAuth {
                     context.session?["hexaville.oauth_token"] = requestToken.oauthToken
                     let location = try provider.createAuthorizeURL(requestToken: requestToken).absoluteString
 
-                    return Response(status: .found, headers: ["Location": location])
+                    var headers = context.responseHeaders
+                    headers["Location"] = location
+                    return Response(status: .found, headers: headers)
                 }
 
                 router.use(.get, provider.oauth.callbackURL.path) { request, context in