class Cucumber::Core::Gherkin::AstBuilder::DocumentBuilder

def all_comments
  attributes[:comments].map do |comment|
    Ast::Comment.new(
      Ast::Location.new(file, comment[:location][:line]),
      comment[:text]
    )
  end
end

def feature
  return Ast::NullFeature.new unless attributes[:feature]
  feature_builder = FeatureBuilder.new(file, attributes[:feature])
  feature_builder.handle_comments(all_comments)
  feature_builder.result
end

def initialize(file, attributes)
  @file = file
  @attributes = rubify_keys(attributes.dup)
end