Sha256: 917381c4b5e256fe1aaa57756a17ae2a6457ffd8a670ee793b44992f0c1bc9f1
Contents?: true
Size: 1.15 KB
Versions: 16
Compression:
Stored size: 1.15 KB
Contents
# frozen_string_literal: true
require 'psych'
module Hako
class YamlLoader
def initialize
@current_path = nil
end
def load(path)
class_loader = Psych::ClassLoader.new
scanner = Psych::ScalarScanner.new(class_loader)
prev_path = @current_path
@current_path = path
visitor = Visitor.new(scanner, class_loader) do |_, val|
load(@current_path.parent.join(val))
end
path.open do |f|
visitor.accept(Psych.parse(f))
end
ensure
@current_path = prev_path
end
class Visitor < Psych::Visitors::ToRuby
INCLUDE_TAG = 'tag:include'
SHOVEL = '<<'
def initialize(scanner, class_loader, &block)
super(scanner, class_loader)
@domain_types[INCLUDE_TAG] = [INCLUDE_TAG, block]
end
def revive_hash(hash, o)
super(hash, o).tap do |r|
if r[SHOVEL].is_a?(Hash)
h2 = {}
r.each do |k, v|
if k == SHOVEL
h2.merge!(v)
else
h2[k] = v
end
end
r.replace(h2)
end
end
end
end
end
end
Version data entries
16 entries across 16 versions & 1 rubygems