ManageIQ/manageiq

module ManageIQ::Providers
  class PhysicalInfraManager < BaseManager
    include SupportsFeatureMixin

    has_many :physical_chassis,  :foreign_key => "ems_id", :dependent => :destroy, :inverse_of => :ext_management_system
    has_many :physical_racks,    :foreign_key => "ems_id", :dependent => :destroy, :inverse_of => :ext_management_system
    has_many :physical_servers,  :foreign_key => "ems_id", :dependent => :destroy, :inverse_of => :ext_management_system
    has_many :physical_switches, :foreign_key => "ems_id", :dependent => :destroy, :inverse_of => :ext_management_system
    has_many :physical_storages, :foreign_key => "ems_id", :dependent => :destroy, :inverse_of => :ext_management_system

    # Asset details
    has_many :physical_server_details,  :through => :physical_servers,  :source => :asset_detail
    has_many :physical_chassis_details, :through => :physical_chassis,  :source => :asset_detail
    has_many :physical_storage_details, :through => :physical_storages, :source => :asset_detail
    has_many :physical_switch_details,  :through => :physical_switches, :source => :asset_detail

    # Computer systems
    has_many :physical_server_computer_systems,  :through => :physical_servers,  :source => :computer_system
    has_many :physical_chassis_computer_systems, :through => :physical_chassis,  :source => :computer_system
    has_many :physical_storage_computer_systems, :through => :physical_storages, :source => :canister_computer_systems

    # Hardwares
    has_many :physical_server_hardwares,  :through => :physical_server_computer_systems,  :source => :hardware
    has_many :physical_chassis_hardwares, :through => :physical_chassis_computer_systems, :source => :hardware
    has_many :physical_storage_hardwares, :through => :physical_storage_computer_systems, :source => :hardware
    has_many :physical_switch_hardwares,  :through => :physical_switches,                 :source => :hardware

    # Guest devices
    has_many :physical_server_network_devices,     :through => :physical_server_hardwares,  :source => :nics
    has_many :physical_server_storage_adapters,    :through => :physical_server_hardwares,  :source => :storage_adapters
    has_many :physical_server_management_devices,  :through => :physical_server_hardwares,  :source => :management_devices
    has_many :physical_chassis_management_devices, :through => :physical_chassis_hardwares, :source => :management_devices
    has_many :physical_storage_management_devices, :through => :physical_storage_hardwares, :source => :management_devices

    # Firmwares
    has_many :physical_server_firmwares,                 :through => :physical_server_hardwares,        :source => :firmwares
    has_many :physical_server_network_device_firmwares,  :through => :physical_server_network_devices,  :source => :firmwares
    has_many :physical_server_storage_adapter_firmwares, :through => :physical_server_storage_adapters, :source => :firmwares
    has_many :physical_switch_firmwares,                 :through => :physical_switch_hardwares,        :source => :firmwares

    # Network
    has_many :physical_server_networks,  :through => :physical_server_management_devices,  :source => :network
    has_many :physical_chassis_networks, :through => :physical_chassis_management_devices, :source => :network
    has_many :physical_storage_networks, :through => :physical_storage_management_devices, :source => :network
    has_many :physical_switch_networks,  :through => :physical_switch_hardwares,           :source => :networks

    # Physical network ports
    has_many :physical_server_network_ports,  :through => :physical_server_network_devices,     :source => :physical_network_ports
    has_many :physical_switch_network_ports,  :through => :physical_switches,                   :source => :physical_network_ports
    has_many :physical_storage_network_ports, :through => :physical_storage_management_devices, :source => :physical_network_ports

    # Physical disks
    has_many :physical_disks, :through => :physical_storages

    # Canisters
    has_many :canisters, :through => :physical_storages

    virtual_total :total_physical_chassis,  :physical_chassis
    virtual_total :total_physical_racks,    :physical_racks
    virtual_total :total_physical_servers,  :physical_servers
    virtual_total :total_physical_switches, :physical_switches
    virtual_total :total_physical_storages, :physical_storages

    virtual_column :total_hosts, :type => :integer
    virtual_column :total_vms,   :type => :integer

    virtual_column :total_valid, :type => :integer
    virtual_column :total_warning, :type => :integer
    virtual_column :total_critical, :type => :integer
    virtual_column :health_state_info, :type => :json

    virtual_column :total_racks, :type => :integer
    virtual_column :total_resources, :type => :integer
    virtual_column :resources_info, :type => :json

    class << model_name
      define_method(:route_key) { "ems_physical_infras" }
      define_method(:singular_route_key) { "ems_physical_infra" }
    end

    supports :authentication_status

    def self.ems_type
      @ems_type ||= "physical_infra_manager".freeze
    end

    def self.description
      @description ||= "PhysicalInfraManager".freeze
    end

    def count_health_state(state)
      count = 0
      count += physical_chassis.where(:health_state => state).count
      count += physical_servers.where(:health_state => state).count
      count += physical_switches.where(:health_state => state).count
      count + physical_storages.where(:health_state => state).count
    end

    def assign_health_states
      {
        :total_valid    => count_health_state("Valid"),
        :total_warning  => count_health_state("Warning"),
        :total_critical => count_health_state("Critical"),
      }
    end

    alias health_state_info assign_health_states

    def count_resources(component = nil)
      count = 0

      if component
        component.count
      else
        count += physical_racks.count
        count += physical_chassis.count
        count += physical_servers.count
        count += physical_switches.count
        count + physical_storages.count
      end
    end

    def assign_resources_info
      {
        :total_racks     => count_resources(physical_racks),
        :total_resources => count_resources,
      }
    end

    alias resources_info assign_resources_info

    def count_physical_servers_with_host
      physical_servers.inject(0) { |t, physical_server| physical_server.host.nil? ? t : t + 1 }
    end

    alias total_hosts count_physical_servers_with_host

    def count_vms
      physical_servers.inject(0) { |t, physical_server| physical_server.host.nil? ? t : t + physical_server.host.vms.size }
    end

    alias total_vms count_vms

    def self.firmware_update_class
      self::FirmwareUpdateTask
    end

    def self.display_name(number = 1)
      n_('Physical Infrastructure Manager', 'Physical Infrastructure Managers', number)
    end
  end
end