frapposelli/vagrant-vcloud

module VagrantPlugins
  module VCloud
    module Action
      class PowerOffVApp
        def initialize(app, env)
          @app = app
          @logger = Log4r::Logger.new('vagrant_vcloud::action::poweroff_vapp')
        end

        def call(env)
          cfg = env[:machine].provider_config
          cnx = cfg.vcloud_cnx.driver

          vapp_id = env[:machine].get_vapp_id

          test_vapp = cnx.get_vapp(vapp_id)

          @logger.debug(
            "Number of VMs in the vApp: #{test_vapp[:vms_hash].count}"
          )

          # this is a helper to get vapp_edge_ip into cache for later destroy
          # of edge gateway rules
          if cfg.network_bridge.nil? && cfg.networks.nil?
            vapp_edge_ip = cnx.get_vapp_edge_public_ip(vapp_id)
          end

          # Poweroff vApp
          env[:ui].info('Single VM left in the vApp, Powering off vApp...')
          vapp_stop_task = cnx.poweroff_vapp(vapp_id)
          vapp_stop_wait = cnx.wait_task_completion(vapp_stop_task)

          unless vapp_stop_wait[:errormsg].nil?
            fail Errors::StopVAppError, :message => vapp_stop_wait[:errormsg]
          end

          @app.call env
        end
      end
    end
  end
end