require "rubygems"
require "rubygems/gem_runner"

require "log4r"

module VagrantPlugins
  module CommandPlugin
    # This class provides methods to help with calling out to the
    # `gem` command but using the RubyGems API.
    class GemHelper
      def initialize(gem_home)
        @gem_home = gem_home.to_s
        @logger   = Log4r::Logger.new("vagrant::plugins::plugincommand::gemhelper")
      end

      # This executes the `gem` command with the given arguments. Under
      # the covers this is actually using the RubyGems API directly,
      # instead of shelling out, which allows for more fine-grained control.
      #
      # @param [Array<String>] argv The arguments to send to the `gem` command.
      def cli(argv)
        # Initialize the UI to use for RubyGems. This allows us to capture
        # the stdout/stderr without actually going to the real STDOUT/STDERR.
        # The final "false" here tells RubyGems we're not a TTY, so don't
        # ask us things.
        gem_ui = Gem::StreamUI.new(StringIO.new, StringIO.new, StringIO.new, false)

        # Set the GEM_HOME so that it is installed into our local gems path
        with_environment do
          @logger.info("Calling gem with argv: #{argv.inspect}")
          Gem::DefaultUserInteraction.use_ui(gem_ui) do
            Gem::GemRunner.new.run(argv)
          end
        end
      rescue Gem::SystemExitException => e
        # This means that something forced an exit within RubyGems.
        # We capture this to check whether it succeeded or not by
        # checking the "exit_code"
        raise Vagrant::Errors::PluginGemError,
          :output => gem_ui.errs.string.chomp if e.exit_code != 0
      ensure
        # Log the output properly
        @logger.debug("Gem STDOUT: #{gem_ui.outs.string}")
        @logger.debug("Gem STDERR: #{gem_ui.errs.string}")
      end

      # This will yield the given block with the proper ENV setup so
      # that RubyGems only sees the gems in the Vagrant-managed gem
      # path.
      def with_environment
        old_gem_home = ENV["GEM_HOME"]
        ENV["GEM_HOME"] = @gem_home
        @logger.debug("Set GEM_HOME to: #{ENV["GEM_HOME"]}")

        # Clear paths so that it reads the new GEM_HOME setting
        Gem.clear_paths

        yield
      ensure
        # Restore the old GEM_HOME
        ENV["GEM_HOME"] = old_gem_home
      end
    end
  end
end