Plugins » Katello » Hammer CLI CSV

module HammerCLICsv

  class SystemsCommand < BaseCommand

    ORGANIZATION = 'Organization'

    ENVIRONMENT = 'Environment'

    CONTENTVIEW = 'Content View'

    def export

      CSV.open(option_csv_file || '/dev/stdout', 'wb', {:force_quotes => false}) do |csv|

        csv << [NAME, COUNT, ORGANIZATION, ENVIRONMENT, CONTENTVIEW, SYSTEMGROUPS, VIRTUAL, HOST,

        @api.resource(:organizations).call(:index, {:per_page => 999999})['results'].each do |organization|

          @api.resource(:systems).call(:index, {

                                'per_page' => 999999,

              column << system['systemGroups'].collect do |systemgroup|

                systemgroup['name']

              end

            virtual = system['facts']['virt.is_guest'] == 'true' ? 'Yes' : 'No'

            host = system['host']

            operatingsystem = "#{system['facts']['distribution.name']} " if system['facts']['distribution.name']

            sockets = system['facts']['cpu.cpu_socket(s)']

            ram = system['facts']['memory.memtotal']

            cores = system['facts']['cpu.core(s)_per_socket']

            products = CSV.generate do |column|

              column << system['installedProducts'].collect do |product|

                "#{product['productId']}|#{product['productName']}"

              end

            subscriptions = CSV.generate do |column|

              column << @api.resource(:subscriptions).call(:index, {

                                                    'system_id' => system['uuid']

                                                  })['results'].collect do |subscription|

                "#{subscription['product_id']}|#{subscription['product_name']}"

              end

            csv << [name, count, organization_name, environment, contentview, systemgroups, virtual, host,

                    operatingsystem, architecture, sockets, ram, cores, sla, products, subscriptions]

          end

        create_systems_from_csv(line)

      end

      @host_guests.each do |host_id, guest_ids|

        @api.resource(:systems).call(:update, {

                               'id' => host_id,

                               'guest_ids' => guest_ids

                             })

      end

    end

    def create_systems_from_csv(line)

                                                     'organization_id' => katello_organization(:name => line[ORGANIZATION]),

                                                     'environment_id' => katello_environment(line[ORGANIZATION], :name => line[ENVIRONMENT]),

                                                     'content_view_id' => katello_contentview(line[ORGANIZATION], :name => line[CONTENTVIEW]),

                                                     'installed_products' => products(line),

                                                     'type' => 'system'

                                                   })['uuid']

                                                       'name' => name,

                                                       'environment_id' => katello_environment(line[ORGANIZATION], :name => line[ENVIRONMENT]),

                                                       'content_view_id' => katello_contentview(line[ORGANIZATION], :name => line[CONTENTVIEW]),

                                                       'installed_products' => products(line)

                                                     }

                                                   })['uuid']

        set_system_groups(system_id, line)

      end

    rescue RuntimeError => e

      raise "#{e}\n       #{line}"

    private

      facts = {}

      facts['cpu.core(s)_per_socket'] = line[CORES]

      facts['cpu.cpu_socket(s)'] = line[SOCKETS]

        (facts['distribution.name'], facts['distribution.version']) = ['RHEL', line[OPERATINGSYSTEM]]

      end

      facts['virt.is_guest'] = line[VIRTUAL] == 'Yes' ? true : false

      facts

    end

    end

    def products(line)

      products = CSV.parse_line(line[PRODUCTS]).collect do |product_details|

        product = {}

        # TODO: these get passed straight through to candlepin; probably would be better to process in server

    end

    def subscriptions(line)

      subscriptions = CSV.parse_line(line[SUBSCRIPTIONS]).collect do |subscription_details|

        subscription = {}

        (subscription[:number], subscription[:name]) = subscription_details.split('|')

      end

      subscriptions

    end

  end

end