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
|
# frozen_string_literal: true
require 'puppet/resource_api/transport'
require 'hocon'
require 'hocon/config_syntax'
# Puppet::ResourceApi::Transport::Wrapper` to interface between the Util::NetworkDevice
class Puppet::ResourceApi::Transport::Wrapper
attr_reader :transport, :schema
def initialize(name, url_or_config_or_transport)
if url_or_config_or_transport.is_a? String
url = URI.parse(url_or_config_or_transport)
raise "Unexpected url '#{url_or_config_or_transport}' found. Only file:/// URLs for configuration supported at the moment." unless url.scheme == 'file'
raise "Trying to load config from '#{url.path}, but file does not exist." if url && !File.exist?(url.path)
config = self.class.deep_symbolize(Hocon.load(url.path, syntax: Hocon::ConfigSyntax::HOCON) || {})
elsif url_or_config_or_transport.is_a? Hash
config = url_or_config_or_transport
elsif transport_class?(name, url_or_config_or_transport)
@transport = url_or_config_or_transport
end
@transport ||= Puppet::ResourceApi::Transport.connect(name, config)
@schema = Puppet::ResourceApi::Transport.list[name]
end
def transport_class?(name, transport)
class_name = name.split('_').map { |e| e.capitalize }.join
expected = Puppet::Transport.const_get(class_name).to_s
expected == transport.class.to_s
end
def facts
context = Puppet::ResourceApi::PuppetContext.new(@schema)
# @transport.facts + custom_facts # look into custom facts work by TP
@transport.facts(context)
end
def respond_to_missing?(name, _include_private)
(@transport.respond_to? name) || super
end
def method_missing(method_name, *args, &block)
if @transport.respond_to? method_name
@transport.send(method_name, *args, &block)
else
super
end
end
# From https://stackoverflow.com/a/11788082/4918
def self.deep_symbolize(obj)
return obj.each_with_object({}) { |(k, v), memo| memo[k.to_sym] = deep_symbolize(v); } if obj.is_a? Hash
return obj.each_with_object([]) { |v, memo| memo << deep_symbolize(v); } if obj.is_a? Array
obj
end
end
|
