class ForemanInventoryUpload::Generators::Slice

# File lib/foreman_inventory_upload/generators/slice.rb, line 12
def initialize(hosts, output = [], slice_id = Foreman.uuid)
  @stream = JsonStream.new(output)
  @hosts = hosts
  @slice_id = slice_id
  @hosts_count = 0
end

# File lib/foreman_inventory_upload/generators/slice.rb, line 19
def render
  report_slice(@hosts)
  @stream.out
end

# File lib/foreman_inventory_upload/generators/slice.rb, line 53
def insights_minimal_data_collection(host)
  @stream.simple_field('account', account_id(host.organization).to_s)
  @stream.simple_field('subscription_manager_id', uuid_value!(host.subscription_facet&.uuid))
  @stream.simple_field('insights_id', uuid_value(fact_value(host, 'insights_id')))
  @stream.simple_field('bios_uuid', bios_uuid(host))
  @stream.simple_field('bios_vendor', fact_value(host, 'dmi::bios::vendor'))
  @stream.simple_field('bios_version', fact_value(host, 'dmi::bios::version'))
  @stream.simple_field('arch', host.architecture&.name)
  @stream.simple_field(
    'infrastructure_type',
    ActiveModel::Type::Boolean.new.cast(fact_value(host, 'virt::is_guest')) ? 'virtual' : 'physical'
  )
  @stream.object_field('system_profile') do
    unless (installed_products = host.subscription_facet&.installed_products).empty?
      @stream.array_field('installed_products') do
        @stream.raw(installed_products.map do |product|
          {
            'name': product.name,
            'id': product.cp_product_id,
          }.to_json
        end.join(', '))
      end
    end
    @stream.simple_field('cores_per_socket', fact_value(host, 'cpu::core(s)_per_socket')) { |v| v.to_i } if fact_value(host, 'cpu::core(s)_per_socket').present?
    @stream.simple_field('system_memory_bytes', fact_value(host, 'memory::memtotal')) { |v| kilobytes_to_bytes(v.to_i) } if fact_value(host, 'memory::memtotal').present?
    @stream.simple_field('number_of_cpus', fact_value(host, 'cpu::cpu(s)')) { |v| v.to_i } if fact_value(host, 'cpu::cpu(s)').present?
    @stream.simple_field('number_of_sockets', fact_value(host, 'cpu::cpu_socket(s)'), :last) { |v| v.to_i } if fact_value(host, 'cpu::cpu_socket(s)').present?
  end
  if host.subscription_facet.hypervisor?
    @stream.simple_field('cpu_socket(s)', fact_value(host, 'cpu::cpu_socket(s)'))
    @stream.simple_field('hypervisor_type', fact_value(host, 'hypervisor::type'))
    @stream.simple_field('hypervisor_version', fact_value(host, 'hypervisor::version'), :last)
  else
    @stream.simple_field('cpu_socket(s)', fact_value(host, 'cpu::cpu_socket(s)'), :last)
  end
end

# File lib/foreman_inventory_upload/generators/slice.rb, line 250
def os_release_value(name, version, codename)
  "#{name} #{version} (#{codename})"
end

# File lib/foreman_inventory_upload/generators/slice.rb, line 43
def report_conversions(host)
  @stream.simple_field('convert2rhel_through_foreman', host.subscription_facet&.convert2rhel_through_foreman)
  @stream.simple_field('activity', fact_value(host, 'conversions::activity'))
  @stream.simple_field('packages_0_nevra', fact_value(host, 'conversions::packages::0::nevra'))
  @stream.simple_field('packages_0_signature', fact_value(host, 'conversions::packages::0::signature'))
  @stream.simple_field('activity_started', fact_value(host, 'conversions::activity_started'))
  @stream.simple_field('activity_ended', fact_value(host, 'conversions::activity_ended'))
  @stream.simple_field('success', ActiveModel::Type::Boolean.new.cast(fact_value(host, 'conversions::success')), :last)
end

# File lib/foreman_inventory_upload/generators/slice.rb, line 90
def report_host(host)
  host_ips_cache = host_ips(host)
  @stream.object do
    if Setting[:insights_minimal_data_collection] && !ForemanRhCloud.with_iop_smart_proxy?
      insights_minimal_data_collection(host)
    else
      @stream.simple_field('fqdn', fqdn(host))
      @stream.simple_field('account', account_id(host.organization).to_s)
      @stream.simple_field('subscription_manager_id', uuid_value!(host.subscription_facet&.uuid))
      @stream.simple_field('satellite_id', uuid_value!(host.subscription_facet&.uuid))
      if host.subscription_facet&.convert2rhel_through_foreman.present?
        @stream.object_field('conversions') do
          @stream.object_field('source_os') do
            @stream.simple_field('name', fact_value(host, 'conversions::source_os::name'))
            @stream.simple_field('version', fact_value(host, 'conversions::source_os::version') || 'Unknown - Fact not reported', :last)
          end
          @stream.object_field('target_os') do
            @stream.simple_field('name', fact_value(host, 'conversions::target_os::name'))
            @stream.simple_field('version', fact_value(host, 'conversions::target_os::version') || 'Unknown - Fact not reported', :last)
          end
          report_conversions(host)
        end
      end
      @stream.simple_field('bios_uuid', bios_uuid(host))
      @stream.simple_field('vm_uuid', uuid_value(fact_value(host, 'virt::uuid')))
      @stream.simple_field('insights_id', uuid_value(fact_value(host, 'insights_id')))
      report_ip_addresses(host, host_ips_cache)
      report_mac_addresses(host)
      @stream.object_field('system_profile') do
        report_system_profile(host, host_ips_cache)
      end

      @stream.array_field('facts') do
        @stream.object do
          @stream.simple_field('namespace', SATELLITE_NAMESPACE)
          @stream.object_field('facts', :last) do
            report_satellite_facts(host)
          end
        end
      end

      @stream.array_field('tags', :last) do
        tags_generator = Tags.new(host)
        host_params_tags = tags_generator.generate_parameters
        host_params_tags.each do |key, value|
          report_tag(SATELLITE_PARAMS_NAMESPACE, key, value)
        end

        tags = tags_generator.generate
        last_index = tags.count - 1
        tags.each_with_index do |pair, index|
          key, value = pair
          report_tag(SATELLITE_NAMESPACE, key, value, index == last_index)
        end
      end
    end
  end
end

# File lib/foreman_inventory_upload/generators/slice.rb, line 238
def report_ip_addresses(host, host_ips_cache)
  ip_addresses = host.interfaces.map { |nic| host_ips_cache[nic.ip] }.compact

  @stream.string_array_value('ip_addresses', ip_addresses)
end

# File lib/foreman_inventory_upload/generators/slice.rb, line 244
def report_mac_addresses(host)
  macs = host.interfaces.map { |nic| nic.mac }.compact

  @stream.string_array_value('mac_addresses', macs)
end

# File lib/foreman_inventory_upload/generators/slice.rb, line 222
def report_satellite_facts(host)
  @stream.simple_field('virtual_host_name', host.subscription_facet&.hypervisor_host&.name)
  @stream.simple_field('virtual_host_uuid', host.subscription_facet&.hypervisor_host&.subscription_facet&.uuid)
  if defined?(ForemanThemeSatellite)
    @stream.simple_field('satellite_version', ForemanThemeSatellite::SATELLITE_VERSION)
  end
  @stream.simple_field('system_purpose_usage', host.subscription_facet.purpose_usage)
  @stream.simple_field('system_purpose_role', host.subscription_facet.purpose_role)
  @stream.simple_field('system_purpose_sla', host.subscription_facet.service_level)
  @stream.simple_field('distribution_version', fact_value(host, 'distribution::version'))
  @stream.simple_field('satellite_instance_id', Foreman.try(:instance_id))
  @stream.simple_field('is_simple_content_access', true)
  @stream.simple_field('is_hostname_obfuscated', !!obfuscate_hostname?(host))
  @stream.simple_field('organization_id', host.organization_id, :last)
end

# File lib/foreman_inventory_upload/generators/slice.rb, line 26
def report_slice(hosts_batch)
  @stream.object do
    @stream.simple_field('report_slice_id', uuid_value!(@slice_id))
    @stream.array_field('hosts', :last) do
      first = true
      hosts_batch.each do |host|
        next unless host&.subscription_facet
        @stream.comma unless first
        if report_host(host)
          first = false
          @hosts_count += 1
        end
      end
    end
  end
end

# File lib/foreman_inventory_upload/generators/slice.rb, line 158
def report_system_profile(host, host_ips_cache)
  @stream.simple_field('number_of_cpus', fact_value(host, 'cpu::cpu(s)')) { |v| v.to_i }
  @stream.simple_field('number_of_sockets', fact_value(host, 'cpu::cpu_socket(s)')) { |v| v.to_i }
  @stream.simple_field('cores_per_socket', fact_value(host, 'cpu::core(s)_per_socket')) { |v| v.to_i }
  @stream.simple_field('system_memory_bytes', fact_value(host, 'memory::memtotal')) { |v| kilobytes_to_bytes(v.to_i) }
  @stream.array_field('network_interfaces') do
    @stream.raw(host.interfaces.reject { |nic| nic.identifier.empty? }.map do |nic|
      {
        'ipv4_addresses': [host_ips_cache[nic.ip]].reject(&:empty?),
        'ipv6_addresses': [nic.ip6].reject(&:empty?),
        'mtu': nic.try(:mtu) && nic.mtu.to_i,
        'mac_address': nic.mac,
        'name': nic.identifier,
      }.compact.to_json
    end.join(', '))
  end
  @stream.simple_field('bios_vendor', fact_value(host, 'dmi::bios::vendor'))
  @stream.simple_field('bios_version', fact_value(host, 'dmi::bios::version'))
  @stream.simple_field('bios_release_date', fact_value(host, 'dmi::bios::relase_date'))
  if (cpu_flags = fact_value(host, 'lscpu::flags'))
    @stream.array_field('cpu_flags') do
      @stream.raw(cpu_flags.split.map do |flag|
        @stream.stringify_value(flag)
      end.join(', '))
    end
  end
  @stream.simple_field(
    'os_release', [
      fact_value(host, 'distribution::name'),
      fact_value(host, 'distribution::version'),
      fact_value(host, 'distribution::id'),
    ]
  ) { |v| os_release_value(*v) }
  @stream.simple_field('os_kernel_version', fact_value(host, 'uname::release'))
  @stream.simple_field('arch', host.architecture&.name)
  @stream.simple_field(
    'infrastructure_type',
    ActiveModel::Type::Boolean.new.cast(fact_value(host, 'virt::is_guest')) ? 'virtual' : 'physical'
  )
  @stream.simple_field('cloud_provider', cloud_provider(host))
  unless (installed_products = host.subscription_facet&.installed_products).empty?
    @stream.array_field('installed_products') do
      @stream.raw(installed_products.map do |product|
        {
          'name': product.name,
          'id': product.cp_product_id,
        }.to_json
      end.join(', '))
    end
  end
  if ForemanRhCloud.with_iop_smart_proxy? || (!Setting[:insights_minimal_data_collection] && !Setting[:exclude_installed_packages])
    @stream.array_field('installed_packages') do
      first = true
      host.installed_packages.each do |package|
        @stream.raw("#{first ? '' : ', '}#{@stream.stringify_value(package.nvrea)}")
        first = false
      end
    end

    report_yum_repos(host)
  end
  @stream.simple_field('satellite_managed', true, :last)
end

# File lib/foreman_inventory_upload/generators/slice.rb, line 149
def report_tag(namespace, key, value, last = nil)
  @stream.object do
    @stream.simple_field('namespace', namespace)
    @stream.simple_field('key', key)
    @stream.simple_field('value', value.to_s, :last)
  end
  @stream.comma unless last
end

# File lib/foreman_inventory_upload/generators/slice.rb, line 265
def report_yum_repo(url_prefix, repo)
  @stream.object do
    @stream.simple_field('name', repo.content.name)
    @stream.simple_field('id', repo.content.label)
    @stream.simple_field('base_url', "#{url_prefix}/#{repo.generate_repo_path(repo.content.content_url)}")
    @stream.simple_field('enabled', true)
    @stream.simple_field('gpgcheck', true, :last)
  end
end

# File lib/foreman_inventory_upload/generators/slice.rb, line 254
def report_yum_repos(host)
  return unless host&.content_facet&.bound_repositories&.any?

  @stream.array_field('yum_repos') do
    host.content_facet.bound_repositories.each_with_index do |repo, index|
      report_yum_repo(host.content_source&.load_balancer_pulp_content_url || ::SmartProxy.pulp_primary.pulp_content_url, repo)
      @stream.comma unless index == host.content_facet.bound_repositories.count - 1
    end
  end
end