aboutsummaryrefslogtreecommitdiffstats
path: root/lib/active_storage/engine.rb
blob: 5f0b62809e96ada1f7c2d5c103d9582958a04cb7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
require "rails/engine"

module ActiveStorage
  class Engine < Rails::Engine # :nodoc:
    config.active_storage = ActiveSupport::OrderedOptions.new

    config.eager_load_namespaces << ActiveStorage

    initializer "active_storage.logger" do
      require "active_storage/service"

      config.after_initialize do |app|
        ActiveStorage::Service.logger = app.config.active_storage.logger || Rails.logger
      end
    end

    initializer "active_storage.routes" do
      require "active_storage/disk_controller"
      require "active_storage/direct_uploads_controller"

      config.after_initialize do |app|
        app.routes.prepend do
          eval(File.read(File.expand_path("../routes.rb", __FILE__)))
        end
      end
    end

    initializer "active_storage.attached" do
      require "active_storage/attached"

      ActiveSupport.on_load(:active_record) do
        extend ActiveStorage::Attached::Macros
      end
    end

    config.after_initialize do |app|
      if config_choice = app.config.active_storage.service
        config_file = Pathname.new(Rails.root.join("config/storage_services.yml"))
        raise("Couldn't find Active Storage configuration in #{config_file}") unless config_file.exist?

        require "yaml"
        require "erb"

        configs =
          begin
            YAML.load(ERB.new(config_file.read).result) || {}
          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}"
          end

        ActiveStorage::Blob.service =
          begin
            ActiveStorage::Service.configure config_choice, configs
          rescue => e
            raise e, "Cannot load `Rails.config.active_storage.service`:\n#{e.message}", e.backtrace
          end
      else
        raise "No storage service specified for current env (#{Rails.env}). " \
              "Add config.active_storage.service = :local into your config/environments/#{Rails.env}.rb."
      end
    end
  end
end