class Cucumber::Core::Gherkin::Writer::Gherkin

def build
  instance_exec(&@source)
  Document.new(@uri, @feature.build.join("\n"))
end

def comment(line)
  comment_lines << "# #{line}"
end

def comment_lines
  @comment_lines ||= []
end

def feature(*args, &source)
  @feature = Feature.new(comment_lines, *args).tap do |builder|
    builder.instance_exec(&source) if source
  end
  self
end

def initialize(uri, &source)
  @uri, @source = uri, source
end