rivetsql 0.1.0__tar.gz

rivetsql-0.1.0/.gitignore

@@ -0,0 +1,25 @@
+.DS_Store
+.env
+.env.*
+*.pem
+*.key
+secrets/
+target/
+build/
+dist/
+out/
+.next/
+node_modules/
+__pycache__/
+*.pyc
+.pytest_cache/
+.hypothesis/
+.cargo/
+*.class
+bin/
+obj/
+.ralph-logs/
+mcp.json
+docs/reference/
+docs/_build/
+*.zip

rivetsql-0.1.0/PKG-INFO

@@ -0,0 +1,161 @@
+Metadata-Version: 2.4
+Name: rivetsql
+Version: 0.1.0
+Summary: SQL pipeline framework — declare what, not how
+Project-URL: Homepage, https://github.com/rivetsql/rivet
+Project-URL: Documentation, https://rivetsql.github.io/rivet
+Project-URL: Repository, https://github.com/rivetsql/rivet
+Project-URL: Changelog, https://github.com/rivetsql/rivet/blob/main/CHANGELOG.md
+Author: Rivet Contributors
+License-Expression: MIT
+Keywords: cli,data-pipeline,dbt-alternative,elt,etl,sql
+Classifier: Development Status :: 4 - Beta
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Database
+Requires-Python: >=3.11
+Requires-Dist: rivetsql-core>=0.1.0
+Requires-Dist: textual>=1.0
+Provides-Extra: all
+Requires-Dist: rivetsql-aws>=0.1.0; extra == 'all'
+Requires-Dist: rivetsql-databricks>=0.1.0; extra == 'all'
+Requires-Dist: rivetsql-duckdb>=0.1.0; extra == 'all'
+Requires-Dist: rivetsql-polars>=0.1.0; extra == 'all'
+Requires-Dist: rivetsql-postgres>=0.1.0; extra == 'all'
+Requires-Dist: rivetsql-pyspark>=0.1.0; extra == 'all'
+Provides-Extra: aws
+Requires-Dist: rivetsql-aws>=0.1.0; extra == 'aws'
+Provides-Extra: databricks
+Requires-Dist: rivetsql-databricks>=0.1.0; extra == 'databricks'
+Provides-Extra: duckdb
+Requires-Dist: rivetsql-duckdb>=0.1.0; extra == 'duckdb'
+Provides-Extra: polars
+Requires-Dist: rivetsql-polars>=0.1.0; extra == 'polars'
+Provides-Extra: postgres
+Requires-Dist: rivetsql-postgres>=0.1.0; extra == 'postgres'
+Provides-Extra: pyspark
+Requires-Dist: rivetsql-pyspark>=0.1.0; extra == 'pyspark'
+Description-Content-Type: text/markdown
+
+<!-- Logo / Banner -->
+<p align="center">
+    <img src="logo.png" alt="Rivet SQL" width="400">
+</p>
+
+<!-- Badges -->
+<p align="center">
+  <a href="https://pypi.org/project/rivetsql/"><img src="https://img.shields.io/pypi/v/rivetsql" alt="PyPI version"></a>
+  <a href="https://pypi.org/project/rivetsql/"><img src="https://img.shields.io/pypi/pyversions/rivetsql" alt="Python versions"></a>
+  <a href="https://github.com/rivetsql/rivet/blob/main/LICENSE"><img src="https://img.shields.io/github/license/rivetsql/rivet" alt="License"></a>
+  <a href="https://github.com/rivetsql/rivet/actions/workflows/ci.yml"><img src="https://img.shields.io/github/actions/workflow/status/rivetsql/rivet/ci.yml?branch=main&label=CI" alt="CI status"></a>
+  <a href="https://codecov.io/gh/rivetsql/rivet"><img src="https://codecov.io/gh/rivetsql/rivet/branch/main/graph/badge.svg" alt="Coverage"></a>
+</p>
+
+<p align="center"><strong>Declarative SQL pipelines — define what, not how.</strong></p>
+
+---
+
+Rivet separates *what* to compute (joints) from *how* to compute (engines) and *where* data lives (catalogs). Define your pipeline once — run it on DuckDB, Polars, PySpark, or Postgres without changing your logic.
+
+## Features
+
+- **Multi-engine execution** — swap between DuckDB, Polars, PySpark, and Postgres without rewriting pipelines
+- **Declarative pipelines** — define joints in SQL, YAML, or Python
+- **Quality checks** — assertions before write, audits after write
+- **Built-in testing** — offline fixture-based tests with `rivet test`
+- **Plugin architecture** — install only the engines you need
+- **Interactive REPL** — explore and debug pipelines interactively
+
+## Quick Start
+
+Install Rivet and the DuckDB plugin:
+
+```bash
+pip install rivetsql[duckdb]
+```
+
+Create and run a pipeline:
+
+```bash
+mkdir my_pipeline && cd my_pipeline
+rivet init
+rivet run
+```
+
+A minimal SQL joint (`joints/transform_orders.sql`):
+
+```sql
+SELECT
+    order_id,
+    customer_name,
+    amount * 1.1 AS amount_with_tax
+FROM {{ ref('raw_orders') }}
+```
+
+## Installation
+
+```bash
+# Full CLI + core
+pip install rivetsql
+
+# Core library only (no CLI)
+pip install rivetsql-core
+
+# With a specific plugin
+pip install rivetsql[duckdb]
+
+# All plugins
+pip install rivetsql[all]
+```
+
+## Plugins
+
+| Plugin | Install | Engine |
+|--------|---------|--------|
+| DuckDB | `pip install rivetsql[duckdb]` | In-process analytical SQL |
+| Postgres | `pip install rivetsql[postgres]` | PostgreSQL databases |
+| Polars | `pip install rivetsql[polars]` | DataFrame-based compute |
+| PySpark | `pip install rivetsql[pyspark]` | Distributed Spark execution |
+| Databricks | `pip install rivetsql[databricks]` | Databricks SQL warehouses |
+| AWS | `pip install rivetsql[aws]` | S3 + Glue catalog integration |
+
+## Pipeline Visualization
+
+```mermaid
+graph LR
+    subgraph Joints["Joints (What)"]
+        S[Source] --> T[Transform]
+        T --> K[Sink]
+    end
+
+    subgraph Engines["Engines (How)"]
+        DK[DuckDB]
+        PL[Polars]
+        PS[PySpark]
+        PG[Postgres]
+    end
+
+    subgraph Catalogs["Catalogs (Where)"]
+        FS[Filesystem]
+        DB[Database]
+        S3[Object Store]
+    end
+
+    T --> DK
+    T --> PL
+    T --> PS
+    T --> PG
+
+    K --> FS
+    K --> DB
+    K --> S3
+```
+
+## Links
+
+- [Documentation](https://rivetsql.github.io/rivet)
+- [Contributing Guide](CONTRIBUTING.md)
+- [License (MIT)](LICENSE)
+- [Changelog](CHANGELOG.md)

rivetsql-0.1.0/README.md

@@ -0,0 +1,120 @@
+<!-- Logo / Banner -->
+<p align="center">
+    <img src="logo.png" alt="Rivet SQL" width="400">
+</p>
+
+<!-- Badges -->
+<p align="center">
+  <a href="https://pypi.org/project/rivetsql/"><img src="https://img.shields.io/pypi/v/rivetsql" alt="PyPI version"></a>
+  <a href="https://pypi.org/project/rivetsql/"><img src="https://img.shields.io/pypi/pyversions/rivetsql" alt="Python versions"></a>
+  <a href="https://github.com/rivetsql/rivet/blob/main/LICENSE"><img src="https://img.shields.io/github/license/rivetsql/rivet" alt="License"></a>
+  <a href="https://github.com/rivetsql/rivet/actions/workflows/ci.yml"><img src="https://img.shields.io/github/actions/workflow/status/rivetsql/rivet/ci.yml?branch=main&label=CI" alt="CI status"></a>
+  <a href="https://codecov.io/gh/rivetsql/rivet"><img src="https://codecov.io/gh/rivetsql/rivet/branch/main/graph/badge.svg" alt="Coverage"></a>
+</p>
+
+<p align="center"><strong>Declarative SQL pipelines — define what, not how.</strong></p>
+
+---
+
+Rivet separates *what* to compute (joints) from *how* to compute (engines) and *where* data lives (catalogs). Define your pipeline once — run it on DuckDB, Polars, PySpark, or Postgres without changing your logic.
+
+## Features
+
+- **Multi-engine execution** — swap between DuckDB, Polars, PySpark, and Postgres without rewriting pipelines
+- **Declarative pipelines** — define joints in SQL, YAML, or Python
+- **Quality checks** — assertions before write, audits after write
+- **Built-in testing** — offline fixture-based tests with `rivet test`
+- **Plugin architecture** — install only the engines you need
+- **Interactive REPL** — explore and debug pipelines interactively
+
+## Quick Start
+
+Install Rivet and the DuckDB plugin:
+
+```bash
+pip install rivetsql[duckdb]
+```
+
+Create and run a pipeline:
+
+```bash
+mkdir my_pipeline && cd my_pipeline
+rivet init
+rivet run
+```
+
+A minimal SQL joint (`joints/transform_orders.sql`):
+
+```sql
+SELECT
+    order_id,
+    customer_name,
+    amount * 1.1 AS amount_with_tax
+FROM {{ ref('raw_orders') }}
+```
+
+## Installation
+
+```bash
+# Full CLI + core
+pip install rivetsql
+
+# Core library only (no CLI)
+pip install rivetsql-core
+
+# With a specific plugin
+pip install rivetsql[duckdb]
+
+# All plugins
+pip install rivetsql[all]
+```
+
+## Plugins
+
+| Plugin | Install | Engine |
+|--------|---------|--------|
+| DuckDB | `pip install rivetsql[duckdb]` | In-process analytical SQL |
+| Postgres | `pip install rivetsql[postgres]` | PostgreSQL databases |
+| Polars | `pip install rivetsql[polars]` | DataFrame-based compute |
+| PySpark | `pip install rivetsql[pyspark]` | Distributed Spark execution |
+| Databricks | `pip install rivetsql[databricks]` | Databricks SQL warehouses |
+| AWS | `pip install rivetsql[aws]` | S3 + Glue catalog integration |
+
+## Pipeline Visualization
+
+```mermaid
+graph LR
+    subgraph Joints["Joints (What)"]
+        S[Source] --> T[Transform]
+        T --> K[Sink]
+    end
+
+    subgraph Engines["Engines (How)"]
+        DK[DuckDB]
+        PL[Polars]
+        PS[PySpark]
+        PG[Postgres]
+    end
+
+    subgraph Catalogs["Catalogs (Where)"]
+        FS[Filesystem]
+        DB[Database]
+        S3[Object Store]
+    end
+
+    T --> DK
+    T --> PL
+    T --> PS
+    T --> PG
+
+    K --> FS
+    K --> DB
+    K --> S3
+```
+
+## Links
+
+- [Documentation](https://rivetsql.github.io/rivet)
+- [Contributing Guide](CONTRIBUTING.md)
+- [License (MIT)](LICENSE)
+- [Changelog](CHANGELOG.md)

rivetsql-0.1.0/pyproject.rivetsql.toml

@@ -0,0 +1,53 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "rivetsql"
+version = "0.1.0"
+description = "SQL pipeline framework — declare what, not how"
+requires-python = ">=3.11"
+license = "MIT"
+readme = "README.md"
+authors = [{ name = "Rivet Contributors" }]
+keywords = ["sql", "data-pipeline", "etl", "elt", "dbt-alternative", "cli"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "License :: OSI Approved :: MIT License",
+    "Topic :: Database",
+]
+dependencies = [
+    "rivetsql-core>=0.1.0",
+    "textual>=1.0",
+]
+
+[project.urls]
+Homepage = "https://github.com/rivetsql/rivet"
+Documentation = "https://rivetsql.github.io/rivet"
+Repository = "https://github.com/rivetsql/rivet"
+Changelog = "https://github.com/rivetsql/rivet/blob/main/CHANGELOG.md"
+
+[project.scripts]
+rivet = "rivet_cli:main"
+
+[project.optional-dependencies]
+duckdb = ["rivetsql-duckdb>=0.1.0"]
+postgres = ["rivetsql-postgres>=0.1.0"]
+polars = ["rivetsql-polars>=0.1.0"]
+pyspark = ["rivetsql-pyspark>=0.1.0"]
+databricks = ["rivetsql-databricks>=0.1.0"]
+aws = ["rivetsql-aws>=0.1.0"]
+all = [
+    "rivetsql-duckdb>=0.1.0",
+    "rivetsql-postgres>=0.1.0",
+    "rivetsql-polars>=0.1.0",
+    "rivetsql-pyspark>=0.1.0",
+    "rivetsql-databricks>=0.1.0",
+    "rivetsql-aws>=0.1.0",
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/rivet_cli"]

rivetsql-0.1.0/pyproject.toml

@@ -0,0 +1,53 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "rivetsql"
+version = "0.1.0"
+description = "SQL pipeline framework — declare what, not how"
+requires-python = ">=3.11"
+license = "MIT"
+readme = "README.md"
+authors = [{ name = "Rivet Contributors" }]
+keywords = ["sql", "data-pipeline", "etl", "elt", "dbt-alternative", "cli"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "License :: OSI Approved :: MIT License",
+    "Topic :: Database",
+]
+dependencies = [
+    "rivetsql-core>=0.1.0",
+    "textual>=1.0",
+]
+
+[project.urls]
+Homepage = "https://github.com/rivetsql/rivet"
+Documentation = "https://rivetsql.github.io/rivet"
+Repository = "https://github.com/rivetsql/rivet"
+Changelog = "https://github.com/rivetsql/rivet/blob/main/CHANGELOG.md"
+
+[project.scripts]
+rivet = "rivet_cli:main"
+
+[project.optional-dependencies]
+duckdb = ["rivetsql-duckdb>=0.1.0"]
+postgres = ["rivetsql-postgres>=0.1.0"]
+polars = ["rivetsql-polars>=0.1.0"]
+pyspark = ["rivetsql-pyspark>=0.1.0"]
+databricks = ["rivetsql-databricks>=0.1.0"]
+aws = ["rivetsql-aws>=0.1.0"]
+all = [
+    "rivetsql-duckdb>=0.1.0",
+    "rivetsql-postgres>=0.1.0",
+    "rivetsql-polars>=0.1.0",
+    "rivetsql-pyspark>=0.1.0",
+    "rivetsql-databricks>=0.1.0",
+    "rivetsql-aws>=0.1.0",
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/rivet_cli"]

rivetsql-0.1.0/scripts/audit_dependencies.py

@@ -0,0 +1,176 @@
+"""Cross-reference declared dependencies against actual imports.
+
+Scans all pyproject.toml files and verifies each declared dependency
+has at least one corresponding import in the relevant source scope.
+"""
+
+import re
+import sys
+from pathlib import Path
+
+# Map from PyPI package name to Python import name(s)
+PACKAGE_TO_IMPORT = {
+    "pyarrow": ["pyarrow"],
+    "sqlglot": ["sqlglot"],
+    "pyyaml": ["yaml"],
+    "textual": ["textual"],
+    "textual-textarea": ["textual_textarea"],
+    "textual-fastdatatable": ["textual_fastdatatable"],
+    "pytest": ["pytest"],
+    "ruff": ["ruff"],
+    "mypy": ["mypy"],
+    "hypothesis": ["hypothesis"],
+    "rivet-core": ["rivet_core"],
+    "duckdb": ["duckdb"],
+    "psycopg": ["psycopg"],
+    "polars": ["polars"],
+    "deltalake": ["deltalake"],
+    "pyspark": ["pyspark"],
+    "boto3": ["boto3", "botocore"],
+    "requests": ["requests"],
+    "hatchling": [],  # build-system only
+}
+
+IMPORT_RE = re.compile(r"^\s*(?:import|from)\s+([\w.]+)", re.MULTILINE)
+# Match quoted dependency strings like "pkg>=1.0" or "pkg[extra]>=1.0"
+DEP_STR_RE = re.compile(r'"([^"]+)"')
+PKG_NAME_RE = re.compile(r'^([a-zA-Z0-9_-]+)')
+
+
+def extract_imports(src_dir: Path) -> set[str]:
+    """Extract all top-level import names from .py files under src_dir."""
+    imports = set()
+    for py_file in src_dir.rglob("*.py"):
+        text = py_file.read_text(errors="ignore")
+        for m in IMPORT_RE.finditer(text):
+            imports.add(m.group(1).split(".")[0])
+    return imports
+
+
+def parse_dep_strings(text: str) -> list[str]:
+    """Extract package names from a TOML array of dependency strings."""
+    names = []
+    for dep_str in DEP_STR_RE.findall(text):
+        m = PKG_NAME_RE.match(dep_str)
+        if m:
+            names.append(m.group(1))
+    return names
+
+
+def find_toml_array(text: str, start: int) -> str:
+    """Extract content of a TOML array starting at '[', handling nested brackets."""
+    depth = 0
+    i = start
+    while i < len(text):
+        if text[i] == "[":
+            depth += 1
+        elif text[i] == "]":
+            depth -= 1
+            if depth == 0:
+                return text[start + 1 : i]
+        elif text[i] == '"':
+            i += 1
+            while i < len(text) and text[i] != '"':
+                i += 1
+        i += 1
+    return text[start + 1 :]
+
+
+def parse_deps_from_toml(text: str) -> dict[str, list[str]]:
+    """Parse dependencies from pyproject.toml text."""
+    result: dict[str, list[str]] = {}
+
+    # Core dependencies
+    dep_match = re.search(r'\[project\].*?(?:^|\n)dependencies\s*=\s*(\[)', text, re.DOTALL)
+    if dep_match:
+        array_content = find_toml_array(text, dep_match.start(1))
+        result["dependencies"] = parse_dep_strings(array_content)
+
+    # Optional dependencies sections
+    opt_section = re.search(
+        r'\[project\.optional-dependencies\]\s*\n(.*?)(?=\n\[(?!^\s)|\Z)', text, re.DOTALL
+    )
+    if opt_section:
+        section_text = opt_section.group(1)
+        for m in re.finditer(r'(\w+)\s*=\s*(\[)', section_text):
+            section_name = m.group(1)
+            abs_start = opt_section.start(1) + m.start(2)
+            array_content = find_toml_array(text, abs_start)
+            result[f"optional[{section_name}]"] = parse_dep_strings(array_content)
+
+    return result
+
+
+def main() -> int:
+    rivet_root = Path(__file__).resolve().parent.parent
+    src_root = rivet_root / "src"
+    all_findings: list[str] = []
+
+    # 1. Root pyproject.toml
+    root_pyproject = rivet_root / "pyproject.toml"
+    print(f"=== Auditing {root_pyproject.relative_to(rivet_root)} against src/ ===")
+    deps = parse_deps_from_toml(root_pyproject.read_text())
+    all_imports = extract_imports(src_root)
+
+    for section, pkg_names in deps.items():
+        for pkg in pkg_names:
+            pkg_lower = pkg.lower()
+            import_names = PACKAGE_TO_IMPORT.get(pkg_lower, [pkg_lower.replace("-", "_")])
+            if not import_names:
+                continue
+            if section == "optional[cli]":
+                scope_imports = extract_imports(src_root / "rivet_cli")
+                scope = "rivet_cli"
+            elif section == "optional[dev]":
+                continue  # dev deps used in tests, not src
+            else:
+                scope_imports = all_imports
+                scope = "all src"
+            found = any(name in scope_imports for name in import_names)
+            status = "OK" if found else "UNUSED"
+            print(f"  [{status}] {section} :: {pkg} (imports: {import_names}, scope: {scope})")
+            if status == "UNUSED":
+                all_findings.append(
+                    f"root :: {section} :: {pkg} (imports: {import_names}, scope: {scope})"
+                )
+
+    # 2. Per-plugin pyproject.toml files
+    for plugin_dir in sorted(src_root.iterdir()):
+        plugin_pyproject = plugin_dir / "pyproject.toml"
+        if not plugin_pyproject.exists():
+            continue
+        label = plugin_dir.name
+        print(f"\n=== Auditing {label}/pyproject.toml against {label}/ ===")
+        deps = parse_deps_from_toml(plugin_pyproject.read_text())
+        plugin_imports = extract_imports(plugin_dir)
+
+        for section, pkg_names in deps.items():
+            for pkg in pkg_names:
+                pkg_lower = pkg.lower()
+                import_names = PACKAGE_TO_IMPORT.get(
+                    pkg_lower, [pkg_lower.replace("-", "_")]
+                )
+                if not import_names:
+                    continue
+                if pkg_lower == "rivet-core":
+                    continue  # self-reference
+                found = any(name in plugin_imports for name in import_names)
+                status = "OK" if found else "UNUSED"
+                print(f"  [{status}] {section} :: {pkg} (imports: {import_names})")
+                if status == "UNUSED":
+                    all_findings.append(f"{label} :: {section} :: {pkg}")
+
+    # Summary
+    print("\n" + "=" * 60)
+    if all_findings:
+        print(f"FINDINGS: {len(all_findings)} dependency(ies) with zero import references:")
+        for f in all_findings:
+            print(f"  - {f}")
+        return 1
+    else:
+        print("All declared dependencies have matching imports. No issues found.")
+        return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())

rivetsql-0.1.0/scripts/check_module_boundaries.py

@@ -0,0 +1,96 @@
+#!/usr/bin/env python3
+"""Scan all `from rivet_*` / `import rivet_*` imports under rivet/src/ and verify
+boundary compliance.
+
+Exit code 0 means no violations; exit code 1 means violations found.
+
+Boundary rules (from rivet steering doc):
+  - rivet_core: only stdlib + pyarrow + sqlglot (no rivet_* imports)
+  - rivet_config: only rivet_core
+  - rivet_bridge: only rivet_core + rivet_config
+  - rivet_cli: only rivet_core + rivet_config + rivet_bridge
+  - Plugins (rivet_duckdb, rivet_postgres, rivet_polars, rivet_pyspark,
+    rivet_aws, rivet_databricks): only rivet_core
+"""
+
+from __future__ import annotations
+
+import re
+import sys
+from dataclasses import dataclass
+from pathlib import Path
+
+ALLOWED_RIVET_IMPORTS: dict[str, set[str]] = {
+    "rivet_core": set(),
+    "rivet_config": {"rivet_core"},
+    "rivet_bridge": {"rivet_core", "rivet_config"},
+    "rivet_cli": {"rivet_core", "rivet_config", "rivet_bridge"},
+    # Plugins — only rivet_core
+    "rivet_duckdb": {"rivet_core"},
+    "rivet_postgres": {"rivet_core"},
+    "rivet_polars": {"rivet_core"},
+    "rivet_pyspark": {"rivet_core"},
+    "rivet_aws": {"rivet_core"},
+    "rivet_databricks": {"rivet_core"},
+}
+
+_IMPORT_RE = re.compile(r"(?:from|import)\s+(rivet_[a-z_]+)")
+
+
+@dataclass(frozen=True)
+class Violation:
+    file: str
+    line_no: int
+    source_module: str
+    imported_module: str
+    line_text: str
+
+
+def scan_boundary_violations(src_dir: str | Path) -> list[Violation]:
+    """Return all boundary violations found under *src_dir*."""
+    src_dir = Path(src_dir)
+    violations: list[Violation] = []
+
+    for module_dir in sorted(src_dir.iterdir()):
+        if not module_dir.is_dir() or not module_dir.name.startswith("rivet_"):
+            continue
+        module_name = module_dir.name
+        allowed = ALLOWED_RIVET_IMPORTS.get(module_name, {"rivet_core"})
+
+        for py_file in sorted(module_dir.rglob("*.py")):
+            rel = str(py_file.relative_to(src_dir))
+            for line_no, raw_line in enumerate(py_file.read_text().splitlines(), 1):
+                stripped = raw_line.strip()
+                if stripped.startswith("#"):
+                    continue
+                m = _IMPORT_RE.search(stripped)
+                if not m:
+                    continue
+                imported = m.group(1)
+                if imported == module_name:
+                    continue
+                if imported.startswith("rivet_") and imported not in allowed:
+                    violations.append(
+                        Violation(rel, line_no, module_name, imported, stripped)
+                    )
+
+    return violations
+
+
+def main() -> int:
+    src_dir = Path(__file__).resolve().parent.parent / "src"
+    violations = scan_boundary_violations(src_dir)
+
+    if not violations:
+        print("✅ No module boundary violations found.")
+        return 0
+
+    print(f"❌ Found {len(violations)} module boundary violation(s):\n")
+    for v in violations:
+        print(f"  {v.file}:{v.line_no}  [{v.source_module} → {v.imported_module}]")
+        print(f"    {v.line_text}\n")
+    return 1
+
+
+if __name__ == "__main__":
+    sys.exit(main())