-
Notifications
You must be signed in to change notification settings - Fork 580
/
loadjson.rb
65 lines (59 loc) · 2.45 KB
/
loadjson.rb
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
62
63
64
65
# frozen_string_literal: true
#
# loadjson.rb
#
module Puppet::Parser::Functions
newfunction(:loadjson, type: :rvalue, arity: -2, doc: <<-'DOC') do |args|
@summary
Load a JSON file containing an array, string, or hash, and return the data
in the corresponding native data type.
The first parameter can be a file path or a URL.
The second parameter is the default value. It will be returned if the file
was not found or could not be parsed.
@return [Array|String|Hash]
The data stored in the JSON file, the type depending on the type of data that was stored.
@example Example Usage:
$myhash = loadjson('/etc/puppet/data/myhash.json')
$myhash = loadjson('https://example.local/my_hash.json')
$myhash = loadjson('https://username:password@example.local/my_hash.json')
$myhash = loadjson('no-file.json', {'default' => 'value'})
DOC
raise ArgumentError, 'Wrong number of arguments. 1 or 2 arguments should be provided.' unless args.length >= 1
require 'open-uri'
begin
if args[0].start_with?('http://', 'https://')
http_options = {}
if (match = args[0].match(%r{(http\://|https\://)(.*):(.*)@(.*)}))
# If URL is in the format of https://username:password@example.local/my_hash.yaml
protocol, username, password, path = match.captures
url = "#{protocol}#{path}"
http_options[:http_basic_authentication] = [username, password]
elsif (match = args[0].match(%r{(http\:\/\/|https\:\/\/)(.*)@(.*)}))
# If URL is in the format of https://username@example.local/my_hash.yaml
protocol, username, path = match.captures
url = "#{protocol}#{path}"
http_options[:http_basic_authentication] = [username, '']
else
url = args[0]
end
begin
contents = OpenURI.open_uri(url, **http_options)
rescue OpenURI::HTTPError => err
res = err.io
warning("Can't load '#{url}' HTTP Error Code: '#{res.status[0]}'")
args[1]
end
PSON.load(contents) || args[1]
elsif File.exists?(args[0]) # rubocop:disable Lint/DeprecatedClassMethods : Changing to .exist? breaks the code
content = File.read(args[0])
PSON.load(content) || args[1]
else
warning("Can't load '#{args[0]}' File does not exist!")
args[1]
end
rescue StandardError => e
raise e unless args[1]
args[1]
end
end
end