From 1dea4d0cebda1b1ed7899b9cff1f0a3aa74ee09d Mon Sep 17 00:00:00 2001 From: Lamont Granquist Date: Wed, 15 Jul 2020 19:00:11 -0700 Subject: extracting shell_out helper to mixlib-shellout Signed-off-by: Lamont Granquist --- lib/mixlib/shellout/helper.rb | 208 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 208 insertions(+) create mode 100644 lib/mixlib/shellout/helper.rb (limited to 'lib') diff --git a/lib/mixlib/shellout/helper.rb b/lib/mixlib/shellout/helper.rb new file mode 100644 index 0000000..50ee702 --- /dev/null +++ b/lib/mixlib/shellout/helper.rb @@ -0,0 +1,208 @@ +#-- +# Author:: Daniel DeLeo () +# Copyright:: Copyright (c) Chef Software Inc. +# License:: Apache License, Version 2.0 +# +# 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. + +require_relative "../shellout" +require "chef-utils/dsl/path_sanity" +require "chef-utils/internal" + +module Mixlib + class ShellOut + module Helper + include ChefUtils::Internal + include ChefUtils::DSL::PathSanity + + # PREFERRED APIS: + # + # all consumers should now call shell_out!/shell_out. + # + # the shell_out_compacted/shell_out_compacted! APIs are private but are intended for use + # in rspec tests, and should ideally always be used to make code refactoring that do not + # change behavior easier: + # + # allow(provider).to receive(:shell_out_compacted!).with("foo", "bar", "baz") + # provider.shell_out!("foo", [ "bar", nil, "baz"]) + # provider.shell_out!(["foo", nil, "bar" ], ["baz"]) + # + # note that shell_out_compacted also includes adding the magical timeout option to force + # people to setup expectations on that value explicitly. it does not include the default_env + # mangling in order to avoid users having to setup an expectation on anything other than + # setting `default_env: false` and allow us to make tweak to the default_env without breaking + # a thousand unit tests. + # + + def shell_out(*args, **options) + options = options.dup + options = __maybe_add_timeout(self, options) + if options.empty? + shell_out_compacted(*__clean_array(*args)) + else + shell_out_compacted(*__clean_array(*args), **options) + end + end + + def shell_out!(*args, **options) + options = options.dup + options = __maybe_add_timeout(self, options) + if options.empty? + shell_out_compacted!(*__clean_array(*args)) + else + shell_out_compacted!(*__clean_array(*args), **options) + end + end + + private + + # helper sugar for resources that support passing timeouts to shell_out + # + # module method to not pollute namespaces, but that means we need self injected as an arg + # @api private + def __maybe_add_timeout(obj, options) + options = options.dup + # historically resources have not properly declared defaults on their timeouts, so a default default of 900s was enforced here + default_val = 900 + return options if options.key?(:timeout) + + # FIXME: need to nuke descendent tracker out of Chef::Provider so we can just define that class here without requiring the + # world, and then just use symbol lookup + if obj.class.ancestors.map(&:name).include?("Chef::Provider") && obj.respond_to?(:new_resource) && obj.new_resource.respond_to?(:timeout) && !options.key?(:timeout) + options[:timeout] = obj.new_resource.timeout ? obj.new_resource.timeout.to_f : default_val + end + options + end + + # helper function to mangle options when `default_env` is true + # + # @api private + def __apply_default_env(options) + options = options.dup + default_env = options.delete(:default_env) + default_env = true if default_env.nil? + if default_env + env_key = options.key?(:env) ? :env : :environment + options[env_key] = { + "LC_ALL" => __config[:internal_locale], + "LANGUAGE" => __config[:internal_locale], + "LANG" => __config[:internal_locale], + __env_path_name => sanitized_path, + }.update(options[env_key] || {}) + end + options + end + + # this SHOULD be used for setting up expectations in rspec, see banner comment at top. + # + # the private constraint is meant to avoid code calling this directly, rspec expectations are fine. + # + def shell_out_compacted(*args, **options) + options = __apply_default_env(options) + if options.empty? + __shell_out_command(*args) + else + __shell_out_command(*args, **options) + end + end + + # this SHOULD be used for setting up expectations in rspec, see banner comment at top. + # + # the private constraint is meant to avoid code calling this directly, rspec expectations are fine. + # + def shell_out_compacted!(*args, **options) + options = __apply_default_env(options) + cmd = if options.empty? + __shell_out_command(*args) + else + __shell_out_command(*args, **options) + end + cmd.error! + cmd + end + + # Helper for subclasses to reject nil out of an array. It allows + # using the array form of shell_out (which avoids the need to surround arguments with + # quote marks to deal with shells). + # + # Usage: + # shell_out!(*clean_array("useradd", universal_options, useradd_options, new_resource.username)) + # + # universal_options and useradd_options can be nil, empty array, empty string, strings or arrays + # and the result makes sense. + # + # keeping this separate from shell_out!() makes it a bit easier to write expectations against the + # shell_out args and be able to omit nils and such in the tests (and to test that the nils are + # being rejected correctly). + # + # @param args [String] variable number of string arguments + # @return [Array] array of strings with nil and null string rejection + + def __clean_array(*args) + args.flatten.compact.map(&:to_s) + end + + def __shell_out_command(*args, **options) + if __config.target_mode? + FakeShellOut.new(args, options, __transport_connection.run_command(args.join(" "))) # FIXME: train should accept run_command(*args) + else + cmd = if options.empty? + Mixlib::ShellOut.new(*args) + else + Mixlib::ShellOut.new(*args, **options) + end + cmd.live_stream ||= __io_for_live_stream + cmd.run_command + cmd + end + end + + def __io_for_live_stream + if STDOUT.tty? && !__config[:daemon] && __log.debug? + STDOUT + else + nil + end + end + + def __env_path_name + if ChefUtils.windows? + "Path" + else + "PATH" + end + end + + class FakeShellOut + attr_reader :stdout, :stderr, :exitstatus, :status + + def initialize(args, options, result) + @args = args + @options = options + @stdout = result.stdout + @stderr = result.stderr + @exitstatus = result.exit_status + @status = OpenStruct.new(success?: ( exitstatus == 0 )) + end + + def error? + exitstatus != 0 + end + + def error! + raise Mixlib::ShellOut::ShellCommandFailed, "Unexpected exit status of #{exitstatus} running #{@args}" if error? + end + end + end + end +end -- cgit v1.2.1