summaryrefslogtreecommitdiff
path: root/lib/dslkeywords/only_when.rb
blob: 2f493b2321b876783f6914d448f5a210297eb4f4 (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
module RCM
  # OnlyWhen (e.g. run on host foo)
  class OnlyWhen
    require 'socket'

    def initialize
      @conds = {}
    end

    def is(arg)
      arg
    end

    def method_missing(method_name, *args, &block)
      @conds[method_name] = args.first
    end

    def respond_to_missing?
      true
    end

    def met?
      return false if @conds.key?(:hostname) && Socket.gethostname != @conds[:hostname].to_s

      true
    end
  end

  # Add 'only_when' to DSL
  class DSL
    def only_when(&block)
      conds = OnlyWhen.new
      conds.instance_eval(&block)
      @conds_met = conds.met?
    end
  end
end