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}