diff --git a/lib/bundler/injector.rb b/lib/bundler/injector.rb index 2cdda578e21..6694642246a 100644 --- a/lib/bundler/injector.rb +++ b/lib/bundler/injector.rb @@ -180,10 +180,21 @@ def remove_gems_from_dependencies(builder, gems, gemfile_path) def remove_gems_from_gemfile(gems, gemfile_path) patterns = /gem\s+(['"])#{Regexp.union(gems)}\1|gem\s*\((['"])#{Regexp.union(gems)}\2\)/ - # remove lines which match the regex - new_gemfile = IO.readlines(gemfile_path).reject {|line| line.match(patterns) } + new_gemfile = [] + multiline_removal = false + IO.readlines(gemfile_path).each do |line| + if line.match(patterns) + multiline_removal = line.rstrip.end_with?(",") + # skip lines which match the regex + next + end + + # skip followup lines until line does not end with ',' + new_gemfile << line unless multiline_removal + multiline_removal = line.rstrip.end_with?(",") if multiline_removal + end - # remove lone \n and append them with other strings + # remove line \n and append them with other strings new_gemfile.each_with_index do |_line, index| if new_gemfile[index + 1] == "\n" new_gemfile[index] += new_gemfile[index + 1] diff --git a/spec/commands/remove_spec.rb b/spec/commands/remove_spec.rb index 402faaf1f3e..995ae6c794a 100644 --- a/spec/commands/remove_spec.rb +++ b/spec/commands/remove_spec.rb @@ -44,6 +44,30 @@ source "#{file_uri_for(gem_repo1)}" G end + + context "when gem is specified in multiple lines" do + it "shows success for removed gem" do + gemfile <<-G + source '#{file_uri_for(gem_repo1)}' + + gem 'git' + gem 'rack', + git: 'https://github.com/rack/rack', + branch: 'master' + gem 'nokogiri' + G + + bundle! "remove rack" + + expect(out).to include("rack was removed.") + gemfile_should_be <<-G + source '#{file_uri_for(gem_repo1)}' + + gem 'git' + gem 'nokogiri' + G + end + end end context "when gem is not present in gemfile" do