-
Notifications
You must be signed in to change notification settings - Fork 311
/
yaml_erb_parser.rb
48 lines (42 loc) · 1.46 KB
/
yaml_erb_parser.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
require 'yaml'
require 'erb'
require 'pathname'
module Hashie
module Extensions
module Parsers
class YamlErbParser
def initialize(file_path, options = {})
@content = File.read(file_path)
@file_path = file_path.is_a?(Pathname) ? file_path.to_s : file_path
@options = options
end
def perform
template = ERB.new(@content)
template.filename = @file_path
permitted_classes = @options.fetch(:permitted_classes) { [] }
permitted_symbols = @options.fetch(:permitted_symbols) { [] }
aliases = @options.fetch(:aliases) { true }
yaml_safe_load(template, permitted_classes, permitted_symbols, aliases)
end
def self.perform(file_path, options = {})
new(file_path, options).perform
end
private
if Gem::Version.new(Psych::VERSION) >= Gem::Version.new('3.1.0') # Ruby 2.6+
def yaml_safe_load(template, permitted_classes, permitted_symbols, aliases)
YAML.safe_load(
template.result,
permitted_classes: permitted_classes,
permitted_symbols: permitted_symbols,
aliases: aliases
)
end
else
def yaml_safe_load(template, permitted_classes, permitted_symbols, aliases)
YAML.safe_load(template.result, permitted_classes, permitted_symbols, aliases)
end
end
end
end
end
end