lib/rocktumbler/gem.rb in rocktumbler-0.1.2 vs lib/rocktumbler/gem.rb in rocktumbler-0.1.3

@@ -1,62 +1,90 @@
 require_relative 'exceptions'
 
 module Rocktumbler
+  # The Gem is a class responsible for handling outputting info on a Rubygem
   class Gem
-    def initialize(gem_dependency)
-      @gem_dependency = gem_dependency
+    def initialize(gem_dep, opts)
+      @gem_dep = gem_dep
+      @opts = opts
     end
 
-    def print(prepend="")
-      gem_info = retrieve_gem_information(@gem_dependency.name)
-      str = "#{prepend}# #{gem_info.summary}\n"
-      str += "#{prepend}# #{gem_info.homepage}\n"
-      str += "#{prepend}gem '#{@gem_dependency.name}'"
-      str += ",'#{@gem_dependency.requirement}'" unless @gem_dependency.requirement.none?
-      str += generate_gem_source(@gem_dependency.source.options) if @gem_dependency.source
-      str += generate_require(@gem_dependency.autorequire) if @gem_dependency.autorequire
-      str += "\n\n"
+    def print(prepend = '')
+      max_length = 80 - 2 - prepend.chars.count
+      gem_info = retrieve_gem_information(@gem_dep.name, max_length)
+      formatted_line(prepend, gem_info)
     end
 
     private
 
-    def generate_require(autorequire)
+    def formatted_line(prepend, gem_info)
+      if @opts.gem_info
+        str = "#{prepend}# #{gem_info.summary.strip}\n"
+      else
+        str = ''
+      end
+      str += format_homepage(gem_info.homepage, prepend) if @opts.gem_homepage
+      str += "#{prepend}gem '#{@gem_dep.name}'"
+      str += format_requirement || ''
+      str += format_source
+      str += format_require(@gem_dep.autorequire)
+      str
+    end
+
+    def format_requirement
+      ", '#{@gem_dep.requirement}'" unless @gem_dep.requirement.none?
+    end
+
+    def format_homepage(homepage, prepend)
+      (homepage.nil?) ? "#{prepend}#\n" : "#{prepend}# #{homepage}\n"
+    end
+
+    def format_require(autorequire)
+      return '' unless autorequire
       if autorequire.count > 1
         # Array.to_s will output double quotes so we gsub them to singles.
-        return ", require: #{autorequire.to_s.gsub(/"/,'\'')}"
+        return ", require: #{autorequire.to_s.tr('"', '\'')}"
       elsif autorequire.first.is_a?(String)
         return ", require: '#{autorequire.first}'"
       else
         return ", require: #{!autorequire.empty?}"
       end
     end
 
-    def generate_gem_source(source_options)
-      ['git','github','git_source','source','path','gist','bitbucket'].each do | source_type |
-        return ", #{source_type}: '#{source_options[source_type]}'" if source_options[source_type]
+    def format_source
+      return '' unless @gem_dep.source
+      source_options = @gem_dep.source.options
+      %w(git github git_source source path gist bitbucket).each do |source_type|
+        next unless source_options[source_type]
+        return ", #{source_type}: '#{source_options[source_type]}'"
       end
     end
 
-    def retrieve_gem_information(gem_name)
-      bundled_gem = Bundler.definition.specs.select{|g| g.name == gem_name}.first
+    def retrieve_gem_information(gem_name, max_length = 80)
+      bundled_gem = Bundler.definition.specs.find { |g| g.name == gem_name }
       if bundled_gem
-        OpenStruct.new(summary: bundled_gem.summary.slice(0,80), homepage: bundled_gem.homepage)
+        OpenStruct.new(summary: bundled_gem.summary.slice(0, max_length),
+                       homepage: bundled_gem.homepage)
       else
-        rubygems_gem_information(gem_name)
+        rubygems_gem_info(gem_name, max_length)
       end
     end
 
-    def rubygems_gem_information(gem_name)
-      begin
-        json = open("https://rubygems.org/api/v1/gems/#{gem_name}.json").read
-        gem_info = JSON.parse(json)
-        OpenStruct.new(summary: gem_info['info'].slice(0,80), homepage: gem_info['homepage_uri'])
-      rescue OpenURI::HTTPError => e
-        if e.message == '404 Not Found'
-          raise GemInformationNotFoundError, "No details found locally or on rubygems for '#{gem_name}'"
-        else
-          raise e
-        end
+    def rubygems_gem_info(gem_name, max_length = 80)
+      gem_info = rubygems_gem_info_json(gem_name)
+      OpenStruct.new(summary: gem_info['info'].slice(0, max_length),
+                     homepage: gem_info['homepage_uri'])
+    rescue OpenURI::HTTPError => e
+      if e.message == '404 Not Found'
+        raise GemInformationNotFoundError,
+              "No details found locally or on rubygems for '#{gem_name}'"
+      else
+        raise e
       end
+    end
+
+    def rubygems_gem_info_json(gem_name)
+      json = open("https://rubygems.org/api/v1/gems/#{gem_name}.json").read
+      JSON.parse(json)
     end
   end
 end