Skip to content

typescript

TREE_SITTER_QUERY = '(function_declaration name: (identifier) @function.name) @function.definition(lexical_declaration (variable_declarator name: (identifier) @function.name value: (function_expression) @function.definition))(lexical_declaration (variable_declarator name: (identifier) @function.name value: (arrow_function) @function.definition))(class_declaration name: (type_identifier) @class.name body: (class_body (method_definition name: (property_identifier) @function.name) @function.definition))(pair key: (property_identifier) @function.name value: (function_expression) @function.definition)(pair key: (property_identifier) @function.name value: (arrow_function) @function.definition)(pair key: (string) @function.name value: (function_expression) @function.definition)(pair key: (string) @function.name value: (arrow_function) @function.definition)' module-attribute

Tree-sitter query for extracting function names and definitions.

TypeScriptExtractor

Bases: TreeSitterExtractor

Source code extractor for extracting TypeScript functions.

Source code in src/codablellm/languages/typescript.py 
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
class TypeScriptExtractor(TreeSitterExtractor):
    """
    Source code extractor for extracting TypeScript functions.
    """

    def __init__(self, include_extended: bool = True) -> None:
        super().__init__("TypeScript", TREE_SITTER_QUERY)
        self._tsx_extractor = (
            TypeScriptExtendedExtractor() if include_extended else None
        )

    def extract(
        self, file_path: PathLike, repo_path: Optional[PathLike] = None
    ) -> Sequence[SourceFunction]:
        if self._tsx_extractor and Path(file_path).suffix == ".tsx":
            return self._tsx_extractor.extract(file_path, repo_path=repo_path)
        return super().extract(file_path, repo_path)

    def get_extractable_files(self, path: PathLike) -> Set[Path]:
        files = rglob_file_extensions(path, [".ts"])
        if self._tsx_extractor:
            files.update(self._tsx_extractor.get_extractable_files(path))
        return files

    @requires_extra(
        "typescript", "TypeScript source code extraction", "tree_sitter_typescript"
    )
    def get_language(self) -> Language:
        return Language(tst.language_typescript())  # type: ignore

    def is_installed(self) -> bool:
        return tst is not None