Extract configuration into config/storage_configuration.yml

1 files changed, 29 insertions, 0 deletions

diff --git a/lib/active_storage/railtie.rb b/lib/active_storage/railtie.rb
index 15ab8aa096..76894c2e16 100644
--- a/lib/active_storage/railtie.rb
+++ b/lib/active_storage/railtie.rb
@@ -23,5 +23,34 @@ module ActiveStorage
         extend ActiveStorage::Attached::Macros
       end
     end
+
+    config.after_initialize do |app|
+      config_choice = app.config.active_storage.service
+      config_file   = Pathname.new(Rails.root.join("config/storage_services.yml"))
+
+      if config_choice
+        raise("Couldn't find Active Storage configuration in #{config_file}") unless config_file.exist?
+
+        begin
+          require "yaml"
+          require "erb"
+          configs = YAML.load(ERB.new(config_file.read).result) || {}
+
+          if service_configuration = configs[config_choice.to_s].symbolize_keys
+            service_name = service_configuration.delete(:service)
+
+            ActiveStorage::Blob.service = ActiveStorage::Service.configure(service_name, service_configuration)
+          else
+            raise "Couldn't configure Active Storage as #{config_choice} was not found in #{config_file}"
+          end
+        rescue Psych::SyntaxError => e
+          raise "YAML syntax error occurred while parsing #{config_file}. " \
+                "Please note that YAML must be consistently indented using spaces. Tabs are not allowed. " \
+                "Error: #{e.message}"
+        rescue => e
+          raise e, "Cannot load `Rails.config.active_storage.service`:\n#{e.message}", e.backtrace
+        end
+      end
+    end
   end
 end