lib/importmap/engine.rb in importmap-rails-0.6.3 vs lib/importmap/engine.rb in importmap-rails-0.7.0

@@ -1,18 +1,37 @@
 require "importmap/map"
 
+# Use Rails.application.importmap to access the map
+Rails::Application.send(:attr_accessor, :importmap)
+
 module Importmap
   class Engine < ::Rails::Engine
-    config.importmap = Importmap::Map.new.draw("config/importmap.rb")
+    config.importmap = ActiveSupport::OrderedOptions.new
+    config.importmap.sweep_cache = Rails.env.development? || Rails.env.test?
+
     config.autoload_once_paths = %W( #{root}/app/helpers )
 
+    initializer "importmap" do |app|
+      app.importmap = Importmap::Map.new.draw("config/importmap.rb")
+    end
+
     initializer "importmap.reloader" do |app|
       app.config.paths.add "config/importmap.rb"
 
       Importmap::Reloader.new.tap do |reloader|
         reloader.execute
         app.reloaders << reloader
         app.reloader.to_run { reloader.execute }
+      end
+    end
+
+    initializer "importmap.cache_sweeper" do |app|
+      if app.config.importmap.sweep_cache
+        app.importmap.cache_sweeper watches: app.root.join("app/javascript")
+
+        ActiveSupport.on_load(:action_controller_base) do
+          before_action { Rails.application.importmap.cache_sweeper.execute_if_updated }
+        end
       end
     end
 
     initializer "importmap.assets" do
       if Rails.application.config.respond_to?(:assets)