class Cucumber::Core::Gherkin::Parser

def document(document)
  messages = ::Gherkin.from_source(document.uri, document.body, gherkin_options(document))
  messages.each do |message|
    event_bus.envelope(message)
    gherkin_query.update(message)
    if !message.gherkin_document.nil?
      event_bus.gherkin_source_parsed(message.gherkin_document)
    elsif !message.pickle.nil?
      receiver.pickle(message.pickle)
    elsif message.parse_error
      raise Core::Gherkin::ParseError.new("#{document.uri}: #{message.parse_error.message}")
    else
      raise "Unknown message: #{message.to_hash}"
    end
  end
end

def done
  receiver.done
  self
end

def gherkin_options(document)
  {
    default_dialect: document.language,
    include_source: false,
    include_gherkin_document: true,
    include_pickles: true
  }
end

def initialize(receiver, event_bus, gherkin_query)
  @receiver = receiver
  @event_bus = event_bus
  @gherkin_query = gherkin_query
end