Fix rails 5 deprecation warnings

Fixes rails 5 deprecation warnings in `config/` files

2 files changed, 16 insertions, 9 deletions

diff --git a/config/initializers/active_record_avoid_type_casting_in_uniqueness_validator.rb b/config/initializers/active_record_avoid_type_casting_in_uniqueness_validator.rb
index d9418caf68b..ef4abb77bd7 100644
--- a/config/initializers/active_record_avoid_type_casting_in_uniqueness_validator.rb
+++ b/config/initializers/active_record_avoid_type_casting_in_uniqueness_validator.rb
@@ -21,8 +21,6 @@
 # This bug was fixed in Rails 5.1 by https://github.com/rails/rails/pull/24745/commits/aa062318c451512035c10898a1af95943b1a3803
 
 if Gitlab.rails5?
-  ActiveSupport::Deprecation.warn("#{__FILE__} is a monkey patch which must be removed when upgrading to Rails 5.1")
-
   if Rails.version.start_with?("5.1")
     raise "Remove this monkey patch: #{__FILE__}"
   end
diff --git a/config/initializers/static_files.rb b/config/initializers/static_files.rb
index 6c28686e69a..a0b8b68f3ef 100644
--- a/config/initializers/static_files.rb
+++ b/config/initializers/static_files.rb
@@ -1,17 +1,26 @@
 app = Rails.application
 
-if app.config.serve_static_files
+if (Gitlab.rails5? && app.config.public_file_server.enabled) || app.config.serve_static_files
   # The `ActionDispatch::Static` middleware intercepts requests for static files
   # by checking if they exist in the `/public` directory.
   # We're replacing it with our `Gitlab::Middleware::Static` that does the same,
   # except ignoring `/uploads`, letting those go through to the GitLab Rails app.
 
-  app.config.middleware.swap(
-    ActionDispatch::Static,
-    Gitlab::Middleware::Static,
-    app.paths["public"].first,
-    app.config.static_cache_control
-  )
+  if Gitlab.rails5?
+    app.config.middleware.swap(
+      ActionDispatch::Static,
+      Gitlab::Middleware::Static,
+      app.paths["public"].first,
+      headers: app.config.public_file_server.headers
+    )
+  else
+    app.config.middleware.swap(
+      ActionDispatch::Static,
+      Gitlab::Middleware::Static,
+      app.paths["public"].first,
+      app.config.static_cache_control
+    )
+  end
 
   # If webpack-dev-server is configured, proxy webpack's public directory
   # instead of looking for static assets