Class: Prism::LexCompat::Heredoc::DashHeredoc

Inherits:
Object
  • Object
show all
Defined in:
lib/prism/lex_compat.rb

Overview

Dash heredocs are a little more complicated. They are a list of tokens that need to be split on “\n” to mimic Ripper’s behavior. We also need to keep track of the state that the heredoc was opened in.

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(split) ⇒ DashHeredoc

Returns a new instance of DashHeredoc.



319
320
321
322
 
# File 'lib/prism/lex_compat.rb', line 319

def initialize(split)
  @split = split
  @tokens = []
end

Instance Attribute Details

#split ⇒ Object (readonly)

:nodoc:



317
318
319
 
# File 'lib/prism/lex_compat.rb', line 317

def split
  @split
end

#tokens ⇒ Object (readonly)

:nodoc:



317
318
319
 
# File 'lib/prism/lex_compat.rb', line 317

def tokens
  @tokens
end

Instance Method Details

#<<(token) ⇒ Object 



324
325
326
 
# File 'lib/prism/lex_compat.rb', line 324

def <<(token)
  tokens << token
end

#to_a ⇒ Object 



328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
 
# File 'lib/prism/lex_compat.rb', line 328

def to_a
  embexpr_balance = 0

  tokens.each_with_object([]) do |token, results| #$ Array[Token]
    case token.event
    when :on_embexpr_beg
      embexpr_balance += 1
      results << token
    when :on_embexpr_end
      embexpr_balance -= 1
      results << token
    when :on_tstring_content
      if embexpr_balance == 0
        lineno = token[0][0]
        column = token[0][1]

        if split
          # Split on "\\\n" to mimic Ripper's behavior. Use a lookbehind
          # to keep the delimiter in the result.
          token.value.split(/(?<=[^\\]\\\n)|(?<=[^\\]\\\r\n)/).each_with_index do |value, index|
            column = 0 if index > 0
            results << Token.new([[lineno, column], :on_tstring_content, value, token.state])
            lineno += value.count("\n")
          end
        else
          results << token
        end
      else
        results << token
      end
    else
      results << token
    end
  end
end