foreman_omaha 0.0.3 → 1.0.0

data/app/models/foreman_omaha/omaha_group.rb

@@ -0,0 +1,34 @@
+module ForemanOmaha
+  class OmahaGroup < ApplicationRecord
+    include Authorizable
+
+    def self.humanize_class_name
+      N_('Omaha Channel')
+    end
+
+    has_many :omaha_facets, :class_name => 'ForemanOmaha::OmahaFacet', :foreign_key => :omaha_group_id,
+                            :inverse_of => :omaha_group, :dependent => :restrict_with_exception
+    has_many :hosts, :class_name => '::Host::Managed', :through => :omaha_facets,
+                     :inverse_of => :omaha_group
+
+    scoped_search :on => :name, :complete_value => :true, :default_order => true
+
+    validates_lengths_from_database
+
+    before_validation :ensure_uuid
+
+    def operatingsystems
+      Coreos.where(:release_name => name.downcase)
+    end
+
+    def latest_operatingsystem
+      operatingsystems.order(:major, :minor).last
+    end
+
+    private
+
+    def ensure_uuid
+      self.uuid ||= SecureRandom.uuid
+    end
+  end
+end

data/app/models/foreman_omaha/omaha_report.rb

@@ -1,7 +1,6 @@
 module ForemanOmaha
   class OmahaReport < ::Report
-    enum :status => [:unknown, :complete, :downloading, :downloaded,
-                     :installed, :instance_hold, :error]
+    enum :status => OmahaFacet::VALID_OMAHA_STATUSES
 
     scoped_search :on => :omaha_version, :rename => :version, :complete_value => true
     scoped_search :on => :status, :complete_value => statuses
@@ -14,25 +13,12 @@ module ForemanOmaha
       'status'
     end
 
-    def to_label
-      case status.to_sym
-      when :complete
-        _('Complete')
-      when :downloading
-        _('Downloading')
-      when :downloaded
-        _('Downloaded')
-      when :installed
-        _('Installed')
-      when :instance_hold
-        _('Instance Hold')
-      when :error
-        _('Error')
-      else
-        _('unknown')
-      end
+    def self.humanize_class_name
+      N_('Omaha Report')
     end
 
+    delegate :to_label, :to_description, :to => :status_mapper
+
     def operatingsystem
       return if omaha_version.blank?
       args = { :type => 'Coreos', :major => osmajor, :minor => osminor }
@@ -41,14 +27,20 @@ module ForemanOmaha
 
     def osmajor
       omaha_version.gsub(/^(\d+)\.\d\.\d$/, '\1')
-    rescue
+    rescue StandardError
       nil
     end
 
     def osminor
       omaha_version.gsub(/^\d+\.(\d\.\d)$/, '\1')
-    rescue
+    rescue StandardError
       nil
     end
+
+    private
+
+    def status_mapper
+      StatusMapper.new(status)
+    end
   end
 end

data/app/models/host_status/omaha_status.rb

@@ -1,27 +1,17 @@
 module HostStatus
   class OmahaStatus < HostStatus::Status
-    def last_report
-      self.last_report = host.last_omaha_report_object unless @last_report_set
-      @last_report
-    end
-
-    def last_report=(report)
-      @last_report_set = true
-      @last_report = report
-    end
-
     def self.status_name
       N_('Omaha Status')
     end
 
     def to_status(_options = {})
-      return ::ForemanOmaha::OmahaReport.statuses[:unknown] unless relevant?
-      ::ForemanOmaha::OmahaReport.statuses[last_report.status]
+      return ::ForemanOmaha::OmahaFacet.statuses[:unknown] unless relevant?
+      ::ForemanOmaha::OmahaFacet.statuses[host.omaha_facet.status]
     end
 
     def to_global(_options = {})
-      return ::ForemanOmaha::OmahaReport.statuses[:unknown] unless relevant?
-      case last_report.status.to_sym
+      return ::ForemanOmaha::OmahaFacet.statuses[:unknown] unless relevant?
+      case host.omaha_facet.status.to_sym
       when :complete, :downloaded, :downloading, :installed
         HostStatus::Global::OK
       when :instance_hold
@@ -34,11 +24,11 @@ module HostStatus
     end
 
     def to_label(_options = {})
-      last_report.to_label
+      host.omaha_facet.to_status_label
     end
 
     def relevant?(_options = {})
-      host && last_report.present?
+      host && !!host.omaha_facet
     end
   end
 end

data/app/services/foreman_omaha/charts/oem_distribution.rb

@@ -0,0 +1,24 @@
+module ForemanOmaha
+  module Charts
+    class OemDistribution
+      attr_accessor :hosts
+
+      def initialize(hosts = nil)
+        @hosts = hosts || Host.authorized(:view_hosts, Host).joins(:omaha_facet)
+      end
+
+      def query
+        @query ||= hosts.group(:oem).count
+      end
+
+      def to_chart_data
+        query.map do |oem, count|
+          {
+            :label => oem,
+            :data => count
+          }
+        end
+      end
+    end
+  end
+end

data/app/services/foreman_omaha/charts/status_distribution.rb

@@ -0,0 +1,57 @@
+module ForemanOmaha
+  module Charts
+    class StatusDistribution
+      attr_accessor :hosts
+
+      def initialize(hosts = nil)
+        @hosts = hosts || Host.authorized(:view_hosts, Host).joins(:omaha_facet)
+      end
+
+      def query
+        @query ||= hosts.group(:status).count.transform_keys { |k| ForemanOmaha::OmahaFacet.statuses.key(k).to_sym }
+      end
+
+      def to_chart_data
+        query.map do |status, count|
+          {
+            :label => status_label(status),
+            :data => count,
+            :color => status_color(status)
+          }
+        end
+      end
+
+      def to_a
+        query.map do |status, count|
+          {
+            :status => status,
+            :label => status_label(status),
+            :count => count
+          }
+        end
+      end
+
+      private
+
+      def color_map
+        {
+          :unknown => '#92A8CD',
+          :complete => '#89A54E',
+          :downloading => '#3D96AE',
+          :downloaded => '#4572A7',
+          :installed => '#DB843D',
+          :instance_hold => '#80699B',
+          :error => '#AA4643'
+        }
+      end
+
+      def status_color(status)
+        color_map[status]
+      end
+
+      def status_label(status)
+        ForemanOmaha::StatusMapper.new(status).to_label
+      end
+    end
+  end
+end

data/app/services/foreman_omaha/charts/version_distribution.rb

@@ -0,0 +1,24 @@
+module ForemanOmaha
+  module Charts
+    class VersionDistribution
+      attr_accessor :hosts
+
+      def initialize(hosts = nil)
+        @hosts = hosts || Host.authorized(:view_hosts, Host).joins(:omaha_facet)
+      end
+
+      def query
+        hosts.group(:version).count
+      end
+
+      def to_chart_data
+        query.map do |version, count|
+          {
+            :label => version,
+            :data => count
+          }
+        end
+      end
+    end
+  end
+end

data/app/services/foreman_omaha/container_linux_config_transpiler.rb

@@ -0,0 +1,33 @@
+require 'open3'
+
+module ForemanOmaha
+  class ContainerLinuxConfigTranspiler
+    class TranspilerNotFound < StandardError; end
+    class TranspileError < StandardError; end
+
+    attr_accessor :input, :output, :errors, :status
+
+    def initialize(input)
+      self.input = input
+    end
+
+    def run!
+      check_transpiler
+      transpile
+      raise TranspileError, "Could not transpile container linux config to ignition: #{errors}" unless status.success?
+      output
+    end
+
+    def transpile
+      self.output, self.errors, self.status = Open3.capture3(cmd, :stdin_data => input)
+    end
+
+    def check_transpiler
+      raise TranspilerNotFound, 'The ct binary was not found in your PATH.' if cmd.blank?
+    end
+
+    def cmd
+      @cmd ||= Foreman::Util.which('ct')
+    end
+  end
+end

data/app/services/foreman_omaha/fact_parser.rb

@@ -3,9 +3,9 @@ module ForemanOmaha
     def operatingsystem
       args = { :name => facts['platform'], :major => facts['osmajor'], :minor => facts['osminor'] }
       description = "#{facts['platform']} #{facts['version']}"
-      Operatingsystem.where(args).first ||
-        Operatingsystem.create!(args.merge(:description => description,
-                                           :release_name => facts['track']))
+      Coreos.where(args).first ||
+        create_coreos_version(args.merge(:description => description,
+                                         :release_name => facts['track']))
     end
 
     def architecture
@@ -28,5 +28,21 @@ module ForemanOmaha
     def support_interfaces_parsing?
       false
     end
+
+    private
+
+    def create_coreos_version(attrs)
+      previous_version = previous_coreos_version
+      return Coreos.create!(attrs) unless previous_coreos_version
+      os = previous_version.deep_clone(
+        :include => [:ptables, :media, :os_default_templates, :architectures]
+      )
+      os.update_attributes(attrs)
+      os
+    end
+
+    def previous_coreos_version
+      Coreos.all.sort_by { |os| Gem::Version.new(os.release) }.last
+    end
   end
 end

data/app/services/foreman_omaha/group_version_breakdown.rb

@@ -0,0 +1,30 @@
+module ForemanOmaha
+  class GroupVersionBreakdown
+    attr_accessor :omaha_group
+
+    def initialize(opts = {})
+      self.omaha_group = opts.fetch(:omaha_group)
+    end
+
+    def version_breakdown
+      facets.group(:version).count.sort_by { |version, _| Gem::Version.new(version) }.reverse.map do |version, count|
+        {
+          :version => version,
+          :count => count,
+          :percentage => percentage(count)
+        }
+      end
+    end
+
+    def facets
+      ForemanOmaha::OmahaFacet.where(:omaha_group => omaha_group)
+    end
+
+    private
+
+    def percentage(count)
+      return 0 if count.zero? || facets.count.zero?
+      (count.to_f * 100 / facets.count).round(2)
+    end
+  end
+end

data/app/services/foreman_omaha/omaha_report_importer.rb

@@ -10,19 +10,68 @@ module ForemanOmaha
 
     private
 
+    def host
+      @host ||= ForemanOmaha::OmahaFacet.find_by(:machineid => machineid).try(:host) if machineid.present?
+      @name ||= @host.try(:name)
+      @host || super
+    end
+
     def create_report_and_logs
       super
-      @report.omaha_version = omaha_version
-      @report.save
-      @report
+      return report unless report.persisted?
+      report.omaha_version = omaha_version
+      report.save
+      update_omaha_facet!
+      report
+    end
+
+    def omaha_facet
+      host.omaha_facet || host.build_omaha_facet
+    end
+
+    def update_omaha_facet!
+      return unless omaha_facet.last_report.nil? || omaha_facet.last_report.utc < time
+      omaha_facet.update(
+        :last_report => time,
+        :status => report_status,
+        :machineid => machineid,
+        :oem => oem,
+        :version => omaha_version,
+        :omaha_group => omaha_group
+      )
     end
 
     def report_status
       raw['status']
     end
 
+    def machineid
+      raw['machineid']
+    end
+
+    def oem
+      raw['oem']
+    end
+
     def omaha_version
       raw['omaha_version'] || 'unknown'
     end
+
+    def omaha_group
+      ForemanOmaha::OmahaGroup.find_by(:uuid => raw['omaha_group']) || find_omaha_group_by_version
+    end
+
+    # If the Report does not contain information
+    # about the Omaha Group, we try to find
+    # the correct group by looking at the defined
+    # operating systems
+    def find_omaha_group_by_version
+      version = Gem::Version.new(omaha_version)
+      major = version.segments.first
+      minor = version.segments.last(2).join('.')
+      os = Coreos.find_by(:major => major, :minor => minor)
+      return unless os
+      ForemanOmaha::OmahaGroup.find_by(:uuid => os.release_name)
+    end
   end
 end

data/app/services/foreman_omaha/renderer_methods.rb

@@ -0,0 +1,7 @@
+module ForemanOmaha
+  module RendererMethods
+    def transpile_container_linux_config(input)
+      ForemanOmaha::ContainerLinuxConfigTranspiler.new(input).run!
+    end
+  end
+end

data/app/services/foreman_omaha/status_mapper.rb

@@ -0,0 +1,53 @@
+module ForemanOmaha
+  class StatusMapper
+    attr_accessor :status
+
+    def initialize(status)
+      @status = status
+    end
+
+    def to_label
+      case status.to_sym
+      when :complete
+        N_('Complete')
+      when :downloading
+        N_('Downloading')
+      when :downloaded
+        N_('Downloaded')
+      when :installed
+        N_('Installed')
+      when :instance_hold
+        N_('Instance Hold')
+      when :error
+        N_('Error')
+      else
+        N_('unknown')
+      end
+    end
+
+    # rubocop:disable Metrics/CyclomaticComplexity
+    def to_description(version = nil)
+      case status.to_sym
+      when :complete
+        if version.present?
+          _('The host has been updated successfully and is now running version %s.') % version
+        else
+          _('The host has been updated successfully.')
+        end
+      when :downloading
+        _('The host has just started downloading the update package.')
+      when :downloaded
+        _('The host has downloaded the update package and will install it now.')
+      when :installed
+        _('The host has installed the update package but is not using it yet.')
+      when :instance_hold
+        _('An update for this host is pending but it was put on hold because of the rollout policy.')
+      when :error
+        _('The host reported an error while updating.')
+      else
+        _('The status of this host is unknown.')
+      end
+    end
+    # rubocop:enable Metrics/CyclomaticComplexity
+  end
+end