drake 0.2.0

A tree-sitter based codebase dependency explorer
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

mod tree;

use anyhow::anyhow;
use tree_sitter::{Language, Query};

pub use tree::{Declaration, Definition, Reference, Tree};

// Matches a package name in a Package.swift file
const PACKAGE_NAME_QUERY: &str = include_str!("package_name.scm");
const DECLARATIONS_QUERY: &str = include_str!("declarations.scm");
const REFERENCES_QUERY: &str = include_str!("references.scm");

pub struct Parser {
    language: Language,
    queries: Queries,
}

struct Queries {
    package_name: Query,
    declaration: Query,
    reference: Query,
}

impl Parser {
    pub fn new() -> Self {
        let language = tree_sitter_swift::language();

        let queries = Queries {
            package_name: Query::new(language, PACKAGE_NAME_QUERY)
                .expect("Failed to parse package name query"),
            declaration: Query::new(language, DECLARATIONS_QUERY)
                .expect("Failed to parse declarations query"),
            reference: Query::new(language, REFERENCES_QUERY)
                .expect("Failed to parse references query"),
        };

        Self { language, queries }
    }

    pub fn parse(&self, source: String) -> anyhow::Result<Tree<'_>> {
        let mut parser = tree_sitter::Parser::new();
        parser.set_language(self.language)?;

        let tree = parser
            .parse(&source, None)
            .ok_or_else(|| anyhow!("Could not parse Swift source"))?;

        Ok(Tree {
            source,
            tree,
            parser: self,
        })
    }
}