module VagrantPlugins

    module ProviderVirtualBox

        module Action

            class SnapshotRollback

                def initialize(app, env)
                    @app = app
                end

                def call(env)

                    env[:ui].info I18n.t("vagrant_snap.actions.vm.snapshot_rollback.rolling_back")

                    # Snapshot rollback involves powering off and on the VM
                    #  so we need to find the gui state
                    boot_mode = env[:machine].provider_config.gui ? "gui" : "headless"

                    env[:machine].provider.driver.snapshot_rollback(boot_mode)

                    @app.call(env)

                end

            end

        end

    end

end