Cache asset paths by methods

Author: dhh

lib/propshaft/helper.rb

@@ -6,20 +6,24 @@ def compute_asset_path(path, options = {})
 
     # Add an option to call `stylesheet_link_tag` with `:all` to include every css file found on the load path.
     def stylesheet_link_tag(*sources, **options)
-      if sources.first == :all
-        super(*all_stylesheets_paths, **options)
+      case sources.first
+      when :all
+        super(*all_stylesheets_paths , **options)
+      when :app
+        super(*app_stylesheets_paths , **options)
       else
         super
       end
     end
 
     # Returns a sorted and unique array of logical paths for all stylesheets in the load path.
     def all_stylesheets_paths
-      Rails.application.assets.load_path
-        .assets(content_types: [ Mime::EXTENSION_LOOKUP["css"] ])
-        .collect { |css| css.logical_path.to_s }
-        .sort
-        .uniq
+      Rails.application.assets.load_path.asset_paths_by_type("css")
+    end
+
+    # Returns a sorted and unique array of logical paths for all stylesheets in app/assets/stylesheets.
+    def app_stylesheets_path
+      Rails.application.assets.load_path.assets_path_by_glob("**/app/assets/stylesheets/**/*.css")
     end
   end
 end

lib/propshaft/load_path.rb

@@ -15,12 +15,18 @@ def find_referenced_by(asset)
     compilers.referenced_by(asset).delete(self)
   end
 
-  def assets(content_types: nil)
-    if content_types
-      assets_by_path.values.select { |asset| asset.content_type.in?(content_types) }
-    else
-      assets_by_path.values
-    end
+  def assets
+    assets_by_path.values
+  end
+
+  def asset_paths_by_type(content_type)
+    (@cached_asset_paths_by_type ||= Hash.new)[content_type] ||=
+      extract_logical_paths_from(assets.select { |a| a.content_type == Mime::EXTENSION_LOOKUP[content_type] })
+  end
+
+  def asset_paths_by_glob(glob)
+    (@cached_asset_paths_by_glob ||= Hash.new)[glob] ||=
+      extract_logical_paths_from(assets.select { |a| a.path.fnmatch?(glob) })
   end
 
   def manifest
@@ -61,12 +67,18 @@ def all_files_from_tree(path)
       path.children.flat_map { |child| child.directory? ? all_files_from_tree(child) : child }
     end
 
+    def extract_logical_paths_from(assets)
+      assets.collect { |asset| asset.logical_path.to_s }.sort
+    end
+
     def without_dotfiles(files)
       files.reject { |file| file.basename.to_s.starts_with?(".") }
     end
 
     def clear_cache
       @cached_assets_by_path = nil
+      @cached_asset_paths_by_type = nil
+      @cached_asset_paths_by_glob = nil
     end
 
     def dedup(paths)

test/propshaft/load_path_test.rb

@@ -29,13 +29,6 @@ class Propshaft::LoadPathTest < ActiveSupport::TestCase
     assert_not_includes @load_path.assets, find_asset(".stuff")
   end
 
-  test "assets by given content types" do
-    assert_not_includes @load_path.assets(content_types: [ Mime[:js] ]), find_asset("one.txt")
-    assert_includes @load_path.assets(content_types: [ Mime[:js] ]), find_asset("again.js")
-    assert_includes @load_path.assets(content_types: [ Mime[:js], Mime[:css] ]), find_asset("again.js")
-    assert_includes @load_path.assets(content_types: [ Mime[:js], Mime[:css] ]), find_asset("another.css")
-  end
-
   test "manifest" do
     @load_path.manifest.tap do |manifest|
       assert_equal "one-f2e1ec14.txt", manifest["one.txt"]
@@ -70,6 +63,18 @@ class Propshaft::LoadPathTest < ActiveSupport::TestCase
     assert_equal Pathname.new("app/assets"), paths.last
   end
 
+  test "asset paths by type" do
+    assert_equal \
+      ["another.css", "dependent/a.css", "dependent/b.css", "dependent/c.css", "file-already-abcdefVWXYZ0123456789_-.digested.css", "file-already-abcdefVWXYZ0123456789_-.digested.debug.css", "file-not.digested.css"],
+      @load_path.asset_paths_by_type("css")
+  end
+
+  test "asset paths by glob" do
+    assert_equal \
+      ["dependent/a.css", "dependent/b.css", "dependent/c.css"],
+      @load_path.asset_paths_by_glob("**/dependent/*.css")
+  end
+
   private
     def find_asset(logical_path)
       root_path = Pathname.new("#{__dir__}/../fixtures/assets/first_path")