Module: Puppet::Util::Yaml

Defined in:
lib/puppet/util/yaml.rb

Defined Under Namespace

Classes: YamlLoadError

Constant Summary collapse

YamlLoadExceptions =
[::StandardError, ::Psych::Exception]

Class Method Summary collapse

Class Method Details

.dump(structure, filename) ⇒ Object


45
46
47
48
49
# File 'lib/puppet/util/yaml.rb', line 45

def self.dump(structure, filename)
  Puppet::FileSystem.replace_file(filename, 0660) do |fh|
    YAML.dump(structure, fh)
  end
end

.safe_load(yaml, allowed_classes = [], filename = nil) ⇒ Object

Safely load the content as YAML. By default only the following classes can be deserialized:

  • TrueClass

  • FalseClass

  • NilClass

  • Numeric

  • String

  • Array

  • Hash

Attempting to deserialize other classes will raise an YamlLoadError exception unless they are specified in the array of allowed_classes.

Parameters:

  • yaml (String)

    The yaml content to parse.

  • allowed_classes (Array) (defaults to: [])

    Additional list of classes that can be deserialized.

  • filename (String) (defaults to: nil)

    The filename to load from, used if an exception is raised.

Returns:

  • The parsed YAML, which can be Hash, Array or scalar types.

Raises:

  • (YamlLoadException)

    If deserialization fails.


26
27
28
29
30
31
32
33
34
35
# File 'lib/puppet/util/yaml.rb', line 26

def self.safe_load(yaml, allowed_classes = [], filename = nil)
  data = YAML.safe_load(yaml, allowed_classes, [], true, filename)
  data = false if data.nil?
  data
rescue ::Psych::DisallowedClass => detail
  path = filename ? "(#{filename})" : "(<unknown>)"
  raise YamlLoadError.new("#{path}: #{detail.message}", detail)
rescue *YamlLoadExceptions => detail
  raise YamlLoadError.new(detail.message, detail)
end

.safe_load_file(filename, allowed_classes = []) ⇒ Object

Safely load the content from a file as YAML.

See Also:


40
41
42
43
# File 'lib/puppet/util/yaml.rb', line 40

def self.safe_load_file(filename, allowed_classes = [])
  yaml = Puppet::FileSystem.read(filename, :encoding => 'bom|utf-8')
  safe_load(yaml, allowed_classes, filename)
end