Initial commit sweeping change to gem and ruby gem objects

Author: betatetra

lib/gempage.rb

@@ -1,10 +1,12 @@
-require "gempage/version"
-
 require 'erb'
 require 'fileutils'
-require 'gempage/gemfile_processor'
-require 'gempage/ruby_gem_info'
+require "gemnasium/parser"
+
+require "gempage/version"
+require 'gempage/gem'
+require 'gempage/ruby_gem'
 require 'gempage/configuration'
+
 Gempage.send :extend, Gempage::Configuration
 
 module Gempage
@@ -15,20 +17,28 @@ def generate(options = nil)
     end
 
     def result
-      return @result if defined? @result
-      @result = gem_listing ? merge_gem_and_info : false
+      gem_listing.group_by(&:groups)
     end
 
-    def gem_listing
-      return @gem_listing if defined? @gem_listing
-      @gem_listing = Gempage::GemfileProcessor.new(gemfile_path).gem_list
+    def process_gems(data)
+      gems = []
+      data.dependencies.each do |dependency|
+        requirements = dependency.requirement.as_list
+        first_req = requirements.nil? || requirements.first == ">= 0" ? nil : requirements.first
+        if first_req
+          first_req = first_req.gsub(/^\=\s/, '')
+          first_req = "'#{first_req}'"
+        end
+        gems << Gempage::Gem.new(dependency.name, groups: dependency.groups.sort, requirements: first_req)
+      end
+
+      gems
     end
 
-    def merge_gem_and_info
-      gem_listing.each_with_index do |listed_gem, index|
-        ruby_gem_json = Gempage::RubyGemInfo.new(listed_gem[:name]).gem_json
-        gem_listing[index].merge!(ruby_gem_json)
-      end
+
+    def gem_listing
+      gemfile = File.exists?(gemfile_path) ? File.read(gemfile_path) : false
+      process_gems(Gemnasium::Parser.gemfile(gemfile)) if gemfile
     end
 
     def format(options)
@@ -64,8 +74,7 @@ def css_file_exists?(file_name)
 
     def output_message
       if result
-        category_count = result.group_by{ |result| result[:category] }.length
-        "Gem list generated for #{category_count} gem groups and #{result.length} gems to #{gempage_path}."
+        "Gem list generated for #{result.length} gem groups and #{gem_listing.length} gems to #{gempage_path}."
       else
         "There is no Gemfile to process, how odd..."
       end

lib/gempage/gem.rb

@@ -0,0 +1,19 @@
+include Forwardable
+
+module Gempage
+  class Gem
+    extend Forwardable
+
+    attr_accessor :name, :groups, :requirements, :error
+
+    def_delegators :@ruby_gem, :info, :homepage_uri, :documentation_uri, :source_code_uri, :error
+
+    def initialize(name, groups:, requirements:)
+      @name = name
+      @groups = groups
+      @requirements = requirements
+      @ruby_gem = Gempage::RubyGem.new(name)
+      @error = nil
+    end
+  end
+end

lib/gempage/gemfile_processor.rb

@@ -0,0 +1,67 @@
+require 'json'
+
+module Gempage
+  class RubyGem
+    class FetchError < StandardError; end
+
+    GITHUB_LINK_PATTERN = /github\.com/
+
+    attr_accessor :name
+
+    def initialize(name)
+      @name = name
+    end
+
+    def info
+      rubygem_content["info"]
+    end
+
+    def homepage_uri
+      rubygem_content["homepage_uri"]
+    end
+
+    def documentation_uri
+      documentation_uri = rubygem_content["documentation_uri"]
+      documentation_uri = nil if documentation_uri == source_code_uri
+      documentation_uri
+    end
+
+    def source_code_uri
+      source_code_uri = rubygem_content["source_code_uri"]
+      source_code_uri = homepage_uri if !source_code_uri && homepage_uri.match(GITHUB_LINK_PATTERN)
+      source_code_uri
+    end
+
+    def error
+      nil
+    end
+
+    private
+
+    def rubygem_json_url
+      "https://rubygems.org/api/v1/gems/#{name}.json"
+    end
+
+    def rubygem_content
+      @rubygem_content ||= begin
+        response_body = fetch_http(rubygem_json_url)
+        JSON.parse(response_body)
+      end
+    end
+
+    def fetch_http(url, request_max = 5)
+      raise FetchError.new("Too many redirects #{url}") if request_max <= 0
+
+      response = Net::HTTP.get_response(URI(url))
+
+      case response.code.to_i
+      when 200
+        response.body
+      when 301..303
+        fetch_http(URI(response["location"]), request_max - 1)
+      else
+        raise FetchError.new("Bad response #{url}: #{response.message} #{response.code}", )
+      end
+    end
+  end
+end

lib/gempage/ruby_gem.rb

@@ -0,0 +1,33 @@
+require 'spec_helper'
+
+describe Gempage::Gem do
+  before(:each) do
+    ruby_gem_json_double = double(
+      name: "httparty",
+      info: "Makes http fun! Also, makes consuming restful web services dead easy.",
+      homepage_uri: "http://jnunemaker.github.com/httparty",
+      documentation_uri: "",
+      source_code_uri: "http://github.com/jnunemaker/httparty",
+      error: ""
+    )
+    allow(Gempage::RubyGem).to receive(:new).with("httparty").and_return(ruby_gem_json_double)
+  end
+
+  it "turns all arguments into public methods" do
+    httparty_gem = Gempage::Gem.new("httparty", groups: [:default], requirements: ["~> 3.2.3"])
+
+    expect(httparty_gem.name).to eql("httparty")
+    expect(httparty_gem.groups).to eql([:default])
+    expect(httparty_gem.requirements).to eql(["~> 3.2.3"])
+  end
+
+  it "delegates methods to RubyGem" do
+    httparty_gem = Gempage::Gem.new("httparty", groups: [:default], requirements: ["~> 3.2.3"])
+
+    expect(httparty_gem.info).to eql("Makes http fun! Also, makes consuming restful web services dead easy.")
+    expect(httparty_gem.homepage_uri).to eql("http://jnunemaker.github.com/httparty")
+    expect(httparty_gem.documentation_uri).to eql("")
+    expect(httparty_gem.source_code_uri).to eql("http://github.com/jnunemaker/httparty")
+    expect(httparty_gem.error).to eql("")
+  end
+end