Allow redirects from and for subclasses of page and document (#204)

Merge pull request 204

Author: Radoslav Karlík

lib/jekyll-redirect-from/generator.rb

@@ -16,7 +16,7 @@ def generate(site)
 
       # Must duplicate pages to modify while in loop
       (site.docs_to_write + site.pages.dup).each do |doc|
-        next unless JekyllRedirectFrom::CLASSES.include?(doc.class)
+        next unless redirectable_document?(doc)
 
         generate_redirect_from(doc)
         generate_redirect_to(doc)
@@ -54,6 +54,10 @@ def generate_redirects_json
       site.pages << page
     end
 
+    def redirectable_document?(doc)
+      doc.is_a?(Jekyll::Document) || doc.is_a?(Jekyll::Page)
+    end
+
     def generate_redirects_json?
       site.config.dig("redirect_from", "json") != false
     end

spec/jekyll_redirect_from/generator_spec.rb

@@ -154,4 +154,18 @@
       end
     end
   end
+
+  context "redirectable_document?" do
+    let(:generator) { JekyllRedirectFrom::Generator.new }
+
+    it "accepts subclasses of Jekyll::Document" do
+      SubclassOfJekyllDocument = Class.new(Jekyll::Document) { define_method(:initialize) {} }
+      expect(generator.send(:redirectable_document?, SubclassOfJekyllDocument.new)).to be_truthy
+    end
+
+    it "accepts subclasses of Jekyll::Page" do
+      SubclassOfJekyllPage = Class.new(Jekyll::Page) { define_method(:initialize) {} }
+      expect(generator.send(:redirectable_document?, SubclassOfJekyllPage.new)).to be_truthy
+    end
+  end
 end