##Note: this is currently not used anymore, it seems no one is using it

require 'veewee/util/shell'
require 'veewee/util/tcp'

module Veewee
  module Builder
    module Virtualbox

      def remove_snapshot_vmachine(vmname,snapname)
        Veewee::Shell.execute("#{@vboxcmd} snapshot '#{vmname}' delete '#{snapname}'")
      end

      def create_snapshot_vmachine(vmname,snapname)
        Veewee::Shell.execute("#{@vboxcmd} snapshot '#{vmname}' take '#{snapname}'")
      end

      def load_snapshot_vmachine(vmname,snapname)
        #if it running , shutdown first
        if (state_vmachine(vmname)=="running")
          stop_vmachine(vmname)
        end

        Veewee::Shell.execute("#{@vboxcmd} snapshot '#{vmname}' restore '#{snapname}'")
        #sometimes it takes some time to shutdown
        sleep 2
        Veewee::Shell.execute("#{@vboxcmd} startvm '#{vmname}'")
      end

      def snapshot_exists(vmname,snapname)
        list_snapshots(vmname).each { |name|
          if (name==snapname) then
            return true
          end
        }
        return false
      end

      def snapshot_version_exists(vmname,snapname)
        list_snapshots(vmname).each { |name|
          if name.match(/^#{snapname}-/) then
            return true
          end
        }
        return false
      end

      def self.rollback_snapshot(vmname,snapname)
        delete_flag=false
        @vboxcmd=Veewee::Environment.determine_vboxcmd

        savestate_recover=false
        if (state_vmachine(vmname)=="running")
          Veewee::Shell.execute("#{@vboxcmd} controlvm '#{vmname}' savestate")
          savestate_recover=true
        end

        list_snapshots(vmname).each { |name|
          if name.match(/^#{snapname}-/) then
            delete_flag=true
          end
          if (delete_flag) then
            remove_snapshot_vmachine(vmname,name)
            sleep 1
          end
        }


        sleep 2

        Veewee::Shell.execute("#{@vboxcmd} startvm '#{vmname}'")

        if (savestate_recover)
          #Recovering from savestate nukes the network! This trick seem to work
          #Also within the vm /etc/init.d/networking restart , but that is OS specific
          #http://www.virtualbox.org/ticket/5666
          #http://www.virtualbox.org/ticket/5654
          #This is supposed to be fixed: http://www.virtualbox.org/changeset/25205 but alas
          Veewee::Shell.execute("#{@vboxcmd} controlvm '#{vmname}' nic1 nat")
          Veewee::Shell.execute("#{@vboxcmd} controlvm '#{vmname}' setlinkstate1 off")
          Veewee::Shell.execute("#{@vboxcmd} controlvm '#{vmname}' setlinkstate1 on")
          sleep 2

          #hmmm, virtualbox => when recovering from a restore , it looses the nat settings!!! So we need to do this again!!
          #thebox.ssh_enable_vmachine({:hostport => host_port , :guestport => 22} )
          #http://www.virtualbox.org/changeset/25402
          #[25402]: NAT: re-establish port-forwarding after savestate / restore state

        end

      end


      def self.list_snapshots(vmname)
        @vboxcmd=Veewee::Environment.determine_vboxcmd

        snapshotresult=Veewee::Shell.execute("#{@vboxcmd} showvminfo --machinereadable '#{vmname}' |grep ^SnapshotName| cut -d '=' -f 2").stdout
        snapshotlist=snapshotresult.gsub(/\"/,'').split(/\n/)
        return snapshotlist
      end

    end #Module
  end #Module
end #Module