codemov-parser 0.1.0

Tree-sitter based source code parser for codemov
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

pub mod rust;
mod tests;
pub mod typescript;

use codemov_core::{ImportEdge, Language, Symbol};
use thiserror::Error;

#[derive(Debug, Error)]
pub enum ParseError {
    #[error("unsupported language: {0:?}")]
    UnsupportedLanguage(Language),
    #[error("parse error: {0}")]
    Parse(String),
    #[error("utf8 error: {0}")]
    Utf8(#[from] std::str::Utf8Error),
}

pub fn extract_symbols(source: &[u8], language: Language) -> Result<Vec<Symbol>, ParseError> {
    match language {
        Language::Rust => rust::extract(source),
        Language::TypeScript | Language::JavaScript => typescript::extract(source, language),
        Language::Unknown => Ok(vec![]),
    }
}

pub fn extract_imports(source: &[u8], language: Language) -> Result<Vec<ImportEdge>, ParseError> {
    match language {
        Language::Rust => rust::extract_imports(source),
        Language::TypeScript | Language::JavaScript => {
            typescript::extract_imports(source, language)
        }
        Language::Unknown => Ok(vec![]),
    }
}