vagrant-multiprovider-snap 0.0.12 → 0.0.13

data/lib/vagrant-multiprovider-snap/providers/hyperv/driver/scripts/list_snapshots.ps1

@@ -0,0 +1,12 @@
+Param(
+    [Parameter(Mandatory=$true)]
+    [string]$VmId
+)
+
+$VM = Get-VM -Id $VmId -ErrorAction "Stop"
+$Snapshots = @(Get-VMSnapshot $VM | Select-Object Name)
+$result = ConvertTo-json $Snapshots
+
+Write-Host "===Begin-Output==="
+Write-Host $result
+Write-Host "===End-Output==="

data/lib/vagrant-multiprovider-snap/providers/hyperv/driver/scripts/rollback_vm.ps1

@@ -0,0 +1,8 @@
+Param(
+    [Parameter(Mandatory=$true)]
+    [string]$VmId,
+    [string]$SnapName
+)
+
+$VM = Get-VM -Id $VmId -ErrorAction "Stop"
+Restore-VMSnapshot $VM -Name $SnapName -Confirm:$false

data/lib/vagrant-multiprovider-snap/providers/hyperv/driver/scripts/snapshot_delete.ps1

@@ -0,0 +1,8 @@
+Param(
+    [Parameter(Mandatory=$true)]
+    [string]$VmId,
+    [string]$SnapName
+)
+
+$VM = Get-VM -Id $VmId -ErrorAction "Stop"
+Remove-VMSnapshot $VM -Name $SnapName 

data/lib/vagrant-multiprovider-snap/providers/hyperv/driver/scripts/snapshot_vm.ps1

@@ -0,0 +1,8 @@
+Param(
+    [Parameter(Mandatory=$true)]
+    [string]$VmId,
+    [string]$SnapName
+)
+
+$VM = Get-VM -Id $VmId -ErrorAction "Stop"
+Checkpoint-VM $VM -SnapshotName $SnapName 

data/lib/vagrant-multiprovider-snap/providers/virtualbox/action.rb

@@ -8,8 +8,10 @@ module VagrantPlugins
 
             autoload :SnapshotTake,              File.expand_path("../action/snapshot_take.rb", __FILE__)
             autoload :SnapshotRollback,          File.expand_path("../action/snapshot_rollback.rb", __FILE__)
+            autoload :SnapshotDelete,            File.expand_path("../action/snapshot_delete.rb", __FILE__)
             autoload :HasSnapshot,               File.expand_path("../action/has_snapshot.rb", __FILE__)
             autoload :MessageSnapshotNotCreated, File.expand_path("../action/message_snapshot_not_created.rb", __FILE__)
+            autoload :MessageSnapshotNotDeleted, File.expand_path("../action/message_snapshot_not_deleted.rb", __FILE__)
 
             def self.action_snapshot_take
                 Vagrant::Action::Builder.new.tap do |b|
@@ -25,6 +27,30 @@ module VagrantPlugins
                 end
             end
 
+            def self.action_snapshot_delete
+                Vagrant::Action::Builder.new.tap do |b|
+                    b.use CheckVirtualbox
+                    b.use Call, Created do |env, b2|
+                        if env[:result]
+                            b2.use Call, HasSnapshot do |env2, b3|
+                                if env2[:result]
+                                    b3.use CheckAccessible
+                                    b3.use Call, SnapshotDelete do |env3,b4|
+                                        unless env3[:result]
+                                            b4.use MessageSnapshotNotDeleted
+                                        end
+                                    end
+                                else
+                                    b3.use MessageSnapshotNotCreated
+                                end
+                            end
+                        else
+                            b2.use MessageNotCreated
+                        end
+                    end
+                end
+            end
+
             def self.action_snapshot_rollback
                 Vagrant::Action::Builder.new.tap do |b|
                     b.use CheckVirtualbox

data/lib/vagrant-multiprovider-snap/providers/virtualbox/action/has_snapshot.rb

@@ -12,7 +12,11 @@ module VagrantPlugins
 
                 def call(env)
 
-                    env[:result] = env[:machine].provider.driver.has_snapshot?
+                    if env[:snap_name].nil?
+                        env[:snap_name] = env[:machine].provider.driver.snapshot_list.last
+                    end
+
+                    env[:result] = env[:machine].provider.driver.has_snapshot?(env[:snap_name])
 
                     @app.call(env)
 

data/lib/vagrant-multiprovider-snap/providers/virtualbox/action/message_snapshot_not_created.rb

@@ -12,7 +12,13 @@ module VagrantPlugins
 
                 def call(env)
 
-                    env[:ui].info I18n.t("vagrant_snap.actions.vm.snapshot_not_created")
+                    if env[:snap_name].nil?
+                        env[:ui].info I18n.t("vagrant_snap.actions.vm.snapshot_not_created.not_created")
+                    else
+                        env[:ui].info(I18n.t("vagrant_snap.actions.vm.snapshot_not_created.named_not_exist",
+                            :snapshot => env[:snap_name]),
+                        )
+                    end
 
                     @app.call(env)
 

data/lib/vagrant-multiprovider-snap/providers/virtualbox/action/message_snapshot_not_deleted.rb

@@ -0,0 +1,29 @@
+module VagrantPlugins
+
+    module ProviderVirtualBox
+
+        module Action
+
+            class MessageSnapshotNotDeleted
+
+                def initialize(app, env)
+                    @app = app
+                end
+
+                def call(env)
+
+                    env[:ui].info(I18n.t("vagrant_snap.actions.vm.snapshot_not_deleted.not_deleted",
+                        :snapshot => env[:snap_name]),
+                    )
+
+                    @app.call(env)
+
+                end
+
+            end
+
+        end
+
+    end
+
+end

data/lib/vagrant-multiprovider-snap/providers/virtualbox/action/snapshot_delete.rb

@@ -0,0 +1,31 @@
+module VagrantPlugins
+
+    module ProviderVirtualBox
+
+        module Action
+
+            class SnapshotDelete
+
+                def initialize(app, env)
+                    @app = app
+                end
+
+                def call(env)
+
+                    env[:ui].info(I18n.t("vagrant_snap.actions.vm.snapshot_delete.deleting",
+                        :snapshot => env[:snap_name]),
+                    )
+
+                    env[:result] = env[:machine].provider.driver.snapshot_delete(env[:snap_name])
+
+                    @app.call(env)
+
+                end
+
+            end
+
+        end
+
+    end
+
+end

data/lib/vagrant-multiprovider-snap/providers/virtualbox/action/snapshot_rollback.rb

@@ -12,12 +12,20 @@ module VagrantPlugins
 
                 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"
 
+                    if env[:snap_name].nil?
+                        env[:snap_name] = env[:machine].provider.driver.snapshot_list.last
+                        env[:ui].info I18n.t("vagrant_snap.actions.vm.snapshot_rollback.rolling_back")
+                    else
+                        env[:ui].info(I18n.t("vagrant_snap.actions.vm.snapshot_rollback.rolling_back_named",
+                            :snapshot => env[:snap_name]),
+                        )
+                    end
+
                     env[:machine].provider.driver.snapshot_rollback(boot_mode,env[:snap_name])
 
                     @app.call(env)

data/lib/vagrant-multiprovider-snap/providers/virtualbox/action/snapshot_take.rb

@@ -12,7 +12,14 @@ module VagrantPlugins
 
                 def call(env)
 
-                    env[:ui].info I18n.t("vagrant_snap.actions.vm.snapshot_take.taking")
+                    if env[:snap_name].nil?
+                        env[:ui].info I18n.t("vagrant_snap.actions.vm.snapshot_take.taking")
+                    else
+                        env[:ui].info(I18n.t("vagrant_snap.actions.vm.snapshot_take.taking_named",
+                            :snapshot => env[:snap_name]),
+                        )
+                    end
+
                     env[:machine].provider.driver.snapshot_take(env[:snap_name])
 
                     @app.call(env)

data/lib/vagrant-multiprovider-snap/providers/virtualbox/driver/base.rb

@@ -10,6 +10,10 @@ module VagrantPlugins
                     execute("snapshot", @uuid, "take", name || "vagrant-snap-#{Time.now.to_i}", "--pause")
                 end
 
+                def snapshot_delete(name)
+                    execute("snapshot", @uuid, "delete", name)
+                end
+
                 def snapshot_rollback(bootmode, name)
                     # don't try to power off if we're already off
                     unless [:poweroff, :aborted].include?(read_state)
@@ -31,8 +35,12 @@ module VagrantPlugins
                     snapshots
                 end
 
-                def has_snapshot?
-                    snapshot_list.length > 0
+                def has_snapshot?(name)
+                    if name.nil?
+                        return true unless snapshot_list.empty?
+                    else
+                        return true if snapshot_list.include? "#{name}"
+                    end
                 end
 
             end

data/lib/vagrant-multiprovider-snap/providers/virtualbox/driver/meta.rb

@@ -7,6 +7,7 @@ module VagrantPlugins
             class Meta
 
                 def_delegators :@driver, :snapshot_take,
+                        :snapshot_delete,
                         :snapshot_rollback,
                         :snapshot_list,
                         :has_snapshot?

data/lib/vagrant-multiprovider-snap/providers/vmware_fusion/action.rb

@@ -8,8 +8,10 @@ module HashiCorp
 
             autoload :SnapshotTake,              File.expand_path("../action/snapshot_take.rb", __FILE__)
             autoload :SnapshotRollback,          File.expand_path("../action/snapshot_rollback.rb", __FILE__)
+            autoload :SnapshotDelete,            File.expand_path("../action/snapshot_delete.rb", __FILE__)
             autoload :HasSnapshot,               File.expand_path("../action/has_snapshot.rb", __FILE__)
             autoload :MessageSnapshotNotCreated, File.expand_path("../action/message_snapshot_not_created.rb", __FILE__)
+            autoload :MessageSnapshotNotDeleted, File.expand_path("../action/message_snapshot_not_deleted.rb", __FILE__)
 
             def self.action_snapshot_take
                 Vagrant::Action::Builder.new.tap do |b|
@@ -24,6 +26,29 @@ module HashiCorp
                 end
             end
 
+            def self.action_snapshot_delete
+                Vagrant::Action::Builder.new.tap do |b|
+                    b.use CheckVMware
+                    b.use Call, Created do |env, b2|
+                        if env[:result]
+                            b2.use Call, HasSnapshot do |env2, b3|
+                                if env2[:result]
+                                    b3.use Call, SnapshotDelete do |env3,b4|
+                                        unless env3[:result]
+                                            b4.use MessageSnapshotNotDeleted
+                                        end
+                                    end
+                                else
+                                    b3.use MessageSnapshotNotCreated
+                                end
+                            end
+                        else
+                            b2.use MessageNotCreated
+                        end
+                    end
+                end
+            end
+
             def self.action_snapshot_rollback
                 Vagrant::Action::Builder.new.tap do |b|
                     b.use CheckVMware
@@ -43,6 +68,7 @@ module HashiCorp
                     end
                 end
             end
+
        end
 
     end

data/lib/vagrant-multiprovider-snap/providers/vmware_fusion/action/has_snapshot.rb

@@ -12,7 +12,13 @@ module HashiCorp
 
                 def call(env)
 
-                    env[:result] = env[:machine].provider.driver.has_snapshot?
+                    if env[:snap_name].nil?
+                        snap_name = env[:machine].provider.driver.snapshot_list.last
+                    else
+                        snap_name = env[:snap_name]
+                    end
+
+                    env[:result] = env[:machine].provider.driver.has_snapshot?(snap_name)
 
                     @app.call(env)
 

data/lib/vagrant-multiprovider-snap/providers/vmware_fusion/action/message_snapshot_not_created.rb

@@ -12,7 +12,13 @@ module HashiCorp
 
                 def call(env)
 
-                    env[:ui].info I18n.t("vagrant_snap.actions.vm.snapshot_not_created")
+                    if env[:snap_name].nil?
+                        env[:ui].info I18n.t("vagrant_snap.actions.vm.snapshot_not_created.not_created")
+                    else
+                        env[:ui].info(I18n.t("vagrant_snap.actions.vm.snapshot_not_created.named_not_exist",
+                            :snapshot => env[:snap_name]),
+                        )
+                    end
 
                     @app.call(env)
 

data/lib/vagrant-multiprovider-snap/providers/vmware_fusion/action/message_snapshot_not_deleted.rb

@@ -0,0 +1,29 @@
+module HashiCorp
+
+    module VagrantVMwarefusion
+
+        module Action
+
+            class MessageSnapshotNotDeleted
+
+                def initialize(app, env)
+                    @app = app
+                end
+
+                def call(env)
+
+                    env[:ui].info(I18n.t("vagrant_snap.actions.vm.snapshot_not_deleted.not_deleted",
+                        :snapshot => env[:snap_name]),
+                    )
+
+                    @app.call(env)
+
+                end
+
+            end
+
+        end
+
+    end
+
+end

data/lib/vagrant-multiprovider-snap/providers/vmware_fusion/action/snapshot_delete.rb

@@ -0,0 +1,31 @@
+module HashiCorp
+
+    module VagrantVMwarefusion
+
+        module Action
+
+            class SnapshotDelete
+
+                def initialize(app, env)
+                    @app = app
+                end
+
+                def call(env)
+
+                    env[:ui].info(I18n.t("vagrant_snap.actions.vm.snapshot_delete.deleting",
+                        :snapshot => env[:snap_name]),
+                    )
+
+                    env[:result] = env[:machine].provider.driver.snapshot_delete(env[:snap_name])
+
+                    @app.call(env)
+
+                end
+
+            end
+
+        end
+
+    end
+
+end

data/lib/vagrant-multiprovider-snap/providers/vmware_fusion/action/snapshot_rollback.rb

@@ -12,13 +12,23 @@ module HashiCorp
 
                 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" : "nogui"
-
-                    env[:machine].provider.driver.snapshot_rollback(boot_mode, env[:snap_name])
+                    # Snapshot rollback involves powering off and on the VM, so we need to find the gui
+                    # state ... however, the start() method in VagrantVMwarefusion.Driver.Fusion takes a
+                    # bool, not a string, so setting this to "gui" or "nogui" will therefore always start
+                    # in gui mode. Setting this to a bool instead
+
+                    boot_mode = env[:machine].provider_config.gui
+
+                    if env[:snap_name].nil?
+                        env[:snap_name] = env[:machine].provider.driver.snapshot_list.last
+                        env[:ui].info I18n.t("vagrant_snap.actions.vm.snapshot_rollback.rolling_back")
+                    else
+                        env[:ui].info(I18n.t("vagrant_snap.actions.vm.snapshot_rollback.rolling_back_named",
+                            :snapshot => env[:snap_name]),
+                        )
+                    end
+
+                    env[:machine].provider.driver.snapshot_rollback(boot_mode,env[:snap_name])
 
                     @app.call(env)
 

data/lib/vagrant-multiprovider-snap/providers/vmware_fusion/action/snapshot_take.rb

@@ -12,7 +12,14 @@ module HashiCorp
 
                 def call(env)
 
-                    env[:ui].info I18n.t("vagrant_snap.actions.vm.snapshot_take.taking")
+                    if env[:snap_name].nil?
+                        env[:ui].info I18n.t("vagrant_snap.actions.vm.snapshot_take.taking")
+                    else
+                        env[:ui].info(I18n.t("vagrant_snap.actions.vm.snapshot_take.taking_named",
+                            :snapshot => env[:snap_name]),
+                        )
+                    end
+
                     env[:machine].provider.driver.snapshot_take(env[:snap_name])
 
                     @app.call(env)