gem-depclean 0.2.3 → 0.2.4

data/VERSION

@@ -1 +1 @@
-0.2.3
+0.2.4

data/lib/commands/world_command.rb

@@ -3,17 +3,16 @@ require 'rubygems/requirement'
 require 'yaml'
 
 class Gem::Commands::WorldCommand < Gem::Command
-  Gem.post_install do |installer|
-    command = Gem::CommandManager.instance['install']
-    name = installer.spec.name
-
-    begin
-      if command.get_all_gem_names.include?(name)
-        new.add name, command.options[:version]
-      end
-    rescue Gem::CommandLineError
-      # called by 'gem update'
-    end
+  def world_path
+    File.join(Gem.user_dir, 'world')
+  end
+
+  def world_gems
+    YAML.load_file(world_path).inject([]) {|acc, (name, versions)|
+      acc + versions.map {|v|
+        Gem.source_index.find_name(name, v).sort_by(&:version).last
+      }.compact
+    }
   end
 
   def initialize
@@ -30,7 +29,7 @@ class Gem::Commands::WorldCommand < Gem::Command
 
   def execute
     if options[:init]
-      init_world
+      generate
     elsif options[:edit]
       edit
     else
@@ -38,7 +37,28 @@ class Gem::Commands::WorldCommand < Gem::Command
     end
   end
 
+  def generate
+    if File.exist?(world_path)
+      terminate_interaction unless ask_yes_no "'#{world_path}' is already exists. overwrite?"
+    end
+
+    open(world_path, 'w') do |f|
+      f << Gem.source_index.map(&:last).select {|spec|
+        spec.dependent_gems.empty?
+      }.group_by(&:name).inject({}) {|h, (name, specs)|
+        versions = specs.map(&:version).sort
+        versions[-1] = Gem::Requirement.default
+
+        h.merge(name => versions.map(&:to_s))
+      }.to_yaml
+    end
+
+    say "'#{world_path}' was successfully initialized."
+  end
+
   def edit
+    terminate_if_world_is_missing
+
     unless editor = ENV['VISUAL'] || ENV['EDITOR']
       alert_error 'Please set VISUAL or EDITOR variable.'
       terminate_interaction
@@ -47,6 +67,16 @@ class Gem::Commands::WorldCommand < Gem::Command
     system editor, world_path
   end
 
+  def list
+    terminate_if_world_is_missing
+
+    YAML.load_file(world_path).sort_by {|name, versions|
+      name.downcase
+    }.each do |name, versions|
+      say "#{name} (#{versions.join(', ')})"
+    end
+  end
+
   def add(name, version)
     open(world_path, 'r+') do |f|
       f << YAML.load(f).tap {|world|
@@ -59,36 +89,25 @@ class Gem::Commands::WorldCommand < Gem::Command
     end
   end
 
-  def world_path
-    File.join(Gem.user_dir, 'world')
-  end
-
-  def world_gems
-    YAML.load_file(world_path).inject([]) {|acc, (name, versions)|
-      acc + versions.map {|v|
-        Gem.source_index.find_name(name, v).sort_by(&:version).last
-      }.compact
-    }
-  end
+  private
 
-  def list
-    YAML.load_file(world_path).sort_by {|name, versions|
-      name.downcase
-    }.each do |name, versions|
-      say "#{name} (#{versions.join(', ')})"
+  def terminate_if_world_is_missing
+    unless File.exist?(world_path)
+      alert_error "'#{world_path}' is missing. Please execute 'gem world --init'."
+      terminate_interaction
     end
   end
+end
 
-  def init_world
-    open(world_path, 'w') do |f|
-      f << Gem.source_index.map(&:last).select {|spec|
-        spec.dependent_gems.empty?
-      }.group_by(&:name).inject({}) {|h, (name, specs)|
-        versions = specs.map(&:version).sort
-        versions[-1] = Gem::Requirement.default
+Gem.post_install do |installer|
+  command = Gem::CommandManager.instance['install']
+  name = installer.spec.name
 
-        h.merge(name => versions.map(&:to_s))
-      }.to_yaml
+  begin
+    if command.get_all_gem_names.include?(name)
+      Gem::Commands::WorldCommand.new.add name, command.options[:version]
     end
+  rescue Gem::CommandLineError
+    # called by 'gem update'
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: gem-depclean
 version: !ruby/object:Gem::Version 
-  version: 0.2.3
+  version: 0.2.4
 platform: ruby
 authors: 
 - Keita Urashima