Skip to content

Base

Base classes and interfaces for tree-sitter analysis.

This module defines the core data structures and interfaces for tree-sitter analysis. It provides: - Entity type definitions for tree-sitter nodes - Metadata structures for tree-sitter nodes. - Base tree-sitter analysis interface

EntityType

Bases: Enum

Types of code entities that can be extracted.

Source code in src/codemap/processor/tree_sitter/base.py 
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
55
56
57
class EntityType(Enum):
	"""Types of code entities that can be extracted."""

	# File-level entities
	MODULE = auto()
	NAMESPACE = auto()
	PACKAGE = auto()

	# Type definitions
	CLASS = auto()
	INTERFACE = auto()
	PROTOCOL = auto()  # Similar to interface but for structural typing
	STRUCT = auto()
	ENUM = auto()
	TYPE_ALIAS = auto()

	# Functions and methods
	FUNCTION = auto()
	METHOD = auto()
	PROPERTY = auto()  # For getter/setter methods
	TEST_CASE = auto()
	TEST_SUITE = auto()

	# Variables and constants
	VARIABLE = auto()
	CONSTANT = auto()
	CLASS_FIELD = auto()  # For class-level variables/fields

	# Code organization
	IMPORT = auto()
	DECORATOR = auto()

	# Documentation
	COMMENT = auto()
	DOCSTRING = auto()

	# Special cases
	UNKNOWN = auto()

MODULE class-attribute instance-attribute 

MODULE = auto()

NAMESPACE class-attribute instance-attribute 

NAMESPACE = auto()

PACKAGE class-attribute instance-attribute 

PACKAGE = auto()

CLASS class-attribute instance-attribute 

CLASS = auto()

INTERFACE class-attribute instance-attribute 

INTERFACE = auto()

PROTOCOL class-attribute instance-attribute 

PROTOCOL = auto()

STRUCT class-attribute instance-attribute 

STRUCT = auto()

ENUM class-attribute instance-attribute 

ENUM = auto()

TYPE_ALIAS class-attribute instance-attribute 

TYPE_ALIAS = auto()

FUNCTION class-attribute instance-attribute 

FUNCTION = auto()

METHOD class-attribute instance-attribute 

METHOD = auto()

PROPERTY class-attribute instance-attribute 

PROPERTY = auto()

TEST_CASE class-attribute instance-attribute 

TEST_CASE = auto()

TEST_SUITE class-attribute instance-attribute 

TEST_SUITE = auto()

VARIABLE class-attribute instance-attribute 

VARIABLE = auto()

CONSTANT class-attribute instance-attribute 

CONSTANT = auto()

CLASS_FIELD class-attribute instance-attribute 

CLASS_FIELD = auto()

IMPORT class-attribute instance-attribute 

IMPORT = auto()

DECORATOR class-attribute instance-attribute 

DECORATOR = auto()

COMMENT class-attribute instance-attribute 

COMMENT = auto()

DOCSTRING class-attribute instance-attribute 

DOCSTRING = auto()

UNKNOWN class-attribute instance-attribute 

UNKNOWN = auto()