feat: add clipboard copy support with -c/--copy and --copy-only flags

Author: nikolay-e

docs/Clipboard Copy.md

@@ -18,6 +18,8 @@ class ParsedArgs:
     max_depth: Optional[int]
     no_content: bool
     max_file_bytes: Optional[int]
+    copy: bool
+    copy_only: bool
 
 
 DEFAULT_IGNORES_HELP = """
@@ -56,6 +58,8 @@ def parse_args() -> ParsedArgs:
     parser.add_argument("--max-depth", type=int, default=None, metavar="N", help="Maximum traversal depth")
     parser.add_argument("--no-content", action="store_true", help="Skip file contents (structure only)")
     parser.add_argument("--max-file-bytes", type=int, default=None, metavar="N", help="Skip files larger than N bytes")
+    parser.add_argument("-c", "--copy", action="store_true", help="Copy output to clipboard")
+    parser.add_argument("--copy-only", action="store_true", help="Copy to clipboard only (no stdout/file output)")
     parser.add_argument(
         "-v",
         "--verbosity",
@@ -109,4 +113,6 @@ def parse_args() -> ParsedArgs:
         max_depth=args.max_depth,
         no_content=args.no_content,
         max_file_bytes=args.max_file_bytes,
+        copy=args.copy or args.copy_only,
+        copy_only=args.copy_only,
     )

src/treemapper/cli.py

@@ -0,0 +1,57 @@
+from __future__ import annotations
+
+import os
+import platform
+import shutil
+import subprocess
+
+
+class ClipboardError(Exception):
+    pass
+
+
+def detect_clipboard_command() -> list[str] | None:
+    system = platform.system()
+    if system == "Darwin":
+        if shutil.which("pbcopy"):
+            return ["pbcopy"]
+        return None
+    if system == "Windows":
+        if shutil.which("clip"):
+            return ["clip"]
+        return None
+    if system in ("Linux", "FreeBSD"):
+        if os.environ.get("WAYLAND_DISPLAY") and shutil.which("wl-copy"):
+            return ["wl-copy"]
+        if os.environ.get("DISPLAY") and shutil.which("xclip"):
+            return ["xclip", "-selection", "clipboard"]
+        return None
+    return None
+
+
+def copy_to_clipboard(text: str) -> None:
+    cmd = detect_clipboard_command()
+    if cmd is None:
+        raise ClipboardError("No clipboard tool found")
+
+    system = platform.system()
+    if system == "Windows":
+        encoded = text.encode("utf-16le")
+    else:
+        encoded = text.encode("utf-8")
+
+    try:
+        proc = subprocess.Popen(cmd, stdin=subprocess.PIPE, stderr=subprocess.PIPE)
+        _, stderr = proc.communicate(input=encoded, timeout=5)
+        if proc.returncode != 0:
+            raise ClipboardError(stderr.decode(errors="replace").strip() or f"Command failed with code {proc.returncode}")
+    except subprocess.TimeoutExpired:
+        proc.kill()
+        proc.communicate()
+        raise ClipboardError("Clipboard operation timed out")
+    except OSError as e:
+        raise ClipboardError(f"Failed to execute clipboard command: {e}")
+
+
+def clipboard_available() -> bool:
+    return detect_clipboard_command() is not None

src/treemapper/clipboard.py

@@ -1,10 +1,17 @@
 import errno
+import logging
 import os
 import sys
 from contextlib import contextmanager
 from typing import Generator
 
 
+def _format_size(size_bytes: int) -> str:
+    if size_bytes < 1024:
+        return f"{size_bytes} B"
+    return f"{size_bytes / 1024:.1f} KB"
+
+
 @contextmanager
 def _handle_broken_pipe() -> Generator[None, None, None]:
     try:
@@ -34,10 +41,11 @@ def _suppress_broken_pipe() -> None:
 
 def main() -> None:
     from .cli import parse_args
+    from .clipboard import ClipboardError, copy_to_clipboard
     from .ignore import get_ignore_specs
     from .logger import setup_logging
     from .tree import TreeBuildContext, build_tree
-    from .writer import write_tree_to_file
+    from .writer import tree_to_string, write_tree_to_file
 
     args = parse_args()
     setup_logging(args.verbosity)
@@ -57,6 +65,17 @@ def main() -> None:
         "children": build_tree(args.root_dir, ctx),
     }
 
+    if args.copy:
+        output_content = tree_to_string(directory_tree, args.output_format)
+        try:
+            copy_to_clipboard(output_content)
+            print(f"Copied to clipboard ({_format_size(len(output_content))})", file=sys.stderr)
+        except ClipboardError as e:
+            logging.warning(f"Clipboard: {e}")
+
+    if args.copy_only and args.output_file is None:
+        return
+
     if args.output_file is None:
         with _handle_broken_pipe():
             write_tree_to_file(directory_tree, args.output_file, args.output_format)

src/treemapper/treemapper.py

@@ -99,6 +99,17 @@ def write_tree_text(file: TextIO, tree: Dict[str, Any]) -> None:
         _write_tree_text_node(file, child, "", i == len(children) - 1)
 
 
+def tree_to_string(tree: Dict[str, Any], output_format: str = "yaml") -> str:
+    buf = io.StringIO()
+    if output_format == "json":
+        write_tree_json(buf, tree)
+    elif output_format == "text":
+        write_tree_text(buf, tree)
+    else:
+        write_tree_yaml(buf, tree)
+    return buf.getvalue()
+
+
 def write_tree_to_file(tree: Dict[str, Any], output_file: Optional[Path], output_format: str = "yaml") -> None:
     def write_tree_content(f: TextIO) -> None:
         if output_format == "json":