use tree_sitter;
use tree_sitter_cpp;

pub fn main() {
  let source_code = "
    // this one has two comments
    // ooOoOoOOOo
    extern int baz;

    // this one has a comment
    extern int bar;

    extern int foo;
  ";

  let query_strings = vec![
    "
(
  (comment)*
  .
  (declaration)
) @decl
    ",
    "
(translation_unit 
  (comment) @comment
  .
  (declaration) @decl
)
    ",
    "
(translation_unit 
  (comment)* @comment
  .
  (declaration) @decl
)
    ",
    "
(translation_unit 
  (
    (comment)* 
    .
    (declaration)
  ) @decl
)
    ",
  ];

  let mut parser = tree_sitter::Parser::new();
  parser.set_language(&tree_sitter_cpp::LANGUAGE.into()).unwrap();

  let tree = parser.parse(source_code, None).unwrap();
  let root_node = tree.root_node();

  println!("s-expression:\n{}\n", root_node.to_sexp());

  let mut cursor = tree_sitter::QueryCursor::new();
  for query_string in query_strings {
    println!("== QUERY ==\n{}\n", query_string);
    let query = tree_sitter::Query::new(
      &tree_sitter_cpp::LANGUAGE.into(), query_string).unwrap();
    let matches = cursor.matches(&query, root_node, source_code.as_bytes());
    for m in matches {
      println!("-----------");
      for c in m.captures {
        println!("{}", c.node.utf8_text(source_code.as_bytes()).unwrap());
      }
      println!("-----------\n");
    }
    println!("===========\n");
  }
}