<%#
kind: job_template
name: Package Action - SSH Default
model: JobTemplate
job_category: Packages
description_format: "%{action} package(s) %{package}"
provider_type: SSH
template_inputs:
- name: pre_script
  description: A script to run prior to the package action
  input_type: user
  required: false
  advanced: true
- name: action
  description: 'The package action: install, update, or remove'
  input_type: user
  required: true
  options: "install\nupdate\nremove\ngroup install\ngroup remove"
- name: package
  description: The name of the package, if any
  input_type: user
  required: false
- name: post_script
  description: A script to run after the package action
  input_type: user
  required: false
  advanced: true
%>

<%
  supported_families = ['Redhat', 'Debian', 'Suse']
  render_error(N_('Unsupported or no operating system found for this host.')) unless @host.operatingsystem && supported_families.include?(@host.operatingsystem.family)
  action = input("action")

  if @host.operatingsystem.family == 'Redhat'
    package_manager = 'yum'
  elsif @host.operatingsystem.family == 'Debian'
    package_manager = 'apt'
  elsif @host.operatingsystem.family == 'Suse'
    package_manager = 'zypper'
  end
-%>
#!/bin/bash

# Helper function that exits with a particular message and code.
#
# Usage:
#   exit_with_message "Could not do a thing" 2
#
exit_with_message () {
  echo "${1}, exiting..."
  exit $2
}

<% if package_manager == 'zypper' -%>
handle_zypp_res_codes () {
  RETVAL=$1

  # See https://github.com/openSUSE/zypper/blob/master/src/main.h
  declare -A ZYPP_RES_CODES
  ZYPP_RES_CODES[100]='Updates are needed'
  ZYPP_RES_CODES[101]='Security updates are needed'
  ZYPP_RES_CODES[102]='Reboot needed after install/upgrade'
  ZYPP_RES_CODES[103]='Restart of package manager itself needed'
  ZYPP_RES_CODES[104]='given capability not found'
  ZYPP_RES_CODES[105]='SIGINT or SIGTERM received'
  ZYPP_RES_CODES[106]='Some repos have been skipped due to refresh errors'
  ZYPP_RES_CODES[107]='Some rpm %post configuration script failed'

  if [ "${ZYPP_RES_CODES[$RETVAL]}" != "" ]; then
    echo ${ZYPP_RES_CODES[$RETVAL]}
    RETVAL=0
  fi
  return $RETVAL
}
<% end -%>

<% unless input("pre_script").blank? -%>
  # Pre Script
  <%= input("pre_script") %>
  RETVAL=$?
  [ $RETVAL -eq 0 ] || exit_with_message "Pre script failed" $RETVAL
<% end -%>

# Action
<% if package_manager == 'yum' -%>
  yum -y <%= action %> <%= input("package") %>
<% elsif package_manager == 'apt' -%>
  <%-
    action = 'install' if action == 'group install'
    action = 'remove' if action == 'group remove'
    if action == 'group update' || action == 'update'
      if input('package').blank?
        action = 'upgrade'
      else
        action = '--only-upgrade install'
      end
    end
  -%>
  apt-get -y update
  apt-get -o Dpkg::Options::="--force-confdef" -o Dpkg::Options::="--force-confold" -y <%= action %> <%= input("package") %>
<% elsif package_manager == 'zypper' -%>
  <%-
    if action == "group install"
      action = "install -t pattern"
    elsif action == "group remove"
      action = "remove -t pattern"
    end
  -%>
  zypper refresh
  zypper -n <%= action %> <%= input("package") %>
  handle_zypp_res_codes $?
<% end -%>
RETVAL=$?
[ $RETVAL -eq 0 ] || exit_with_message "Package action failed" $RETVAL

<% unless input("post_script").blank? -%>
  # Post Script
  <%= input("post_script") %>
  RETVAL=$?
  [ $RETVAL -eq 0 ] || exit_with_message "Post script failed" $RETVAL
<% end -%>