module ADSP::Stream::ReaderHelpers

# File lib/adsp/stream/reader_helpers.rb, line 214
def self.included(klass)
  klass.extend ClassMethods
end

# File lib/adsp/stream/reader_helpers.rb, line 18
def each_byte(&block)
  each_string method(:getbyte), &block
end

# File lib/adsp/stream/reader_helpers.rb, line 71
def each_char(&block)
  each_string method(:getc), &block
end

# File lib/adsp/stream/reader_helpers.rb, line 139
def each_line(&block)
  each_string method(:gets), &block
end

# File lib/adsp/stream/reader_helpers.rb, line 13
def getbyte
  read 1
end

# File lib/adsp/stream/reader_helpers.rb, line 40
def getc
  if @external_encoding.nil?
    byte = getbyte
    return nil if byte.nil?

    return transcode_to_internal byte
  end

  char = ::String.new :encoding => ::Encoding::BINARY

  # Read one byte until valid string will appear.
  loop do
    byte = getbyte
    return nil if byte.nil?

    char << byte

    char.force_encoding @external_encoding
    return transcode_to_internal char if char.valid_encoding?

    char.force_encoding ::Encoding::BINARY
  end
end

# File lib/adsp/stream/reader_helpers.rb, line 84
def gets(separator = $OUTPUT_RECORD_SEPARATOR, limit = nil)
  # Limit can be a first argument.
  if separator.is_a? ::Numeric
    limit     = separator
    separator = $OUTPUT_RECORD_SEPARATOR
  end

  line_ending =
    if separator.nil?
      nil
    else
      Validation.validate_string separator
      ::String.new separator, :encoding => target_encoding
    end

  Validation.validate_positive_integer limit unless limit.nil?

  line = ::String.new :encoding => target_encoding

  loop do
    char = getc

    if char.nil?
      return nil if line.empty?

      break
    end

    line << char

    break if
      (!line_ending.nil? && line.end_with?(line_ending)) ||
      (!limit.nil? && line.length >= limit)
  end

  @lineno += 1

  line
end

# File lib/adsp/stream/reader_helpers.rb, line 24
def readbyte
  readstring method(:getbyte)
end

# File lib/adsp/stream/reader_helpers.rb, line 66
def readchar
  readstring method(:getc)
end

# File lib/adsp/stream/reader_helpers.rb, line 126
def readline
  readstring method(:gets)
end

# File lib/adsp/stream/reader_helpers.rb, line 131
def readlines
  lines = []
  each_line { |line| lines << line }

  lines
end

# File lib/adsp/stream/reader_helpers.rb, line 29
def ungetbyte(byte)
  Validation.validate_string byte

  @buffer.prepend byte

  nil
end

# File lib/adsp/stream/reader_helpers.rb, line 76
def ungetc(char)
  ungetstring char
end

# File lib/adsp/stream/reader_helpers.rb, line 146
def ungetline(line)
  ungetstring line

  @lineno -= 1

  nil
end

# File lib/adsp/stream/reader_helpers.rb, line 166
          def each_string(each_proc, &block)
  return enum_for __method__, each_proc unless block.is_a? ::Proc

  loop do
    string = each_proc.call
    break if string.nil?

    yield string
  end

  nil
end

# File lib/adsp/stream/reader_helpers.rb, line 158
          def readstring(each_proc)
  string = each_proc.call
  raise ::EOFError if string.nil?

  string
end

# File lib/adsp/stream/reader_helpers.rb, line 180
          def ungetstring(string)
  Validation.validate_string string

  string = ::String.new string, :encoding => @internal_encoding unless @internal_encoding.nil?
  string = transcode_to_external string unless @external_encoding.nil?

  string.force_encoding ::Encoding::BINARY
  @buffer.prepend string

  nil
end