garcun 0.0.2

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

@@ -0,0 +1,265 @@
+# 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 HouseKeeping < Chef::Resource
+      include Garcon
+
+      # Chef attributes
+      identity_attr :path
+      provides :house_keeping
+
+      # Actions
+      actions :purge
+      default_action :purge
+
+      # Attributes
+      attribute :path,
+        kind_of:String,
+        callbacks:{ path: ->(path) { validate_path(path, :dir) }},
+        name_attribute: true
+      attribute :exclude,
+        kind_of: [String, Regexp],
+        default: nil
+      attribute :manage_symlink_source,
+        kind_of: [TrueClass, FalseClass],
+        default: true
+      attribute :force_unlink,
+        kind_of: [TrueClass, FalseClass],
+        default: true
+      attribute :age,
+        kind_of: Integer,
+        default: nil
+      attribute :size,
+        kind_of: String,
+        regex:   /\d+[kmgt]{1}b{0,1}/i,
+        default: nil
+      attribute :directory_size,
+        kind_of: String,
+        regex:   /\d+[kmgtpe]{1}b{0,1}/i,
+        default: nil
+      attribute :recursive,
+        kind_of: [TrueClass, FalseClass],
+        default: false
+    end
+  end
+
+  class Provider
+    class HouseKeeping < Chef::Provider
+      include Garcon
+
+      provides :house_keeping
+
+      # 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::HouseKeeping]
+      #
+      # @api private
+      def load_current_resource(r = new_resource)
+        @current_resource ||= PurgeFlushBinge.new(r)
+        @current_resource.exclude(r.exclude) if r.exclude
+        @current_resource
+      end
+
+      def action_purge(r = new_resource)
+        if @current_resource.num_files > 1
+          path  = r.path
+          age   = r.age
+          size  = r.size
+          dsize = r.directory_size
+          num   = @current_resource.num_files
+          purge = {}
+          purge.merge!(@current_resource.older_than(age))    unless age.nil?
+          purge.merge!(@current_resource.larger_than(size))  unless size.nil?
+          purge.merge!(@current_resource.to_dir_size(dsize)) unless dsize.nil?
+          num_purged = purge.length
+
+          converge_by "Processed #{num} files from #{path}, " \
+                      "#{num_purged} purged" do
+            if purge.length > 1
+              purge.each do |file, data|
+                file file do
+                  force_unlink          r.force_unlink
+                  manage_symlink_source r.manage_symlink_source
+                  backup  false
+                  action :delete
+                end
+              end
+              r.updated_by_last_action(true)
+            end
+          end
+        end
+      end
+
+      private #   P R O P R I E T À   P R I V A T A   Vietato L'accesso
+
+      class PurgeFlushBinge
+        def initialize(new_resource)
+          search_path = if new_resource.recursive
+            ::File.join(new_resource.path, '**', '*')
+          else
+            ::File.join(new_resource.path, '*')
+          end
+          @file_map = Hash.new
+          @dir_size = 0
+
+          Dir[search_path].select { |f| ::File.file?(f) }.map do |file|
+            fstat = ::File.stat(file)
+            @dir_size += fstat.size
+            @file_map.store(file,
+              ctime: fstat.ctime.to_i,
+              mtime: fstat.mtime.to_i,
+              size:  fstat.size)
+          end
+        end
+
+        def older_than(days)
+          purge_time = Time.now - (60 * 60 * 24 * days.to_i)
+          @file_map.select { |file, data| data[:mtime] < purge_time.to_i }
+        end
+
+        def larger_than(size)
+          @file_map.select do |file, data|
+            c = Humanize.new(size.to_s)
+            data[:size].to_f > c.to_size(:b).to_f
+          end
+        end
+
+        def to_dir_size(size)
+          sorted = @file_map.sort_by { |k,v| v[:mtime] }
+          c = Humanize.new(size.to_s)
+          delete_size = @dir_size - c.to_size(:b).to_f
+          return {} if delete_size <= 0
+
+          f_size = 0
+          list   = {}
+          sorted.each do |f|
+            list[f[0]] = f[1]
+            f_size += f[1][:size]
+            break if f_size >= delete_size
+          end
+          list
+        end
+
+        def exclude(regexp)
+          @file_map.delete_if { |file| file.match(Regexp.new(regexp)) }
+        end
+
+        def num_files
+          @file_map.length
+        end
+      end
+
+      class Humanize
+        def initialize(size)
+          @units = { b: 1, kb: 1024**1, mb: 1024**2, gb: 1024**3, tb: 1024**4 }
+          @size_int = size.partition(/\D{1,2}/).at(0).to_i
+          unit = size.partition(/\D{1,2}/).at(1).to_s.downcase
+          case
+          when unit.match(/[kmgtpe]{1}/)
+            @size_unit = unit.concat('b')
+          when unit.match(/[kmgtpe]{1}b/)
+            @size_unit = unit
+          else
+            @size_unit = 'b'
+          end
+        end
+
+        def to_size(unit, places = 1)
+          unit_val = @units[unit.to_s.downcase.to_sym]
+          bytes    = @size_int * @units[@size_unit.to_sym]
+          size     = bytes.to_f / unit_val.to_f
+          sprintf("%.#{places}f", size).to_f
+        end
+
+        def from_size(places = 1)
+          unit_val = @units[@size_unit.to_s.downcase.to_sym]
+          sprintf("%.#{places}f", @size_int * unit_val)
+        end
+      end
+    end
+  end
+end
+
+# Chef::Platform mapping for resource and providers
+#
+# @return [undefined]
+#
+# @api private.
+Chef::Platform.set(
+  platform: :amazon,
+  resource: :house_keeping,
+  provider:  Chef::Provider::HouseKeeping
+)
+
+# Chef::Platform mapping for resource and providers
+#
+# @return [undefined]
+#
+# @api private.
+Chef::Platform.set(
+  platform: :centos,
+  resource: :house_keeping,
+  provider:  Chef::Provider::HouseKeeping
+)
+
+# Chef::Platform mapping for resource and providers
+#
+# @return [undefined]
+#
+# @api private.
+Chef::Platform.set(
+  platform: :oracle,
+  resource: :house_keeping,
+  provider:  Chef::Provider::HouseKeeping
+)
+
+# Chef::Platform mapping for resource and providers
+#
+# @return [undefined]
+#
+# @api private.
+Chef::Platform.set(
+  platform: :redhat,
+  resource: :house_keeping,
+  provider:  Chef::Provider::HouseKeeping
+)
+
+# Chef::Platform mapping for resource and providers
+#
+# @return [undefined]
+#
+# @api private.
+Chef::Platform.set(
+  platform: :scientific,
+  resource: :house_keeping,
+  provider:  Chef::Provider::HouseKeeping
+)

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

@@ -0,0 +1,31 @@
+# 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'
+
+# Cache the node object for profit and gross revenue margins.
+#
+class Chef::Resource::NodeCache < Chef::Resource
+  include Garcon(blender: true)
+  attribute(:name,  kind_of: String, name_attribute: true)
+  attribute(:cache, kinf_of: Hash,   default: lazy { node })
+  action(:run) do
+    Garcon.config.stash.load(node: new_resource.node)
+  end
+end

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

@@ -0,0 +1,183 @@
+# 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'
+require 'digest/md5'
+
+class Chef
+  class Resource
+    class Partial < Chef::Resource
+      include Garcon
+
+      # Chef attributes
+      identity_attr :name
+      provides :partial
+
+      # Actions
+      actions :run
+      default_action :run
+
+      # Attributes
+      attribute :run_list,
+        kind_of: String
+      attribute :save,
+        kind_of: [TrueClass, FalseClass],
+        default: true
+      attribute :arguments,
+        kind_of: Hash
+      attribute :attributes,
+        kind_of: Hash
+    end
+  end
+
+  class Provider
+    class Partial < Chef::Provider
+      include Garcon
+
+      # 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::Partial]
+      #
+      # @api private
+      def load_current_resource
+        @current_resource ||= Chef::Resource::Partial.new(new_resource.name)
+        @current_resource
+      end
+
+      def action_run(r = new_resource)
+        converge_by "Executing partial run list: #{r.run_list}" do
+          partial = r.name
+          runner  = Chef::PartialRun.new(partial, r.attributes, r.arguments)
+          runner.partial_run(r.run_list)
+          unless runner.run_status.updated_resources.empty?
+            r.updated_by_last_action(true)
+          end
+        end
+
+        if r.save
+          node.consume_attributes(runner.clean_attrs)
+          runner.save_updated_node
+        end
+      end
+    end
+  end
+
+  class PartialRun < Client
+    def initialize(partial, attributes = nil, arguments = {})
+      super(attributes, arguments)
+      @partial = partial
+    end
+
+    def partial_run(run_list)
+      cache_path = file_cache_path
+      begin
+        digest = Digest::MD5.hexdigest(@partial)
+        Chef::Config[:file_cache_path] = file_cache_path("partial-#{digest}")
+        run_ohai
+        register unless Chef::Config[:solo]
+        load_node
+        run_list_items = run_list.split(',').collect do |item|
+          Chef::RunList::RunListItem.new(item)
+        end
+        node.run_list(*run_list_items)
+        build_node
+        run_context = setup_run_context
+        converge(run_context)
+      ensure
+        Chef::Config[:file_cache_path] = cache_path
+      end
+    end
+
+    # Clean the node data
+    #
+    def clean_attrs
+      data = node.to_hash
+      %w(run_list recipes roles).each { |k| data.delete(k) }
+      data
+    end
+
+    def run_completed_successfully
+      # do nothing
+    end
+  end
+end
+
+# Chef::Platform mapping for resource and providers
+#
+# @return [undefined]
+#
+# @api private.
+Chef::Platform.set(
+  platform: :amazon,
+  resource: :partial,
+  provider:  Chef::Provider::Partial
+)
+
+# Chef::Platform mapping for resource and providers
+#
+# @return [undefined]
+#
+# @api private.
+Chef::Platform.set(
+  platform: :centos,
+  resource: :partial,
+  provider:  Chef::Provider::Partial
+)
+
+# Chef::Platform mapping for resource and providers
+#
+# @return [undefined]
+#
+# @api private.
+Chef::Platform.set(
+  platform: :oracle,
+  resource: :partial,
+  provider:  Chef::Provider::Partial
+)
+
+# Chef::Platform mapping for resource and providers
+#
+# @return [undefined]
+#
+# @api private.
+Chef::Platform.set(
+  platform: :redhat,
+  resource: :partial,
+  provider:  Chef::Provider::Partial
+)
+
+# Chef::Platform mapping for resource and providers
+#
+# @return [undefined]
+#
+# @api private.
+Chef::Platform.set(
+  platform: :scientific,
+  resource: :partial,
+  provider:  Chef::Provider::Partial
+)