foreman_patch 1.1.1

data/app/controllers/foreman_patch/concerns/api/v2/hosts_controller_extensions.rb

@@ -0,0 +1,21 @@
+module ForemanPatch
+  module Concerns
+    module Api::V2::HostsControllerExtensions
+      extend ActiveSupport::Concern
+
+      included do
+        after_action :reschedule_patching, only: :update
+
+        def reschedule_patching
+          return if @host.group_facet.nil?
+          return unless @host.group_facet.saved_change_to_attribute?(:group_id)
+
+          schedule = params.dig('host', 'group_facet_attributes', 'schedule') || 'future-only'
+
+          ForemanTasks.async_task(Actions::ForemanPatch::Host::Reschedule, @host, include_active: schedule == 'all') unless schedule == 'none'
+        end
+
+      end
+    end
+  end
+end

data/app/controllers/foreman_patch/concerns/hosts_controller_extensions.rb

@@ -0,0 +1,61 @@
+module ForemanPatch
+  module Concerns
+    module HostsControllerExtensions
+      extend ActiveSupport::Concern
+
+      module Overrides
+        def action_permission
+          case params[:action]
+          when 'select_multiple_patch_group', 'update_multiple_patch_group'
+            :edit
+          else
+            super
+          end
+        end
+      end
+
+      included do
+        prepend Overrides
+
+        after_action :reschedule_patching, only: :update
+
+        def select_multiple_patch_group
+          find_multiple
+        end
+
+        def update_multiple_patch_group
+          find_multiple
+
+          if (params['patch_group']['id'].blank?)
+            @hosts.each do |host|
+              group_facet = host.group_facet
+              group_facet.group = nil unless group_facet.blank?
+              host.save!
+            end
+          else
+            patch_group = ForemanPatch::Group.find(params['patch_group']['id'])
+
+            @hosts.each do |host|
+              group_facet = host.group_facet || host.build_group_facet
+              group_facet.group = patch_group
+              host.save!
+            end
+          end
+
+          ForemanTasks.async_task(Actions::ForemanPatch::Host::Reschedule, @hosts, include_active: params['patch_group']['include_active'])
+
+          success _('Updated hosts: changed patch group')
+          redirect_back_or_to hosts_path
+        end
+
+        def reschedule_patching
+          return if @host.group_facet.nil?
+          return unless @host.group_facet.saved_change_to_attribute?(:group_id)
+
+          ForemanTasks.async_task(Actions::ForemanPatch::Host::Reschedule, @host)
+        end
+
+      end
+    end
+  end
+end

data/app/controllers/foreman_patch/cycles_controller.rb

@@ -0,0 +1,34 @@
+module ForemanPatch
+  class CyclesController < ApplicationController
+    include Foreman::Controller::AutoCompleteSearch
+
+    before_action :find_resource, only: [:show, :destroy]
+
+    def index
+      params[:order] ||= 'start_date DESC'
+      @cycles = resource_base_search_and_page
+    end
+
+    def show
+    end
+
+    def destroy
+      if @cycle.destroy
+        process_success object: @cycle
+      else
+        process_error object: @cycle
+      end
+    end
+
+    def resource_class
+      ForemanPatch::Cycle
+    end
+
+    private
+
+    def cycle_params
+      params.require(:cycle).permit(:name, :description, :start_date)
+    end
+
+  end
+end

data/app/controllers/foreman_patch/groups_controller.rb

@@ -0,0 +1,68 @@
+module ForemanPatch
+  class GroupsController < ApplicationController
+    include Foreman::Controller::AutoCompleteSearch
+    include Foreman::Controller::CsvResponder
+
+    before_action :find_resource, only: [:edit, :update, :destroy]
+    before_action :setup_search_options, only: :index
+
+    def index
+      respond_to do |format|
+        format.html do
+          @groups = resource_base_search_and_page
+          render :index
+        end
+        format.csv do
+          csv_response(resource_base_with_search)
+        end
+      end
+    end
+
+    def new
+      @group = Group.new
+    end
+
+    def create
+      @group = Group.new(group_params)
+      if @group.save
+        process_success object: @group
+      else
+        process_error object: @group
+      end
+    end
+
+    def edit
+    end
+
+    def update
+      if @group.update(group_params)
+        process_success object: @group
+      else
+        process_error object: @group
+      end
+    end
+
+    def destroy
+      if @group.destroy
+        process_success object: @group
+      else
+        process_error object: @group
+      end
+    end
+
+    def resource_class
+      ForemanPatch::Group
+    end
+
+    def csv_columns
+      [:name, :label, :default_window_plan, :hosts_count]
+    end
+
+    private
+
+    def group_params
+      params.require(:group).permit(:name, :description, :default_window_plan_id, :max_unavailable, :priority)
+    end
+
+  end
+end

data/app/controllers/foreman_patch/invocations_controller.rb

@@ -0,0 +1,51 @@
+module ForemanPatch
+  class InvocationsController < ApplicationController
+    include Foreman::Controller::AutoCompleteSearch
+
+    before_action :find_resource
+
+    helper ForemanPatch::PatchingHelper
+
+    def show
+      @invocation = ForemanPatch::Invocation.find(params[:id])
+      @host = @invocation.host
+
+      @round = @invocation.round
+      @window = @round.window
+      @cycle = @window.cycle
+
+      @auto_refresh = !@invocation.complete?
+      @since = params[:since].to_f if params[:since].present?
+
+      @invocation.phases.each do |phase|
+        @line_sets = phase.live_output
+        @line_sets = @line_sets.drop_while { |o| o['timestamp'].to_f <= @since } if @since
+
+        unless @line_sets.empty?
+          @phase = phase.label.demodulize.underscore
+          break
+        end
+      end
+
+      @line_counter = params[:line_counter].to_i
+    end
+
+    def destroy
+      if @invocation.destroy!
+      else
+        process_error
+      end
+    end
+
+    def resource_class
+      ForemanPatch::Invocation
+    end
+
+    private
+
+    def create_override
+      ForemanPatch::Override.create!(source: @invocation, user: User.current, reason: params[:reason])
+    end
+
+  end
+end

data/app/controllers/foreman_patch/plans_controller.rb

@@ -0,0 +1,59 @@
+module ForemanPatch
+  class PlansController < ApplicationController
+
+    before_action :find_resource, only: [:show, :edit, :iterate, :update, :destroy]
+
+    def index
+      @plans = resource_base_search_and_page
+    end
+
+    def new
+      @plan = Plan.new
+    end
+
+    def create
+      @plan = Plan.new(plan_params)
+    end
+
+    def show
+    end
+
+    def edit
+    end
+
+    def iterate
+      @plan.iterate
+    end
+
+    def update
+      if @plan.update(plan_params)
+        process_success success_hash
+      else
+        process_error
+      end
+    end
+
+    def destroy
+      if @plan.destroy
+        process_success success_hash
+      else
+        process_error
+      end
+    end
+
+    def resource_class
+      ForemanPatch::Plan
+    end
+
+    private
+
+    def plan_params
+      params.require(:plan).permit(:name, :description, :cycle_name, :start_date, :interval, :units, :active_count)
+    end
+
+    def success_hash
+      { success_redirect: params[:redirect].presence }
+    end
+
+  end
+end

data/app/controllers/foreman_patch/rounds_controller.rb

@@ -0,0 +1,18 @@
+module ForemanPatch
+  class RoundsController < ApplicationController
+
+    before_action :find_resource, only: [:move, :chart, :show]
+
+    helper ForemanPatch::PatchingHelper
+
+    def show
+    end
+
+    def resource_class
+      ForemanPatch::Round
+    end
+
+    private
+
+  end
+end

data/app/controllers/foreman_patch/ticket_fields_controller.rb

@@ -0,0 +1,37 @@
+module ForemanPatch
+  class TicketFieldsController < ::LookupKeysController
+    include Foreman::Controller::AutoCompleteSearch
+    include ForemanPatch::Parameters::TicketField
+
+    before_action :find_resource, only: [:edit, :update, :destroy], if: proc { params[:id] }
+
+    def index
+      @ticket_fields = resource_base_search_and_page
+    end
+    
+    def new
+      @ticket_field = TicketField.new
+    end
+
+    def create
+      @ticket_field = TicketField.new(ticket_field_params)
+      if @ticket_field.save
+        process_success
+      else
+        process_error
+      end
+    end
+
+    def resource_class
+      ForemanPatch::TicketField
+    end
+
+    def resource
+      @ticket_field
+    end
+
+    def resource_params
+      ticket_field_params
+    end
+  end
+end

data/app/controllers/foreman_patch/window_plans_controller.rb

@@ -0,0 +1,60 @@
+module ForemanPatch
+  class WindowPlansController < ApplicationController
+    helper ForemanPatch::PlansHelper
+
+    before_action :find_resource, only: [:edit, :update, :destroy]
+    before_action :find_plan, only: [:new, :create]
+
+    def new
+      @window_plan = WindowPlan.new
+    end
+
+    def create
+      @window_plan = WindowPlan.new(window_plan_params)
+    end
+
+    def edit
+      @plan = @window_plan.plan
+    end
+
+    def update
+      if @window_plan.update(window_plan_params)
+        process_success success_hash
+      else
+        process_error
+      end
+    end
+
+    def destroy
+      if @window_plan.destroy
+        process_success success_hash
+      else
+        process_error
+      end
+    end
+
+    def resource_class
+      ForemanPatch::WindowPlan
+    end
+
+    private
+
+    def allowed_nested_id
+      %w(plan_id)
+    end
+
+    def find_plan
+      @plan ||= ForemanPatch::Plan.find(params[:plan_id])
+    end
+
+    def window_plan_params
+      params[:window_plan][:plan_id] = params[:plan_id] unless params[:plan_id].nil?
+      params.require(:window_plan).permit(:name, :description, :state_day, :start_time, :duration, :plan_id)
+    end
+
+    def success_hash
+      { success_redirect: params[:redirect].presence || plan_path(@plan) }
+    end
+
+  end
+end

data/app/controllers/foreman_patch/windows_controller.rb

@@ -0,0 +1,58 @@
+module ForemanPatch
+  class WindowsController < ApplicationController
+
+    before_action :find_cycle, only: [:new, :create]
+    before_action :find_resource, only: [:show, :destroy]
+
+    helper ForemanPatch::WindowPatchingHelper
+    helper ForemanPatch::TicketHelper
+
+    def new
+      @window = Window.new
+    end
+
+    def create
+      @window = Window.new(window_params)
+      if @window.save
+        process_success object: @window
+      else
+        process_error object: @window
+      end
+    end
+
+    def show
+      @cycle = @window.cycle
+
+      @auto_refresh = @window.task.try(:pending?)
+
+      respond_to do |format|
+        format.json
+        format.html
+      end
+    end
+
+    def destroy
+      if @window.destroy
+        process_success object: @window
+      else
+        process_error object: @window
+      end
+    end
+
+    def resource_class
+      ForemanPatch::Window
+    end
+
+    private
+
+    def find_cycle
+      @cycle ||= Cycle.find(params[:cycle_id])
+    end
+
+    def window_params
+      params[:window][:cycle_id] = @cycle.id unless @cycle.nil?
+      params.require(:window).permit(:name, :description, :start_at, :end_by)
+    end
+
+  end
+end

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

@@ -0,0 +1,13 @@
+module ForemanPatch
+  module HostsHelperExtensions
+    extend ActiveSupport::Concern
+
+    included do
+      # execute callbacks
+    end
+
+    # create or overwrite instance methods...
+    def instance_method_name
+    end
+  end
+end

data/app/helpers/foreman_patch/cycles_helper.rb

@@ -0,0 +1,19 @@
+module ForemanPatch
+  module CyclesHelper
+
+    def cycle_length(cycle)
+      if ForemanPatch::Cycle === cycle
+        cycle = cycle.plan
+      end
+
+      cycle.interval.send(cycle.units).value / ActiveSupport::Duration::SECONDS_PER_DAY
+    end
+
+    def cycle_windows(cycle, date)
+      cycle.windows.select do |window|
+        window.start_at.to_date === date
+      end
+    end
+
+  end
+end

data/app/helpers/foreman_patch/hosts_helper.rb

@@ -0,0 +1,52 @@
+module ForemanPatch
+  module HostsHelper
+
+    def patch_host_multiple_actions
+      [
+        {action: [_('Change Patch Group'), select_multiple_patch_group_hosts_path], priority: 1000}
+      ]
+    end
+
+    def patch_groups_for_host(host, options)
+      include_blank = options.fetch(:include_blank, nil)
+      if include_blank == true
+        include_blank = '<option></option>'
+      end
+
+      group_options = ForemanPatch::Group.all.map do |group|
+        selected = host.group.try(:id) == group.id ? 'selected' : ''
+        %(<option #{selected} value="#{group.id}">#{h(group.name)}</option>)
+      end
+
+      group_options = group_options.join
+      group_options.insert(0, include_blank) if include_blank
+      group_options.html_safe
+    end
+
+    def patch_host_overview_fields(host)
+      fields = []
+      if host.group_facet.present?
+        name = host.group_facet.group || 'None'
+        search = (host.group_facet.group.present? ? "patch_group = #{host.group_facet.group}" : 'not has patch_group')
+
+        fields << {
+          field: [
+            _('Patch Group'),
+            link_to(name, hosts_path(search: search)),
+          ],
+          priority: 1001,
+        }
+
+        fields << {
+          field: [
+            _('Last Patch Date'),
+            host.group_facet.last_patched_at,
+          ],
+          priority: 1002,
+        }
+      end
+      fields
+    end
+
+  end
+end

data/app/helpers/foreman_patch/patching_helper.rb

@@ -0,0 +1,116 @@
+module ForemanPatch
+  module PatchingHelper
+
+    def round_hosts_authorizer
+      @round_hosts_authorizer ||= Authorizer.new(User.current, collection: @hosts)
+    end
+
+    def patch_invocation_actions(round, invocation)
+      host = invocation.host
+      task = invocation.task
+
+      links = []
+
+      if authorized_for(main_app.hash_for_host_path(host).merge(auth_object: host, permission: :view_hosts, authorizer: round_hosts_authorizer))
+        links << { title: _('Host Detail'),
+                   action: { href: main_app.host_path(host), 'data-method': 'get', id: "#{host.name}-actions-detail" } }
+      end
+
+      if task.present? && authorized_for(main_app.hash_for_foreman_tasks_task_path(task).merge(auth_object: task, permission: :view_foreman_tasks))
+        links << { title: _('Task Detail'),
+                   action: { href: main_app.foreman_tasks_task_path(task),
+                             'data-method': 'get',
+                             id: "#{host.name}-actions-task" }
+        }
+      end
+
+      unless task.present?
+        links << { title: _('Cancel'),
+                   action: { href: invocation_path(invocation),
+                             'data-method': 'delete',
+                             id: "#{host.name}-actions-cancel" }
+        }
+      end
+
+      if task.present? and task.pending?
+        links << { title: _('Cancel'), 
+                   action: { href: main_app.cancel_foreman_tasks_task_path(task),
+                             'data-method': 'post',
+                             id: "#{host.name}-actions-cancel" }
+        }
+
+        links << { title: _('Abort'),
+                   action: { href: main_app.abort_foreman_tasks_task_path(task),
+                             'data-method': 'post',
+                             id: "#{host.name}-actions-abort" }
+        }
+      end
+
+      links
+    end
+
+    def round_invocation_hosts(round)
+      round.invocations.map do |invocation|
+        {
+          name: invocation.host.name,
+          link: invocation_path(invocation),
+          state: invocation.state,
+          result: invocation.result,
+          actions: patch_invocation_actions(round, invocation)
+        } 
+      end
+    end
+
+    def round_task_buttons(task)
+      buttons = []
+
+      if authorized_for(permission: :view_foreman_tasks, auth_object: task, authorizer: Authorizer.new(User.current, collection: [task]))
+        buttons << link_to(_('Task Details'), main_app.foreman_tasks_task_path(task),
+                           class: 'btn btn-default',
+                           title: _('See the task details'))
+      end
+
+      # if authorized
+      buttons << link_to(_('Cancel'), main_app.cancel_foreman_tasks_task_path(task),
+                         class: 'btn btn-danger',
+                         title: _('Try to cancel the patching round'),
+                         disabled: !task.cancellable?,
+                         method: :post)
+      buttons << link_to(_('Abort'), main_app.abort_foreman_tasks_task_path(task),
+                         class: 'btn btn-danger',
+                         title: _('Try to abort the patching round without waiting for its result'),
+                         disabled: !task.cancellable?,
+                         method: :post)
+      # end
+
+      buttons
+    end
+
+    def invocation_task_buttons(task, invocation)
+      buttons = []
+
+      if task.present? && authorized_for(permission: :view_foreman_tasks, auth_object: task)
+        buttons << link_to(_('Task Details'), main_app.foreman_tasks_task_path(task),
+                           class: 'btn btn-default',
+                           title: _('See the task details'))
+      end
+
+      #if authorized_for... eventually
+      if task.present?
+        buttons << link_to(_('Cancel Patch'), main_app.cancel_foreman_tasks_task_path(task),
+                           class: 'btn btn-danger',
+                           title: _('Try to cancel the patch of the host'),
+                           disabled: !task.cancellable?,
+                           method: :post)
+        buttons << link_to(_('Abort Patch'), main_app.abort_foreman_tasks_task_path(task),
+                           class: 'btn btn-danger',
+                           title: _('Try to abort the patching of the host without waiting for its result'),
+                           disabled: !task.cancellable?,
+                           method: :post)
+      end
+      #end
+      buttons
+    end
+
+  end
+end