Wrap the LayoutReader in the Jekyll module

lib/jekyll/layout_reader.rb

@@ -1,40 +1,40 @@
-class LayoutReader
+module Jekyll
-  attr_reader :site
+  class LayoutReader
-  def initialize(site)
+    attr_reader :site
-    @site = site
+    def initialize(site)
-    @layouts = {}
+      @site = site
-  end
+      @layouts = {}
-
-  def read
-    layout_entries.each do |f|
-      @layouts[layout_name(f)] = Layout.new(site, layout_directory, f)
     end
 
-    @layouts
+    def read
-  end
+      layout_entries.each do |f|
+        @layouts[layout_name(f)] = Layout.new(site, layout_directory, f)
+      end
 
-  private
+      @layouts
-
+    end
-  def layout_entries
+
-    entries = []
+    private
-    within(layout_directory) do
+
-      entries = EntryFilter.new(site).filter(Dir['**/*.*'])
+    def layout_entries
+      entries = []
+      within(layout_directory) do
+        entries = EntryFilter.new(site).filter(Dir['**/*.*'])
+      end
+      entries
+    end
+
+    def layout_name(file)
+      file.split(".")[0..-2].join(".")
+    end
+
+    def within(directory)
+      return unless File.exists?(directory)
+      Dir.chdir(directory) { yield }
+    end
+
+    def layout_directory
+      File.join(site.source, site.config['layouts'])
     end
-    entries
   end
-
-  def layout_name(file)
-    file.split(".")[0..-2].join(".")
-  end
-
-  def within(directory)
-    return unless File.exists?(directory)
-    Dir.chdir(directory) { yield }
-  end
-
-  def layout_directory
-    File.join(site.source, site.config['layouts'])
-  end
-
-
 end