codebeacon 0.1.2__py3-none-any.whl

codebeacon/extract/queries/actix.scm

@@ -0,0 +1,115 @@
+; ── Actix-web / Axum (Rust) ───────────────────────────────────────────────────
+; Grammar: tree-sitter-rust
+;
+; Rust grammar note:
+;   - attribute_item: #[get("/users")]
+;   - attribute: get("/users") — identifier + token_tree
+;   - token_tree: the (...) argument list (unparsed by tree-sitter)
+;   - string_literal / string_content: path string
+;   - function_item: async fn handler
+;   - struct_item: struct definition
+;   - field_declaration: struct field
+;
+; Proc macro note:
+;   #[derive(Queryable, Serialize)] — captured by @entity.derive_trait
+;   Diesel table! macro handled by regexp in entities.py
+;
+; Captures:
+;   @route.proc_macro     - get/post/put/patch/delete (Actix proc macro name)
+;   @route.path           - path string content
+;   @route.func_name      - handler function name
+;   @route.axum_method    - Axum: get/post/put/delete (from axum::routing::get)
+;   @route.axum_path      - Axum: path string in Router::new().route(path, ...)
+;   @entity.struct_name   - struct with derive macros
+;   @entity.derive_trait  - Queryable/DeriveEntityModel/Serialize etc.
+;   @entity.field_name    - struct field name
+;   @entity.field_type    - struct field type
+;   @service.struct_name  - AppState / service struct
+;   @import.path          - use path
+
+; ── Actix: #[get("/users")] async fn handler() ───────────────────────────────
+
+(function_item
+  .
+  (attribute_item
+    (attribute
+      (identifier) @route.proc_macro
+      (#match? @route.proc_macro "^(get|post|put|patch|delete|options|head)$")
+      (token_tree
+        (string_literal
+          (string_content) @route.path
+        )
+      )
+    )
+  )
+  name: (identifier) @route.func_name
+) @route.actix_handler
+
+; ── Axum: Router::new().route("/users", get(handler)) ────────────────────────
+
+(call_expression
+  function: (field_expression
+    field: (field_identifier) @_route_fn
+    (#eq? @_route_fn "route")
+  )
+  arguments: (arguments
+    (string_literal
+      (string_content) @route.axum_path
+    )
+    (call_expression
+      function: (identifier) @route.axum_method
+      (#match? @route.axum_method "^(get|post|put|patch|delete|options|head)$")
+      arguments: (arguments
+        (identifier) @route.axum_handler
+      )
+    )
+  )
+) @route.axum_route
+
+; ── #[derive(Queryable, Serialize, ...)] struct ──────────────────────────────
+
+(struct_item
+  .
+  (attribute_item
+    (attribute
+      (identifier) @_derive (#eq? @_derive "derive")
+      (token_tree
+        (identifier) @entity.derive_trait
+        (#match? @entity.derive_trait "^(Queryable|DeriveEntityModel|DeriveRelation|FromRow|Model|Serialize|Deserialize|sqlx)$")
+      )
+    )
+  )
+  name: (type_identifier) @entity.struct_name
+) @entity.struct
+
+; All struct fields
+(struct_item
+  name: (type_identifier) @entity.struct_name
+  body: (field_declaration_list
+    (field_declaration
+      name: (field_identifier) @entity.field_name
+      type: _ @entity.field_type
+    )
+  )
+) @entity.struct_with_fields
+
+; ── All attribute_items on functions (for proc macro detection) ───────────────
+
+(attribute_item
+  (attribute
+    (identifier) @route.attr_name
+    (#match? @route.attr_name "^(get|post|put|patch|delete|options|head|route|web)$")
+  )
+) @route.attr
+
+; ── AppState / service struct ─────────────────────────────────────────────────
+
+(struct_item
+  name: (type_identifier) @service.struct_name
+) @service.struct
+
+; ── use declarations ─────────────────────────────────────────────────────────
+
+(use_declaration
+  argument: _ @import.path
+) @import.use

codebeacon/extract/queries/angular.scm

@@ -0,0 +1,155 @@
+; ── Angular (TypeScript) ─────────────────────────────────────────────────────
+; Grammar: tree-sitter-typescript
+;
+; TypeScript decorator note:
+;   - decorator: @Component({...})
+;   - call_expression: Component({...}) inside decorator
+;   - object: { selector: "...", templateUrl: "..." }
+;   - property_identifier: key names
+;
+; Angular-specific:
+;   - @Component({ selector, templateUrl/template, styleUrls })
+;   - @Injectable({ providedIn: "root" })
+;   - @NgModule({ declarations, imports, providers, bootstrap })
+;   - RouterModule.forRoot(routes) / RouterModule.forChild(routes)
+;   - constructor(private service: ServiceType) → DI
+;
+; Captures:
+;   @component.class_name   - @Component class name
+;   @component.selector     - selector: "app-root"
+;   @component.template_url - templateUrl: "./foo.component.html"
+;   @service.class_name     - @Injectable class name
+;   @service.inject_type    - constructor injected type
+;   @module.class_name      - @NgModule class name
+;   @module.providers       - providers array items
+;   @route.path             - Routes array { path: "users" }
+;   @route.component        - Routes array { component: UserComponent }
+;   @route.lazy_load        - loadComponent/loadChildren
+;   @import.path            - import source
+
+; ── @Component({...}) class ───────────────────────────────────────────────────
+
+(class_declaration
+  (decorator
+    (call_expression
+      function: (identifier) @_comp (#eq? @_comp "Component")
+      arguments: (arguments
+        (object
+          (pair
+            key: (property_identifier) @_sel (#eq? @_sel "selector")
+            value: (string
+              (string_fragment) @component.selector
+            )
+          )
+        )
+      )
+    )
+  )
+  name: (type_identifier) @component.class_name
+) @component.class
+
+; templateUrl capture
+(decorator
+  (call_expression
+    function: (identifier) @_comp2 (#eq? @_comp2 "Component")
+    arguments: (arguments
+      (object
+        (pair
+          key: (property_identifier) @_turl
+          (#eq? @_turl "templateUrl")
+          value: (string
+            (string_fragment) @component.template_url
+          )
+        )
+      )
+    )
+  )
+) @component.template_url_decorator
+
+; ── @Injectable({ providedIn: "root" }) class ────────────────────────────────
+
+(class_declaration
+  (decorator
+    (call_expression
+      function: (identifier) @_inj (#eq? @_inj "Injectable")
+    )
+  )
+  name: (type_identifier) @service.class_name
+) @service.injectable
+
+; Constructor DI
+(method_definition
+  name: (property_identifier) @_ctor
+  (#eq? @_ctor "constructor")
+  parameters: (formal_parameters
+    (required_parameter
+      pattern: (identifier) @_pname
+      type: (type_annotation
+        (type_identifier) @service.inject_type
+      )
+    )
+  )
+) @service.constructor_di
+
+; ── @NgModule({...}) class ────────────────────────────────────────────────────
+
+(class_declaration
+  (decorator
+    (call_expression
+      function: (identifier) @_mod (#eq? @_mod "NgModule")
+      arguments: (arguments
+        (object
+          (pair
+            key: (property_identifier) @_providers (#eq? @_providers "providers")
+            value: (array
+              (identifier) @module.providers
+            )
+          )
+        )
+      )
+    )
+  )
+  name: (type_identifier) @module.class_name
+) @module.ngmodule
+
+; ── Angular Router routes array ───────────────────────────────────────────────
+
+; Routes array: { path: "users", component: UserListComponent }
+(array
+  (object
+    (pair
+      key: (property_identifier) @_path (#eq? @_path "path")
+      value: (string
+        (string_fragment) @route.path
+      )
+    )
+  )
+) @route.routes
+
+; component reference in route
+(object
+  (pair
+    key: (property_identifier) @_path2 (#eq? @_path2 "path")
+    value: (string (string_fragment) @route.path)
+  )
+  (pair
+    key: (property_identifier) @_comp3
+    (#match? @_comp3 "^(component|redirectTo)$")
+    value: (identifier) @route.component
+  )
+) @route.route_entry
+
+; Lazy loading: loadComponent: () => import("./foo").then(m => m.FooComponent)
+(object
+  (pair
+    key: (property_identifier) @_ll
+    (#match? @_ll "^(loadComponent|loadChildren)$")
+    value: _ @route.lazy_load
+  )
+) @route.lazy
+
+; ── imports ───────────────────────────────────────────────────────────────────
+
+(import_statement
+  source: (string) @import.path
+) @import.decl

codebeacon/extract/queries/aspnet.scm

@@ -0,0 +1,159 @@
+; ── ASP.NET Core (C#) ─────────────────────────────────────────────────────────
+; Grammar: tree-sitter-c-sharp
+;
+; C# grammar note:
+;   - attribute_list: [HttpGet("{id}")] — wraps attribute nodes
+;   - attribute: HttpGet("{id}") — name + attribute_argument_list
+;   - identifier: attribute/class/method names
+;   - string_literal_content: content inside string_literal
+;
+; Special handling in extract/routes.py:
+;   - [Route("api/[controller]")] → replace [controller] with class name minus "Controller"
+;   - Minimal API: app.MapGet("/users", handler)
+;
+; Captures:
+;   @route.class_attr       - [Route("api/[controller]")] on class → path template
+;   @route.class_name       - controller class name
+;   @route.method_attr      - [HttpGet] / [HttpPost] etc. attribute name
+;   @route.method_path      - path in [HttpGet("{id}")]
+;   @route.method_name      - action method name
+;   @route.map_method       - MapGet/MapPost etc. (minimal API)
+;   @route.map_path         - path string (minimal API)
+;   @service.class_name     - service class name
+;   @service.interface      - implemented interface
+;   @di.service_type        - AddScoped<IService, ServiceImpl> → interface type
+;   @di.impl_type           - implementation type
+;   @entity.class_name      - EF Core entity class (DbSet target)
+;   @entity.dbset_name      - DbSet<User> property name
+;   @import.path            - using namespace
+
+; ── Controller class with [Route(...)] ───────────────────────────────────────
+
+(class_declaration
+  (attribute_list
+    (attribute
+      name: (identifier) @_route (#eq? @_route "Route")
+      (attribute_argument_list
+        (attribute_argument
+          (string_literal
+            (string_literal_content) @route.class_attr
+          )
+        )
+      )
+    )
+  )
+  name: (identifier) @route.class_name
+) @route.controller_class
+
+; Controller class without route (uses [ApiController] convention)
+(class_declaration
+  (attribute_list
+    (attribute
+      name: (identifier) @_api
+      (#match? @_api "^(ApiController|Controller)$")
+    )
+  )
+  name: (identifier) @route.class_name
+) @route.controller_bare
+
+; ── Method route attributes ───────────────────────────────────────────────────
+
+; [HttpGet("{id}")] — with path
+(method_declaration
+  (attribute_list
+    (attribute
+      name: (identifier) @route.method_attr
+      (#match? @route.method_attr "^(HttpGet|HttpPost|HttpPut|HttpPatch|HttpDelete|HttpOptions|HttpHead|Route)$")
+      (attribute_argument_list
+        (attribute_argument
+          (string_literal
+            (string_literal_content) @route.method_path
+          )
+        )
+      )
+    )
+  )
+  name: (identifier) @route.method_name
+) @route.method_with_path
+
+; [HttpPost] — without path
+(method_declaration
+  (attribute_list
+    (attribute
+      name: (identifier) @route.method_attr
+      (#match? @route.method_attr "^(HttpGet|HttpPost|HttpPut|HttpPatch|HttpDelete|HttpOptions|HttpHead)$")
+    )
+  )
+  name: (identifier) @route.method_name
+) @route.method_bare
+
+; ── Minimal API: app.MapGet("/users", handler) ───────────────────────────────
+
+(expression_statement
+  (invocation_expression
+    expression: (member_access_expression
+      name: (identifier) @route.map_method
+      (#match? @route.map_method "^(MapGet|MapPost|MapPut|MapPatch|MapDelete|MapMethods|Map)$")
+    )
+    (argument_list
+      (argument
+        (string_literal
+          (string_literal_content) @route.map_path
+        )
+      )
+    )
+  )
+) @route.minimal_api
+
+; ── Service class implementing interface ──────────────────────────────────────
+
+(class_declaration
+  name: (identifier) @service.class_name
+  (base_list
+    (identifier) @service.interface
+  )
+) @service.class
+
+; ── DI registration: builder.Services.AddScoped<IFoo, FooImpl>() ─────────────
+
+(invocation_expression
+  expression: (member_access_expression
+    name: (identifier) @_scope
+    (#match? @_scope "^(AddScoped|AddSingleton|AddTransient|AddHostedService)$")
+  )
+  (argument_list)
+) @di.registration
+
+; Generic DI: AddScoped<IFoo, FooImpl>()
+(invocation_expression
+  expression: (generic_name
+    (identifier) @_scope
+    (#match? @_scope "^(AddScoped|AddSingleton|AddTransient)$")
+    (type_argument_list
+      (identifier) @di.service_type
+      (identifier) @di.impl_type
+    )
+  )
+) @di.generic_registration
+
+; ── EF Core DbContext ─────────────────────────────────────────────────────────
+
+(property_declaration
+  type: (generic_name
+    (identifier) @_dbset (#eq? @_dbset "DbSet")
+    (type_argument_list
+      (identifier) @entity.class_name
+    )
+  )
+  name: (identifier) @entity.dbset_name
+) @entity.dbset
+
+; ── using directives ─────────────────────────────────────────────────────────
+
+(using_directive
+  (qualified_name) @import.path
+) @import.using
+
+(using_directive
+  (identifier) @import.path
+) @import.using_simple

codebeacon/extract/queries/django.scm

@@ -0,0 +1,122 @@
+; ── Django (Python) ───────────────────────────────────────────────────────────
+; Grammar: tree-sitter-python
+;
+; Convention note: Django routes live in urls.py, models in models.py.
+; The extractor handles file-name-based dispatch before running these queries.
+;
+; Captures:
+;   @route.path_call    - path("...", view) call
+;   @route.path_str     - URL string
+;   @route.view_name    - view function/class reference
+;   @route.include_path - include("app.urls") string
+;   @route.re_path_str  - re_path("regex") string
+;   @entity.class_name  - models.Model subclass
+;   @entity.field_name  - model field name
+;   @entity.field_type  - model field type (e.g. CharField, ForeignKey)
+;   @view.class_name    - CBV class name
+;   @view.func_name     - FBV function name
+;   @import.path        - import path
+
+; ── urlpatterns path() entries ───────────────────────────────────────────────
+
+(assignment
+  left: (identifier) @_up
+  (#eq? @_up "urlpatterns")
+  right: (list
+    (call
+      function: (identifier) @_path_fn
+      (#match? @_path_fn "^(path|re_path)$")
+      arguments: (argument_list
+        (string) @route.path_str
+        [
+          (identifier) @route.view_name
+          (attribute
+            attribute: (identifier) @route.view_name
+          )
+          (call
+            function: (attribute
+              attribute: (identifier) @_asv
+              (#eq? @_asv "as_view")
+            )
+          )
+        ]
+      )
+    )
+  )
+) @route.urlpatterns
+
+; include() inside urlpatterns
+(call
+  function: (identifier) @_inc
+  (#eq? @_inc "include")
+  arguments: (argument_list
+    (string) @route.include_path
+  )
+) @route.include
+
+; ── Django model ──────────────────────────────────────────────────────────────
+
+(class_definition
+  name: (identifier) @entity.class_name
+  superclasses: (argument_list
+    [
+      (attribute
+        attribute: (identifier) @_model
+        (#match? @_model "^(Model|AbstractModel|AbstractBaseModel)$")
+      )
+      (identifier) @_model
+      (#match? @_model "^(Model|AbstractModel)$")
+    ]
+  )
+  body: (block
+    (expression_statement
+      (assignment
+        left: (identifier) @entity.field_name
+        right: (call
+          function: [
+            (identifier) @entity.field_type
+            (attribute
+              attribute: (identifier) @entity.field_type
+            )
+          ]
+          (#match? @entity.field_type "^(CharField|IntegerField|FloatField|BooleanField|DateField|DateTimeField|ForeignKey|OneToOneField|ManyToManyField|TextField|EmailField|URLField|SlugField|UUIDField|FileField|ImageField|JSONField|AutoField|BigAutoField|SmallIntegerField|PositiveIntegerField)$")
+        )
+      )
+    )
+  )
+) @entity.model
+
+; ── Class-based views ────────────────────────────────────────────────────────
+
+(class_definition
+  name: (identifier) @view.class_name
+  superclasses: (argument_list
+    [
+      (identifier) @_view_base
+      (attribute attribute: (identifier) @_view_base)
+    ]
+    (#match? @_view_base "^(View|ListView|DetailView|CreateView|UpdateView|DeleteView|TemplateView|APIView|GenericAPIView|ModelViewSet|ReadOnlyModelViewSet|ViewSet|FormView)$")
+  )
+) @view.cbv
+
+; ── Function-based views (decorated with @login_required etc.) ───────────────
+
+(decorated_definition
+  definition: (function_definition
+    name: (identifier) @view.func_name
+    parameters: (parameters
+      (identifier) @_req
+      (#eq? @_req "request")
+    )
+  )
+) @view.fbv_decorated
+
+; ── imports ───────────────────────────────────────────────────────────────────
+
+(import_from_statement
+  module_name: _ @import.path
+) @import.from
+
+(import_statement
+  name: _ @import.path
+) @import.plain

codebeacon/extract/queries/express.scm

@@ -0,0 +1,124 @@
+; ── Express / Koa / Fastify (JavaScript/TypeScript) ──────────────────────────
+; Grammar: tree-sitter-javascript / tree-sitter-typescript
+;
+; Captures:
+;   @route.method     - "get", "post", "put", "delete", "patch", "all"
+;   @route.path       - path string literal
+;   @route.object     - router/app object name (for prefix tracking)
+;   @route.use_prefix - path passed to app.use() for prefix mounting
+;   @service.name     - exported class name
+;   @import.path      - import/require path
+
+; ── app.METHOD("/path", handler) ─────────────────────────────────────────────
+
+(expression_statement
+  (call_expression
+    function: (member_expression
+      object: (identifier) @route.object
+      property: (property_identifier) @route.method
+      (#match? @route.method "^(get|post|put|patch|delete|del|options|head|all|use)$")
+    )
+    arguments: (arguments
+      .
+      (string) @route.path
+    )
+  )
+) @route.call
+
+; ── router.route("/path").get(handler).post(handler) ─────────────────────────
+
+(call_expression
+  function: (member_expression
+    object: (call_expression
+      function: (member_expression
+        object: (identifier) @route.object
+        property: (property_identifier) @_route_kw
+        (#eq? @_route_kw "route")
+      )
+      arguments: (arguments
+        .
+        (string) @route.path
+      )
+    )
+    property: (property_identifier) @route.method
+    (#match? @route.method "^(get|post|put|patch|delete|options|head|all)$")
+  )
+) @route.chained
+
+; ── app.use("/prefix", router) — prefix mounting ─────────────────────────────
+
+(expression_statement
+  (call_expression
+    function: (member_expression
+      object: (identifier) @_obj
+      property: (property_identifier) @_use
+      (#eq? @_use "use")
+    )
+    arguments: (arguments
+      .
+      (string) @route.use_prefix
+      .
+      _
+    )
+  )
+) @route.use_mount
+
+; ── const router = express.Router() / Router() ───────────────────────────────
+
+(variable_declarator
+  name: (identifier) @route.router_name
+  value: (call_expression
+    function: [
+      (identifier) @_r (#eq? @_r "Router")
+      (member_expression property: (property_identifier) @_r (#eq? @_r "Router"))
+    ]
+  )
+) @route.router_decl
+
+; ── Fastify: fastify.register(plugin, { prefix: "/api" }) ────────────────────
+
+(call_expression
+  function: (member_expression
+    object: (identifier) @_app
+    property: (property_identifier) @_reg
+    (#eq? @_reg "register")
+  )
+  arguments: (arguments
+    _
+    (object
+      (pair
+        key: (property_identifier) @_k
+        (#eq? @_k "prefix")
+        value: (string) @route.use_prefix
+      )
+    )
+  )
+) @route.fastify_register
+
+; ── Exported class (service) ──────────────────────────────────────────────────
+
+(export_statement
+  declaration: (class_declaration
+    name: (identifier) @service.name
+  )
+) @service.export_class
+
+(class_declaration
+  name: (identifier) @service.name
+) @service.class
+
+; ── imports ───────────────────────────────────────────────────────────────────
+
+; ES module: import ... from "..."
+(import_statement
+  source: (string) @import.path
+) @import.es
+
+; CommonJS: require("...")
+(call_expression
+  function: (identifier) @_req
+  (#eq? @_req "require")
+  arguments: (arguments
+    (string) @import.path
+  )
+) @import.cjs