rbtr-lang-tree-sitter-query 2026.9.0.dev1__tar.gz

rbtr_lang_tree_sitter_query-2026.9.0.dev1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Alejandro Giacometti
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

rbtr_lang_tree_sitter_query-2026.9.0.dev1/PKG-INFO

@@ -0,0 +1,60 @@
+Metadata-Version: 2.4
+Name: rbtr-lang-tree-sitter-query
+Version: 2026.9.0.dev1
+Summary: rbtr — tree-sitter query (.scm) language plugin
+Keywords: code-search,code-index,tree-sitter,static-analysis,semantic-search,developer-tools,tree-sitter-query
+Author: Alejandro Giacometti
+Author-email: Alejandro Giacometti <alejandro.giacometti@gmail.com>
+License-Expression: MIT
+License-File: LICENSE
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Software Development :: Libraries
+Classifier: Topic :: Text Processing :: Indexing
+Classifier: Typing :: Typed
+Requires-Dist: rbtr==2026.9.0.dev1
+Requires-Dist: tree-sitter-query
+Requires-Python: >=3.13
+Project-URL: Homepage, https://github.com/janrito/rbtr
+Project-URL: Repository, https://github.com/janrito/rbtr
+Project-URL: Documentation, https://github.com/janrito/rbtr/tree/main/packages/rbtr-lang-tree-sitter-query#readme
+Project-URL: Issues, https://github.com/janrito/rbtr/issues
+Project-URL: Changelog, https://github.com/janrito/rbtr/releases
+Description-Content-Type: text/markdown
+
+# rbtr-lang-tree-sitter-query
+
+tree-sitter query (`.scm`) support for [rbtr]. Optional plugin —
+install with `pip install rbtr[tree_sitter_query]`.
+
+[rbtr]: https://github.com/janrito/rbtr/tree/main/packages/rbtr#readme
+
+## What it ingests
+
+`.scm` files — the tree-sitter query language rbtr's own plugins are written
+in, and which third-party grammars ship too (`tags.scm`, `highlights.scm`,
+`injections.scm`), so cloned repos are covered. Each top-level pattern becomes
+a **doc-section** chunk, named by its own outer capture where the author gave
+one. Patterns with no outer label are anonymous sections whose nested captures
+stay full-text-searchable.
+
+## Chunks produced
+
+```scm
+; doc_section "function"
+(function_definition name: (identifier) @_fn_name) @function
+
+; doc_section "keyword"
+["if" "else"] @keyword
+```
+
+## Embedded / injected chunks
+
+None. Query files do not embed other languages.
+
+## Grammar & dependencies
+
+Uses the `tree-sitter-query` grammar (`.scm`). No dependency on other language
+plugins.

rbtr_lang_tree_sitter_query-2026.9.0.dev1/README.md

@@ -0,0 +1,34 @@
+# rbtr-lang-tree-sitter-query
+
+tree-sitter query (`.scm`) support for [rbtr]. Optional plugin —
+install with `pip install rbtr[tree_sitter_query]`.
+
+[rbtr]: https://github.com/janrito/rbtr/tree/main/packages/rbtr#readme
+
+## What it ingests
+
+`.scm` files — the tree-sitter query language rbtr's own plugins are written
+in, and which third-party grammars ship too (`tags.scm`, `highlights.scm`,
+`injections.scm`), so cloned repos are covered. Each top-level pattern becomes
+a **doc-section** chunk, named by its own outer capture where the author gave
+one. Patterns with no outer label are anonymous sections whose nested captures
+stay full-text-searchable.
+
+## Chunks produced
+
+```scm
+; doc_section "function"
+(function_definition name: (identifier) @_fn_name) @function
+
+; doc_section "keyword"
+["if" "else"] @keyword
+```
+
+## Embedded / injected chunks
+
+None. Query files do not embed other languages.
+
+## Grammar & dependencies
+
+Uses the `tree-sitter-query` grammar (`.scm`). No dependency on other language
+plugins.

rbtr_lang_tree_sitter_query-2026.9.0.dev1/pyproject.toml

@@ -0,0 +1,57 @@
+[build-system]
+requires = ["uv_build>=0.11.26,<1"]
+build-backend = "uv_build"
+
+[tool.uv.build-backend]
+module-name = "rbtr_lang_tree_sitter_query"
+
+[tool.uv.sources.rbtr]
+workspace = true
+
+[project]
+name = "rbtr-lang-tree-sitter-query"
+version = "2026.9.0-dev1"
+description = "rbtr — tree-sitter query (.scm) language plugin"
+readme = "README.md"
+license = "MIT"
+license-files = ["LICENSE"]
+keywords = [
+    "code-search",
+    "code-index",
+    "tree-sitter",
+    "static-analysis",
+    "semantic-search",
+    "developer-tools",
+    "tree-sitter-query",
+]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "Programming Language :: Python :: 3 :: Only",
+    "Programming Language :: Python :: 3.13",
+    "Topic :: Software Development :: Libraries",
+    "Topic :: Text Processing :: Indexing",
+    "Typing :: Typed",
+]
+requires-python = ">=3.13"
+dependencies = [
+    "rbtr==2026.9.0-dev1",
+    "tree-sitter-query",
+]
+
+[[project.authors]]
+name = "Alejandro Giacometti"
+email = "alejandro.giacometti@gmail.com"
+
+[project.urls]
+Homepage = "https://github.com/janrito/rbtr"
+Repository = "https://github.com/janrito/rbtr"
+Documentation = "https://github.com/janrito/rbtr/tree/main/packages/rbtr-lang-tree-sitter-query#readme"
+Issues = "https://github.com/janrito/rbtr/issues"
+Changelog = "https://github.com/janrito/rbtr/releases"
+
+[project.entry-points."rbtr.languages"]
+tree_sitter_query = "rbtr_lang_tree_sitter_query.plugin:tree_sitter_query"
+
+[dependency-groups]
+dev = ["rbtr[test]"]

rbtr_lang_tree_sitter_query-2026.9.0.dev1/pyproject.toml.orig

@@ -0,0 +1,61 @@
+[build-system]
+requires = ["uv_build>=0.11.26,<1"]
+build-backend = "uv_build"
+
+[tool.uv.build-backend]
+module-name = "rbtr_lang_tree_sitter_query"
+# No source-exclude: the plugin's tests, samples, and snapshots ship with
+# the wheel (they are the language's golden record), and tree_sitter_query.scm
+# ships as package data.
+
+[project]
+name = "rbtr-lang-tree-sitter-query"
+version = "2026.9.0-dev1"
+description = "rbtr — tree-sitter query (.scm) language plugin"
+readme = "README.md"
+license = "MIT"
+license-files = ["LICENSE"]
+authors = [
+  { name = "Alejandro Giacometti", email = "alejandro.giacometti@gmail.com" },
+]
+keywords = [
+  "code-search",
+  "code-index",
+  "tree-sitter",
+  "static-analysis",
+  "semantic-search",
+  "developer-tools",
+  "tree-sitter-query",
+]
+classifiers = [
+  "Development Status :: 4 - Beta",
+  "Intended Audience :: Developers",
+  "Programming Language :: Python :: 3 :: Only",
+  "Programming Language :: Python :: 3.13",
+  "Topic :: Software Development :: Libraries",
+  "Topic :: Text Processing :: Indexing",
+  "Typing :: Typed",
+]
+requires-python = ">=3.13"
+dependencies = [
+  "rbtr==2026.9.0-dev1",
+  "tree-sitter-query",
+]
+
+[project.urls]
+Homepage = "https://github.com/janrito/rbtr"
+Repository = "https://github.com/janrito/rbtr"
+Documentation = "https://github.com/janrito/rbtr/tree/main/packages/rbtr-lang-tree-sitter-query#readme"
+Issues = "https://github.com/janrito/rbtr/issues"
+Changelog = "https://github.com/janrito/rbtr/releases"
+
+[project.entry-points."rbtr.languages"]
+# The value resolves to the language's `LanguageRegistration` (named by its
+# id) — see the rbtr ARCHITECTURE "External plugins" note.
+tree_sitter_query = "rbtr_lang_tree_sitter_query.plugin:tree_sitter_query"
+
+[tool.uv.sources]
+rbtr = { workspace = true }
+
+[dependency-groups]
+dev = ["rbtr[test]"]

rbtr_lang_tree_sitter_query-2026.9.0.dev1/src/rbtr_lang_tree_sitter_query/__init__.py

@@ -0,0 +1 @@
+"""tree-sitter query language plugin package."""

rbtr_lang_tree_sitter_query-2026.9.0.dev1/src/rbtr_lang_tree_sitter_query/plugin.py

@@ -0,0 +1,35 @@
+"""tree-sitter query language plugin.
+
+Indexes `.scm` files — the tree-sitter query language that rbtr's own
+language plugins are written in (see the `.scm` files beside each
+`plugin.py`). Third-party grammars ship `.scm` too (`tags.scm`,
+`highlights.scm`, `injections.scm`), so cloned repos are covered as well.
+
+Each top-level pattern becomes a `doc_section` chunk, named by its own
+outer capture where the author gave one (`tree_sitter_query.scm` pairs
+it as `@_section_name`):
+
+    (function_definition name: (identifier) @_fn_name) @function
+        → section "function"
+
+    ["if" "else"] @keyword
+        → section "keyword"
+
+Patterns with no outer label of their own — a predicate-wrapped injection
+rule, or one anchored at a structural wrapper — are anonymous sections;
+their nested captures stay full-text-searchable.
+"""
+
+from __future__ import annotations
+
+from rbtr.languages.registration import LanguageRegistration, QueryExtraction, load_query
+
+tree_sitter_query = LanguageRegistration(
+    id="tree_sitter_query",
+    extensions=frozenset({".scm"}),
+    grammar_module="tree_sitter_query",
+    extraction=QueryExtraction(
+        query=load_query(__package__, "tree_sitter_query"),
+    ),
+    extraction_serial=2,
+)

rbtr_lang_tree_sitter_query-2026.9.0.dev1/src/rbtr_lang_tree_sitter_query/tests/__snapshots__/test_samples/test_edges_match_snapshot.json

@@ -0,0 +1 @@
+[]

rbtr_lang_tree_sitter_query-2026.9.0.dev1/src/rbtr_lang_tree_sitter_query/tests/__snapshots__/test_samples/test_extraction_matches_snapshot.json

@@ -0,0 +1,230 @@
+[
+  {
+    "blob_sha": "sha1",
+    "file_path": "highlights.scm",
+    "kind": "comment",
+    "name": "",
+    "scope": "",
+    "language": "tree_sitter_query",
+    "file_language": "tree_sitter_query",
+    "content": "; A tree-sitter query file exercising every construct rbtr should index:\n; named nodes with fields, anonymous nodes, alternation lists, groupings,\n; predicates, wildcards, negated fields, anchors, quantifiers, supertypes,\n; and a MISSING-node pattern. Every top-level pattern is one doc section.",
+    "line_start": 1,
+    "line_end": 4,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "91411c21fd288abd"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "highlights.scm",
+    "kind": "doc_section",
+    "name": "definition.function",
+    "scope": "",
+    "language": "tree_sitter_query",
+    "file_language": "tree_sitter_query",
+    "content": "; Named-node patterns with a field and an outer definition capture.\n(function_definition\n  name: (identifier) @function\n  body: (_) @_body) @definition.function",
+    "line_start": 6,
+    "line_end": 9,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "22635e30fc3ec0ec"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "highlights.scm",
+    "kind": "doc_section",
+    "name": "definition.class",
+    "scope": "",
+    "language": "tree_sitter_query",
+    "file_language": "tree_sitter_query",
+    "content": "(class_definition\n  name: (identifier) @type) @definition.class",
+    "line_start": 11,
+    "line_end": 12,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "2f7ccba614073350"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "highlights.scm",
+    "kind": "doc_section",
+    "name": "keyword",
+    "scope": "",
+    "language": "tree_sitter_query",
+    "file_language": "tree_sitter_query",
+    "content": "; Alternation list and a bare anonymous-node matcher.\n[\"if\" \"else\" \"while\" \"return\"] @keyword",
+    "line_start": 14,
+    "line_end": 15,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "68190efc87150b38"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "highlights.scm",
+    "kind": "doc_section",
+    "name": "keyword.coroutine",
+    "scope": "",
+    "language": "tree_sitter_query",
+    "file_language": "tree_sitter_query",
+    "content": "\"async\" @keyword.coroutine",
+    "line_start": 17,
+    "line_end": 17,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "4587431c020385fb"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "highlights.scm",
+    "kind": "doc_section",
+    "name": "",
+    "scope": "",
+    "language": "tree_sitter_query",
+    "file_language": "tree_sitter_query",
+    "content": "; Wildcard child, a negated field, anchors, and quantifiers.\n(call_expression\n  function: (_) @call\n  !type_arguments)",
+    "line_start": 19,
+    "line_end": 22,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "efd9cbebd9a590b9"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "highlights.scm",
+    "kind": "doc_section",
+    "name": "",
+    "scope": "",
+    "language": "tree_sitter_query",
+    "file_language": "tree_sitter_query",
+    "content": "(array . (identifier) @first (_)* @rest)",
+    "line_start": 24,
+    "line_end": 24,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "7d959afd205ade46"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "highlights.scm",
+    "kind": "doc_section",
+    "name": "",
+    "scope": "",
+    "language": "tree_sitter_query",
+    "file_language": "tree_sitter_query",
+    "content": "; Groupings with predicates: regex match, alternation, and an injection.\n((identifier) @constant\n  (#match? @constant \"^[A-Z][A-Z_]+$\"))",
+    "line_start": 26,
+    "line_end": 28,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "0ef0b2c93e4c6519"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "highlights.scm",
+    "kind": "doc_section",
+    "name": "",
+    "scope": "",
+    "language": "tree_sitter_query",
+    "file_language": "tree_sitter_query",
+    "content": "((identifier) @type.builtin\n  (#any-of? @type.builtin \"int\" \"str\" \"bool\"))",
+    "line_start": 30,
+    "line_end": 31,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "d90c7294ccaddc1d"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "highlights.scm",
+    "kind": "doc_section",
+    "name": "",
+    "scope": "",
+    "language": "tree_sitter_query",
+    "file_language": "tree_sitter_query",
+    "content": "((comment) @injection.content\n  (#set! injection.language \"markdown\"))",
+    "line_start": 33,
+    "line_end": 34,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "a89ebd6b36363bf2"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "highlights.scm",
+    "kind": "doc_section",
+    "name": "definition.function",
+    "scope": "",
+    "language": "tree_sitter_query",
+    "file_language": "tree_sitter_query",
+    "content": "; Supertype/subtype match and a MISSING-node pattern.\n(declaration/function_definition) @definition.function",
+    "line_start": 36,
+    "line_end": 37,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "d208be73ec2bbd8a"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "highlights.scm",
+    "kind": "doc_section",
+    "name": "",
+    "scope": "",
+    "language": "tree_sitter_query",
+    "file_language": "tree_sitter_query",
+    "content": "(binary_expression (MISSING identifier))",
+    "line_start": 39,
+    "line_end": 39,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "cbe537008a79992a"
+  }
+]

rbtr_lang_tree_sitter_query-2026.9.0.dev1/src/rbtr_lang_tree_sitter_query/tests/samples/tree_sitter_query/highlights.scm

@@ -0,0 +1,39 @@
+; A tree-sitter query file exercising every construct rbtr should index:
+; named nodes with fields, anonymous nodes, alternation lists, groupings,
+; predicates, wildcards, negated fields, anchors, quantifiers, supertypes,
+; and a MISSING-node pattern. Every top-level pattern is one doc section.
+
+; Named-node patterns with a field and an outer definition capture.
+(function_definition
+  name: (identifier) @function
+  body: (_) @_body) @definition.function
+
+(class_definition
+  name: (identifier) @type) @definition.class
+
+; Alternation list and a bare anonymous-node matcher.
+["if" "else" "while" "return"] @keyword
+
+"async" @keyword.coroutine
+
+; Wildcard child, a negated field, anchors, and quantifiers.
+(call_expression
+  function: (_) @call
+  !type_arguments)
+
+(array . (identifier) @first (_)* @rest)
+
+; Groupings with predicates: regex match, alternation, and an injection.
+((identifier) @constant
+  (#match? @constant "^[A-Z][A-Z_]+$"))
+
+((identifier) @type.builtin
+  (#any-of? @type.builtin "int" "str" "bool"))
+
+((comment) @injection.content
+  (#set! injection.language "markdown"))
+
+; Supertype/subtype match and a MISSING-node pattern.
+(declaration/function_definition) @definition.function
+
+(binary_expression (MISSING identifier))

rbtr_lang_tree_sitter_query-2026.9.0.dev1/src/rbtr_lang_tree_sitter_query/tests/test_samples.py

@@ -0,0 +1,68 @@
+"""tree-sitter query sample extraction: `samples/tree_sitter_query/` through the real pipeline."""
+
+from __future__ import annotations
+
+from pathlib import Path
+from typing import TYPE_CHECKING
+
+import pytest
+from tree_sitter import Parser
+
+from rbtr.domain.models import Chunk, ChunkKind, Edge
+from rbtr.git import FileEntry
+from rbtr.languages.edges import build_resolution_map, infer_import_edges
+from rbtr.languages.extract import extract_file
+from rbtr.languages.manager import get_manager
+from rbtr.testing import render_edges
+
+if TYPE_CHECKING:
+    from syrupy.assertion import SnapshotAssertion
+
+
+@pytest.fixture
+def project() -> list[tuple[str, str]]:
+    root = Path(__file__).parent / "samples" / "tree_sitter_query"
+    return [
+        (str(p.relative_to(root)), p.read_text()) for p in sorted(root.rglob("*")) if p.is_file()
+    ]
+
+
+@pytest.fixture
+def chunks(project: list[tuple[str, str]]) -> list[Chunk]:
+    manager = get_manager()
+    out: list[Chunk] = []
+    for path, text in project:
+        lang = manager.detect_language(path) or "tree_sitter_query"
+        out.extend(extract_file(FileEntry(path, "sha1", text.encode()), lang))
+    return out
+
+
+@pytest.fixture
+def edges(project: list[tuple[str, str]], chunks: list[Chunk]) -> list[Edge]:
+    manager = get_manager()
+    repo_files = {path for path, _ in project}
+    return infer_import_edges(chunks, repo_files, build_resolution_map(manager))
+
+
+def test_emits_expected_kinds(chunks: list[Chunk]) -> None:
+    """Each top-level query pattern is a doc-section chunk."""
+    kinds = {c.kind for c in chunks}
+    assert {ChunkKind.DOC_SECTION, ChunkKind.COMMENT} <= kinds
+
+
+def test_parses_cleanly(project: list[tuple[str, str]]) -> None:
+    manager = get_manager()
+    for path, text in project:
+        grammar = manager.grammar(manager.detect_language(path) or "tree_sitter_query")
+        assert grammar is not None
+        assert not Parser(grammar).parse(text.encode()).root_node.has_error, path
+
+
+def test_extraction_matches_snapshot(chunks: list[Chunk], snapshot_json: SnapshotAssertion) -> None:
+    assert chunks == snapshot_json
+
+
+def test_edges_match_snapshot(
+    chunks: list[Chunk], edges: list[Edge], snapshot_json: SnapshotAssertion
+) -> None:
+    assert render_edges(edges, chunks) == snapshot_json

rbtr_lang_tree_sitter_query-2026.9.0.dev1/src/rbtr_lang_tree_sitter_query/tree_sitter_query.scm

@@ -0,0 +1,14 @@
+; Every top-level pattern is a section. A pattern is named by its own outer
+; label — a capture that is a direct child of the pattern node (e.g.
+; `(function_definition …) @function` → "function"; `["if" "else"] @keyword`
+; → "keyword"). The `?` makes that label optional: predicate-wrapped rules
+; and structural wrappers carry no label of their own and stay anonymous.
+; Captures, predicates, and matched node types inside remain full-text
+; searchable within the section either way.
+; Comments: a `;` block folds into the pattern it precedes, else standalone.
+(comment) @comment
+
+(program (named_node (capture (identifier) @_section_name)?) @doc_section)
+(program (list (capture (identifier) @_section_name)?) @doc_section)
+(program (grouping (capture (identifier) @_section_name)?) @doc_section)
+(program (anonymous_node (capture (identifier) @_section_name)?) @doc_section)