1 files changed, 35 insertions, 28 deletions
diff --git a/fedora_darkfish.rb b/fedora_darkfish.rb
index 28c3946..6bb9825 100644
--- a/fedora_darkfish.rb
+++ b/fedora_darkfish.rb
@@ -2,46 +2,53 @@ require 'rdoc'
 require 'rdoc/generator/darkfish'
 require 'rdoc/generator/json_index'
 
-class RDoc::Generator::Darkfish
-  alias initialize_orig initialize
+module RDoc::Generator::Fedora
+  class RDoc::Generator::Fedora::Darkfish < RDoc::Generator::Darkfish
+    RDoc::RDoc.add_generator(self)
 
-  def initialize store, options
-    options.template_dir = File.join __dir__, 'darkfish'
-    initialize_orig store, options
-  end
+    alias initialize_orig initialize
+
+    def initialize store, options
+      options.template_dir = File.join __dir__, 'darkfish'
 
-  def install_rdoc_static_file source, destination, options # :nodoc:
-    return unless source.exist?
+      initialize_orig store, options
 
-    begin
-      FileUtils.mkdir_p File.dirname(destination), **options
-      FileUtils.ln_sf source, destination, **options
-    rescue
+      @json_index = RDoc::Generator::Fedora::JsonIndex.new self, options
+    end
+
+    def install_rdoc_static_file source, destination, options # :nodoc:
+      return unless source.exist?
+
+      begin
+        FileUtils.mkdir_p File.dirname(destination), **options
+        FileUtils.ln_sf source, destination, **options
+      rescue
+      end
     end
   end
-end
 
-class RDoc::Generator::JsonIndex
-  alias generate_orig generate
+  class RDoc::Generator::Fedora::JsonIndex < RDoc::Generator::JsonIndex
+    alias generate_orig generate
 
-  def generate
-    generate_orig
+    def generate
+      generate_orig
 
-    out_dir = @base_dir + @options.op_dir
+      out_dir = @base_dir + @options.op_dir
 
-    template_dir = File.join __dir__, 'json_index'
+      template_dir = File.join __dir__, 'json_index'
 
-    Dir.chdir template_dir do
-      Dir['**/*.js'].each do |source|
-        dest = File.join out_dir, source
-        source = File.realpath source
-        FileUtils.ln_sf source, dest, :verbose => $DEBUG_RDOC
+      Dir.chdir template_dir do
+	Dir['**/*.js'].each do |source|
+	  dest = File.join out_dir, source
+	  source = File.realpath source
+	  FileUtils.ln_sf source, dest, :verbose => $DEBUG_RDOC
+	end
       end
     end
-  end
 
-  # Don't generate gzipped content. Because we have always shipped also the
-  # original, it just increases the size.
-  def generate_gzipped
+    # Don't generate gzipped content. Because we have always shipped also the
+    # original, it just increases the size.
+    def generate_gzipped
+    end
   end
 end