outdate-bundled-gems.rb: Do not clean the same directory twice

ruby · Jan 9, 2024 · 8f61617 · 8f61617
1 parent db476cc
commit 8f61617
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/tool/outdate-bundled-gems.rb b/tool/outdate-bundled-gems.rb
@@ -23,6 +23,8 @@
 end
 
 class Removal
+  attr_reader :base
+
   def initialize(base = nil)
     @base = (File.join(base, "/") if base)
     @remove = {}
@@ -74,7 +76,7 @@ def each_directory
 end
 
 srcdir = Removal.new(ARGV.shift)
-curdir = Removal.new
+curdir = !srcdir.base || File.identical?(srcdir.base, ".") ? srcdir : Removal.new
 
 srcdir.glob(".bundle/gems/*/") do |dir|
   unless srcdir.exist?("gems/#{File.basename(dir)}.gem")
@@ -131,5 +133,7 @@ def each_directory
 
 srcdir.each_file {|f| fu.rm_f(f)}
 srcdir.each_directory {|d| fu.rm_rf(d)}
-curdir.each_file {|f| fu.rm_f(f)}
-curdir.each_directory {|d| fu.rm_rf(d)}
+unless curdir.equal?(srcdir)
+  curdir.each_file {|f| fu.rm_f(f)}
+  curdir.each_directory {|d| fu.rm_rf(d)}
+end