ManageIQ/manageiq-providers-ibm_cloud

class ManageIQ::Providers::IbmCloud::Inventory::Collector::PowerVirtualServers::TargetCollection < ManageIQ::Providers::IbmCloud::Inventory::Collector::PowerVirtualServers
  def initialize(_manager, _target)
    super

    parse_targets!
  end

  def availability_zones
    []
  end

  def images
    @images ||= references(:miq_templates).map do |ems_ref|
      images_api.pcloud_cloudinstances_images_get(cloud_instance_id, ems_ref)
    rescue IbmCloudPower::ApiError => err
      error_message = JSON.parse(err.response_body)["description"]
      _log.debug("ImageID not found: #{error_message}")
      nil
    end.compact
  end

  def flavors
    []
  end

  def cloud_volume_types
    []
  end

  def volumes
    @volumes ||= references(:cloud_volumes).map do |ems_ref|
      volumes_api.pcloud_cloudinstances_volumes_get(cloud_instance_id, ems_ref)
    rescue IbmCloudPower::ApiError => err
      error_message = JSON.parse(err.response_body)["description"]
      _log.debug("VolumeID not found: #{error_message}")
      nil
    end.compact
  end

  def pvm_instances_by_id
    @pvm_instances_by_id ||= pvm_instances.index_by(&:pvm_instance_id)
  end

  def pvm_instances
    @pvm_instances ||= references(:vms).map do |ems_ref|
      pvm_instances_api.pcloud_pvminstances_get(cloud_instance_id, ems_ref)
    rescue IbmCloudPower::ApiError => err
      error_message = JSON.parse(err.response_body)["description"]
      _log.debug("PVMInstanceID not found: #{error_message}")
      nil
    end.compact
  end

  def networks
    @networks ||= references(:cloud_networks).map do |ems_ref|
      networks_api.pcloud_networks_get(cloud_instance_id, ems_ref)
    rescue IbmCloudPower::ApiError => err
      error_message = JSON.parse(err.response_body)["description"]
      _log.debug("NetworkID not found: #{error_message}")
      nil
    end.compact
  end

  def sshkeys
    []
  end

  def snapshots
    begin
      @snapshots ||= snapshots_api.pcloud_cloudinstances_snapshots_getall(cloud_instance_id).snapshots.select do |snapshot|
        references(:vms).include?(snapshot.pvm_instance_id)
      end
    rescue IbmCloudPower::ApiError => err
      error_message = JSON.parse(err.response_body)["description"]
      _log.debug("Error retrieving snapshots: #{error_message}")
      nil
    end.compact
  end

  private

  def parse_targets!
    # `target` here is an `InventoryRefresh::TargetCollection`.  This contains two types of targets,
    # `InventoryRefresh::Target` which is essentialy an association/manager_ref pair, or an ActiveRecord::Base
    # type object like a Vm.
    #
    # This gives us some flexibility in how we request a resource be refreshed.
    target.targets.each do |target|
      case target
      when MiqTemplate
        add_target!(:miq_templates, target.ems_ref)
      when Vm
        add_target!(:vms, target.ems_ref)
      when Flavor
        add_target!(:flavors, target.ems_ref)
      when ManageIQ::Providers::CloudManager::AuthKeyPair
        add_target!(:auth_key_pairs, target.ems_ref)
      when AvailabilityZone
        add_target!(:availability_zones, target.ems_ref)
      when SecurityGroup
        add_target!(:security_groups, target.ems_ref)
      when CloudNetwork
        add_target!(:cloud_networks, target.ems_ref)
      when CloudSubnet
        add_target!(:cloud_subnets, target.ems_ref)
      when FloatingIp
        add_target!(:floating_ips, target.ems_ref)
      when CloudVolume
        add_target!(:cloud_volumes, target.ems_ref)
      when CloudVolumeType
        add_target!(:cloud_volume_types, target.ems_ref)
      end
    end
  end
end