garcun 0.0.2

data/lib/garcon/chef/handler/devreporter.rb

@@ -0,0 +1,127 @@
+# encoding: UTF-8
+#
+# Author:    Stefano Harding <riddopic@gmail.com>
+# License:   Apache License, Version 2.0
+# Copyright: (C) 2014-2015 Stefano Harding
+#
+# 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 'chef'
+require 'chef/log'
+require 'chef/handler'
+require 'garcon'
+
+class DevReporter < Chef::Handler
+  attr_reader :resources, :immediate, :delayed, :always
+
+  def initialize(opts = {})
+  end
+
+  def full_name(resource)
+    "#{resource.resource_name}[#{resource.name}]"
+  end
+
+  def humanize(seconds)
+    [[60, :seconds],
+     [60, :minutes],
+     [24, :hours],
+     [1000, :days] ].map do |count, name|
+      if seconds > 0
+        seconds, n = seconds.divmod(count)
+        "#{n.to_i} #{name}"
+      end
+    end.compact.reverse.join(' ')
+  end
+
+  def banner
+    puts <<-EOF
+
+
+                          .:  ;:. .:;S;:. .:;.;:. .:;S;:. .:;S;:.
+                          S   ' S S  S    S  S    S     S  /
+                          `:;S;:' `:;S;:' `:;S;:' `:;S;:' `:;S;:'
+          .g8"""bgd
+        .dP'     `M
+        dM'       `  ,6"Yb.  `7Mb,od8 ,p6"bo   ,pW"Wq.`7MMpMMMb.
+        MM          8)   MM    MM' "'6M'  OO  6W'   `Wb MM    MM
+        MM.    `7MMF',pm9MM    MM    8M       8M     M8 MM    MM
+        `Mb.     MM 8M   MM    MM    YM.    , YA.   ,A9 MM    MM
+          `"bmmmdPY `Moo9^Yo..JMML.   YMbmd'   `Ybmd9'.JMML  JMML.
+                                        bog
+                                         od              V #{Garcon::VERSION}
+
+    EOF
+  end
+
+  def report
+    if run_status.success?
+      cookbooks = Hash.new(0)
+      recipes   = Hash.new(0)
+      resources = Hash.new(0)
+      run_time  = humanize(run_status.elapsed_time)
+      updates   = run_status.updated_resources.length
+      total     = run_status.all_resources.length
+
+      all_resources.each do |r|
+        cookbooks[r.cookbook_name]                      += r.elapsed_time
+        recipes["#{r.cookbook_name}::#{r.recipe_name}"] += r.elapsed_time
+        resources["#{r.resource_name}[#{r.name}]"]       = r.elapsed_time
+      end
+
+      @max_time = all_resources.max_by { |r| r.elapsed_time      }.elapsed_time
+      @max      = all_resources.max_by { |r| full_name(r).length }
+
+      # Start droping puts because mash grinder faceplanted Chef::Log.info
+      # logging changes random so excelent thanks Chef!
+      #
+      puts ''
+      puts '*|*|*|*|*|*|*|*|*         <`)))><  	'
+      banner
+      puts 'Elapsed Time  Cookbook             Version'.yellow
+      puts '------------  -------------------  ----------------------'.green
+      cookbooks.sort_by { |k, v| -v }.each do |cookbook, run_time|
+        ver = run_context.cookbook_collection[cookbook.to_sym].version
+        puts '%19f  %-20s %-7s' % [run_time, cookbook, ver]
+      end
+      puts ''
+      puts 'Elapsed Time  Recipe'.orange
+      puts '------------  -------------------------------------------'.green
+      recipes.sort_by { |k, v| -v }.each do |recipe, run_time|
+        puts '%19f  %s' % [run_time, recipe]
+      end
+      puts ''
+      puts 'Elapsed Time  Resource'.orange
+      puts '------------  -------------------------------------------'.green
+      resources.sort_by { |k, v| -v }.each do |resource, run_time|
+        puts '%19f  %s' % [run_time, resource]
+      end
+      puts '+-----------------------------------------------------------------'\
+           '-------------+'.purple
+      puts ''
+      puts "Chef Run Completed in #{run_time} on #{node.name}. Updated " \
+           "#{updates} of #{total} resources."
+
+      cookbooks = run_context.cookbook_collection
+      puts
+      puts "Slowest Resource: #{full_name(@max)} (%.6fs)"%[@max_time]
+      puts ''
+      puts Faker::Hacker.say_something_smart
+      puts ''
+    elsif run_status.failed?
+      banner
+      puts "Chef FAILED in #{run_time} on #{node.name} with exception:".orange
+      puts run_status.formatted_exception
+    end
+  end
+end

data/lib/garcon/chef/log.rb

@@ -0,0 +1,64 @@
+# encoding: UTF-8
+#
+# Author:    Stefano Harding <riddopic@gmail.com>
+# License:   Apache License, Version 2.0
+# Copyright: (C) 2014-2015 Stefano Harding
+#
+# 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 'mixlib/log/formatter'
+
+module Mixlib
+  module Log
+    class Formatter < Logger::Formatter
+      # Set Chef::Log::Formatter.show_time == true/false to enable/disable the
+      # printing of the time with the message.
+
+      def call(severity, time, progname, msg)
+        format % [
+          format_datetime(time).blue,
+          format_severity(severity),
+          msg2str(msg).strip
+        ]
+      end
+
+      private #        P R O P R I E T À   P R I V A T A   Vietato L'accesso
+
+      def format
+        "\n[%s] %5s: %s\n"
+      end
+
+      def format_severity(severity)
+        case severity
+        when 'FATAL'
+          severity.bright_red
+        when 'ERROR'
+          severity.red
+        when 'WARN'
+          severity.yellow
+        when 'DEBUG'
+          severity.light_gray
+        when 'INFO'
+          severity.green
+        else
+          severity
+        end
+      end
+
+      def format_datetime(time)
+        time.strftime('%T')
+      end
+    end
+  end
+end

data/lib/garcon/chef/node.rb

@@ -0,0 +1,100 @@
+# encoding: UTF-8
+#
+# Author:    Stefano Harding <riddopic@gmail.com>
+# License:   Apache License, Version 2.0
+# Copyright: (C) 2014-2015 Stefano Harding
+#
+# 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.
+#
+
+# To get to node we have to go through Chef, happy Rubocop?
+#
+class Chef
+  # Slap some dandy handy methods to stick on a node.
+  #
+  class Node
+    # Boolean to check if a recipe is loaded in the run list.
+    #
+    # @param [String] recipe
+    #   the command to find
+    #
+    # @return [TrueClass, FalseClass]
+    #   true if the command is found in the path, false otherwise
+    #
+    # @api public
+    def has_recipe?(recipe)
+      loaded_recipes.include?(with_default(recipe))
+    end
+    alias_method :include_recipe?,  :has_recipe?
+    alias_method :includes_recipe?, :has_recipe?
+
+    # Determine if the current node is in the given Chef environment
+    # (or matches the given regular expression).
+    #
+    # @param [String, Regex] environment
+    #
+    # @return [Boolean]
+    #
+    # @api public
+    def in?(environment)
+      environment === chef_environment
+    end
+
+    # Recursively searchs a nested datastructure for a key and returns the
+    # value. If a block is provided its value will be returned if the key does
+    # not exist, otherwise UndefinedAttributeError is raised.
+    #
+    # @param [Array<String, Symbol>] keys
+    #   the list of keys to kdeep fetch
+    #
+    # @yield optional block to execute if no value is found
+    #
+    # @return [Object]
+    #
+    # @api public
+    def get(*keys, &block)
+      args.reduce(self) do |obj, arg|
+        begin
+          arg = Integer(arg) if obj.is_a? Array
+          obj.fetch(arg)
+        rescue ArgumentError, IndexError, NoMethodError
+          break block.call(arg) if block
+          raise UndefinedAttributeError
+        end
+      end
+    end
+    alias_method :deep_fetch, :get
+
+    private #        P R O P R I E T À   P R I V A T A   Vietato L'accesso
+
+    # Automatically appends "::default" to recipes that need them.
+    #
+    # @param [String] recipe
+    #
+    # @return [String]
+    #
+    # @api private
+    def with_default(recipe)
+      name.include?('::') ? name : "#{recipe}::default"
+    end
+
+    # The list of loaded recipes on the Chef run (normalized)
+    #
+    # @return [Array<String>]
+    #
+    # @api private
+    def loaded_recipes
+      node.run_context.loaded_recipes.map { |name| with_default(name) }
+    end
+  end
+end

data/lib/garcon/chef/provider/civilize.rb

@@ -0,0 +1,209 @@
+# encoding: UTF-8
+#
+# Author:    Stefano Harding <riddopic@gmail.com>
+# License:   Apache License, Version 2.0
+# Copyright: (C) 2014-2015 Stefano Harding
+#
+# 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 'garcon'
+
+class Chef
+  class Resource
+    class Civilize < Chef::Resource
+      include Garcon
+
+      # Chef attributes
+      identity_attr :name
+      provides      :civilize
+
+      # Actions
+      actions        :run
+      default_action :run
+
+      # Attributes
+      attribute :name,
+        kind_of: String
+      attribute :iptables,
+        kind_of: [TrueClass, FalseClass],
+        default: true
+      attribute :selinux,
+        kind_of: [TrueClass, FalseClass],
+        default: true
+      attribute :dotfiles,
+        kind_of: [TrueClass, FalseClass, String, Array],
+        default: true
+      attribute :ruby,
+        kind_of: [TrueClass, FalseClass],
+        default: true
+      attribute :docker,
+        kind_of: Array,
+        default: %w[tar htop initscripts]
+      attribute :rhel_svcs,
+        kind_of: Array,
+        default: %w[
+          autofs avahi-daemon bluetooth cpuspeed cups gpm haldaemon messagebus
+        ]
+    end
+  end
+
+  class Provider
+    class Civilize < Chef::Provider
+      include Garcon
+
+      # Shortcut to new_resource.
+      #
+      alias_method :r, :new_resource
+
+      # Boolean indicating if WhyRun is supported by this provider
+      #
+      # @return [TrueClass, FalseClass]
+      #
+      # @api private
+      def whyrun_supported?
+        true
+      end
+
+      # Load and return the current resource.
+      #
+      # @return [Chef::Provider::Civilize]
+      #
+      # @api private
+      def load_current_resource
+        @current_resource ||= Chef::Resource::Civilize.new(r.name)
+        @current_resource
+      end
+
+      def action_run
+        civilize_docker if docker? && r.docker
+        rhel_services   if r.rhel_svcs
+        iptables        if !docker? && r.iptables
+        selinux         if selinux? && r.selinux
+        ps1prompt
+        dotfiles        if r.dotfiles
+      end
+
+      private #        P R O P R I E T À   P R I V A T A   Vietato L'accesso
+
+      def civilize_docker
+        r.docker.each { |pkg| package pkg }
+      end
+
+      def rhel_services
+        r.rhel_svcs.each do |svc|
+          service(svc) { action [:stop, :disable] }
+        end
+      end
+
+      def iptables
+        shell_out!('iptables -F')
+      end
+
+      def selinux
+        shell_out!('setenforce 0')
+      end
+
+      def ps1prompt
+        ver = run_context.cookbook_collection[cookbook_name].metadata.version
+        t ||= Chef::Resource::Template.new('/etc/profile.d/ps1.sh', run_context)
+        t.cookbook   'garcon'
+        t.owner      'root'
+        t.group      'root'
+        t.mode        00644
+        t.variables   version: ver
+        t.run_action :create
+      end
+
+      def dotfiles
+        users = if r.dotfiles.is_a?(TrueClass)
+                  Array('root')
+                elsif r.dotfiles.respond_to?(:to_ary)
+                  users = r.dotfiles
+                elsif r.dotfiles.respond_to?(:to_str)
+                  users = Array(r.dotfiles)
+                end
+
+        users.each do |user|
+          home = user =~ /root/ ? '/root' : "/home/#{user}"
+          ['.bashrc', '.inputrc'].each do |dot|
+            file = ::File.join(home, dot)
+            f ||= Chef::Resource::CookbookFile.new(file, run_context)
+            f.source      dot[1..-1]
+            f.cookbook   'garcon'
+            f.backup      false
+            f.owner       user
+            f.group       user
+            f.mode        00644
+            f.run_action :create
+          end
+        end
+      end
+    end
+  end
+end
+
+# Chef::Platform mapping for resource and providers
+#
+# @return [undefined]
+#
+# @api private.
+Chef::Platform.set(
+  platform: :amazon,
+  resource: :civilize,
+  provider:  Chef::Provider::Civilize
+)
+
+# Chef::Platform mapping for resource and providers
+#
+# @return [undefined]
+#
+# @api private.
+Chef::Platform.set(
+  platform: :centos,
+  resource: :civilize,
+  provider:  Chef::Provider::Civilize
+)
+
+# Chef::Platform mapping for resource and providers
+#
+# @return [undefined]
+#
+# @api private.
+Chef::Platform.set(
+  platform: :oracle,
+  resource: :civilize,
+  provider:  Chef::Provider::Civilize
+)
+
+# Chef::Platform mapping for resource and providers
+#
+# @return [undefined]
+#
+# @api private.
+Chef::Platform.set(
+  platform: :redhat,
+  resource: :civilize,
+  provider:  Chef::Provider::Civilize
+)
+
+# Chef::Platform mapping for resource and providers
+#
+# @return [undefined]
+#
+# @api private.
+Chef::Platform.set(
+  platform: :scientific,
+  resource: :civilize,
+  provider:  Chef::Provider::Civilize
+)