ManageIQ/manageiq

module VmOrTemplate::Operations
  extend ActiveSupport::Concern

  include Configuration
  include Power
  include Relocation
  include Snapshot

  alias_method :ruby_clone, :clone

  def raw_clone(_name, _folder, _pool = nil, _host = nil, _datastore = nil, _powerOn = false, _template_flag = false, _transform = nil, _config = nil, _customization = nil, _disk = nil)
    raise NotImplementedError, _("must be implemented in a subclass")
  end

  def clone(name, folder, pool = nil, host = nil, datastore = nil, powerOn = false, template_flag = false, transform = nil, config = nil, customization = nil, disk = nil)
    raise _("VM has no EMS, unable to clone") unless ext_management_system

    raw_clone(name, folder, pool, host, datastore, powerOn, template_flag, transform, config, customization, disk)
  end

  def raw_mark_as_template
    raise NotImplementedError, _("must be implemented in a subclass")
  end

  def mark_as_template
    raise _("VM has no EMS, unable to mark as template") unless ext_management_system

    raw_mark_as_template
  end

  def raw_mark_as_vm(_pool, _host = nil)
    raise NotImplementedError, _("must be implemented in a subclass")
  end

  def mark_as_vm(pool, host = nil)
    raise _("VM has no EMS, unable to mark as vm") unless ext_management_system

    raw_mark_as_vm(pool, host)
  end

  def raw_unregister
    raise NotImplementedError, _("must be implemented in a subclass")
  end

  def unregister_queue
    run_command_via_queue("raw_unregister")
  end

  def unregister
    raise _("VM has no Provider, unable to unregister VM") unless ext_management_system

    check_policy_prevent(:request_vm_unregister, :unregister_queue)
  end

  def raw_destroy
    raise NotImplementedError, _("must be implemented in a subclass")
  end

  def destroy_queue
    run_command_via_queue("raw_destroy")
  end

  def vm_destroy
    raise _("VM has no Provider, unable to destroy VM") unless ext_management_system

    check_policy_prevent(:request_vm_destroy, :destroy_queue)
  end

  def raw_rename(_new_name)
    raise NotImplementedError, _("must be implemented in a subclass")
  end

  def rename(new_name)
    raise _("VM has no Provider, unable to rename VM") unless ext_management_system

    raw_rename(new_name)
  end

  def rename_queue(userid, new_name)
    task_opts = {
      :action => "Renaming VM for user #{userid}",
      :userid => userid
    }

    run_command_via_task(task_opts, :method_name => "rename", :args => [new_name])
  end

  def raw_set_description(_new_description)
    raise NotImplementedError, _("must be implemented in a subclass")
  end

  def set_description(new_description)
    raise _("VM has no Provider, unable to rename VM") unless ext_management_system

    raw_set_description(new_description)
  end

  def set_description_queue(userid, new_description)
    task_opts = {
      :action => "Setting description for user #{userid}",
      :userid => userid
    }

    run_command_via_task(task_opts, :method_name => "set_description", :args => [new_description])
  end

  def raw_set_custom_field(_attribute, _value)
    raise NotImplementedError, _("must be implemented in a subclass")
  end

  def set_custom_field(attribute, value)
    raise _("VM has no EMS, unable to set custom attribute") unless ext_management_system

    raw_set_custom_field(attribute, value)
  end

  def log_user_event(user_event)
    $log.info(user_event)
  end

  #
  # UI button validation methods
  #

  included do
    supports :action do
      if orphaned?
        _("The VM is orphaned")
      elsif archived?
        _("The VM is archived")
      end
    end

    supports :control do
      if retired?
        _('The VM is retired')
      elsif template?
        _('The VM is a template')
      elsif terminated?
        _('The VM is terminated')
      elsif !has_required_host?
        _('The VM is not connected to a Host')
      elsif disconnected?
        _('The VM does not have a valid connection state')
      elsif !has_active_ems?
        _("The VM is not connected to an active Provider")
      end
    end

    supports :vm_control_powered_on do
      if !supports?(:control)
        unsupported_reason(:control)
      elsif current_state != "on"
        "The VM is not powered on"
      end
    end
  end
end