Replace omnibus_updater with chef_client_updater

omnibus_updater is deprecated

cookbooks/chef_client_updater/resources/updater.rb

@@ -0,0 +1,136 @@
+#
+# Cookbook:: chef_client_updater
+# Resource:: updater
+#
+# Copyright:: 2016-2017, Will Jordan
+# Copyright:: 2016-2017, Chef Software, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+provides 'chef_client_updater'
+
+property :channel, [String, Symbol], default: :stable
+property :prevent_downgrade, [true, false], default: false
+property :version, [String, Symbol], default: :latest
+property :post_install_action, String, default: 'exec'
+property :exec_command, String, default: $PROGRAM_NAME.split(' ').first
+property :exec_args, Array, default: ARGV
+
+action :update do
+  if update_necessary? # ~FC023
+    converge_by "Upgraded chef-client #{current_version} to #{desired_version}" do
+      upgrade_command = Mixlib::ShellOut.new(mixlib_install.install_command)
+      upgrade_command.run_command
+      run_post_install_action
+    end
+  end
+end
+
+action_class.class_eval do
+  def load_mixlib_install
+    require 'mixlib/install'
+  rescue LoadError
+    Chef::Log.info('mixlib-install gem not found. Installing now')
+    chef_gem 'mixlib-install' do
+      compile_time true if respond_to?(:compile_time)
+    end
+
+    require 'mixlib/install'
+  end
+
+  def load_mixlib_versioning
+    require 'mixlib/versioning'
+  rescue LoadError
+    Chef::Log.info('mixlib-versioning gem not found. Installing now')
+    chef_gem 'mixlib-versioning' do
+      compile_time true if respond_to?(:compile_time)
+    end
+
+    require 'mixlib/versioning'
+  end
+
+  def mixlib_install
+    load_mixlib_install
+    options = {
+      product_name: 'chef',
+      platform_version_compatibility_mode: true,
+      channel: new_resource.channel.to_sym,
+      product_version: new_resource.version == 'latest' ? :latest : new_resource.version,
+
+    }
+    Mixlib::Install.new(options)
+  end
+
+  # why would we use this when mixlib-install has a current_version method?
+  # well mixlib-version parses the manifest JSON, which might not be there.
+  # ohai handles this better IMO
+  def current_version
+    node['chef_packages']['chef']['version']
+  end
+
+  def desired_version
+    new_resource.version == 'latest' ? mixlib_install.available_versions.last : new_resource.version
+  end
+
+  # why wouldn't we use the built in update_available? method in mixlib-install?
+  # well that would use current_version from mixlib-install and it has no
+  # concept or preventing downgrades
+  def update_necessary?
+    load_mixlib_versioning
+    cur_version = Mixlib::Versioning.parse(current_version)
+    des_version = Mixlib::Versioning.parse(desired_version)
+    new_resource.prevent_downgrade ? (des_version > cur_version) : (des_version != cur_version)
+  end
+
+  def eval_post_install_action
+    return unless new_resource.post_install_action == 'exec'
+
+    if Chef::Config[:interval]
+      Chef::Log.warn 'post_install_action "exec" not supported for long-running client process -- changing to "kill".'
+      new_resource.post_install_action = 'kill'
+    end
+
+    return if Process.respond_to?(:exec)
+    Chef::Log.warn 'post_install_action Process.exec not available -- changing to "kill".'
+    new_resource.post_install_action = 'kill'
+  end
+
+  def run_post_install_action
+    # make sure the passed action will actually work
+    eval_post_install_action
+
+    case new_resource.post_install_action
+    when 'exec'
+      if Chef::Config[:local_mode]
+        Chef::Log.info 'Shutting down local-mode server.'
+        if Chef::Application.respond_to?(:destroy_server_connectivity)
+          Chef::Application.destroy_server_connectivity
+        elsif defined?(Chef::LocalMode) && Chef::LocalMode.respond_to?(:destroy_server_connectivity)
+          Chef::LocalMode.destroy_server_connectivity
+        end
+      end
+      Chef::Log.warn 'Replacing ourselves with the new version of Chef to continue the run.'
+      Kernel.exec(new_resource.exec_command, *new_resource.exec_args)
+    when 'kill'
+      if Chef::Config[:client_fork] && Process.ppid != 1
+        Chef::Log.warn 'Chef client is defined for forked runs. Sending TERM to parent process!'
+        Process.kill('TERM', Process.ppid)
+      end
+      Chef::Log.warn 'New chef-client installed. Forcing chef exit!'
+      exit(213)
+    else
+      raise "Unexpected post_install_action behavior: #{new_resource.post_install_action}"
+    end
+  end
+end