tree_sitter_swift/lib.rs
1//! This crate provides Swift language support for the [tree-sitter][] parsing library.
2//!
3//! Typically, you will use the [language][language func] function to add this language to a
4//! tree-sitter [Parser][], and then use the parser to parse some code:
5//!
6//! ```
7//! let code = r#"
8//! "#;
9//! let mut parser = tree_sitter::Parser::new();
10//! let language = tree_sitter_swift::LANGUAGE;
11//! parser
12//! .set_language(&language.into())
13//! .expect("Error loading Swift parser");
14//! let tree = parser.parse(code, None).unwrap();
15//! assert!(!tree.root_node().has_error());
16//! ```
17//!
18//! [Language]: https://docs.rs/tree-sitter/*/tree_sitter/struct.Language.html
19//! [language func]: fn.language.html
20//! [Parser]: https://docs.rs/tree-sitter/*/tree_sitter/struct.Parser.html
21//! [tree-sitter]: https://tree-sitter.github.io/
22
23use tree_sitter_language::LanguageFn;
24
25extern "C" {
26 fn tree_sitter_swift() -> *const ();
27}
28
29/// The tree-sitter [`LanguageFn`] for this grammar.
30pub const LANGUAGE: LanguageFn = unsafe { LanguageFn::from_raw(tree_sitter_swift) };
31
32/// The content of the [`node-types.json`][] file for this grammar.
33///
34/// [`node-types.json`]: https://tree-sitter.github.io/tree-sitter/using-parsers#static-node-types
35pub const NODE_TYPES: &str = include_str!("../../src/node-types.json");
36
37pub const HIGHLIGHTS_QUERY: &str = include_str!("../../queries/highlights.scm");
38pub const INJECTIONS_QUERY: &str = include_str!("../../queries/injections.scm");
39pub const LOCALS_QUERY: &str = include_str!("../../queries/locals.scm");
40pub const TAGS_QUERY: &str = include_str!("../../queries/tags.scm");
41
42#[cfg(test)]
43mod tests {
44 #[test]
45 fn test_can_load_grammar() {
46 let mut parser = tree_sitter::Parser::new();
47 parser
48 .set_language(&super::LANGUAGE.into())
49 .expect("Error loading Swift parser");
50 }
51
52 #[test]
53 fn test_can_parse_basic_file() {
54 let mut parser = tree_sitter::Parser::new();
55 parser
56 .set_language(&super::LANGUAGE.into())
57 .expect("Error loading Swift parser");
58
59 let tree = parser
60 .parse("_ = \"Hello!\"\n", None)
61 .expect("Unable to parse!");
62
63 assert_eq!(
64 "(source_file (assignment target: (directly_assignable_expression (simple_identifier)) result: (line_string_literal text: (line_str_text))))",
65 tree.root_node().to_sexp(),
66 );
67 }
68}