codebeacon 0.1.2__py3-none-any.whl

codebeacon/extract/queries/fastapi.scm

@@ -0,0 +1,152 @@
+; ── FastAPI (Python) ──────────────────────────────────────────────────────────
+; Grammar: tree-sitter-python
+;
+; Captures:
+;   @route.decorator     - @app.get / @router.post etc. decorator node
+;   @route.object        - app/router object name
+;   @route.method        - HTTP method (get/post/put/delete/patch)
+;   @route.path          - path string
+;   @route.func_name     - handler function name
+;   @route.prefix        - APIRouter(prefix="...") value
+;   @service.func_name   - Depends() dependency function name
+;   @service.class_name  - class with methods using Depends
+;   @entity.class_name   - BaseModel / SQLAlchemy Base subclass
+;   @entity.field_name   - model field name
+;   @entity.field_type   - model field type annotation
+;   @router.include      - app.include_router(router, prefix="...")
+;   @import.path         - import path
+
+; ── @app.get("/path") / @router.post("/path") ────────────────────────────────
+
+(decorated_definition
+  (decorator
+    (call
+      function: (attribute
+        object: (identifier) @route.object
+        attribute: (identifier) @route.method
+        (#match? @route.method "^(get|post|put|delete|patch|options|head)$")
+      )
+      arguments: (argument_list
+        (string) @route.path
+      )
+    )
+  )
+  definition: (function_definition
+    name: (identifier) @route.func_name
+  )
+) @route.handler
+
+; ── APIRouter(prefix="/api/v1") ───────────────────────────────────────────────
+
+(assignment
+  left: (identifier) @route.router_name
+  right: (call
+    function: (identifier) @_apiRouter
+    (#eq? @_apiRouter "APIRouter")
+    arguments: (argument_list
+      (keyword_argument
+        name: (identifier) @_prefix_key
+        (#eq? @_prefix_key "prefix")
+        value: (string) @route.prefix
+      )
+    )
+  )
+) @route.router_decl
+
+; ── app.include_router(router, prefix="...") ─────────────────────────────────
+
+; With prefix keyword argument
+(call
+  function: (attribute
+    object: (identifier) @_app
+    attribute: (identifier) @_include
+    (#eq? @_include "include_router")
+  )
+  arguments: (argument_list
+    (identifier) @router.include_router
+    (keyword_argument
+      name: (identifier) @_pk
+      (#eq? @_pk "prefix")
+      value: (string) @router.include_prefix
+    )
+  )
+) @router.include
+
+; Without prefix keyword argument
+(call
+  function: (attribute
+    object: (identifier) @_app2
+    attribute: (identifier) @_include2
+    (#eq? @_include2 "include_router")
+  )
+  arguments: (argument_list
+    (identifier) @router.include_router
+  )
+) @router.include_no_prefix
+
+; ── Depends() function dependency ────────────────────────────────────────────
+
+; Functions that take parameters (potential service functions)
+(function_definition
+  name: (identifier) @service.func_name
+  parameters: (parameters
+    (typed_parameter
+      (identifier) @_param
+    )
+  )
+) @service.function
+
+; Depends() in function signature
+(call
+  function: (identifier) @_depends
+  (#eq? @_depends "Depends")
+  arguments: (argument_list
+    (identifier) @service.depends_func
+  )
+) @service.depends
+
+; ── BaseModel / SQLAlchemy subclass (entity) ─────────────────────────────────
+
+; Direct identifier base class: class User(BaseModel)
+(class_definition
+  name: (identifier) @entity.class_name
+  superclasses: (argument_list
+    (identifier) @_base
+    (#match? @_base "^(BaseModel|Base|DeclarativeBase|SQLModel)$")
+  )
+) @entity.class
+
+; Attribute base class: class User(db.Model)
+(class_definition
+  name: (identifier) @entity.class_name
+  superclasses: (argument_list
+    (attribute
+      attribute: (identifier) @_base
+      (#match? @_base "^(Model|Base)$")
+    )
+  )
+) @entity.class_attr
+
+; Entity fields (type-annotated class attributes)
+(class_definition
+  body: (block
+    (expression_statement
+      (assignment
+        left: (identifier) @entity.field_name
+        type: (type
+          (identifier) @entity.field_type
+        )
+      )
+    )
+  )
+) @entity.with_fields
+
+; ── imports ───────────────────────────────────────────────────────────────────
+
+(import_from_statement
+  module_name: _ @import.path
+) @import.from
+
+(import_statement
+  name: _ @import.path
+) @import.plain

codebeacon/extract/queries/flask.scm

@@ -0,0 +1,120 @@
+; ── Flask (Python) ────────────────────────────────────────────────────────────
+; Grammar: tree-sitter-python
+;
+; Captures:
+;   @route.decorator     - @app.route / @bp.route decorator
+;   @route.object        - app/blueprint object name
+;   @route.path          - path string
+;   @route.methods       - methods list ["GET","POST"]
+;   @route.func_name     - handler function name
+;   @blueprint.name      - Blueprint variable name
+;   @blueprint.url_prefix - url_prefix value
+;   @blueprint.import_name - blueprint name string
+;   @app.register_bp     - app.register_blueprint(bp, url_prefix="...")
+;   @app.register_prefix - url_prefix in register_blueprint
+;   @entity.class_name   - db.Model subclass name
+;   @entity.field_name   - field name
+;   @entity.field_type   - Column(...) type
+;   @import.path         - import path
+
+; ── @app.route("/path") / @blueprint.route("/path") ─────────────────────────
+
+(decorated_definition
+  (decorator
+    (call
+      function: (attribute
+        object: (identifier) @route.object
+        attribute: (identifier) @_route_attr
+        (#eq? @_route_attr "route")
+      )
+      arguments: (argument_list
+        (string) @route.path
+        (keyword_argument
+          name: (identifier) @_methods_key
+          (#eq? @_methods_key "methods")
+          value: (list
+            (string) @route.methods
+          )
+        )?
+      )
+    )
+  )
+  definition: (function_definition
+    name: (identifier) @route.func_name
+  )
+) @route.handler
+
+; ── Blueprint declaration ─────────────────────────────────────────────────────
+
+(assignment
+  left: (identifier) @blueprint.name
+  right: (call
+    function: (identifier) @_bp
+    (#eq? @_bp "Blueprint")
+    arguments: (argument_list
+      (string) @blueprint.import_name
+      _
+      (keyword_argument
+        name: (identifier) @_upk
+        (#eq? @_upk "url_prefix")
+        value: (string) @blueprint.url_prefix
+      )?
+    )
+  )
+) @blueprint.decl
+
+; ── app.register_blueprint(bp, url_prefix="/api") ────────────────────────────
+
+(call
+  function: (attribute
+    object: (identifier) @_app
+    attribute: (identifier) @_reg
+    (#eq? @_reg "register_blueprint")
+  )
+  arguments: (argument_list
+    (identifier) @app.register_bp
+    (keyword_argument
+      name: (identifier) @_upk
+      (#eq? @_upk "url_prefix")
+      value: (string) @app.register_prefix
+    )?
+  )
+) @app.register
+
+; ── SQLAlchemy / Flask-SQLAlchemy model ───────────────────────────────────────
+
+(class_definition
+  name: (identifier) @entity.class_name
+  superclasses: (argument_list
+    [
+      (attribute
+        attribute: (identifier) @_m
+        (#match? @_m "^(Model|Base)$")
+      )
+      (identifier) @_m
+      (#match? @_m "^(Model|Base)$")
+    ]
+  )
+) @entity.class
+
+; SQLAlchemy Column fields
+(assignment
+  left: (identifier) @entity.field_name
+  right: (call
+    function: [
+      (identifier) @entity.field_type
+      (attribute attribute: (identifier) @entity.field_type)
+    ]
+    (#match? @entity.field_type "^(Column|relationship|backref)$")
+  )
+) @entity.field
+
+; ── imports ───────────────────────────────────────────────────────────────────
+
+(import_from_statement
+  module_name: _ @import.path
+) @import.from
+
+(import_statement
+  name: _ @import.path
+) @import.plain

codebeacon/extract/queries/gin.scm

@@ -0,0 +1,142 @@
+; ── Gin / Echo / Fiber (Go) ───────────────────────────────────────────────────
+; Grammar: tree-sitter-go
+;
+; Go struct tag parsing note:
+;   GORM struct tags are raw string literals: `gorm:"primaryKey"`.
+;   These are parsed by extract/entities.py using regex on the raw text,
+;   not by tree-sitter queries (tree-sitter-go does not parse struct tags).
+;
+; Captures:
+;   @route.method       - GET/POST/PUT/DELETE/PATCH/Any etc.
+;   @route.path         - path string
+;   @route.object       - router/group variable name
+;   @route.group_prefix - r.Group("/api") prefix string
+;   @route.handler_name - handler function identifier
+;   @service.struct_name  - struct name (service)
+;   @service.field_type   - field type (for DI via struct embedding)
+;   @entity.struct_name   - GORM model struct name
+;   @entity.field_name    - struct field name
+;   @entity.field_tag     - raw struct tag string (for GORM parsing)
+;   @import.path          - import path string
+
+; ── r.GET("/path", handler) ──────────────────────────────────────────────────
+
+(expression_statement
+  (call_expression
+    function: (selector_expression
+      operand: (identifier) @route.object
+      field: (field_identifier) @route.method
+      (#match? @route.method "^(GET|POST|PUT|PATCH|DELETE|OPTIONS|HEAD|Any|Handle|UseWithErr|Use)$")
+    )
+    arguments: (argument_list
+      (interpreted_string_literal) @route.path
+      (identifier) @route.handler_name
+    )
+  )
+) @route.call
+
+; Case-insensitive method names (Echo: e.GET, Fiber: app.Get)
+(expression_statement
+  (call_expression
+    function: (selector_expression
+      operand: (identifier) @route.object
+      field: (field_identifier) @route.method
+      (#match? @route.method "^(Get|Post|Put|Patch|Delete|Options|Head|All|Add)$")
+    )
+    arguments: (argument_list
+      (interpreted_string_literal) @route.path
+      (identifier) @route.handler_name
+    )
+  )
+) @route.call_lower
+
+; ── r.Group("/prefix") ───────────────────────────────────────────────────────
+
+(short_var_declaration
+  left: (expression_list
+    (identifier) @route.group_name
+  )
+  right: (expression_list
+    (call_expression
+      function: (selector_expression
+        operand: (identifier) @_router
+        field: (field_identifier) @_group
+        (#eq? @_group "Group")
+      )
+      arguments: (argument_list
+        (interpreted_string_literal) @route.group_prefix
+      )
+    )
+  )
+) @route.group_decl
+
+; ── Service struct ────────────────────────────────────────────────────────────
+
+(type_declaration
+  (type_spec
+    name: (type_identifier) @service.struct_name
+    type: (struct_type
+      (field_declaration_list
+        (field_declaration
+          type: (pointer_type
+            (type_identifier) @service.field_type
+          )
+        )
+      )
+    )
+  )
+) @service.struct
+
+(type_declaration
+  (type_spec
+    name: (type_identifier) @service.struct_name
+    type: (struct_type
+      (field_declaration_list
+        (field_declaration
+          type: (type_identifier) @service.field_type
+        )
+      )
+    )
+  )
+) @service.struct_plain
+
+; ── GORM entity struct ────────────────────────────────────────────────────────
+
+(type_declaration
+  (type_spec
+    name: (type_identifier) @entity.struct_name
+    type: (struct_type
+      (field_declaration_list
+        (field_declaration
+          name: (field_identifier) @entity.field_name
+          type: _ @entity.field_type
+          tag: (raw_string_literal) @entity.field_tag
+        )
+      )
+    )
+  )
+) @entity.struct
+
+; Struct with no tags (still useful for type extraction)
+(type_declaration
+  (type_spec
+    name: (type_identifier) @entity.struct_name
+    type: (struct_type)
+  )
+) @entity.struct_bare
+
+; ── imports ───────────────────────────────────────────────────────────────────
+
+(import_declaration
+  (import_spec_list
+    (import_spec
+      path: (interpreted_string_literal) @import.path
+    )
+  )
+) @import.block
+
+(import_declaration
+  (import_spec
+    path: (interpreted_string_literal) @import.path
+  )
+) @import.single

codebeacon/extract/queries/ktor.scm

@@ -0,0 +1,144 @@
+; ── Ktor (Kotlin) ─────────────────────────────────────────────────────────────
+; Grammar: tree-sitter-kotlin
+;
+; Kotlin grammar note:
+;   - call_expression: f(args) — can have trailing lambda
+;   - annotated_lambda / lambda_literal: trailing { } block
+;   - value_arguments: (args)
+;   - value_argument → string_literal → string_content: path string
+;   - identifier: function name (get, post, route, routing)
+;   - qualified_identifier: import paths
+;
+; Ktor routing DSL nesting:
+;   routing {
+;     get("/a") { }          → GET /a
+;     route("/prefix") {
+;       get("/b") { }        → GET /prefix/b
+;     }
+;   }
+; Path composition is done by extract/routes.py by tracking nesting depth
+; using start_point line numbers. This file captures individual route/method calls.
+;
+; Captures:
+;   @route.method         - get/post/put/patch/delete (HTTP method)
+;   @route.path           - path string content
+;   @route.route_prefix   - route("prefix") path — prefix scope
+;   @route.call_line      - line number for nesting (via @route.call node)
+;   @service.koin_type    - Koin: single/factory/scoped class
+;   @entity.table_name    - Exposed Table object name
+;   @entity.column_name   - Exposed column name
+;   @import.path          - import qualified path
+
+; ── routing { get("/path") { } } ─────────────────────────────────────────────
+
+; Direct HTTP method call: get("/path") { }
+(call_expression
+  (call_expression
+    (identifier) @route.method
+    (#match? @route.method "^(get|post|put|patch|delete|options|head)$")
+    (value_arguments
+      (value_argument
+        (string_literal
+          (string_content) @route.path
+        )
+      )
+    )
+  )
+  (annotated_lambda)?
+) @route.method_call
+
+; Also handle: get("/path") without trailing lambda
+(call_expression
+  (identifier) @route.method
+  (#match? @route.method "^(get|post|put|patch|delete|options|head)$")
+  (value_arguments
+    (value_argument
+      (string_literal
+        (string_content) @route.path
+      )
+    )
+  )
+) @route.method_call_simple
+
+; ── route("/prefix") { } — creates a path prefix scope ───────────────────────
+
+(call_expression
+  (call_expression
+    (identifier) @_route_fn
+    (#eq? @_route_fn "route")
+    (value_arguments
+      (value_argument
+        (string_literal
+          (string_content) @route.route_prefix
+        )
+      )
+    )
+  )
+  (annotated_lambda)
+) @route.prefix_scope
+
+; ── Koin DI: single { UserService(get()) } ───────────────────────────────────
+
+(call_expression
+  (identifier) @_koin_scope
+  (#match? @_koin_scope "^(single|factory|scoped|viewModel|worker)$")
+  (annotated_lambda
+    (lambda_literal
+      (call_expression
+        (identifier) @service.koin_type
+        (value_arguments)?
+      )
+    )
+  )
+) @service.koin_binding
+
+; ── Exposed ORM: object Users : Table() ──────────────────────────────────────
+
+(object_declaration
+  (identifier) @entity.table_name
+  (delegation_specifier
+    (constructor_invocation
+      (user_type (identifier) @_table (#match? @_table "^(Table|IntIdTable|LongIdTable|UUIDTable|IdTable)$"))
+    )
+  )
+) @entity.table
+
+; Exposed columns: val id = integer("id")
+(property_declaration
+  (variable_declaration
+    (simple_identifier) @entity.column_name
+  )
+  (call_expression
+    (identifier) @entity.column_type
+    (#match? @entity.column_type "^(integer|long|varchar|text|bool|double|float|decimal|uuid|reference|optReference|enumeration|enumerationByName|date|datetime|timestamp)$")
+    (value_arguments
+      (value_argument
+        (string_literal
+          (string_content) @entity.column_key
+        )
+      )
+    )
+  )
+) @entity.column
+
+; ── Data class (Kotlin entity / DTO) ─────────────────────────────────────────
+
+(class_declaration
+  (modifiers
+    (class_modifier) @_data
+    (#eq? @_data "data")
+  )
+  (simple_identifier) @entity.class_name
+) @entity.data_class
+
+; ── Regular class (service) ───────────────────────────────────────────────────
+
+(class_declaration
+  (simple_identifier) @service.class_name
+) @service.class
+
+; ── imports ───────────────────────────────────────────────────────────────────
+
+(import
+  (qualified_identifier) @import.path
+) @import.decl