Sha256: 04885ed84e3020926ab3f1f9a71525489b229e7806eaed3d9e224d960e46c6bc
Contents?: true
Size: 1.69 KB
Versions: 9
Compression:
Stored size: 1.69 KB
Contents
require 'retrospec/resource'
class Conditional
attr_reader :test, :value, :statements
# things I need: a key/value store for variables
# types of variables
# those that can be changed
# those that can be influenced (facts, other variables that contain variables)
# takes a subtype of Puppet::AST::Branch that contains conditional logic
def initialize(branch, parameters)
@statements = branch.statements
end
# get the attributes for the given resources found in the type code passed in
# this will return a array of hashes, one for each resource found
def self.all(type)
r_attrs = []
generate_conditionals(type).each do |c|
r_attrs << Resource.all(c.statements)
end
r_attrs.flatten
end
# a array of types the are known to contain conditional code and statements
def self.types
#test, statement, value
# if I don't have a statement that I am part of a bigger code block
# [Puppet::Parser::AST::IfStatement, Puppet::Parser::AST::CaseStatement, Puppet::Parser::AST::Else,
# Puppet::Parser::AST::CaseOpt, Puppet::Parser::AST::Selector]
[Puppet::Parser::AST::IfStatement, Puppet::Parser::AST::Else]
end
# returns a array of branch subtypes
def self.find_conditionals(type)
conds = []
if type.code.respond_to?(:find_all)
conds = type.code.find_all {|c| types.include?(c.class) }
end
conds
end
# find and create an array of conditionals
# we need the type so we can look through the code to find conditional statements
def self.generate_conditionals(type)
conditionals = []
find_conditionals(type).each do |cond|
conditionals << Conditional.new(cond, type.arguments)
end
conditionals
end
end
Version data entries
9 entries across 9 versions & 1 rubygems