Fixed compatibility error with setup command and rdoc plugin on rubygems

1 files changed, 14 insertions, 1 deletions

diff --git a/lib/rdoc/rubygems_hook.rb b/lib/rdoc/rubygems_hook.rb
index 95d33d7a8a..4c2a30d786 100644
--- a/lib/rdoc/rubygems_hook.rb
+++ b/lib/rdoc/rubygems_hook.rb
@@ -270,12 +270,25 @@ end
 # `rubygems/rdoc.rb`.
 module RDoc
   class RubygemsHook
+
+    attr_accessor :generate_rdoc, :generate_ri
+
     def self.default_gem?
       !File.exist?(File.join(__dir__, "..", "rubygems_plugin.rb"))
     end
 
-    def initialize(spec)
+    def initialize(spec, generate_rdoc = false, generate_ri = true)
       @spec = spec
+      @generate_rdoc = generate_rdoc
+      @generate_ri   = generate_ri
+    end
+
+    def generate
+      # Do nothing if this is NOT a default gem.
+      return unless self.class.default_gem?
+
+      # Generate document for compatibility if this is a default gem.
+      RubyGemsHook.new(@spec, @generate_rdoc, @generate_ri).generate
     end
 
     def remove