automatically wire up sitemap manipulators, let extensions control their name, add self-registration method

middleman-core/lib/middleman-core/extensions.rb

@@ -106,6 +106,7 @@ module Middleman
   class Extension
     class_attribute :supports_multiple_instances, :instance_reader => false, :instance_writer => false
     class_attribute :defined_helpers, :instance_reader => false, :instance_writer => false
+    class_attribute :ext_name, :instance_reader => false, :instance_writer => false
 
     class << self
       def config
@@ -123,6 +124,14 @@ module Middleman
         m.module_eval(&block)
         self.defined_helpers << m
       end
+
+      def extension_name
+        self.ext_name || self.name.underscore.split("/").last.to_sym
+      end
+
+      def register(n=self.extension_name)
+        ::Middleman::Extensions.register(n, self)
+      end
     end
 
     attr_accessor :app, :options
@@ -148,11 +157,15 @@ module Middleman
         end
       end
 
-      klass.after_configuration(&method(:after_configuration))
+      klass.after_configuration do
+        if ext.respond_to?(:manipulate_resource_list)
+          ext.app.sitemap.register_resource_list_manipulator(ext.class.extension_name, ext)
         end
 
-    def after_configuration
+        if ext.respond_to?(:after_configuration)
-      nil
+          ext.after_configuration
+        end
+      end
     end
   end
 end

middleman-more/lib/middleman-more.rb

@@ -81,24 +81,18 @@ module Middleman
 
         # AssetHash appends a hash of the file contents to the assets filename
         # to avoid browser caches failing to update to your new content.
-        Middleman::Extensions.register(:asset_hash) do
         require "middleman-more/extensions/asset_hash"
-          Middleman::Extensions::AssetHash
+        Middleman::Extensions::AssetHash.register
-        end
 
         # AssetHost allows you to setup multiple domains to host your static
         # assets. Calls to asset paths in dynamic templates will then rotate
         # through each of the asset servers to better spread the load.
-        Middleman::Extensions.register(:asset_host) do
         require "middleman-more/extensions/asset_host"
-          Middleman::Extensions::AssetHost
+        Middleman::Extensions::AssetHost.register
-        end
 
         # Provide Apache-style index.html files for directories
-        Middleman::Extensions.register(:directory_indexes) do
         require "middleman-more/extensions/directory_indexes"
-          Middleman::Extensions::DirectoryIndexes
+        Middleman::Extensions::DirectoryIndexes.register
-        end
 
         # Lorem provides a handful of helpful prototyping methods to generate
         # words, paragraphs, fake images, names and email addresses.
@@ -108,10 +102,8 @@ module Middleman
         # AutomaticImageSizes inspects the images used in your dynamic templates
         # and automatically adds width and height attributes to their HTML
         # elements.
-        Middleman::Extensions.register(:automatic_image_sizes) do
         require "middleman-more/extensions/automatic_image_sizes"
-          Middleman::Extensions::AutomaticImageSizes
+        Middleman::Extensions::AutomaticImageSizes.register
-        end
       end
     end
   end

middleman-more/lib/middleman-more/extensions/asset_hash.rb

@@ -16,8 +16,6 @@ module Middleman
         # Allow specifying regexes to ignore, plus always ignore apple touch icons
         @ignore = Array(options.ignore) + [/^apple-touch-icon/]
 
-        app.sitemap.register_resource_list_manipulator(:asset_hash, self)
-
         app.use Middleware, :exts => options.exts, :middleman_app => app, :ignore => @ignore
       end
 

middleman-more/lib/middleman-more/extensions/automatic_image_sizes.rb

@@ -10,6 +10,7 @@ module Middleman
 
         # Include 3rd-party fastimage library
         require "middleman-more/extensions/automatic_image_sizes/fastimage"
+      end
 
       helpers do
         # Override default image_tag helper to automatically calculate and include
@@ -44,5 +45,4 @@ module Middleman
       end
     end
   end
-  end
 end

middleman-more/lib/middleman-more/extensions/directory_indexes.rb

@@ -3,34 +3,11 @@ module Middleman
   module Extensions
 
     # Directory Indexes extension
-    module DirectoryIndexes
+    class DirectoryIndexes < ::Middleman::Extension
-
-      # Setup extension
-      class << self
-
-        # Once registered
-        def registered(app)
-          app.after_configuration do
-            sitemap.register_resource_list_manipulator(
-              :directory_indexes,
-              DirectoryIndexManager.new(self)
-            )
-          end
-        end
-
-        alias :included :registered
-      end
-
-      # Central class for managing the directory indexes extension
-      class DirectoryIndexManager
-        def initialize(app)
-          @app = app
-        end
-
       # Update the main sitemap resource list
       # @return [void]
       def manipulate_resource_list(resources)
-          index_file = @app.index_file
+        index_file = app.index_file
         new_index_path = "/#{index_file}"
 
         resources.each do |resource|
@@ -50,5 +27,4 @@ module Middleman
       end
     end
   end
-  end
 end