module VagrantPlugins
  module ProviderVirtualBox
    module Action
      class CheckBox
        def initialize(app, env)
          @app = app
        end

        def call(env)
          box_name = env[:machine].config.vm.box
          raise Vagrant::Errors::BoxNotSpecified if !box_name

          if !env[:box_collection].find(box_name, :virtualbox)
            box_url = env[:machine].config.vm.box_url
            raise Vagrant::Errors::BoxSpecifiedDoesntExist, :name => box_name if !box_url

            # Add the box then reload the box collection so that it becomes
            # aware of it.
            env[:ui].info I18n.t("vagrant.actions.vm.check_box.not_found", :name => box_name)
            env[:action_runner].run(Vagrant::Action.action_box_add, {
              :box_name     => box_name,
              :box_provider => env[:machine].provider_name,
              :box_url      => box_url
            })

            # Reload the environment and set the VM to be the new loaded VM.
            env[:machine] = env[:machine].env.machine(
              env[:machine].name, env[:machine].provider_name, true)
          end

          @app.call(env)
        end
      end
    end
  end
end