Clean code and required changes made

README.md

@@ -441,10 +441,10 @@ strict-transport-security: max-age=31536000; includeSubDomains; preload
 
 If Strict-Transport-Security is not enough to accomplish a forwarding connection to HTTPS from the browsers, you can opt to add an additional middleware who provides this redirection if clients try to reach your site with an HTTP connection.
 
-To use the HSTS Redirect Middleware, you can add the following line in your middlewares, preferably before securityHeadersFactory.build():
+To use the HTTPS Redirect Middleware, you can add the following line in **configure.swift** to enable the middleware. This must be done before `securityHeadersFactory.build()` to ensure HSTS works:
 
 ```swift
-app.middleware.use(securityHeadersFactory.redirectMiddleware)
+app.middleware.use(HTTPSRedirectMiddleware())
 ```
 
 ## Server

Sources/VaporSecurityHeaders/Configurations/HSTSRedirectMiddleware.swift → Sources/VaporSecurityHeaders/Configurations/HTTPSRedirectMiddleware.swift

@@ -1,7 +1,9 @@
 import Vapor
 
-public class HSTSRedirectMiddleware: Middleware {
+public class HTTPSRedirectMiddleware: Middleware {
 
+    public init() {}
+
     public func respond(to request: Request, chainingTo next: Responder) -> EventLoopFuture<Response> {
         if request.application.environment == .development {
             return next.respond(to: request)

Sources/VaporSecurityHeaders/SecurityHeadersFactory.swift

@@ -9,7 +9,6 @@ public class SecurityHeadersFactory {
     var server: ServerConfiguration?
     var referrerPolicy: ReferrerPolicyConfiguration?
     var contentSecurityPolicyReportOnly: ContentSecurityPolicyReportOnlyConfiguration?
-    public var redirectMiddleware = HSTSRedirectMiddleware()
 
     public init() {}
 

Tests/VaporSecurityHeadersTests/RedirectionTest.swift

@@ -65,7 +65,7 @@ class RedirectionTest: XCTestCase {
             application.environment = environment
         }
         if withRedirection == true {
-        application.middleware.use(SecurityHeadersFactory().redirectMiddleware)
+        application.middleware.use(HTTPSRedirectMiddleware())
         }
         try routes(application)
         return try application.responder.respond(to: request).wait()