rbtr-lang-javascript 2026.9.0.dev0__py3-none-any.whl

rbtr_lang_javascript/__init__.py

@@ -0,0 +1 @@
+"""JavaScript and TypeScript language plugin package."""

rbtr_lang_javascript/javascript.scm

@@ -0,0 +1,4 @@
+; JS uses `identifier` for class names (TS uses `type_identifier`).
+(class_declaration
+  name: (identifier) @_cls_name) @class
+

rbtr_lang_javascript/plugin.py

@@ -0,0 +1,190 @@
+"""JavaScript and TypeScript language plugins.
+
+Registers three languages (one per tree-sitter grammar) sharing a
+common import extractor and a shared core query (classes, functions,
+generators, variables, imports, and methods — class/object members
+including get/set accessors).  TS additionally captures interfaces,
+enums, type aliases, abstract classes, and interface/abstract method
+signatures.
+
+Extracted chunks::
+
+    function greet() {}             → function "greet", scope ""
+    const add = (a, b) => a + b;    → function "add", scope ""
+    class Service {}                → class "Service", scope ""
+
+    import { foo } from './models'
+        → import, metadata {module: "models", names: "foo", dots: "1"}
+    import React from 'react'
+        → import, metadata {module: "react", names: "React"}
+    import './styles.css'
+        → import, metadata {module: "styles", dots: "1"}
+"""
+
+from __future__ import annotations
+
+from pathlib import PurePosixPath
+from typing import TYPE_CHECKING
+
+from rbtr.domain.models import ImportMeta
+from rbtr.languages.registration import (
+    ImportResolver,
+    LanguageRegistration,
+    QueryExtraction,
+    load_query,
+    parse_path_relative,
+)
+
+if TYPE_CHECKING:
+    from tree_sitter import Node
+
+_SHARED_QUERIES = load_query(__package__, "shared") + load_query(__package__, "variables")
+_JS_QUERY = load_query(__package__, "javascript") + _SHARED_QUERIES
+_TS_QUERY = load_query(__package__, "typescript") + _SHARED_QUERIES
+
+
+# ── Import extractor (shared by JS and TS) ───────────────────────────
+
+
+def extract_import_meta(
+    resolver: ImportResolver, node: Node, captures: dict[str, list[Node]]
+) -> ImportMeta:
+    """Extract import data from a JS/TS `import_statement` node.
+
+    Reads `@_import_module` from captures, then walks the node
+    for import names (which the query can't capture because
+    they're multi-valued).  Applies `parse_path_relative` and
+    extension stripping to the module path.
+
+    Examples:
+
+        `import { foo } from './models'`:
+            module="models", names="foo", dots="1"
+
+        `import React from 'react'`:
+            module="react", names="React"
+
+        `import * as utils from '../utils'`:
+            module="utils", names="utils", dots="2"
+
+        `import './styles.css'`:
+            module="styles", dots="1"
+    """
+    meta = resolver(node, captures)
+    names: list[str] = []
+
+    # Module comes from @_import_module capture (string_fragment).
+    # Still need to parse relative paths and strip extensions.
+    raw_module = meta.module
+
+    # Import clause (no named fields — child iteration needed).
+    for child in node.children:
+        if child.type != "import_clause":
+            continue
+        for ic in child.children:
+            match ic.type:
+                case "identifier":
+                    if ic.text:
+                        names.append(ic.text.decode())
+                case "named_imports":
+                    for spec in ic.children:
+                        if spec.type == "import_specifier":
+                            sid = spec.child_by_field_name("name")
+                            if sid and sid.text:
+                                names.append(sid.text.decode())
+                case "namespace_import":
+                    for ns in ic.children:
+                        if ns.type == "identifier" and ns.text:
+                            names.append(ns.text.decode())
+                            break
+
+    if raw_module:
+        dots, cleaned = parse_path_relative(raw_module)
+        if dots:
+            meta.dots = str(dots)
+        # Strip file extensions so edges.py can match without guessing.
+        cleaned = str(PurePosixPath(cleaned).with_suffix("")) or cleaned
+        meta.module = cleaned
+
+    if names:
+        meta.names = ",".join(names)
+    return meta
+
+
+# ── Plugin ───────────────────────────────────────────────────────────
+
+
+javascript = LanguageRegistration(
+    id="javascript",
+    extensions=frozenset({".js", ".jsx", ".mjs"}),
+    grammar_module="tree_sitter_javascript",
+    extraction=QueryExtraction(
+        query=_JS_QUERY,
+        scope_types=frozenset({"class_declaration", "function_declaration"}),
+        class_scope_types=frozenset({"class_declaration"}),
+        # Both `/** */` JSDoc and `//` comments land in
+        # the grammar as a single `comment` node type.
+    ),
+    index_files=frozenset({"index.js"}),
+    import_targets=frozenset({"javascript", "css"}),
+    source_roots=("", "src"),
+    extraction_serial=6,
+)
+typescript = LanguageRegistration(
+    id="typescript",
+    extensions=frozenset({".ts"}),
+    grammar_module="tree_sitter_typescript",
+    grammar_entry="language_typescript",
+    extraction=QueryExtraction(
+        query=_TS_QUERY,
+        scope_types=frozenset(
+            {
+                "class_declaration",
+                "abstract_class_declaration",
+                "interface_declaration",
+                "function_declaration",
+                "internal_module",
+                "module",
+                "enum_declaration",
+            }
+        ),
+        class_scope_types=frozenset(
+            {"class_declaration", "abstract_class_declaration", "interface_declaration"}
+        ),
+    ),
+    index_files=frozenset({"index.ts", "index.js"}),
+    import_targets=frozenset({"typescript", "tsx", "javascript", "css"}),
+    source_roots=("", "src"),
+    extraction_serial=6,
+)
+tsx = LanguageRegistration(
+    id="tsx",
+    extensions=frozenset({".tsx"}),
+    grammar_module="tree_sitter_typescript",
+    grammar_entry="language_tsx",
+    extraction=QueryExtraction(
+        query=_TS_QUERY,
+        scope_types=frozenset(
+            {
+                "class_declaration",
+                "abstract_class_declaration",
+                "interface_declaration",
+                "function_declaration",
+                "internal_module",
+                "module",
+                "enum_declaration",
+            }
+        ),
+        class_scope_types=frozenset(
+            {"class_declaration", "abstract_class_declaration", "interface_declaration"}
+        ),
+    ),
+    index_files=frozenset({"index.tsx", "index.ts", "index.js"}),
+    import_targets=frozenset({"tsx", "typescript", "javascript", "css"}),
+    source_roots=("", "src"),
+    extraction_serial=3,
+)
+
+javascript.import_extractor(extract_import_meta)
+typescript.import_extractor(extract_import_meta)
+tsx.import_extractor(extract_import_meta)

rbtr_lang_javascript/shared.scm

@@ -0,0 +1,29 @@
+; Shared patterns — identical across the JS and TS grammars.
+; Comments (JS/TS: single `comment` type for `//` and `/* */`, incl. JSDoc).
+(comment) @comment
+
+; method_definition covers class and object-literal members, generators,
+; and get/set accessors; inside a class it promotes to a method scoped to
+; the class, elsewhere it is a scope-less method.
+(function_declaration
+  name: (identifier) @_fn_name) @function
+
+(generator_function_declaration
+  name: (identifier) @_fn_name) @function
+
+(import_statement
+  source: (string (string_fragment) @_import_module)) @import
+
+(lexical_declaration
+  (variable_declarator
+    name: (identifier) @_fn_name
+    value: (arrow_function))) @function
+
+(method_definition
+  name: (property_identifier) @_method_name) @method
+
+; A re-export reads from another module, so it resolves as an import, and
+; it is how a symbol reaches a consumer that never names its defining
+; file.
+(export_statement
+  source: (string (string_fragment) @_import_module)) @import

rbtr_lang_javascript/tests/__snapshots__/test_samples/test_edges_match_snapshot[javascript].json

@@ -0,0 +1,4 @@
+[
+  "javascript.js::import \"./styles.css\"; -> styles.css::.greeter [imports]",
+  "javascript.js::import { LOCALE } from \"./config.js\"; -> config.js::LOCALE [imports]"
+]

rbtr_lang_javascript/tests/__snapshots__/test_samples/test_edges_match_snapshot[tsx].json

@@ -0,0 +1,3 @@
+[
+  "tsx.tsx::import { INITIAL_LABEL } from \"./labels\"; -> labels.ts::INITIAL_LABEL [imports]"
+]

rbtr_lang_javascript/tests/__snapshots__/test_samples/test_edges_match_snapshot[typescript].json

@@ -0,0 +1,6 @@
+[
+  "typescript.ts::export * from \"./config\"; -> config.ts::LOCALE [imports]",
+  "typescript.ts::export { Locale } from \"./types\"; -> types.ts::Formatter [imports]",
+  "typescript.ts::import type { Formatter } from \"./types\"; -> types.ts::Formatter [imports]",
+  "typescript.ts::import { LOCALE } from \"./config\"; -> config.ts::LOCALE [imports]"
+]

rbtr_lang_javascript/tests/__snapshots__/test_samples/test_extraction_matches_snapshot[javascript].json

@@ -0,0 +1,420 @@
+[
+  {
+    "blob_sha": "sha1",
+    "file_path": "config.js",
+    "kind": "comment",
+    "name": "",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "// Locale configuration for the greeter.",
+    "line_start": 1,
+    "line_end": 1,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "1086b120f17a34b9"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "config.js",
+    "kind": "variable",
+    "name": "LOCALE",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "LOCALE = \"en\"",
+    "line_start": 2,
+    "line_end": 2,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "79b96a86c2a4e227"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "comment",
+    "name": "",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "// Greeter — format greetings for named recipients.\n//\n// The JavaScript plugin extracts function declarations, generator\n// functions, arrow functions bound to consts, classes, module-level\n// const/let variables (including destructuring), imports, and methods\n// (class members, including get/set accessors; object-literal methods are\n// captured without a scope).",
+    "line_start": 1,
+    "line_end": 7,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "cc08702b255e2178"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "import",
+    "name": "import { LOCALE } from \"./config.js\";",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "import { LOCALE } from \"./config.js\";",
+    "line_start": 9,
+    "line_end": 9,
+    "metadata": {
+      "module": "config",
+      "names": "LOCALE",
+      "dots": "1",
+      "language_hint": ""
+    },
+    "id": "19dbd9da06350928"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "import",
+    "name": "import format from \"formatter\";",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "import format from \"formatter\";",
+    "line_start": 10,
+    "line_end": 10,
+    "metadata": {
+      "module": "formatter",
+      "names": "format",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "b9fdecded9f3899d"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "import",
+    "name": "import \"./styles.css\";",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "import \"./styles.css\";",
+    "line_start": 11,
+    "line_end": 11,
+    "metadata": {
+      "module": "styles",
+      "names": "",
+      "dots": "1",
+      "language_hint": ""
+    },
+    "id": "6f2daecdfe38687e"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "variable",
+    "name": "DEFAULT_GREETING",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "DEFAULT_GREETING = \"Hello\"",
+    "line_start": 13,
+    "line_end": 13,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "8486d13cd3f4dad0"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "variable",
+    "name": "locale",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "{ locale, fallback } = LOCALE",
+    "line_start": 14,
+    "line_end": 14,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "9aecdd2952d7477d"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "variable",
+    "name": "fallback",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "{ locale, fallback } = LOCALE",
+    "line_start": 14,
+    "line_end": 14,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "8133eb3ed093cf61"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "comment",
+    "name": "",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "/** Format a greeting for a name. */",
+    "line_start": 16,
+    "line_end": 16,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "f104f05d5b9c37cb"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "function",
+    "name": "formatGreeting",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "function formatGreeting(name) {\n  return `${DEFAULT_GREETING}, ${name} (${locale ?? fallback})`;\n}",
+    "line_start": 17,
+    "line_end": 19,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "de271dc4da07feb0"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "function",
+    "name": "cachedDefault",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "const cachedDefault = () => DEFAULT_GREETING;",
+    "line_start": 21,
+    "line_end": 21,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "6a672d8c65feb8f2"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "comment",
+    "name": "",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "/** Stateful greeter holding a prefix. */",
+    "line_start": 23,
+    "line_end": 23,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "afd95abe4fefa643"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "class",
+    "name": "Greeter",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "class Greeter {\n  constructor(prefix = DEFAULT_GREETING) {\n    this.prefix = prefix;\n  }\n\n  greet(name) {\n    return format(`${this.prefix}, ${name}`);\n  }\n}",
+    "line_start": 24,
+    "line_end": 32,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "5dc8fd6afa366584"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "method",
+    "name": "constructor",
+    "scope": "Greeter",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "constructor(prefix = DEFAULT_GREETING) {\n    this.prefix = prefix;\n  }",
+    "line_start": 25,
+    "line_end": 27,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "f8c2d42e485c98fe"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "method",
+    "name": "greet",
+    "scope": "Greeter",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "greet(name) {\n    return format(`${this.prefix}, ${name}`);\n  }",
+    "line_start": 29,
+    "line_end": 31,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "54cecd87796a0099"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "comment",
+    "name": "",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "/** Yield each recipient name in turn. */",
+    "line_start": 34,
+    "line_end": 34,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "bca733b8c9e46c9d"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "function",
+    "name": "recipients",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "function* recipients(names) {\n  for (const name of names) {\n    yield name;\n  }\n}",
+    "line_start": 35,
+    "line_end": 39,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "9fe04be46ef4242e"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "comment",
+    "name": "",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "/** Greeting helpers grouped as an object literal. */",
+    "line_start": 41,
+    "line_end": 41,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "3107f6745e3b67a3"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "variable",
+    "name": "helpers",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "helpers = {\n  shout(name) {\n    return `${name.toUpperCase()}!`;\n  },\n}",
+    "line_start": 42,
+    "line_end": 46,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "607e8ab8ea7c25fe"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "javascript.js",
+    "kind": "method",
+    "name": "shout",
+    "scope": "",
+    "language": "javascript",
+    "file_language": "javascript",
+    "content": "shout(name) {\n    return `${name.toUpperCase()}!`;\n  }",
+    "line_start": 43,
+    "line_end": 45,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "934493ecda7800b3"
+  },
+  {
+    "blob_sha": "sha1",
+    "file_path": "styles.css",
+    "kind": "class",
+    "name": ".greeter",
+    "scope": "",
+    "language": "css",
+    "file_language": "css",
+    "content": ".greeter {\n  color: #333;\n}",
+    "line_start": 1,
+    "line_end": 3,
+    "metadata": {
+      "module": "",
+      "names": "",
+      "dots": "",
+      "language_hint": ""
+    },
+    "id": "6f956de079e5ad55"
+  }
+]