| Current Path : /usr/share/ruby/vendor_ruby/puppet/node/ |
Linux gator3171.hostgator.com 4.19.286-203.ELK.el7.x86_64 #1 SMP Wed Jun 14 04:33:55 CDT 2023 x86_64 |
| Current File : //usr/share/ruby/vendor_ruby/puppet/node/facts.rb |
require 'time'
require 'puppet/node'
require 'puppet/indirector'
require 'puppet/util/pson'
# Manage a given node's facts. This either accepts facts and stores them, or
# returns facts for a given node.
class Puppet::Node::Facts
# Set up indirection, so that nodes can be looked for in
# the node sources.
extend Puppet::Indirector
extend Puppet::Util::Pson
# We want to expire any cached nodes if the facts are saved.
module NodeExpirer
def save(instance, key = nil, options={})
Puppet::Node.indirection.expire(instance.name, options)
super
end
end
indirects :facts, :terminus_setting => :facts_terminus, :extend => NodeExpirer
attr_accessor :name, :values
def add_local_facts
values["clientcert"] = Puppet.settings[:certname]
values["clientversion"] = Puppet.version.to_s
values["clientnoop"] = Puppet.settings[:noop]
end
def initialize(name, values = {})
@name = name
@values = values
add_timestamp
end
def initialize_from_hash(data)
@name = data['name']
@values = data['values']
# Timestamp will be here in YAML
timestamp = data['values']['_timestamp']
@values.delete_if do |key, val|
key =~ /^_/
end
#Timestamp will be here in pson
timestamp ||= data['timestamp']
timestamp = Time.parse(timestamp) if timestamp.is_a? String
self.timestamp = timestamp
self.expiration = data['expiration']
if expiration.is_a? String
self.expiration = Time.parse(expiration)
end
end
# Convert all fact values into strings.
def stringify
values.each do |fact, value|
values[fact] = value.to_s
end
end
# Sanitize fact values by converting everything not a string, boolean
# numeric, array or hash into strings.
def sanitize
values.each do |fact, value|
values[fact] = sanitize_fact value
end
end
def ==(other)
return false unless self.name == other.name
strip_internal == other.send(:strip_internal)
end
def self.from_data_hash(data)
new_facts = allocate
new_facts.initialize_from_hash(data)
new_facts
end
def self.from_pson(data)
Puppet.deprecation_warning("from_pson is being removed in favour of from_data_hash.")
self.from_data_hash(data)
end
def to_data_hash
result = {
'name' => name,
'values' => strip_internal,
}
if timestamp
if timestamp.is_a? Time
result['timestamp'] = timestamp.iso8601(9)
else
result['timestamp'] = timestamp
end
end
if expiration
if expiration.is_a? Time
result['expiration'] = expiration.iso8601(9)
else
result['expiration'] = expiration
end
end
result
end
# Add internal data to the facts for storage.
def add_timestamp
self.timestamp = Time.now
end
def timestamp=(time)
self.values['_timestamp'] = time
end
def timestamp
self.values['_timestamp']
end
# Strip out that internal data.
def strip_internal
newvals = values.dup
newvals.find_all { |name, value| name.to_s =~ /^_/ }.each { |name, value| newvals.delete(name) }
newvals
end
private
def sanitize_fact(fact)
if fact.is_a? Hash then
ret = {}
fact.each_pair { |k,v| ret[sanitize_fact k]=sanitize_fact v }
ret
elsif fact.is_a? Array then
fact.collect { |i| sanitize_fact i }
elsif fact.is_a? Numeric \
or fact.is_a? TrueClass \
or fact.is_a? FalseClass \
or fact.is_a? String
fact
else
fact.to_s
end
end
end