class Gherkin::Stream::ParserMessageStream

Public Class Methods

new(paths, sources, options) click to toggle source
# File lib/gherkin/stream/parser_message_stream.rb, line 9
def initialize(paths, sources, options)
  @paths = paths
  @sources = sources
  @options = options

  id_generator = options[:id_generator] || Cucumber::Messages::IdGenerator::UUID.new
  @parser = Parser.new(AstBuilder.new(id_generator))
  @compiler = Pickles::Compiler.new(id_generator)
end

Public Instance Methods

messages() click to toggle source
# File lib/gherkin/stream/parser_message_stream.rb, line 19
def messages
  enumerated = false
  Enumerator.new do |y|
    raise DoubleIterationException, "Messages have already been enumerated" if enumerated
    enumerated = true

    sources.each do |source|
      y.yield(Cucumber::Messages::Envelope.new(source: source)) if @options[:include_source]
      begin
        gherkin_document = nil

        if @options[:include_gherkin_document]
          gherkin_document = build_gherkin_document(source)
          y.yield(Cucumber::Messages::Envelope.new(gherkin_document: gherkin_document))
        end
        if @options[:include_pickles]
          gherkin_document ||= build_gherkin_document(source)
          pickles = @compiler.compile(gherkin_document, source)
          pickles.each do |pickle|
            y.yield(Cucumber::Messages::Envelope.new(pickle: pickle))
          end
        end
      rescue CompositeParserException => err
        yield_parse_errors(y, err.errors, source.uri)
      rescue ParserException => err
        yield_parse_errors(y, [err], source.uri)
      end
    end
  end
end

Private Instance Methods

build_gherkin_document(source) click to toggle source
# File lib/gherkin/stream/parser_message_stream.rb, line 84
def build_gherkin_document(source)
  if @options[:default_dialect]
    token_matcher = TokenMatcher.new(@options[:default_dialect])
    gd = @parser.parse(source.data, token_matcher)
  else
    gd = @parser.parse(source.data)
  end
  Cucumber::Messages::GherkinDocument.new(
    uri: source.uri,
    feature: gd.feature,
    comments: gd.comments
  )
end
sources() click to toggle source
# File lib/gherkin/stream/parser_message_stream.rb, line 68
def sources
  Enumerator.new do |y|
    @paths.each do |path|
      source = Cucumber::Messages::Source.new(
        uri: path,
        data: File.open(path, 'r:UTF-8', &:read),
        media_type: 'text/x.cucumber.gherkin+plain'
      )
      y.yield(source)
    end
    @sources.each do |source|
      y.yield(source)
    end
  end
end
yield_parse_errors(y, errors, uri) click to toggle source
# File lib/gherkin/stream/parser_message_stream.rb, line 52
def yield_parse_errors(y, errors, uri)
  errors.each do |err|
    parse_error = Cucumber::Messages::ParseError.new(
      source: Cucumber::Messages::SourceReference.new(
        uri: uri,
        location: Cucumber::Messages::Location.new(
          line: err.location[:line],
          column: err.location[:column]
        )
      ),
      message: err.message
    )
    y.yield(Cucumber::Messages::Envelope.new(parse_error: parse_error))
  end
end