semtree-parse 0.2.0

Tree-sitter parsing and chunk extraction for semtree
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

mod go;
mod javascript;
mod python;
mod rust;
pub(crate) mod shared;
mod typescript;

use semtree_core::{Chunk, Language};
use thiserror::Error;

use crate::extractor::Extractor;
use crate::parser::ParsedTree;

#[derive(Debug, Error)]
#[error("no extractor for language: {0}")]
pub struct ExtractError(pub Language);

pub fn extract(tree: &ParsedTree) -> Result<Vec<Chunk>, ExtractError> {
    match tree.language {
        Language::Rust => Ok(rust::RustExtractor.extract(tree)),
        Language::Python => Ok(python::PythonExtractor.extract(tree)),
        Language::JavaScript => Ok(javascript::JavaScriptExtractor.extract(tree)),
        Language::TypeScript => Ok(typescript::TypeScriptExtractor.extract(tree)),
        Language::Go => Ok(go::GoExtractor.extract(tree)),
        lang => Err(ExtractError(lang)),
    }
}