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
|
# frozen_string_literal: true
$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', '..'))
require 'puppet/provider/elastic_parsedfile'
require 'puppet/util/package'
require 'puppet_x/elastic/hash'
# Provider for yaml-based Elasticsearch configuration files.
class Puppet::Provider::ElasticYaml < Puppet::Provider::ElasticParsedFile
class << self
attr_accessor :metadata
end
# Transform a given string into a Hash-based representation of the
# provider.
def self.parse(text)
yaml = YAML.safe_load text
if yaml
yaml.map do |key, metadata|
{
:name => key,
:ensure => :present,
@metadata => metadata
}
end
else
[]
end
end
# Transform a given list of provider records into yaml-based
# representation.
def self.to_file(records)
yaml = records.map do |record|
# Convert top-level symbols to strings
record.transform_keys(&:to_s)
end
yaml = yaml.reduce({}) do |hash, record|
# Flatten array of hashes into single hash
hash.merge(record['name'] => record.delete(@metadata.to_s))
end
yaml = yaml.extend(Puppet_X::Elastic::SortedHash).to_yaml.split("\n")
yaml.shift if yaml.first =~ %r{---}
yaml = yaml.join("\n")
yaml << "\n"
end
def self.skip_record?(_record)
false
end
# This is ugly, but it's overridden in ParsedFile with abstract
# functionality we don't need for our simple provider class.
# This has been observed to break in Puppet version 3/4 switches.
def self.valid_attr?(klass, attr_name)
klass.is_a? Class ? klass.parameters.include?(attr_name) : true
end
end
|
