Class: SafeYAML::PsychHandler

Inherits:
Psych::Handler
  • Object
show all
Defined in:
lib/safe_yaml/psych_handler.rb

Instance Method Summary collapse

Constructor Details

#initialize(options, &block) ⇒ PsychHandler

Returns a new instance of PsychHandler.



6
7
8
9
10
11
12
13
14
 
# File 'lib/safe_yaml/psych_handler.rb', line 6

def initialize(options, &block)
  @options      = SafeYAML::OPTIONS.merge(options || {})
  @block        = block
  @initializers = @options[:custom_initializers] || {}
  @anchors      = {}
  @stack        = []
  @current_key  = nil
  @result       = nil
end

Instance Method Details

#add_to_current_structure(value, anchor = nil, quoted = nil, tag = nil) ⇒ Object 



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
 
# File 'lib/safe_yaml/psych_handler.rb', line 20

def add_to_current_structure(value, anchor=nil, quoted=nil, tag=nil)
  value = Transform.to_proper_type(value, quoted, tag, @options)

  @anchors[anchor] = value if anchor

  if @result.nil?
    @result = value
    @current_structure = @result
    return
  end

  if @current_structure.respond_to?(:<<)
    @current_structure << value

  elsif @current_structure.respond_to?(:[]=)
    if @current_key.nil?
      @current_key = value

    else
      if @current_key == "<<"
        @current_structure.merge!(value)
      else
        @current_structure[@current_key] = value
      end

      @current_key = nil
    end

  else
    raise "Don't know how to add to a #{@current_structure.class}!"
  end
end

#alias(anchor) ⇒ Object

event handlers



63
64
65
 
# File 'lib/safe_yaml/psych_handler.rb', line 63

def alias(anchor)
  add_to_current_structure(@anchors[anchor])
end

#end_current_structure ⇒ Object 



53
54
55
56
 
# File 'lib/safe_yaml/psych_handler.rb', line 53

def end_current_structure
  @stack.pop
  @current_structure = @stack.last
end

#end_document(implicit) ⇒ Object 



71
72
73
 
# File 'lib/safe_yaml/psych_handler.rb', line 71

def end_document(implicit)
  @block.call(@result)
end

#end_mapping ⇒ Object 



82
83
84
 
# File 'lib/safe_yaml/psych_handler.rb', line 82

def end_mapping
  self.end_current_structure()
end

#end_sequence ⇒ Object 



93
94
95
 
# File 'lib/safe_yaml/psych_handler.rb', line 93

def end_sequence
  self.end_current_structure()
end

#result ⇒ Object 



16
17
18
 
# File 'lib/safe_yaml/psych_handler.rb', line 16

def result
  @result
end

#scalar(value, anchor, tag, plain, quoted, style) ⇒ Object 



67
68
69
 
# File 'lib/safe_yaml/psych_handler.rb', line 67

def scalar(value, anchor, tag, plain, quoted, style)
  add_to_current_structure(value, anchor, quoted, tag)
end

#start_mapping(anchor, tag, implicit, style) ⇒ Object 



75
76
77
78
79
80
 
# File 'lib/safe_yaml/psych_handler.rb', line 75

def start_mapping(anchor, tag, implicit, style)
  map = @initializers.include?(tag) ? @initializers[tag].call : {}
  self.add_to_current_structure(map, anchor)
  @current_structure = map
  @stack.push(map)
end

#start_sequence(anchor, tag, implicit, style) ⇒ Object 



86
87
88
89
90
91
 
# File 'lib/safe_yaml/psych_handler.rb', line 86

def start_sequence(anchor, tag, implicit, style)
  seq = @initializers.include?(tag) ? @initializers[tag].call : []
  self.add_to_current_structure(seq, anchor)
  @current_structure = seq
  @stack.push(seq)
end

#streaming? ⇒ Boolean

Returns:

  • (Boolean) 


58
59
60
 
# File 'lib/safe_yaml/psych_handler.rb', line 58

def streaming?
  true
end