Small changes

Author: nobodywasishere

README.md

@@ -28,6 +28,32 @@ The code used in the [Using Parsers](https://tree-sitter.github.io/tree-sitter/u
 was ported as a spec test at [spec/tree_sitter_spec.cr](spec/tree_sitter_spec.cr), the API documentation is being
 ported as well, not yet on github-pages, but run `crystal doc` and have fun.
 
+```crystal
+require "tree_sitter"
+
+parser = TreeSitter::Parser.new("crystal")
+
+source = <<-CRYSTAL
+class Name
+end
+CRYSTAL
+
+tree = parser.parse nil, source
+
+query = TreeSitter::Query.new(parser.language, <<-SCM)
+(class_def) @class
+
+(constant) @constant
+SCM
+
+cursor = TreeSitter::QueryCursor.new(query)
+cursor.exec(tree.root_node)
+
+cursor.each_capture do |capture|
+  p capture
+end
+```
+
 ## Contributing
 
 1. Fork it (<https://github.com/crystal-lang-tools/crystal-tree-sitter/fork>)

src/tree_sitter/query_cursor.cr

@@ -66,6 +66,12 @@ module TreeSitter
       Capture.new(rule, Node.new(capture.node))
     end
 
+    def each_capture(& : Capture -> Nil)
+      while capture = next_capture
+        yield capture
+      end
+    end
+
     def to_unsafe
       @cursor
     end

src/tree_sitter/repository.cr

@@ -10,7 +10,7 @@ module TreeSitter
       @@language_paths ||= begin
         languages = Hash(String, Path).new
         Config.parser_directories.each do |dir|
-          Dir[dir.join("**", "src", "grammar.json")].each do |grammar_path|
+          Dir[dir.join("*", "src", "grammar.json")].each do |grammar_path|
             languages[$2] = Path.new($1) if grammar_path =~ %r{(.*/tree\-sitter\-([\w\-_]+))/src/grammar.json\z}
           end
         end