Class: Fluent::Config::LiteralParser

Inherits:
BasicParser show all
Defined in:
lib/fluent/config/literal_parser.rb

Direct Known Subclasses

V1Parser

Constant Summary

Constants inherited from BasicParser

BasicParser::LINE_END, BasicParser::LINE_END_WITHOUT_SPACING_AND_COMMENT, BasicParser::SPACING, BasicParser::SPACING_WITHOUT_COMMENT, BasicParser::ZERO_OR_MORE_SPACING

Class Method Summary collapse

Instance Method Summary collapse

Methods inherited from BasicParser

#check, #eof?, #error_sample, #getch, #line_end, #parse_error!, #prev_match, #scan, #skip, #spacing, #spacing_without_comment

Methods included from BasicParser::ClassMethods

#def_literal, #def_symbol, #symbol

Constructor Details

#initialize(strscan, eval_context) ⇒ LiteralParser

Returns a new instance of LiteralParser.



52
53
54
55
56
57
58
59
60
61
62
63
64
65
 
# File 'lib/fluent/config/literal_parser.rb', line 52

def initialize(strscan, eval_context)
  super(strscan)
  @eval_context = eval_context
  unless @eval_context.respond_to?(:use_nil)
    def @eval_context.use_nil
      raise SetNil
    end
  end
  unless @eval_context.respond_to?(:use_default)
    def @eval_context.use_default
      raise SetDefault
    end
  end
end

Class Method Details

.unescape_char(c) ⇒ Object 



29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
 
# File 'lib/fluent/config/literal_parser.rb', line 29

def self.unescape_char(c)
  case c
  when '"'
    '\"'
  when "'"
    "\\'"
  when '\\'
    '\\\\'
  when "\r"
    '\r'
  when "\n"
    '\n'
  when "\t"
    '\t'
  when "\f"
    '\f'
  when "\b"
    '\b'
  else
    c
  end
end

Instance Method Details

#eval_embedded_code(code) ⇒ Object 



183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
 
# File 'lib/fluent/config/literal_parser.rb', line 183

def eval_embedded_code(code)
  if @eval_context.nil?
    parse_error! "embedded code is not allowed in this file"
  end
  # Add hostname and worker_id to code for preventing unused warnings
  code = <<EOM + code
hostname = Socket.gethostname
worker_id = ENV['SERVERENGINE_WORKER_ID'] || ''
EOM
  begin
    @eval_context.instance_eval(code)
  rescue SetNil
    nil
  rescue SetDefault
    :default
  end
end

#eval_escape_char(c) ⇒ Object 



201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
 
# File 'lib/fluent/config/literal_parser.rb', line 201

def eval_escape_char(c)
  case c
  when '"'
    '"'
  when "'"
    "'"
  when "r"
    "\r"
  when "n"
    "\n"
  when "t"
    "\t"
  when "f"
    "\f"
  when "b"
    "\b"
  when "0"
    "\0"
  when /[a-zA-Z0-9]/
    parse_error! "unexpected back-slash escape character '#{c}'"
  else  # symbols
    c
  end
end

#parse_literal(string_boundary_charset = LINE_END) ⇒ Object 



67
68
69
70
71
72
73
74
75
76
77
78
 
# File 'lib/fluent/config/literal_parser.rb', line 67

def parse_literal(string_boundary_charset = LINE_END)
  spacing_without_comment

  value = if skip(/\[/)
            scan_json(true)
          elsif skip(/\{/)
            scan_json(false)
          else
            scan_string(string_boundary_charset)
          end
  value
end

#scan_double_quoted_string ⇒ Object 



90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
 
# File 'lib/fluent/config/literal_parser.rb', line 90

def scan_double_quoted_string
  string = []
  while true
    if skip(/\"/)
      if string.include?(nil)
        return nil
      elsif string.include?(:default)
        return :default
      else
        return string.join
      end
    elsif check(/[^"]#{LINE_END_WITHOUT_SPACING_AND_COMMENT}/)
      if s = check(/[^\\]#{LINE_END_WITHOUT_SPACING_AND_COMMENT}/)
        string << s
      end
      skip(/[^"]#{LINE_END_WITHOUT_SPACING_AND_COMMENT}/)
    elsif s = scan(/\\./)
      string << eval_escape_char(s[1,1])
    elsif skip(/\#\{/)
      string << eval_embedded_code(scan_embedded_code)
      skip(/\}/)
    elsif s = scan(/./)
      string << s
    else
      parse_error! "unexpected end of file in a double quoted string"
    end
  end
end

#scan_embedded_code ⇒ Object 



157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
 
# File 'lib/fluent/config/literal_parser.rb', line 157

def scan_embedded_code
  src = '"#{'+@ss.rest+"\n=begin\n=end\n}"

  seek = -1
  while (seek = src.index('}', seek + 1))
    unless Ripper.sexp(src[0..seek] + '"').nil? # eager parsing until valid expression
      break
    end
  end

  unless seek
    raise Fluent::ConfigParseError, @ss.rest
  end

  code = src[3, seek-3]

  if @ss.rest.length < code.length
    @ss.pos += @ss.rest.length
    parse_error! "expected end of embedded code but $end"
  end

  @ss.pos += code.length

  '"#{' + code + '}"'
end

#scan_json(is_array) ⇒ Object 



226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
 
# File 'lib/fluent/config/literal_parser.rb', line 226

def scan_json(is_array)
  result = nil
  # Yajl does not raise ParseError for incomplete json string, like '[1', '{"h"', '{"h":' or '{"h1":1'
  # This is the reason to use JSON module.

  buffer = (is_array ? "[" : "{")
  line_buffer = ""

  until result
    char = getch

    break if char.nil?

    if char == "#"
      # If this is out of json string literals, this object can be parsed correctly
      # '{"foo":"bar", #' -> '{"foo":"bar"}' (to check)
      parsed = nil
      begin
        parsed = JSON.parse(buffer + line_buffer.rstrip.sub(/,$/, '') + (is_array ? "]" : "}"))
      rescue JSON::ParserError
        # This '#' is in json string literals
      end

      if parsed
        # ignore chars as comment before newline
        while (char = getch) != "\n"
          # ignore comment char
        end
        buffer << line_buffer + "\n"
        line_buffer = ""
      else
        # '#' is a char in json string
        line_buffer << char
      end

      next # This char '#' MUST NOT terminate json object.
    end

    if char == "\n"
      buffer << line_buffer + "\n"
      line_buffer = ""
      next
    end

    line_buffer << char
    begin
      result = JSON.parse(buffer + line_buffer)
    rescue JSON::ParserError
      # Incomplete json string yet
    end
  end

  unless result
    parse_error! "got incomplete JSON #{is_array ? 'array' : 'hash'} configuration"
  end

  JSON.dump(result)
end

#scan_nonquoted_string(boundary_charset = LINE_END) ⇒ Object 



136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
 
# File 'lib/fluent/config/literal_parser.rb', line 136

def scan_nonquoted_string(boundary_charset = LINE_END)
  charset = /(?!#{boundary_charset})./

  string = []
  while true
    if s = scan(/\#/)
      string << '#'
    elsif s = scan(charset)
      string << s
    else
      break
    end
  end

  if string.empty?
    return nil
  end

  string.join
end

#scan_single_quoted_string ⇒ Object 



119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
 
# File 'lib/fluent/config/literal_parser.rb', line 119

def scan_single_quoted_string
  string = []
  while true
    if skip(/\'/)
      return string.join
    elsif s = scan(/\\'/)
      string << "'"
    elsif s = scan(/\\\\/)
      string << "\\"
    elsif s = scan(/./)
      string << s
    else
      parse_error! "unexpected end of file in a single quoted string"
    end
  end
end

#scan_string(string_boundary_charset = LINE_END) ⇒ Object 



80
81
82
83
84
85
86
87
88
 
# File 'lib/fluent/config/literal_parser.rb', line 80

def scan_string(string_boundary_charset = LINE_END)
  if skip(/\"/)
    return scan_double_quoted_string
  elsif skip(/\'/)
    return scan_single_quoted_string
  else
    return scan_nonquoted_string(string_boundary_charset)
  end
end