ManageIQ/manageiq

class VmMigrateTask < MiqRequestTask
  alias_attribute :vm, :source

  validate :validate_request_type, :validate_state
  default_value_for :request_type, "vm_migrate"

  AUTOMATE_DRIVES = true

  def self.base_model
    VmMigrateTask
  end

  def self.get_description(req_obj)
    name = nil
    if req_obj.source.nil?
      # Single source has not been selected yet
      if req_obj.options[:src_ids].length == 1
        v = Vm.find_by(:id => req_obj.options[:src_ids].first)
        name = v.nil? ? "" : v.name
      else
        name = "Multiple VMs"
      end
    else
      name = req_obj.source.name
    end

    new_settings = []
    host_name = req_obj.get_option_last(:placement_host_name)
    new_settings << "Host: #{host_name}" if host_name.present?
    respool_name = req_obj.get_option_last(:placement_rp_name)
    new_settings << "Resource Pool: #{respool_name}" if respool_name.present?
    folder_name = req_obj.get_option_last(:placement_folder_name)
    new_settings << "Folder: #{folder_name}" if folder_name.present?
    storage = req_obj.get_option_last(:placement_ds_name)
    new_settings << "Storage: #{storage}" if storage.present?
    "#{request_class::TASK_DESCRIPTION} for: #{name} - #{new_settings.join(", ")}"
  end

  def statemachine_task_status
    if %w[finished migrated].include?(state)
      status.to_s.downcase == "error" ? "error" : "ok"
    else
      "retry"
    end
  end

  def after_request_task_create
    update_attribute(:description, get_description)
  end

  def do_request
    host_id = get_option(:placement_host_name)
    host = Host.find_by(:id => host_id)

    respool_id = get_option(:placement_rp_name)
    respool = ResourcePool.find_by(:id => respool_id)

    folder_id = get_option(:placement_folder_name)
    folder = EmsFolder.find_by(:id => folder_id)

    datastore_id = get_option(:placement_ds_name)
    datastore = Storage.find_by(:id => datastore_id)

    disk_transform = get_option(:disk_format)

    # Determine if we call migrate for relocate
    vc_method = if datastore || disk_transform
                  :relocate
                elsif respool && host.nil?
                  :relocate
                elsif host
                  :migrate
                end

    _log.warn("Calling VM #{vc_method} for #{vm.id}:#{vm.name}")

    begin
      case vc_method
      when :migrate  then vm.migrate(host, respool)
      when :relocate then vm.relocate(host, respool, datastore, nil, disk_transform)
      end

      vm.move_into_folder(folder) if folder.present?
    rescue => err
      update_and_notify_parent(:state => 'finished', :status => 'Error', :message => "Failed. Reason[#{err.message}]")
      return
    end

    if AUTOMATE_DRIVES
      update_and_notify_parent(:state => 'migrated', :message => "Finished #{request_class::TASK_DESCRIPTION}", :status => 'Ok')
    else
      update_and_notify_parent(:state => 'finished', :message => "#{request_class::TASK_DESCRIPTION} complete", :status => 'Ok')
    end
  end
end