gemrage 0.1.2 → 0.2.0

data/VERSION

@@ -1 +1 @@
-0.1.2
+0.2.0

data/gemrage.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{gemrage}
-  s.version = "0.1.2"
+  s.version = "0.2.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Daniel Huckstep"]

data/lib/rubygems/commands/scan_command.rb

@@ -15,6 +15,8 @@ Bundler.require
 class Gem::Commands::ScanCommand < Gem::Command
   include Gem::GemcutterUtilities
 
+  GemrageHost = 'http://gemrage.com/'
+
   def initialize
     super('scan', description, :dir => '')
   end
@@ -32,17 +34,20 @@ class Gem::Commands::ScanCommand < Gem::Command
   end
 
   def execute
-    p platform
+    p system_scan
   end
 
 private
 
   def system_scan
     if windows? && pik?
+      notify("Scanning with pik")
       pik_scan
     elsif !windows? && rvm?
+      notify("Scanning with RVM...this might take a while")
       rvm_scan
     else
+      notify("Scanning with basic gem support")
       basic_scan
     end
   end
@@ -52,19 +57,72 @@ private
   end
 
   def rvm_scan
+    h = {}
     RVM.list_gemsets.map do |config|
-      RVM.use(config)
-      platform = rvm_platform
-      parse_gem_list(RVM.perform_set_operation(:gem, 'list').stdout)
+      notify('Scanning RVM config ', config)
+      begin
+        rd, wr = IO.pipe
+        pid = fork do
+          begin
+            # Clear this because shenanigans ensue
+            ENV['BUNDLE_GEMFILE'] = nil
+            RVM.use(config)
+            wr.write("#{rvm_platform}\n")
+            # Have to do it this way to get stdout instead of just the okay from RVM
+            wr.write(RVM.perform_set_operation(:gem, 'list').stdout)
+          ensure
+            # No, mister superman no here...
+            RVM.reset_current!
+            # Ensure we close these inside the fork
+            rd.close unless rd.closed?
+            wr.close unless wr.closed?
+          end
+        end
+        # Close the write pipe so eof works and we don't hang
+        wr.close unless wr.closed?
+        Process.waitpid(pid)
+        plat = rd.readline.chomp.to_sym
+        h = merge_gem_list(h, parse_gem_list(rd.read, plat))
+      rescue => boom
+        # Don't care right now
+        # Might be an IO problem or whatever, just skip it
+      ensure
+        # Ensure we close these outside the fork
+        rd.close unless rd.closed?
+        wr.close unless wr.closed?
+      end
     end
+    h
+  rescue => boom
+    notify('There was an error scanning: ', boom.message)
+    notify('On line: ', boom.backtrace.first)
+    notify('Please report this at http://gemrage.com/ so we can fix it!')
+    {}
   ensure
     RVM.reset_current!
   end
 
+  def merge_gem_list(*hashes)
+    h = {}
+    hashes.each do |hash|
+      hash.each do |name, platform_hash|
+        h[name] ||= {}
+        platform_hash.each do |platform, versions|
+          h[name][platform] = [h[name][platform], versions].compact.uniq.join(',')
+        end
+      end
+    end
+    h
+  end
+
   def basic_scan
+    plat = platform
+    h = {}
     Gem.source_index.map do |name, spec|
-      spec.name
+      h[spec.name] ||= {}
+      h[spec.name][plat] = [spec.version.to_s, h[spec.name][plat]].compact.uniq.join(',')
     end
+    h
   end
 
   def pik?
@@ -79,15 +137,23 @@ private
     Digest::SHA1.hexdigest(Mac.addr)
   end
 
-  def parse_gem_list(stdout)
-
+  def parse_gem_list(stdout, plat = platform)
+    h = {}
+    stdout.split("\n").each do |line|
+      name, versions = line.match(/^([\w\-_]+) \((.*)\)$/)[1,2] rescue next
+      versions = versions.split(',').map { |version| version.strip.split.first }
+      h[name] ||= {}
+      h[name][plat] = [h[name][plat], versions].compact.uniq.join(',')
+    end
+    h
   end
 
   def rvm_platform
-    engine = RVM.ruby('print RUBY_ENGINE').stdout and engine = engine.empty? ? nil : engine
-    description = RVM.ruby('print RUBY_DESCRIPTION').stdout and description = description.empty? ? nil : description
-    version = RVM.ruby('print RUBY_VERSION').stdout and version = version.empty? ? nil : version
-    platform(engine, description, version)
+    platform(rvm_const('RUBY_ENGINE'), rvm_const('RUBY_DESCRIPTION'), rvm_const('RUBY_VERSION'))
+  end
+
+  def rvm_const(rc)
+    c = RVM.ruby("print #{rc}").stdout.strip and c = c.empty? ? nil : c
   end
 
   def windows?
@@ -95,8 +161,8 @@ private
   end
 
   def platform(engine = (defined?(RUBY_ENGINE) ? RUBY_ENGINE : nil),
-         description = (defined?(RUBY_DESCRIPTION) ? RUBY_DESCRIPTION : nil),
-         version = (defined?(RUBY_VERSION) ? RUBY_VERSION : nil))
+               description = (defined?(RUBY_DESCRIPTION) ? RUBY_DESCRIPTION : nil),
+               version = (defined?(RUBY_VERSION) ? RUBY_VERSION : nil))
     if engine && engine == 'jruby'
       :jruby
     elsif engine && engine == 'ironruby'
@@ -121,4 +187,9 @@ private
       :unknown
     end
   end
+
+  def notify(*messages)
+    print(*messages)
+    print("\n")
+  end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: gemrage
 version: !ruby/object:Gem::Version 
-  hash: 31
+  hash: 23
   prerelease: false
   segments: 
   - 0
-  - 1
   - 2
-  version: 0.1.2
+  - 0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Daniel Huckstep