summaryrefslogtreecommitdiff
path: root/lib/chef/guard_interpreter/resource_guard_interpreter.rb
blob: c4b6d33a16128e8858adf5368246474ad81d5575 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
#
# Author:: Adam Edwards (<adamed@getchef.com>)
# Copyright:: Copyright (c) 2014 Opscode, 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 'chef/guard_interpreter/default_guard_interpreter'

class Chef
  class GuardInterpreter
    class ResourceGuardInterpreter < DefaultGuardInterpreter

      def translate_command_block(command, opts, &block)
        merge_inherited_attributes

        if command && ! block_given?
          block_attributes = opts.merge({:code => command})

          # Handles cases like powershell_script where default
          # attributes are different when used in a guard vs. not. For
          # powershell_script in particular, this will go away when
          # the one attribue that causes this changes its default to be
          # the same after some period to prepare for deprecation
          if @resource.class.respond_to?(:get_default_attributes)
            block_attributes = @resource.class.send(:get_default_attributes, opts).merge(block_attributes)
          end

          translated_block = to_block(block_attributes)
          [nil, translated_block]
        else
          super
        end
      end

      def initialize(resource_symbol, parent_resource)
        @parent_resource = parent_resource

        resource_class = get_resource_class(parent_resource, resource_symbol)

        raise ArgumentError, "Specified guard_interpreter resource #{resource_symbol.to_s} unknown for this platform" if resource_class.nil?

        empty_events = Chef::EventDispatch::Dispatcher.new
        anonymous_run_context = Chef::RunContext.new(parent_resource.node, {}, empty_events)

        @resource = resource_class.new('Guard resource', anonymous_run_context)

        if ! @resource.kind_of?(Chef::Resource::Script)
          raise ArgumentError, "Specified guard interpreter class #{resource_class} must be a kind of Chef::Resource::Script resource"
        end
      end

      protected

      def evaluate_action(action=nil, &block)
        @resource.instance_eval(&block)

        run_action = action || @resource.action

        begin
          @resource.run_action(run_action)
          resource_updated = @resource.updated
        rescue Mixlib::ShellOut::ShellCommandFailed
          resource_updated = nil
        end

        resource_updated
      end

      def to_block(attributes, action=nil)
        resource_block = block_from_attributes(attributes)
        Proc.new do
          evaluate_action(action, &resource_block)
        end
      end

      def get_resource_class(parent_resource, resource_symbol)
        if parent_resource.nil? || parent_resource.node.nil?
          raise ArgumentError, "Node for guard resource parent must not be nil"
        end
        Chef::Resource.resource_for_node(resource_symbol, parent_resource.node)
      end

      def block_from_attributes(attributes)
        Proc.new do
          attributes.keys.each do |attribute_name|
            send(attribute_name, attributes[attribute_name]) if respond_to?(attribute_name)
          end
        end
      end

      def merge_inherited_attributes
        inherited_attributes = []

        if @parent_resource.class.respond_to?(:guard_inherited_attributes)
          inherited_attributes = @parent_resource.class.send(:guard_inherited_attributes)
        end

        if inherited_attributes && !inherited_attributes.empty?
          inherited_attributes.each do |attribute|
            if @parent_resource.respond_to?(attribute) && @resource.respond_to?(attribute)
              parent_value = @parent_resource.send(attribute)
              child_value = @resource.send(attribute)
              if parent_value || child_value
                @resource.send(attribute, parent_value)
              end
            end
          end
        end
      end
    end
  end
end