blob: c15836838eec1325a500e66411666be75b895de9 (
plain)
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
|
require 'psych/handler'
module Psych
###
# This class builds an in-memory parse tree tree that represents a YAML
# document.
#
# See Psych::Handler for documentation on the event methods used in this
# class.
class TreeBuilder < Psych::Handler
def initialize
@stack = []
end
def root
@stack.first
end
%w{
Document
Sequence
Mapping
}.each do |node|
class_eval %{
def start_#{node.downcase}(*args)
super
n = Nodes::#{node}.new(*args)
@stack.last.children << n
@stack.push n
end
def end_#{node.downcase}(*args)
super
@stack.pop
end
}
end
def start_stream encoding
super
@stack.push Nodes::Stream.new(encoding)
end
def scalar(*args)
super
@stack.last.children << Nodes::Scalar.new(*args)
end
def alias(*args)
super
@stack.last.children << Nodes::Alias.new(*args)
end
end
end
|