ManageIQ/manageiq

module ManageIQ::Providers
  class Inventory::Persister
    class Builder
      class InfraManager < ::ManageIQ::Providers::Inventory::Persister::Builder
        def networks
          add_properties(
            :manager_ref                  => %i[hardware ipaddress ipv6address],
            :parent_inventory_collections => %i[vms miq_templates]
          )
        end

        def host_networks
          add_properties(
            :model_class                  => ::Network,
            :manager_ref                  => %i[hardware ipaddress],
            :parent_inventory_collections => %i[hosts]
          )
        end

        def guest_devices
          add_properties(
            :manager_ref                  => %i[hardware uid_ems],
            :parent_inventory_collections => %i[vms miq_templates]
          )
        end

        def host_guest_devices
          add_properties(
            :model_class                  => ::GuestDevice,
            :manager_ref                  => %i[hardware uid_ems],
            :parent_inventory_collections => %i[hosts]
          )
        end

        def host_hardwares
          add_properties(
            :model_class                  => ::Hardware,
            :manager_ref                  => %i[host],
            :parent_inventory_collections => %i[hosts]
          )
        end

        def host_disks
          add_properties(
            :model_class                  => ::Disk,
            :manager_ref                  => %i[hardware device_name],
            :parent_inventory_collections => %i[hosts]
          )
        end

        def host_system_services
          add_properties(
            :model_class                  => ::SystemService,
            :manager_ref                  => %i[host name],
            :parent_inventory_collections => %i[hosts]
          )
        end

        def host_operating_systems
          add_properties(
            :model_class                  => ::OperatingSystem,
            :manager_ref                  => %i[host],
            :parent_inventory_collections => %i[hosts]
          )
        end

        def ems_custom_attributes
          add_properties(
            :model_class                  => ::CustomAttribute,
            :manager_ref                  => %i[resource name],
            :parent_inventory_collections => %i[vms miq_templates]
          )
        end

        def vm_and_template_ems_custom_fields
          skip_auto_inventory_attributes

          add_properties(
            :model_class                  => ::CustomAttribute,
            :manager_ref                  => %i[resource name],
            :parent_inventory_collections => %i[vms miq_templates]
          )

          add_inventory_attributes(%i[section name value source resource])
        end

        def ems_folders
          skip_sti

          add_properties(
            :manager_ref          => %i[uid_ems],
            :attributes_blacklist => %i[parent]
          )
          add_common_default_values
        end

        def datacenters
          add_properties(:attributes_blacklist => %i[parent])
          add_common_default_values
        end

        def clusters
          add_properties(
            :assocation           => :ems_clusters,
            :attributes_blacklist => %i[datacenter_id parent]
          )
          add_inventory_attributes(%i[datacenter_id])
          add_common_default_values
        end

        def storages
          add_properties(:attributes_blacklist => %i[parent])
          add_common_default_values
        end

        def hosts
          add_properties(
            :attributes_blacklist   => %i[parent],
            :delete_method          => :disconnect_inv,
            :custom_reconnect_block => INVENTORY_RECONNECT_BLOCK
          )
          add_common_default_values
        end

        def vms
          super
          vm_template_infra_shared_properties
        end

        def miq_templates
          super
          vm_template_infra_shared_properties
        end

        def host_storages
          add_properties(
            :manager_ref                  => %i[host storage],
            :parent_inventory_collections => %i[hosts]
          )
        end

        def host_switches
          add_properties(
            :manager_ref                  => %i[host switch],
            :parent_inventory_collections => %i[hosts]
          )
        end

        def host_virtual_switches
          add_properties(
            :manager_ref                  => %i[host uid_ems],
            :model_class                  => Switch,
            :parent_inventory_collections => %i[hosts]
          )
        end

        def distributed_virtual_switches
          add_properties(
            :manager_ref          => %i[uid_ems],
            :attributes_blacklist => %i[parent],
            :secondary_refs       => {:by_switch_uuid => %i[switch_uuid]}
          )
          add_common_default_values
        end

        def lans
          add_properties(
            :manager_ref                  => %i[switch uid_ems],
            :parent_inventory_collections => %i[hosts]
          )
        end

        def distributed_virtual_lans
          add_properties(
            :model_class                  => Lan,
            :manager_ref                  => %i[switch uid_ems],
            :parent_inventory_collections => %i[distributed_virtual_switches]
          )
        end

        def host_virtual_lans
          add_properties(
            :model_class                  => Lan,
            :manager_ref                  => %i[switch uid_ems],
            :parent_inventory_collections => %i[hosts]
          )
        end

        def subnets
          add_properties(
            :manager_ref                  => %i[lan ems_ref],
            :parent_inventory_collections => %i[hosts]
          )
        end

        def customization_specs
          add_properties(:manager_ref => %i[name])

          add_common_default_values
        end

        def miq_scsi_luns
          add_properties(
            :manager_ref                  => %i[miq_scsi_target uid_ems],
            :parent_inventory_collections => %i[hosts]
          )
        end

        def miq_scsi_targets
          add_properties(
            :manager_ref                  => %i[guest_device uid_ems],
            :parent_inventory_collections => %i[hosts]
          )
        end

        def storage_profiles
          add_common_default_values
        end

        def storage_profile_storages
          add_properties(
            :manager_ref                  => %i[storage_profile storage],
            :parent_inventory_collections => %i[storage_profiles]
          )
        end

        def ems_extensions
          add_common_default_values
        end

        def ems_licenses
          add_common_default_values
        end

        def orchestration_templates
          add_common_default_values
        end

        def root_folder_relationship
          skip_auto_inventory_attributes
          skip_model_class

          add_properties(
            :custom_save_block => root_folder_relationship_save_block
          )

          add_dependency_attributes(
            :ems_folders => [persister.collections[:ems_folders]]
          )
        end

        def parent_blue_folders
          skip_auto_inventory_attributes
          skip_model_class

          add_properties(
            :custom_save_block => relationship_save_block(:relationship_key => :parent)
          )

          dependency_collections = %i[clusters ems_folders datacenters hosts resource_pools storages]
          dependency_attributes = dependency_collections.index_with { |collection| [persister.collections[collection]].compact }
          add_dependency_attributes(dependency_attributes)
        end

        def vm_parent_blue_folders
          skip_auto_inventory_attributes
          skip_model_class

          add_properties(
            :custom_save_block => relationship_save_block(:relationship_key => :parent, :parent_type => "EmsFolder")
          )

          add_dependency_attributes(:vms =>persister.collections.values_at(:vms, :miq_templates, :vms_and_templates).compact)
        end

        def iso_images
          add_properties(
            :parent_inventory_collections => %i[storages],
            :manager_ref                  => %i[name],
            :model_class                  => IsoImage
          )
        end

        private

        def root_folder_relationship_save_block
          lambda do |ems, inventory_collection|
            folder_inv_collection = inventory_collection.dependency_attributes[:ems_folders]&.first
            return if folder_inv_collection.nil?

            # All folders must have a parent except for the root folder
            root_folder_obj = folder_inv_collection.data.detect { |obj| obj.data[:parent].nil? }
            return if root_folder_obj.nil?

            root_folder = folder_inv_collection.model_class.find(root_folder_obj.id)
            root_folder.with_relationship_type(:ems_metadata) { root_folder.parent = ems }
          end
        end

        def vm_template_infra_shared_properties
          add_inventory_attributes(%i[resource_pool])
        end
      end
    end
  end
end