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[:box_collection].add(box_name, box_url)
            env[:box_collection].reload!

            # Reload the environment and set the VM to be the new loaded VM.
            env[:machine].env.reload!
            env[:machine] = env[:machine].env.vms[env[:machine].name]
          end

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