refactor: middleware_enabled_rails_envs to

middleware_enabled
SonicGarden · Nov 8, 2023 · 8fcd0c2 · 8fcd0c2
1 parent 4101c7a
commit 8fcd0c2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/lib/sg_fargate_rails/config.rb b/lib/sg_fargate_rails/config.rb
@@ -1,15 +1,15 @@
 module SgFargateRails
   class Config
     attr_reader :proxy_ip_addresses
-    attr_accessor :middleware_enabled_rails_envs
+    attr_accessor :middleware_enabled
 
     # NOTE: good_jobダッシュボードへのアクセスをproxy経由のアクセスに制限するかどうか
     attr_accessor :restrict_access_to_good_job_dashboard
 
     def initialize
       self.proxy_ip_addresses = ENV['SG_PROXY_IP_ADDRESSES']
       self.restrict_access_to_good_job_dashboard = Rails.env.production?
-      self.middleware_enabled_rails_envs = %w[production staging]
+      self.middleware_enabled = !Rails.env.development? && !Rails.env.test?
     end
 
     def proxy_ip_addresses=(ip_addresses)

diff --git a/lib/sg_fargate_rails/railtie.rb b/lib/sg_fargate_rails/railtie.rb
@@ -11,7 +11,7 @@ class Railtie < ::Rails::Railtie
     end
 
     initializer :initialize_sg_fargate_rails, after: :load_config_initializers do |app|
-      if SgFargateRails.config.middleware_enabled_rails_envs.include?(Rails.env)
+      if SgFargateRails.config.middleware_enabled
         app.config.middleware.insert 0, SgFargateRails::AdjustCloudfrontHeaders
         app.config.middleware.insert 1, SgFargateRails::Healthcheck
         app.config.middleware.swap ActionDispatch::RemoteIp, SgFargateRails::RemoteIp, app.config.action_dispatch.ip_spoofing_check, app.config.action_dispatch.trusted_proxies