consistent use of Bundler.ui in inline installs

4 files changed, 12 insertions, 3 deletions

diff --git a/lib/bundler/cli.rb b/lib/bundler/cli.rb
index cf30589673..e679e5e172 100644
--- a/lib/bundler/cli.rb
+++ b/lib/bundler/cli.rb
@@ -19,7 +19,6 @@ module Bundler
       custom_gemfile = options[:gemfile] || Bundler.settings[:gemfile]
       ENV['BUNDLE_GEMFILE']   = File.expand_path(custom_gemfile) if custom_gemfile
       Bundler.settings[:retry] = options[:retry] if options[:retry]
-      Bundler.rubygems.ui = UI::RGProxy.new(Bundler.ui)
       auto_install if AUTO_INSTALL_CMDS.include?(current_cmd)
     rescue UnknownArgumentError => e
       raise InvalidOption, e.message
diff --git a/lib/bundler/cli/install.rb b/lib/bundler/cli/install.rb
index f7c046d433..445040e1ea 100644
--- a/lib/bundler/cli/install.rb
+++ b/lib/bundler/cli/install.rb
@@ -103,6 +103,7 @@ module Bundler
 
       definition = Bundler.definition
       definition.validate_ruby!
+
       Installer.install(Bundler.root, definition, options)
       Bundler.load.cache if Bundler.app_cache.exist? && !options["no-cache"] && !Bundler.settings[:frozen]
 
diff --git a/lib/bundler/inline.rb b/lib/bundler/inline.rb
index b38a925df2..6f674822d2 100644
--- a/lib/bundler/inline.rb
+++ b/lib/bundler/inline.rb
@@ -37,10 +37,19 @@ def gemfile(install = false, &gemfile)
 
   builder = Bundler::Dsl.new
   builder.instance_eval(&gemfile)
+
   definition = builder.to_definition(nil, true)
   def definition.lock(file); end
   definition.validate_ruby!
-  Bundler::Installer.install(Bundler.root, definition, :system => true) if install
+
+  if install
+    Bundler.ui = Bundler::UI::Shell.new
+    Bundler::Installer.install(Bundler.root, definition, :system => true)
+    Bundler::Installer.post_install_messages.each do |name, message|
+      Bundler.ui.info "Post install message from #{name}:\n#{message}"
+    end
+  end
+
   runtime = Bundler::Runtime.new(nil, definition)
   runtime.setup_environment
   runtime.setup.require
diff --git a/spec/runtime/inline_spec.rb b/spec/runtime/inline_spec.rb
index 176c962953..1e07692636 100644
--- a/spec/runtime/inline_spec.rb
+++ b/spec/runtime/inline_spec.rb
@@ -82,7 +82,7 @@ describe "bundler/inline#gemfile" do
       end
     RUBY
 
-    expect(out).to eq("Rack's post install message")
+    expect(out).to include("Rack's post install message")
     expect(exitstatus).to be_zero if exitstatus
   end
 end