5d081e6a62c801f85b0a559928b484b76c08f3fa
[mirror/dsa-puppet.git] / 3rdparty / modules / stdlib / lib / puppet / parser / functions / parseyaml.rb
1 #
2 # parseyaml.rb
3 #
4 module Puppet::Parser::Functions
5   newfunction(:parseyaml, :type => :rvalue, :doc => <<-DOC
6     @summary
7       This function accepts YAML as a string and converts it into the correct
8       Puppet structure.
9
10     @return
11       converted YAML into Puppet structure
12
13     > *Note:*
14       The optional second argument can be used to pass a default value that will
15       be returned if the parsing of YAML string have failed.
16   DOC
17              ) do |arguments|
18     raise ArgumentError, 'Wrong number of arguments. 1 or 2 arguments should be provided.' unless arguments.length >= 1
19     require 'yaml'
20
21     begin
22       YAML.load(arguments[0]) || arguments[1] # rubocop:disable Security/YAMLLoad : using YAML.safe_load causes the code to break
23       # in ruby 1.9.3 Psych::SyntaxError is a RuntimeException
24       # this still needs to catch that and work also on rubies that
25       # do not have Psych available.
26     rescue StandardError, Psych::SyntaxError => e # rubocop:disable Lint/ShadowedException : See above
27       raise e unless arguments[1]
28       arguments[1]
29     end
30   end
31 end
32
33 # vim: set ts=2 sw=2 et :