struct ExtendSessionMiddleware<A: Authenticatable>: AsyncMiddleware {
    let authenticableType: A.Type

    init(authenticableType: A.Type) {
        self.authenticableType = authenticableType
    }

    func respond(to req: Request, chainingTo next: AsyncResponder) async throws -> Response {
        guard
            req.auth.has(authenticableType),
            let rememberMe = Bool(req.session.data["remeberMe"] ?? "")
        else { return try await next.respond(to: req) }

        let response = try await next.respond(to: req)

        let extendTime: TimeInterval = rememberMe ? 7 * 24 * 60 * 60 : 24 * 60 * 60
        let expiresAt = Date().addingTimeInterval(extendTime)

        let cookieName = req.application.sessions.configuration.cookieName
        if var cookie = req.cookies[cookieName] {
            cookie.expires = expiresAt
            cookie.maxAge = Int(extendTime)
            response.cookies[cookieName] = cookie
        }

        return response
    }
}

var config = SessionsConfiguration.default()
config.cookieFactory = { sessionID, session in
    let rememberMe = Bool(session.data["remeberMe"] ?? "")
    let time: TimeInterval = (rememberMe ?? false) ? 7 * 24 * 60 * 60 : 24 * 60 * 60
    let expiresAt = Date().addingTimeInterval(time)
    return .init(string: sessionID.string, expires: expiresAt)
}
application.middleware.use(SessionsMiddleware(session: application.sessions.driver, configuration: config))