Module: AwesomePrint::Puppet

Defined in:
lib/awesome_print/ext/awesome_puppet.rb

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.included(base) ⇒ Object 



19
20
21
22
 
# File 'lib/awesome_print/ext/awesome_puppet.rb', line 19

def self.included(base)
  base.send :alias_method, :cast_without_puppet_resource, :cast
  base.send :alias_method, :cast, :cast_with_puppet_resource
end

Instance Method Details

#awesome_puppet_resource(object) ⇒ Object 



41
42
43
44
45
 
# File 'lib/awesome_print/ext/awesome_puppet.rb', line 41

def awesome_puppet_resource(object)
  return '' if object.nil?
  resource_object = object.to_ral
  awesome_puppet_type(resource_object)
end

#awesome_puppet_type(object) ⇒ Object 



47
48
49
50
51
52
53
54
55
56
57
58
 
# File 'lib/awesome_print/ext/awesome_puppet.rb', line 47

def awesome_puppet_type(object)
  return '' if object.nil?
  return object.to_s unless object.respond_to?(:name) && object.respond_to?(:title) && object.respond_to?(:to_hash)
  if Array.new.respond_to?(:to_h)
    # to_h is only supported in ruby 2.1+
    h = object.to_hash.merge(name: object.name, title: object.title).sort.to_h
  else
    h = object.to_hash.merge(name: object.name, title: object.title)
  end
  res_str = awesome_hash(JSON.parse(h.to_json)) #converting to json removes symbols
  "#{object.class} #{res_str}"
end

#cast_with_puppet_resource(object, type) ⇒ Object

this tells ap how to cast our object so we can be specific about printing different puppet objects



26
27
28
29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/awesome_print/ext/awesome_puppet.rb', line 26

def cast_with_puppet_resource(object, type)
  cast = cast_without_puppet_resource(object, type)
  # check the object to see if it has an acestor (< ) of the specified type
  if defined?(::Puppet::Type) && (object.class < ::Puppet::Type)
    cast = :puppet_type
  elsif defined?(::Puppet::Pops::Types) && (object.class < ::Puppet::Pops::Types)
    cast = :puppet_type
  elsif defined?(::Puppet::Parser::Resource) && (object.class < ::Puppet::Parser::Resource)
    cast = :puppet_resource
  elsif /Puppet::Pops::Types/.match(object.class.to_s)
    cast = :puppet_type
  end
  cast
end