foreman_omaha 0.0.1

data/Rakefile

@@ -0,0 +1,47 @@
+#!/usr/bin/env rake
+begin
+  require 'bundler/setup'
+rescue LoadError
+  puts 'You must `gem install bundler` and `bundle install` to run rake tasks'
+end
+begin
+  require 'rdoc/task'
+rescue LoadError
+  require 'rdoc/rdoc'
+  require 'rake/rdoctask'
+  RDoc::Task = Rake::RDocTask
+end
+
+RDoc::Task.new(:rdoc) do |rdoc|
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title    = 'ForemanOmaha'
+  rdoc.options << '--line-numbers'
+  rdoc.rdoc_files.include('README.rdoc')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+
+APP_RAKEFILE = File.expand_path('../test/dummy/Rakefile', __FILE__)
+
+Bundler::GemHelper.install_tasks
+
+require 'rake/testtask'
+
+Rake::TestTask.new(:test) do |t|
+  t.libs << 'lib'
+  t.libs << 'test'
+  t.pattern = 'test/**/*_test.rb'
+  t.verbose = false
+end
+
+task default: :test
+
+begin
+  require 'rubocop/rake_task'
+  RuboCop::RakeTask.new
+rescue => _
+  puts 'Rubocop not loaded.'
+end
+
+task :default do
+  Rake::Task['rubocop'].execute
+end

data/app/controllers/api/v2/omaha_reports_controller.rb

@@ -0,0 +1,86 @@
+module Api
+  module V2
+    class OmahaReportsController < V2::BaseController
+      include Api::Version2
+      include Foreman::Controller::SmartProxyAuth
+
+      before_action :find_resource, :only => [:show, :destroy]
+      before_action :setup_search_options, :only => [:index, :last]
+
+      add_smart_proxy_filters :create,
+                              :features => proc { ::ForemanOmaha::OmahaReportImporter.authorized_smart_proxy_features }
+
+      def_param_group :omaha_report do
+        param :report, Hash, :required => true, :action_aware => true do
+          param :host, String, :required => true, :desc => N_('Hostname or certname')
+          param :status, String,
+                :required => true,
+                :desc => N_('Omaha status, can be one of %s') % ::ForemanOmaha::OmahaReport.statuses.keys.to_sentence
+          param :omaha_version, String, :required => true, :desc => N_('Omaha OS version')
+        end
+      end
+
+      api :GET, '/omaha_reports/', N_('List all omaha reports')
+      param_group :search_and_pagination, ::Api::V2::BaseController
+
+      def index
+        @omaha_reports = resource_scope_for_index.my_reports
+        @total = resource_scope_for_index.my_reports.count
+      end
+
+      api :GET, '/omaha_reports/:id/', N_('Show a omaha report')
+      param :id, :identifier, :required => true
+
+      def show
+      end
+
+      api :POST, '/omaha_reports', N_('Create a omaha report')
+      param_group :omaha_report, :as => :create
+
+      def create
+        @omaha_report = resource_class.import(params[:omaha_report], detected_proxy.try(:id))
+        process_response @omaha_report.errors.empty?
+      rescue ::Foreman::Exception => e
+        render_message(e.to_s, :status => :unprocessable_entity)
+      end
+
+      api :DELETE, '/omaha_reports/:id/', N_('Delete a omaha report')
+      param :id, String, :required => true
+
+      def destroy
+        process_response @omaha_report.destroy
+      end
+
+      api :GET, '/hosts/:host_id/omaha_reports/last', N_('Show the last report for a host')
+      param :id, :identifier, :required => true
+
+      def last
+        if params[:host_id].present?
+          conditions = { :host_id => resource_finder(Host.authorized(:view_hosts), params[:host_id]).try(:id) }
+        end
+        max_id = resource_scope.where(conditions).maximum(:id)
+        @omaha_report = resource_scope.find(max_id)
+        render :show
+      end
+
+      private
+
+      def resource_class
+        ::ForemanOmaha::OmahaReport
+      end
+
+      def resource_scope(options = {})
+        super(options.merge(:permission => :view_omaha_reports)).my_reports
+      end
+
+      def action_permission
+        case params[:action]
+        when 'last'
+          'view'
+        else
+          super
+        end
+      end
+    end
+  end
+end

data/app/controllers/omaha_reports_controller.rb

@@ -0,0 +1,57 @@
+class OmahaReportsController < ApplicationController
+  include Foreman::Controller::AutoCompleteSearch
+
+  before_action :setup_search_options, :only => :index
+
+  def model_of_controller
+    ::ForemanOmaha::OmahaReport
+  end
+
+  def index
+    @omaha_reports = resource_base_search_and_page(:host)
+  end
+
+  def show
+    # are we searching for the last report?
+    if params[:id] == 'last'
+      conditions = {
+        :host_id => resource_finder(Host.authorized(:view_hosts), params[:host_id]).try(:id)
+      } if params[:host_id].present?
+      params[:id] = resource_base.where(conditions).maximum(:id)
+    end
+
+    return not_found if params[:id].blank?
+
+    @omaha_report = resource_base.includes(:logs => [:message, :source]).find(params[:id])
+    @offset = @omaha_report.reported_at - @omaha_report.created_at
+  end
+
+  def destroy
+    @omaha_report = resource_base.find(params[:id])
+    if @omaha_report.destroy
+      process_success(
+        :success_msg => _('Successfully deleted report.'),
+        :success_redirect => omaha_reports_path
+      )
+    else
+      process_error
+    end
+  end
+
+  # TODO: Remove for Foreman 1.14+
+  def resource_base_with_search
+    resource_base.search_for(params[:search], :order => params[:order])
+  end
+
+  # TODO: Remove for Foreman 1.14+
+  def resource_base_search_and_page(tables = [])
+    base = tables.empty? ? resource_base_with_search : resource_base_with_search.eager_load(*tables)
+    base.paginate(:page => params[:page], :per_page => params[:per_page])
+  end
+
+  private
+
+  def resource_base
+    super.my_reports
+  end
+end

data/app/helpers/concerns/foreman_omaha/hosts_helper_extensions.rb

@@ -0,0 +1,15 @@
+module ForemanOmaha
+  module HostsHelperExtensions
+    extend ActiveSupport::Concern
+
+    included do
+      alias_method_chain :show_appropriate_host_buttons, :foreman_omaha
+    end
+
+    def show_appropriate_host_buttons_with_foreman_omaha(host)
+      buttons = show_appropriate_host_buttons_without_foreman_omaha(host)
+      buttons << (link_to_if_authorized(_('Omaha'), hash_for_host_omaha_reports_path(:host_id => host), :title => _('Browse host omaha reports'), :class => 'btn btn-default') if host.omaha_reports.any?)
+      buttons.compact
+    end
+  end
+end

data/app/helpers/omaha_reports_helper.rb

@@ -0,0 +1,28 @@
+module OmahaReportsHelper
+  def iconclass_for_omaha_status(status)
+    case status.to_sym
+    when :complete
+      'pficon pficon-ok'
+    when :downloading
+      'fa fa-download text-info'
+    when :downloaded
+      'fa fa-download text-success'
+    when :installed
+      'fa fa-sign-in text-success'
+    when :instance_hold
+      'fa fa-pause-circle-o text-warning'
+    when :error
+      'pficon pficon-error-circle-o'
+    else
+      'th'
+    end
+  end
+
+  def operatingsystem_link(operatingsystem)
+    link_to_if_authorized(os_name(operatingsystem),
+                          hash_for_edit_operatingsystem_path(
+                            :id => operatingsystem
+                          ).merge(:auth_object => operatingsystem,
+                                  :authorizer => authorizer))
+  end
+end

data/app/models/concerns/foreman_omaha/host_extensions.rb

@@ -0,0 +1,11 @@
+module ForemanOmaha
+  module HostExtensions
+    extend ActiveSupport::Concern
+    included do
+      has_many :omaha_reports, :class_name => '::ForemanOmaha::OmahaReport',
+                               :foreign_key => :host_id
+      has_one :last_omaha_report_object, -> { order("#{Report.table_name}.id DESC") },
+              :foreign_key => :host_id, :class_name => '::ForemanOmaha::OmahaReport'
+    end
+  end
+end

data/app/models/foreman_omaha/fact_name.rb

@@ -0,0 +1,7 @@
+module ForemanOmaha
+  class FactName < ::FactName
+    def origin
+      'Omaha'
+    end
+  end
+end

data/app/models/foreman_omaha/omaha_report.rb

@@ -0,0 +1,54 @@
+module ForemanOmaha
+  class OmahaReport < ::Report
+    enum :status => [:unknown, :complete, :downloading, :downloaded,
+              :installed, :instance_hold, :error]
+
+    scoped_search :on => :omaha_version, :rename => :version, :complete_value => true
+    scoped_search :on => :status, :complete_value => statuses
+
+    def self.import(report, proxy_id = nil)
+      OmahaReportImporter.import(report, proxy_id)
+    end
+
+    def self.report_status_column
+      '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
+    end
+
+    def operatingsystem
+      return unless omaha_version.present?
+      args = { :type => 'Coreos', :major => osmajor, :minor => osminor }
+      Operatingsystem.find_by(args)
+    end
+
+    def osmajor
+      omaha_version.gsub(/^(\d+)\.\d\.\d$/, '\1')
+    rescue
+      nil
+    end
+
+    def osminor
+      omaha_version.gsub(/^\d+\.(\d\.\d)$/, '\1')
+    rescue
+      nil
+    end
+  end
+end

data/app/models/host_status/omaha_status.rb

@@ -0,0 +1,44 @@
+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]
+    end
+
+    def to_global(_options = {})
+      return ::ForemanOmaha::OmahaReport.statuses[:unknown] unless relevant?
+      case last_report.status.to_sym
+      when :complete, :downloaded, :downloading, :installed
+        HostStatus::Global::OK
+      when :instance_hold
+        HostStatus::Global::WARN
+      when :error
+        HostStatus::Global::ERROR
+      else
+        HostStatus::Global::OK
+      end
+    end
+
+    def to_label(_options = {})
+      last_report.to_label
+    end
+
+    def relevant?(_options = {})
+      host && last_report.present?
+    end
+  end
+end

data/app/services/foreman_omaha/fact_importer.rb

@@ -0,0 +1,11 @@
+module ForemanOmaha
+  class FactImporter < ::FactImporter
+    def self.authorized_smart_proxy_features
+      'Omaha'
+    end
+
+    def fact_name_class
+      ForemanOmaha::FactName
+    end
+  end
+end

data/app/services/foreman_omaha/fact_parser.rb

@@ -0,0 +1,32 @@
+module ForemanOmaha
+  class FactParser < ::FactParser
+    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']))
+    end
+
+    def architecture
+      name = nil
+      name = 'x86_64' if facts['board'] == 'amd64-usr'
+      name = 'arm64' if facts['board'] == 'arm64-usr'
+      Architecture.where(:name => name).first_or_create unless name.nil?
+    end
+
+    def environment; end
+
+    def model; end
+
+    def domain; end
+
+    def ipmi_interface; end
+
+    def certname; end
+
+    def support_interfaces_parsing?
+      false
+    end
+  end
+end

data/app/services/foreman_omaha/omaha_report_importer.rb

@@ -0,0 +1,28 @@
+module ForemanOmaha
+  class OmahaReportImporter < ::ReportImporter
+    def report_name_class
+      OmahaReport
+    end
+
+    def self.authorized_smart_proxy_features
+      @authorized_smart_proxy_features ||= super + ['Omaha']
+    end
+
+    private
+
+    def create_report_and_logs
+      super
+      @report.omaha_version = omaha_version
+      @report.save
+      @report
+    end
+
+    def report_status
+      raw['status']
+    end
+
+    def omaha_version
+      raw['omaha_version'] || 'unknown'
+    end
+  end
+end

data/app/views/api/v2/omaha_reports/base.json.rabl

@@ -0,0 +1,3 @@
+object @omaha_report
+
+attributes :id, :host_id, :host_name, :reported_at, :status

data/app/views/api/v2/omaha_reports/create.json.rabl

@@ -0,0 +1,3 @@
+object @omaha_report
+
+extends "api/v2/omaha_reports/show"

data/app/views/api/v2/omaha_reports/index.json.rabl

@@ -0,0 +1,3 @@
+collection @omaha_reports
+
+extends "api/v2/omaha_reports/main"

data/app/views/api/v2/omaha_reports/main.json.rabl

@@ -0,0 +1,5 @@
+object @omaha_report
+
+extends "api/v2/omaha_reports/base"
+
+attributes :omaha_version, :created_at, :updated_at

data/app/views/api/v2/omaha_reports/show.json.rabl

@@ -0,0 +1,3 @@
+object @omaha_report
+
+extends "api/v2/omaha_reports/main"

data/app/views/api/v2/omaha_reports/update.json.rabl

@@ -0,0 +1,3 @@
+object @omaha_report
+
+extends "api/v2/omaha_reports/show"

data/app/views/omaha_reports/_details.html.erb

@@ -0,0 +1,28 @@
+<div class="container-fluid container-cards-pf">
+  <div class="row row-cards-pf">
+    <div class="col-xs-6 col-sm-4 col-md-4">
+      <div class="card-pf card-pf-accented card-pf-aggregate-status">
+        <h2 class="card-pf-title">
+          <span class="fa fa-cube"></span>Version
+        </h2>
+        <div class="card-pf-body">
+          <p class="card-pf-aggregate-status-notifications">
+            <span class="card-pf-aggregate-status-notification"><%= version %></span>
+          </p>
+        </div>
+      </div>
+    </div>
+    <div class="col-xs-6 col-sm-4 col-md-4">
+      <div class="card-pf card-pf-accented card-pf-aggregate-status">
+        <h2 class="card-pf-title">
+          <span class="fa fa-download"></span>Status
+        </h2>
+        <div class="card-pf-body">
+          <p class="card-pf-aggregate-status-notifications">
+          <span class="card-pf-aggregate-status-notification"><span class="<%= iconclass_for_omaha_status(status) %>"></span> <%= label %></span>
+          </p>
+        </div>
+      </div>
+    </div>
+  </div>
+</div>

data/app/views/omaha_reports/_list.html.erb

@@ -0,0 +1,34 @@
+<table class="<%= table_css_classes('table-fixed') %>">
+  <thead>
+    <tr>
+      <% unless params[:host_id] %>
+      <th><%= sort :host, :as => _("Host") %></th>
+      <% end %>
+      <th class="col-md-2"><%= sort :reported, :as => _("Last report") %></th>
+      <th class="col-md-2"><%= sort :version, :as => _("Version") %></th>
+      <th class="col-md-2"><%= _("Status") %></th>
+      <th class="col-md-2"><%= _('Actions') %></th>
+    </tr>
+  </thead>
+  <tbody>
+    <% for report in @omaha_reports %>
+      <tr>
+        <% if params[:host_id].nil? %>
+          <% if report.host.nil? %>
+              <td></td>
+          <% else %>
+              <td class="ellipsis"><%= link_to report.host, host_omaha_reports_path(report.host) %></td>
+          <% end %>
+        <% end %>
+        <td><%= display_link_if_authorized(_("%s ago") % time_ago_in_words(report.reported_at), hash_for_omaha_report_path(:id => report.id)) %></td>
+        <td><%= report.operatingsystem.present? ? operatingsystem_link(report.operatingsystem) : _('unknown') %></td>
+        <td><span class="<%= iconclass_for_omaha_status(report.status) %>"></span> <%= report.to_label %></td>
+        <td>
+          <%= action_buttons(display_delete_if_authorized hash_for_omaha_report_path(:id => report).merge(:auth_object => report, :authorizer => authorizer),
+                                           :confirm => _("Delete report for %s?") % report.host.try(:name)) %>
+        </td>
+      </tr>
+    <% end %>
+  </tbody>
+</table>
+<%= will_paginate_with_info @omaha_reports %>

data/app/views/omaha_reports/index.html.erb

@@ -0,0 +1,2 @@
+<% title _("Omaha Reports") %>
+<%= render :partial => 'list' %>

data/app/views/omaha_reports/show.html.erb

@@ -0,0 +1,21 @@
+<%= javascript 'reports' %>
+<% title "#{@omaha_report.host}"%>
+
+<p class='ra'> <%= _("Reported at %s ") % @omaha_report.reported_at %> </p>
+<% if @offset > 10 %>
+  <div class="alert alert-block alert-danger alert-dismissable">
+    <%= alert_close %>
+    <h3><%= _("Host times seems to be adrift!") %></h3>
+    <%= (_("Host reported time is <em>%s</em>") % @omaha_report.reported_at).html_safe %> <br/>
+    <%= (_("Foreman report creation time is <em>%s</em>") % @omaha_report.created_at).html_safe %> <br/>
+    <%= (_("Which is an offset of <b>%s</b>") % distance_of_time_in_words(@omaha_report.reported_at, @omaha_report.created_at, :include_seconds => true)).html_safe %>
+  </div>
+<% end %>
+
+<%= render 'details', :label => @omaha_report.to_label, :status => @omaha_report.status, :version => @omaha_report.omaha_version %>
+
+<%= title_actions link_to(_('Back'), :back, :class => 'btn btn-default'),
+  display_delete_if_authorized(hash_for_omaha_report_path(:id => @omaha_report), :class=> "btn btn-danger"),
+  link_to(_("Host details"), @omaha_report.host, :class => 'btn btn-default'),
+  link_to(_("Other reports for this host"), host_omaha_reports_path(@omaha_report.host), :class => 'btn btn-default')
+%>

data/app/views/omaha_reports/welcome.html.erb

@@ -0,0 +1,14 @@
+<div class="blank-slate-pf">
+  <div class="blank-slate-pf-icon">
+    <%= icon_text("book", "", :kind => "fa") %>
+  </div>
+  <h1><%= _('Omaha Reports') %></h1>
+  <p>
+  <%= _("You don't seem to have any omaha reports.") %></br>
+  <%= _('If you wish to configure CoreOS to forward reports to Foreman, please follow') %>
+  <%= link_to _("the documentation"), 'https://github.com/timogoebel/foreman_omaha', :rel => "external" %>.
+  </p>
+  <div class="blank-slate-pf-main-action">
+    <%= link_to _('Documentation'), 'https://github.com/timogoebel/foreman_omaha', :rel => 'external', :class => 'btn btn-primary btn-lg' %>
+  </div>
+</div>

data/config/routes.rb

@@ -0,0 +1,29 @@
+Rails.application.routes.draw do
+  namespace :api, :defaults => { :format => 'json' } do
+    scope '(:apiv)', :module => :v2,
+                     :defaults => { :apiv => 'v2' },
+                     :apiv => /v1|v2/,
+                     :constraints => ApiConstraints.new(:version => 2) do
+      constraints(:id => /[^\/]+/) do
+        resources :omaha_reports, :only => [:index, :show, :destroy] do
+          get :last, :on => :collection
+        end
+      end
+      resources :omaha_reports, :only => [:create]
+    end
+  end
+
+  resources :omaha_reports, :only => [:index, :show, :destroy] do
+    collection do
+      get 'auto_complete_search'
+    end
+  end
+
+  constraints(:id => /[^\/]+/) do
+    resources :hosts do
+      constraints(:host_id => /[^\/]+/) do
+        resources :omaha_reports, :only => [:index, :show]
+      end
+    end
+  end
+end

data/db/migrate/20160812083100_add_omaha_fields_to_reports.foreman_omaha.rb

@@ -0,0 +1,5 @@
+class AddOmahaFieldsToReports < ActiveRecord::Migration
+  def change
+    add_column :reports, :omaha_version, :string
+  end
+end

data/db/seeds.d/60_omaha_proxy_feature.rb

@@ -0,0 +1,2 @@
+f = Feature.where(:name => 'Omaha').first_or_create
+raise "Unable to create proxy feature: #{format_errors f}" if f.nil? || f.errors.any?