ManageIQ/manageiq-providers-vmware

class ManageIQ::Providers::Vmware::InfraManager::OrchestrationTemplate < ::OrchestrationTemplate
  belongs_to :ext_management_system, :foreign_key => "ems_id", :class_name => "ManageIQ::Providers::Vmware::InfraManager", :inverse_of => false

  delegate :allowed_resource_pools, :allowed_folders, :allowed_hosts, :allowed_storages, :allowed_datacenters, :refresh_field_values, :to => :workflow_helper

  include ProviderObjectMixin

  SPEC_KEY_MAPPING = {
    "ems_folder"    => "folder_id",
    "host"          => "host_id",
    "storage"       => "default_datastore_id",
    "resource_pool" => "resource_pool_id"
  }.freeze

  def deploy(options = {})
    require 'vsphere-automation-vcenter'

    with_provider_connection(:service => :cis) do |api_client|
      request_body = VSphereAutomation::VCenter::VcenterOvfLibraryItemDeploy.new(deployment_spec(options))
      api_instance = VSphereAutomation::VCenter::OvfLibraryItemApi.new(api_client)
      api_instance.deploy(ems_ref, request_body)
    end
  end

  def deployment_spec(opts)
    opts = opts.with_indifferent_access
    raise _("Resource pool is required for content library item deployment.") if opts[:resource_pool_id].blank?
    raise _("accept_all_eula is required for content library item deployment.") if opts[:accept_all_eula].nil?

    spec = {"accept_all_EULA" => opts[:accept_all_eula]}
    spec["name"] = opts[:vm_name] if opts[:vm_name].present?
    spec["storage_provisioning"] = opts[:disk_format] if opts[:disk_format].present?

    if opts[:network_id].present?
      network = Lan.find_by(:id => opts[:network_id])
      spec["network_mappings"] = [{"key" => network.name, "value" => network.ems_ref}] if network.present?
    end

    if opts[:storage_id].present?
      resource = spec_resource("storage", opts)
      spec[SPEC_KEY_MAPPING["storage"]] = resource.ems_ref if resource.present?
    end

    target = {}
    %w[resource_pool ems_folder host].each do |r|
      resource = spec_resource(r, opts)
      target[SPEC_KEY_MAPPING[r]] = resource.ems_ref if resource.present?
    end

    if target["folder_id"].blank? && opts[:datacenter_id].present?
      dc = Datacenter.find_by(:id => opts[:datacenter_id])
      vm_folder = dc.folders.find { |f| f.name == 'vm' } if dc
      target["folder_id"] = vm_folder.ems_ref if vm_folder
    end

    deploy_options = {"deployment_spec" => spec, "target" => target}
    _log.info("Content Library deployment request body: #{deploy_options}")

    deploy_options
  end

  def spec_resource(klass_name, opts)
    key = "#{klass_name}_id"
    return if opts[key].blank?

    klass_name.camelize.constantize.find_by(:id => opts[key])
  end

  def workflow_helper
    @workflow_helper ||= MiqProvisionOrchWorkflow.new({:src_vm_id => [id]}, User.current_user, :initial_pass => true)
  end

  def allowed_vlans
    # workflow_helper.allowed_vlans returns back host level networks which have no ems_ref
    Lan.where(:name => [workflow_helper.allowed_vlans&.keys]).where.not(:ems_ref => nil).uniq(&:name).pluck(:id, :name)
  end

  def target_name_valid?(name, ems_folder_id = nil)
    # TODO: need a way to tell the ovf template is for VM template (where target is a VM) or vApp template (where target is a resource pool with VMs)
    folder = EmsFolder.find_by(:id => ems_folder_id)
    if folder
      folder.vms.none? { |vm| vm.name == name }
    else
      !ext_management_system.vms.where(:name => name).exists?
    end
  end

  def unique_md5?
    false
  end

  def self.display_name(number = 1)
    n_('OVF Template', 'OVF Templates', number)
  end
end