\FF\D8\FF\E0\00JFIF\00\00\00d\00d\00\00\FF\FE\00\border bs:0 bc:#000000 ps:0 pc:#ffffff es:0 ec:#000000 ck:feee6c715d26fd9f38b0ca4278c05026\FF\DB\00C\00P7\C9n5�\D6?\BD�\9Ds\EBp\9F[`8m\B7)o\B5\E8\E6I\99\FE3]]A2\BA\8Cw\D6E\93\\DEv\C8\009\F2\F1NI?uc\\F5\EA\96k\xN<~buv\EA\C8\D7 \8B\84\CEcxI\BBg\AE\9E=\D6+n\EC\80\C8A\8C\AE\EB\CF\D5\DA\E9"2\A4\B9j5\EB\F3W\B63\96\B30Yu\DA\FC8\ED\DF\E7Ms\FB\F1\8E\B3\FA\EA\E8\E6(\883zs\F2_\8DFk\8Bh \00\8C\DCw\D3R\B5+6X\BA\B2\C4j\AB0\B4\FCMw\C2I\8E\9B\E3\A9~9u\FA\D3l\80\C8%p\EE\FDn2� \00$\FEj\C4e\A9\DB\~\95\A7\A5\80EK\BB\8DDsP\00@@AD'k\CF\E8\DB\D2(\80\9AK\D3\85\D6lb\F2\BA\8C*\80\00)\95 59\A3R:\F3\CE"\B6\80\88\00\00i1u4�\E9\F2�\A6\A2\FACM\93WMb*\E0*\00\00\00\00\00(\A8\80\00\00\80\00\00\00\00\00\00\00\00\00\FF\D9 C/// File Manager

File Manager

Path: /opt/chef.upgrade/embedded/lib/ruby/gems/2.3.0/gems/cheffish-5.0.1/lib/cheffish/

Viewing File: chef_run.rb

require "cheffish/basic_chef_client"

module Cheffish
  class ChefRun
    #
    # @param chef_config A hash with symbol keys that looks suspiciously similar to `Chef::Config`.
    #        Some possible options:
    #        - stdout: <IO object> - where to stream stdout to
    #        - stderr: <IO object> - where to stream stderr to
    #        - log_level: :debug|:info|:warn|:error|:fatal
    #        - log_location: <path|IO object> - where to stream logs to
    #        - verbose_logging: true|false - true if you want verbose logging in :debug
    #
    def initialize(chef_config = {})
      @chef_config = chef_config || {}
    end

    attr_reader :chef_config

    class StringIOTee < StringIO
      def initialize(*streams)
        super()
        @streams = streams.flatten.select { |s| !s.nil? }
      end

      attr_reader :streams

      def write(*args, &block)
        super
        streams.each { |s| s.write(*args, &block) }
      end
    end

    def client
      @client ||= begin
        chef_config = self.chef_config.dup
        chef_config[:log_level] ||= :debug if !chef_config.has_key?(:log_level)
        chef_config[:verbose_logging] = false if !chef_config.has_key?(:verbose_logging)
        chef_config[:stdout] = StringIOTee.new(chef_config[:stdout])
        chef_config[:stderr] = StringIOTee.new(chef_config[:stderr])
        chef_config[:log_location] = StringIOTee.new(chef_config[:log_location])
        @client = ::Cheffish::BasicChefClient.new(nil,
          [ event_sink, Chef::Formatters.new(:doc, chef_config[:stdout], chef_config[:stderr]) ],
          chef_config
        )
      end
    end

    def event_sink
      @event_sink ||= EventSink.new
    end

    #
    # output
    #
    def stdout
      @client ? client.chef_config[:stdout].string : nil
    end

    def stderr
      @client ? client.chef_config[:stderr].string : nil
    end

    def logs
      @client ? client.chef_config[:log_location].string : nil
    end

    def logged_warnings
      logs.lines.select { |l| l =~ /^\[[^\]]*\] WARN:/ }.join("\n")
    end

    def logged_errors
      logs.lines.select { |l| l =~ /^\[[^\]]*\] ERROR:/ }.join("\n")
    end

    def logged_info
      logs.lines.select { |l| l =~ /^\[[^\]]*\] INFO:/ }.join("\n")
    end

    def resources
      client.run_context.resource_collection
    end

    def compile_recipe(&recipe)
      client.load_block(&recipe)
    end

    def converge
      begin
        client.converge
        @converged = true
      rescue RuntimeError => e
        @raised_exception = e
        raise
      end
    end

    def reset
      @client = nil
      @converged = nil
      @stdout = nil
      @stderr = nil
      @logs = nil
      @raised_exception = nil
    end

    def converged?
      !!@converged
    end

    def converge_failed?
      @raised_exception.nil? ? false : true
    end

    def updated?
      client.updated?
    end

    def up_to_date?
      !client.updated?
    end

    def output_for_failure_message
      message = ""
      if stdout && !stdout.empty?
        message << "---                    ---\n"
        message << "--- Chef Client Output ---\n"
        message << "---                    ---\n"
        message << stdout
        message << "\n" if !stdout.end_with?("\n")
      end
      if stderr && !stderr.empty?
        message << "---                          ---\n"
        message << "--- Chef Client Error Output ---\n"
        message << "---                          ---\n"
        message << stderr
        message << "\n" if !stderr.end_with?("\n")
      end
      if logs && !logs.empty?
        message << "---                  ---\n"
        message << "--- Chef Client Logs ---\n"
        message << "---                  ---\n"
        message << logs
      end
      message
    end

    class EventSink
      def initialize
        @events = []
      end

      attr_reader :events

      def method_missing(method, *args)
        @events << [ method, *args ]
      end

      def respond_to_missing?(method_name, include_private = false)
        # Chef::EventDispatch::Dispatcher calls #respond_to? to see (basically) if we'll accept an event;
        # obviously, per above #method_missing, we'll accept whatever we're given. if there's a problem, it
        # will surface higher up the stack.
        true
      end
    end
  end
end