treemapper 1.6.1__tar.gz → 2.2.0__tar.gz

treemapper-2.2.0/.gitignore

@@ -0,0 +1,53 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# Default output files
+tree.yaml
+tree.json
+tree.md
+tree.txt
+
+# Virtual environments
+.venv/
+venv/
+ENV/
+
+# Tooling caches
+.pytest_cache/
+.mypy_cache/
+.ruff_cache/
+.hypothesis/
+
+# IDE / OS
+.idea/
+.vscode/
+.DS_Store
+
+# Local-only planning
+TODO.md
+test-repos

treemapper-2.2.0/CHANGELOG.md

@@ -0,0 +1,64 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog 1.1.0](https://keepachangelog.com/en/1.1.0/),
+and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+
+## [Unreleased]
+
+## [2.2.0] - 2026-06-20
+
+### Changed
+
+- Bumped the dependency floor to `diffctx>=1.10.0,<2.0`. This pulls in the
+  diffctx 1.10 engine: the calibrated default `--tau` of 0.12 (tighter
+  diff-context selection by default), the 256 KB MCP file cap, and the
+  document/import edge correctness fixes. No treemapper API changes.
+
+## [2.1.0] - 2026-06-14
+
+### Changed
+
+- Bumped the dependency floor to `diffctx>=1.9.1,<2.0`. This pulls in the diffctx
+  1.9 diff-context improvements for `treemapper . --diff`: an orientation header
+  (`commit_message` + `changed_files`) at the top of every format, a `role:
+  changed` marker on fragments overlapping the diff with changed code emitted
+  first, context ordered by descending per-file relevance, and merging of
+  line-contiguous same-role fragments. The `>=1.9.1` floor also remains a hard
+  requirement for the branded `run(prog=…, version=…)` entry — the pre-1.8
+  fallback path was removed, so there is no unbranded code path.
+- `treemapper` drives the engine through the public
+  `diffctx.run(prog="treemapper", version=…)` entry, so `--help`, `--version`,
+  and error prefixes are branded as `treemapper`. The `treemapper-mcp` install
+  hint is likewise branded via the engine's `prog`/`extra` parameters.
+
+### Added
+
+- GitHub Actions release workflow (`cd.yml`) that builds the pure wheel + sdist
+  and publishes to PyPI via OIDC trusted publishing.
+
+## [2.0.0]
+
+### Changed
+
+- **TreeMapper is now a thin wrapper over the [`diffctx`](https://pypi.org/project/diffctx/)
+  engine.** All directory traversal, serialization, and git-diff context
+  selection are delegated to `diffctx` (pinned `>=1.7,<2.0`). TreeMapper itself
+  ships only the `treemapper` command, the `treemapper-mcp` server entry point,
+  and a Python API that re-exports `map_directory`, `build_diff_context`, and
+  the `to_yaml`/`to_json`/`to_text`/`to_markdown` serializers. No engine logic
+  is duplicated.
+- Pure-Python wheel (built with hatchling). The native engine arrives through
+  the `diffctx` dependency.
+
+### Note
+
+Releases `1.0.0` through `1.6.1` shipped TreeMapper as a self-contained package;
+that lineage was renamed to `diffctx` at `diffctx` `1.7.0`. TreeMapper `2.0.0`
+re-establishes the `treemapper` name as the product layer on top of that engine.
+
+[Unreleased]: https://github.com/nikolay-e/treemapper/compare/v2.2.0...HEAD
+[2.2.0]: https://github.com/nikolay-e/treemapper/compare/v2.1.0...v2.2.0
+[2.1.0]: https://github.com/nikolay-e/treemapper/compare/v2.0.0...v2.1.0
+[2.0.0]: https://github.com/nikolay-e/treemapper/releases/tag/v2.0.0

treemapper-2.2.0/PKG-INFO

@@ -0,0 +1,120 @@
+Metadata-Version: 2.4
+Name: treemapper
+Version: 2.2.0
+Summary: Map a codebase to YAML/JSON/Markdown/text and select smart git-diff context for LLMs
+Project-URL: Changelog, https://github.com/nikolay-e/treemapper/releases
+Project-URL: Homepage, https://github.com/nikolay-e/treemapper
+Project-URL: Issues, https://github.com/nikolay-e/treemapper/issues
+Project-URL: Repository, https://github.com/nikolay-e/treemapper
+Author-email: Nikolay Eremeev <nikolay.eremeev@outlook.com>
+License-Expression: Apache-2.0
+License-File: LICENSE
+Keywords: ai,claude,code-context,codebase,context-selection,diff-context,directory-tree,git-diff,json,llm,mcp,tree,yaml
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Topic :: Software Development
+Classifier: Topic :: Software Development :: Version Control :: Git
+Classifier: Topic :: Utilities
+Classifier: Typing :: Typed
+Requires-Python: >=3.10
+Requires-Dist: diffctx<2.0,>=1.10.0
+Provides-Extra: dev
+Requires-Dist: mypy<3.0,>=2.1.0; extra == 'dev'
+Requires-Dist: pre-commit<5.0,>=3.0; extra == 'dev'
+Requires-Dist: pytest-timeout<3.0,>=2.4.0; extra == 'dev'
+Requires-Dist: pytest<10.0,>=7.0; extra == 'dev'
+Requires-Dist: pyyaml<8.0,>=6.0.2; extra == 'dev'
+Requires-Dist: ruff<1.0,>=0.4; extra == 'dev'
+Requires-Dist: types-pyyaml<7.0,>=6.0.12.20260518; extra == 'dev'
+Provides-Extra: full
+Requires-Dist: diffctx[full]<2.0,>=1.10.0; extra == 'full'
+Provides-Extra: mcp
+Requires-Dist: diffctx[mcp]<2.0,>=1.10.0; extra == 'mcp'
+Provides-Extra: tree-sitter
+Requires-Dist: diffctx[tree-sitter]<2.0,>=1.10.0; extra == 'tree-sitter'
+Description-Content-Type: text/markdown
+
+# TreeMapper
+
+[![PyPI](https://img.shields.io/pypi/v/treemapper)](https://pypi.org/project/treemapper/)
+[![License](https://img.shields.io/pypi/l/treemapper)](https://github.com/nikolay-e/treemapper/blob/main/LICENSE)
+
+**Map a codebase into LLM-ready context.** TreeMapper serializes an entire
+directory tree — structure plus file contents — to YAML, JSON, Markdown, or
+text, and selects the minimal smart git-diff context needed to understand a
+change. Paste the output into Claude, ChatGPT, or any LLM.
+
+TreeMapper is a thin command-line product built on the
+[`diffctx`](https://pypi.org/project/diffctx/) engine. All traversal,
+serialization, and diff-context selection live in `diffctx`; TreeMapper
+re-exposes them under the `treemapper` name so there is a single source of
+truth and no duplicated logic.
+
+## Installation
+
+```bash
+pipx install treemapper                  # recommended: isolated, no venv needed
+pip install treemapper                   # or with pip
+pip install 'treemapper[tree-sitter]'    # + AST parsing for smarter diff context
+pip install 'treemapper[mcp]'            # + MCP server for AI assistants
+```
+
+## Usage
+
+```bash
+treemapper .                     # map current directory to YAML (stdout)
+treemapper /path/to/project      # map a specific directory
+treemapper . -f json             # output as JSON
+treemapper . -f md --save        # save as tree.md
+treemapper . --no-content        # structure only, no file contents
+treemapper . -c                  # copy output to clipboard
+treemapper . --diff HEAD~1       # smart context for the last commit
+treemapper . --diff main..HEAD   # smart context for a branch range
+treemapper graph .               # project dependency graph (mermaid)
+```
+
+### Two modes
+
+- **Tree mapping** (`treemapper .`) — walks the directory tree, respects
+  hierarchical ignore patterns (`.gitignore`, `.diffctx/ignore`), reads file
+  contents with binary/encoding detection, and serializes the result.
+- **Diff context** (`treemapper . --diff`) — analyzes a git diff and selects
+  the minimal set of code fragments needed to understand the change, instead
+  of dumping whole files.
+
+Run `treemapper --help` for the full flag reference.
+
+## Python API
+
+```python
+import treemapper
+
+tree = treemapper.map_directory(".", no_content=False)
+print(treemapper.to_yaml(tree))
+
+context = treemapper.build_diff_context(root_dir=".", diff_range="HEAD~1")
+
+# treemapper.run drives the full branded CLI from Python (same surface as the command)
+treemapper.run(["graph", "."], prog="treemapper", version=treemapper.__version__)
+```
+
+Every run reports an **exact `tiktoken` token count** (not a `chars / 4`
+estimate), so you know the real context cost before you paste.
+
+## Relationship to diffctx
+
+TreeMapper is the user-facing distribution; `diffctx` is the reusable engine.
+Pin compatibility is `diffctx>=1.10.0,<2.0`. If you are embedding the engine in
+your own tool, depend on `diffctx` directly.
+
+## License
+
+Apache 2.0

treemapper-2.2.0/README.md

@@ -0,0 +1,76 @@
+# TreeMapper
+
+[![PyPI](https://img.shields.io/pypi/v/treemapper)](https://pypi.org/project/treemapper/)
+[![License](https://img.shields.io/pypi/l/treemapper)](https://github.com/nikolay-e/treemapper/blob/main/LICENSE)
+
+**Map a codebase into LLM-ready context.** TreeMapper serializes an entire
+directory tree — structure plus file contents — to YAML, JSON, Markdown, or
+text, and selects the minimal smart git-diff context needed to understand a
+change. Paste the output into Claude, ChatGPT, or any LLM.
+
+TreeMapper is a thin command-line product built on the
+[`diffctx`](https://pypi.org/project/diffctx/) engine. All traversal,
+serialization, and diff-context selection live in `diffctx`; TreeMapper
+re-exposes them under the `treemapper` name so there is a single source of
+truth and no duplicated logic.
+
+## Installation
+
+```bash
+pipx install treemapper                  # recommended: isolated, no venv needed
+pip install treemapper                   # or with pip
+pip install 'treemapper[tree-sitter]'    # + AST parsing for smarter diff context
+pip install 'treemapper[mcp]'            # + MCP server for AI assistants
+```
+
+## Usage
+
+```bash
+treemapper .                     # map current directory to YAML (stdout)
+treemapper /path/to/project      # map a specific directory
+treemapper . -f json             # output as JSON
+treemapper . -f md --save        # save as tree.md
+treemapper . --no-content        # structure only, no file contents
+treemapper . -c                  # copy output to clipboard
+treemapper . --diff HEAD~1       # smart context for the last commit
+treemapper . --diff main..HEAD   # smart context for a branch range
+treemapper graph .               # project dependency graph (mermaid)
+```
+
+### Two modes
+
+- **Tree mapping** (`treemapper .`) — walks the directory tree, respects
+  hierarchical ignore patterns (`.gitignore`, `.diffctx/ignore`), reads file
+  contents with binary/encoding detection, and serializes the result.
+- **Diff context** (`treemapper . --diff`) — analyzes a git diff and selects
+  the minimal set of code fragments needed to understand the change, instead
+  of dumping whole files.
+
+Run `treemapper --help` for the full flag reference.
+
+## Python API
+
+```python
+import treemapper
+
+tree = treemapper.map_directory(".", no_content=False)
+print(treemapper.to_yaml(tree))
+
+context = treemapper.build_diff_context(root_dir=".", diff_range="HEAD~1")
+
+# treemapper.run drives the full branded CLI from Python (same surface as the command)
+treemapper.run(["graph", "."], prog="treemapper", version=treemapper.__version__)
+```
+
+Every run reports an **exact `tiktoken` token count** (not a `chars / 4`
+estimate), so you know the real context cost before you paste.
+
+## Relationship to diffctx
+
+TreeMapper is the user-facing distribution; `diffctx` is the reusable engine.
+Pin compatibility is `diffctx>=1.10.0,<2.0`. If you are embedding the engine in
+your own tool, depend on `diffctx` directly.
+
+## License
+
+Apache 2.0

treemapper-2.2.0/pyproject.toml

@@ -0,0 +1,107 @@
+[build-system]
+build-backend = "hatchling.build"
+requires = [ "hatchling>=1.25,<2.0" ]
+
+[project]
+name = "treemapper"
+dynamic = [ "version" ]
+description = "Map a codebase to YAML/JSON/Markdown/text and select smart git-diff context for LLMs"
+readme = "README.md"
+keywords = [
+  "ai",
+  "claude",
+  "code-context",
+  "codebase",
+  "context-selection",
+  "diff-context",
+  "directory-tree",
+  "git-diff",
+  "json",
+  "llm",
+  "mcp",
+  "tree",
+  "yaml",
+]
+license = "Apache-2.0"
+authors = [
+  { name = "Nikolay Eremeev", email = "nikolay.eremeev@outlook.com" },
+]
+requires-python = ">=3.10"
+classifiers = [
+  "Development Status :: 5 - Production/Stable",
+  "Environment :: Console",
+  "Intended Audience :: Developers",
+  "Operating System :: OS Independent",
+  "Programming Language :: Python :: 3 :: Only",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
+  "Programming Language :: Python :: 3.14",
+  "Topic :: Software Development",
+  "Topic :: Software Development :: Version Control :: Git",
+  "Topic :: Utilities",
+  "Typing :: Typed",
+]
+dependencies = [
+  "diffctx>=1.10.0,<2.0",
+]
+optional-dependencies.dev = [
+  "mypy>=2.1.0,<3.0",
+  "pre-commit>=3.0,<5.0",
+  "pytest>=7.0,<10.0",
+  "pytest-timeout>=2.4.0,<3.0",
+  "pyyaml>=6.0.2,<8.0",
+  "ruff>=0.4,<1.0",
+  "types-pyyaml>=6.0.12.20260518,<7.0",
+]
+optional-dependencies.full = [
+  "diffctx[full]>=1.10.0,<2.0",
+]
+optional-dependencies.mcp = [
+  "diffctx[mcp]>=1.10.0,<2.0",
+]
+optional-dependencies.tree-sitter = [
+  "diffctx[tree-sitter]>=1.10.0,<2.0",
+]
+urls.Changelog = "https://github.com/nikolay-e/treemapper/releases"
+urls.Homepage = "https://github.com/nikolay-e/treemapper"
+urls.Issues = "https://github.com/nikolay-e/treemapper/issues"
+urls.Repository = "https://github.com/nikolay-e/treemapper"
+scripts.treemapper = "treemapper.cli:main"
+scripts.treemapper-mcp = "treemapper.mcp_main:main"
+
+[tool.hatch.version]
+path = "src/treemapper/version.py"
+
+[tool.hatch.build.targets.wheel]
+packages = [ "src/treemapper" ]
+
+[tool.hatch.build.targets.sdist]
+include = [
+  "src/treemapper",
+  "tests",
+  "README.md",
+  "CHANGELOG.md",
+  "LICENSE",
+]
+
+[tool.ruff]
+target-version = "py310"
+line-length = 130
+lint.select = [ "C901", "E", "F", "I", "N", "RUF", "UP", "W" ]
+lint.ignore = [ "E501" ]
+
+[tool.pytest.ini_options]
+testpaths = [ "tests" ]
+pythonpath = [ "src" ]
+timeout = 30
+
+[tool.mypy]
+python_version = "3.10"
+strict = true
+files = [ "src" ]
+
+[[tool.mypy.overrides]]
+module = [ "diffctx.*" ]
+ignore_missing_imports = true

treemapper-2.2.0/src/treemapper/__init__.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+from diffctx import (
+    build_diff_context,
+    map_directory,
+    run,
+    to_json,
+    to_markdown,
+    to_text,
+    to_yaml,
+)
+
+from .version import __version__
+
+__all__ = [
+    "__version__",
+    "build_diff_context",
+    "map_directory",
+    "run",
+    "to_json",
+    "to_markdown",
+    "to_text",
+    "to_yaml",
+]

treemapper-2.2.0/src/treemapper/__main__.py

@@ -0,0 +1,6 @@
+from __future__ import annotations
+
+from .cli import main
+
+if __name__ == "__main__":
+    main()

treemapper-2.2.0/src/treemapper/cli.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+import diffctx
+
+from .version import __version__
+
+
+def main(argv: list[str] | None = None) -> None:
+    diffctx.run(argv, prog="treemapper", version=__version__)
+
+
+if __name__ == "__main__":
+    main()

treemapper-2.2.0/src/treemapper/mcp_main.py

@@ -0,0 +1,11 @@
+from __future__ import annotations
+
+from diffctx.mcp.__main__ import main as _engine_mcp_main
+
+
+def main() -> None:
+    _engine_mcp_main(prog="treemapper-mcp", extra="treemapper[mcp]")
+
+
+if __name__ == "__main__":
+    main()

treemapper-2.2.0/src/treemapper/version.py

@@ -0,0 +1 @@
+__version__ = "2.2.0"

treemapper-2.2.0/tests/conftest.py

@@ -0,0 +1,59 @@
+from __future__ import annotations
+
+import contextlib
+import io
+import subprocess
+from dataclasses import dataclass
+from pathlib import Path
+
+import pytest
+
+from treemapper.cli import main
+
+
+@dataclass
+class CliResult:
+    stdout: str
+    stderr: str
+    exit_code: int
+
+
+def run_cli(argv: list[str]) -> CliResult:
+    out, err = io.StringIO(), io.StringIO()
+    exit_code = 0
+    with contextlib.redirect_stdout(out), contextlib.redirect_stderr(err):
+        try:
+            main(argv)
+        except SystemExit as exc:
+            code = exc.code
+            exit_code = code if isinstance(code, int) else (0 if code is None else 1)
+    return CliResult(stdout=out.getvalue(), stderr=err.getvalue(), exit_code=exit_code)
+
+
+@pytest.fixture
+def sample_project(tmp_path: Path) -> Path:
+    (tmp_path / "alpha.py").write_text("def alpha():\n    return 1\n", encoding="utf-8")
+    (tmp_path / "readme.txt").write_text("hello tree\n", encoding="utf-8")
+    nested = tmp_path / "pkg"
+    nested.mkdir()
+    (nested / "beta.py").write_text("def beta():\n    return 2\n", encoding="utf-8")
+    return tmp_path
+
+
+def _git(repo: Path, *args: str) -> None:
+    subprocess.run(["git", *args], cwd=repo, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+
+
+@pytest.fixture
+def git_repo(tmp_path: Path) -> Path:
+    _git(tmp_path, "init")
+    _git(tmp_path, "config", "user.email", "test@example.com")
+    _git(tmp_path, "config", "user.name", "Test")
+    target = tmp_path / "module.py"
+    target.write_text("def compute(x):\n    return x + 1\n", encoding="utf-8")
+    _git(tmp_path, "add", "-A")
+    _git(tmp_path, "commit", "-m", "initial")
+    target.write_text("def compute(x):\n    return x + 2\n\n\ndef extra(y):\n    return y * 2\n", encoding="utf-8")
+    _git(tmp_path, "add", "-A")
+    _git(tmp_path, "commit", "-m", "change compute and add extra")
+    return tmp_path

treemapper-2.2.0/tests/test_api.py

@@ -0,0 +1,30 @@
+from __future__ import annotations
+
+from pathlib import Path
+
+import diffctx
+
+import treemapper
+
+
+def test_public_api_is_exported() -> None:
+    for name in ("map_directory", "build_diff_context", "run", "to_yaml", "to_json", "to_text", "to_markdown"):
+        assert hasattr(treemapper, name), name
+
+
+def test_run_is_the_engine_entry() -> None:
+    assert treemapper.run is diffctx.run
+
+
+def test_map_directory_round_trips_to_yaml(sample_project: Path) -> None:
+    tree = treemapper.map_directory(str(sample_project))
+    assert tree["type"] == "directory"
+    rendered = treemapper.to_yaml(tree)
+    assert "alpha.py" in rendered
+
+
+def test_build_diff_context_on_real_repo(git_repo: Path) -> None:
+    context = treemapper.build_diff_context(root_dir=git_repo, diff_range="HEAD~1")
+    assert isinstance(context, dict)
+    rendered = treemapper.to_yaml(context)
+    assert "module.py" in rendered

treemapper-2.2.0/tests/test_cli.py

@@ -0,0 +1,82 @@
+from __future__ import annotations
+
+import json
+import subprocess
+import sys
+from pathlib import Path
+
+import yaml
+
+from treemapper.version import __version__
+
+from .conftest import CliResult, run_cli
+
+
+def test_version_is_treemapper_branded() -> None:
+    result = run_cli(["--version"])
+    assert result.exit_code == 0
+    assert result.stdout.strip() == f"treemapper {__version__}"
+
+
+def test_help_renders_and_is_treemapper_branded() -> None:
+    result = run_cli(["--help"])
+    assert result.exit_code == 0
+    assert "--diff" in result.stdout
+    assert "usage: treemapper" in result.stdout.lower()
+    assert "usage: diffctx" not in result.stdout.lower()
+
+
+def test_yaml_tree_includes_files_and_content(sample_project: Path) -> None:
+    result = run_cli([str(sample_project)])
+    assert result.exit_code == 0
+    tree = yaml.safe_load(result.stdout)
+    assert tree["type"] == "directory"
+    names = {child["name"] for child in tree["children"]}
+    assert "alpha.py" in names
+    assert "pkg" in names
+    assert "def alpha()" in result.stdout
+
+
+def test_json_format(sample_project: Path) -> None:
+    result = run_cli([str(sample_project), "-f", "json"])
+    assert result.exit_code == 0
+    tree = json.loads(result.stdout)
+    assert tree["type"] == "directory"
+
+
+def test_no_content_omits_file_bodies(sample_project: Path) -> None:
+    result = run_cli([str(sample_project), "--no-content"])
+    assert result.exit_code == 0
+    assert "def alpha()" not in result.stdout
+    assert "alpha.py" in result.stdout
+
+
+def test_save_writes_file(sample_project: Path, monkeypatch) -> None:
+    monkeypatch.chdir(sample_project)
+    result = run_cli([str(sample_project), "-f", "md", "--save"])
+    assert result.exit_code == 0
+    assert (sample_project / "tree.md").is_file()
+
+
+def test_diff_mode_selects_changed_context(git_repo: Path, monkeypatch) -> None:
+    monkeypatch.chdir(git_repo)
+    result = run_cli([str(git_repo), "--diff", "HEAD~1"])
+    assert result.exit_code == 0
+    assert "module.py" in result.stdout
+    assert "extra" in result.stdout
+
+
+def test_console_script_entry_point(sample_project: Path) -> None:
+    proc = subprocess.run(
+        [sys.executable, "-m", "treemapper", str(sample_project), "--no-content"],
+        capture_output=True,
+        text=True,
+        check=False,
+    )
+    assert proc.returncode == 0
+    assert "alpha.py" in proc.stdout
+
+
+def test_result_dataclass_shape() -> None:
+    result = run_cli(["--version"])
+    assert isinstance(result, CliResult)