Class: PrettyPrint::Buffer::ArrayBuffer

Inherits:
DefaultBuffer show all
Defined in:
lib/syntax_tree/prettyprint.rb

Overview

This is an output buffer that wraps an array output object. It provides a trim! method that trims off trailing whitespace from the last element in the array if it’s an unfrozen string using the same method as the StringBuffer.

Instance Attribute Summary

Attributes inherited from DefaultBuffer

#output

Instance Method Summary collapse

Methods inherited from DefaultBuffer

#<<

Constructor Details

#initialize(output = []) ⇒ ArrayBuffer

Returns a new instance of ArrayBuffer.



305
306
307
 
# File 'lib/syntax_tree/prettyprint.rb', line 305

def initialize(output = [])
  super(output)
end

Instance Method Details

#trim! ⇒ Object 



309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
 
# File 'lib/syntax_tree/prettyprint.rb', line 309

def trim!
  return 0 if output.empty?

  trimmed = 0

  while output.any? && output.last.is_a?(String) &&
          output.last.match?(/\A[\t ]*\z/)
    trimmed += output.pop.length
  end

  if output.any? && output.last.is_a?(String) && !output.last.frozen?
    length = output.last.length
    output.last.gsub!(/[\t ]*\z/, "")
    trimmed += length - output.last.length
  end

  trimmed
end