oni-cli 1.0.0__py3-none-any.whl

oni/oni.py

@@ -0,0 +1,384 @@
+"""
+ONI (鬼)
+"""
+
+from __future__ import annotations
+
+import os
+import subprocess
+import sys
+from pathlib import Path
+from typing import Annotated, List, Optional, Set
+
+import pyfiglet
+import questionary
+import typer
+from questionary import Style as QStyle
+from rich import box
+from rich.align import Align
+from rich.console import Console, Group
+from rich.panel import Panel
+from rich.progress import (
+    BarColumn,
+    Progress,
+    SpinnerColumn,
+    TaskProgressColumn,
+    TextColumn,
+)
+from rich.syntax import Syntax
+from rich.table import Table
+from rich.text import Text
+
+# ---------------------------------------------------------
+# Palette
+# ---------------------------------------------------------
+COLOR_CRIMSON = "#e60033"
+COLOR_EMBER = "#ff6a00"
+COLOR_GOLD = "#e0a96d"
+COLOR_BONE = "#f4f1de"
+COLOR_IRON = "#3d3a45"
+COLOR_ASH = "#6c757d"
+
+console = Console()
+
+CLI_DESCRIPTION = "Oni is a fast CLI tool that packages your files into a single context window."
+
+app = typer.Typer(
+    name="oni",
+    help=f"[bold {COLOR_CRIMSON}]ONI (鬼)[/] :: {CLI_DESCRIPTION}",
+    rich_markup_mode="rich",
+    no_args_is_help=False,
+)
+
+# Standard ignore sets spared from the maw
+DEFAULT_IGNORES = {
+    ".git",
+    "node_modules",
+    "__pycache__",
+    ".venv",
+    "venv",
+    ".pytest_cache",
+    ".mypy_cache",
+    "dist",
+    "build",
+    ".DS_Store",
+    ".idea",
+    ".vscode",
+}
+
+# Extension-to-syntax mapping for code fences
+EXT_MAP = {
+    ".py": "python",
+    ".js": "javascript",
+    ".ts": "typescript",
+    ".tsx": "tsx",
+    ".jsx": "jsx",
+    ".json": "json",
+    ".yaml": "yaml",
+    ".yml": "yaml",
+    ".toml": "toml",
+    ".md": "markdown",
+    ".rs": "rust",
+    ".go": "go",
+    ".html": "html",
+    ".css": "css",
+    ".sh": "bash",
+    ".zsh": "bash",
+    ".sql": "sql",
+    ".c": "c",
+    ".cpp": "cpp",
+    ".h": "c",
+    ".hpp": "cpp",
+    ".swift": "swift",
+    ".kt": "kotlin",
+}
+
+# Questionary style tuned to iron and crimson
+ONI_PROMPT_STYLE = QStyle(
+    [
+        ("qmark", f"fg:{COLOR_CRIMSON} bold"),
+        ("question", f"fg:{COLOR_BONE} bold"),
+        ("answer", f"fg:{COLOR_EMBER} bold"),
+        ("pointer", f"fg:{COLOR_CRIMSON} bold"),
+        ("text", f"fg:{COLOR_BONE}"),
+    ]
+)
+
+
+def render_banner() -> None:
+    ascii_art = pyfiglet.figlet_format("ONI", font="alligator2")
+    header_text = Text(ascii_art, style=f"bold {COLOR_CRIMSON}")
+    sub_title = Text(
+        "INSTANT CONTEXT WINDOWS FROM YOUR FILES",
+        style=f"bold {COLOR_EMBER}",
+    )
+    console.print(
+        Panel(
+            Align.center(Group(header_text, sub_title)),
+            border_style=COLOR_CRIMSON,
+            box=box.DOUBLE,
+            padding=(0, 2),
+        )
+    )
+
+
+def is_binary(path: Path) -> bool:
+    """Detect binary files that would choke the prompt buffer."""
+    try:
+        with open(path, "rb") as f:
+            chunk = f.read(1024)
+            return b"\x00" in chunk
+    except Exception:
+        return True
+
+
+def copy_to_macos_clipboard(payload: str) -> None:
+    """Feeds the amalgamated payload directly to the macOS pbcopy buffer."""
+    process = subprocess.Popen(["pbcopy"], stdin=subprocess.PIPE, close_fds=True)
+    process.communicate(payload.encode("utf-8"))
+    if process.returncode != 0:
+        raise RuntimeError("pbcopy buffer rejected the payload.")
+
+
+# ---------------------------------------------------------
+# Core Execution
+# ---------------------------------------------------------
+@app.callback(invoke_without_command=True)
+def devour(
+    ctx: typer.Context,
+    patterns: Annotated[
+        Optional[List[str]],
+        typer.Argument(
+            help="File paths or glob patterns (e.g. 'src/**/*.py', 'main.py', '*.json'). Prompts if omitted."
+        ),
+    ] = None,
+    output_meta: Annotated[
+        bool,
+        typer.Option(
+            "--meta/--raw",
+            help="Include repository overview and file breakdown in LLM context header.",
+        ),
+    ] = True,
+    max_file_size_kb: Annotated[
+        int,
+        typer.Option(
+            "--max-kb", "-m", help="Skip files larger than this threshold in KB."
+        ),
+    ] = 512,
+    preview: Annotated[
+        bool,
+        typer.Option(
+            "--preview", "-p", help="Echo sample of amalgamated context to terminal."
+        ),
+    ] = False,
+):
+    """
+    Read codebase files and amalgamate them directly into your macOS clipboard.
+    """
+    render_banner()
+    root_dir = Path.cwd()
+
+    # Interactive prompt if no patterns provided
+    if not patterns:
+        console.print(
+            f"[bold {COLOR_EMBER}]Specify file patterns to search for...[/]\n"
+        )
+        user_input = questionary.text(
+            "Enter target paths or globs to fetch (space-separated):",
+            default="**/*",
+            style=ONI_PROMPT_STYLE,
+        ).ask()
+
+        if not user_input or not user_input.strip():
+            console.print(f"[{COLOR_ASH}]Aborted.[/]")
+            raise typer.Exit()
+        patterns = user_input.strip().split()
+
+    console.print(
+        f"[bold {COLOR_ASH}]Root Directory:[/] [bold {COLOR_BONE}]{root_dir}[/]"
+    )
+    console.print(
+        f"[bold {COLOR_ASH}]Glob Patterns:[/] [bold {COLOR_GOLD}]{', '.join(patterns)}[/]\n"
+    )
+
+    # Step 1: Scan & resolve files
+    matching_files: Set[Path] = set()
+
+    with console.status(
+        f"[bold {COLOR_CRIMSON}]Fetching matching files across directories...",
+        spinner="aesthetic",
+    ):
+        for pattern in patterns:
+            target = root_dir / pattern
+            if target.is_file():
+                matching_files.add(target.resolve())
+            elif target.is_dir():
+                for p in target.rglob("*"):
+                    if p.is_file():
+                        matching_files.add(p.resolve())
+            else:
+                for p in root_dir.glob(pattern):
+                    if p.is_file():
+                        matching_files.add(p.resolve())
+
+    # Filter out ignored directories
+    filtered_files: List[Path] = []
+    for f in matching_files:
+        rel = f.relative_to(root_dir)
+        if any(ignored in rel.parts for ignored in DEFAULT_IGNORES):
+            continue
+        filtered_files.append(f)
+
+    filtered_files.sort()
+
+    if not filtered_files:
+        console.print(
+            Panel(
+                f"[bold {COLOR_CRIMSON}]Nothing found.[/]\n"
+                f"[{COLOR_ASH}]No files matched the specified patterns or all were filtered by ignore rules (.git, node_modules, etc).[/]",
+                border_style=COLOR_CRIMSON,
+                box=box.ROUNDED,
+            )
+        )
+        raise typer.Exit()
+
+    # Step 2: Ingest, format, and display progress
+    payload_parts: List[str] = []
+    devour_table = Table(
+        box=box.HORIZONTALS,
+        border_style=COLOR_IRON,
+        title=f"[bold {COLOR_CRIMSON}]LOG[/]",
+        expand=True,
+    )
+    devour_table.add_column("#", style=f"bold {COLOR_ASH}", width=5)
+    devour_table.add_column("File", style=f"bold {COLOR_BONE}")
+    devour_table.add_column("Lines", style=COLOR_GOLD, justify="right")
+    devour_table.add_column("Weight", style=COLOR_EMBER, justify="right")
+    devour_table.add_column("Fate", style=COLOR_CRIMSON, justify="center")
+
+    if output_meta:
+        payload_parts.append(
+            f"# FILE CONTEXT\n"
+            f"- Project Root: `{root_dir.name}`\n"
+            f"- Total Files: {len(filtered_files)}\n\n"
+            f"The following files have been added into the context window "
+            f"for full comprehension and context.\n\n"
+            f"---\n\n"
+        )
+
+    total_bytes = 0
+    total_lines = 0
+
+    with Progress(
+        SpinnerColumn("bouncingBar", style=f"bold {COLOR_CRIMSON}"),
+        TextColumn(f"[bold {COLOR_BONE}]{{task.description}}"),
+        BarColumn(
+            bar_width=35,
+            style=COLOR_IRON,
+            complete_style=COLOR_CRIMSON,
+            finished_style=COLOR_EMBER,
+        ),
+        TaskProgressColumn(style=f"bold {COLOR_GOLD}"),
+        console=console,
+    ) as progress:
+        task = progress.add_task("Reading files...", total=len(filtered_files))
+
+        for idx, file_path in enumerate(filtered_files, start=1):
+            rel_path = file_path.relative_to(root_dir)
+            size_kb = file_path.stat().st_size / 1024
+
+            if is_binary(file_path):
+                devour_table.add_row(
+                    f"{idx:02d}",
+                    str(rel_path),
+                    "—",
+                    f"{size_kb:.1f} KB",
+                    f"[{COLOR_ASH}](BINARY)[/]",
+                )
+                progress.update(task, advance=1)
+                continue
+
+            if size_kb > max_file_size_kb:
+                devour_table.add_row(
+                    f"{idx:02d}",
+                    str(rel_path),
+                    "—",
+                    f"{size_kb:.1f} KB",
+                    f"[{COLOR_EMBER}]SPAT OUT (> {max_file_size_kb}KB)[/]",
+                )
+                progress.update(task, advance=1)
+                continue
+
+            try:
+                content = file_path.read_text(encoding="utf-8", errors="replace")
+                lines_count = len(content.splitlines())
+                total_lines += lines_count
+                total_bytes += file_path.stat().st_size
+                lang = EXT_MAP.get(file_path.suffix.lower(), "")
+
+                # LLM code fence formatting
+                shard_block = f"## File: `{rel_path}`\n```{lang}\n{content}\n```\n\n"
+                payload_parts.append(shard_block)
+
+                devour_table.add_row(
+                    f"{idx:02d}",
+                    str(rel_path),
+                    f"{lines_count:,}",
+                    f"{size_kb:.1f} KB",
+                    f"[bold {COLOR_CRIMSON}]DEVOUR[/]",
+                )
+            except Exception as e:
+                devour_table.add_row(
+                    f"{idx:02d}", str(rel_path), "ERR", "0 KB", f"[red]{str(e)[:15]}[/]"
+                )
+
+            progress.update(task, advance=1)
+
+    console.print()
+    console.print(devour_table)
+
+    final_payload = "".join(payload_parts)
+
+    if not final_payload.strip():
+        console.print(f"[{COLOR_CRIMSON}]No source content was found.[/]")
+        raise typer.Exit()
+
+    # Step 3: Direct Ingestion into macOS Clipboard
+    with console.status(
+        f"[bold {COLOR_CRIMSON}]Binding context into macOS clipboard...", spinner="line"
+    ):
+        try:
+            copy_to_macos_clipboard(final_payload)
+        except Exception as err:
+            console.print(f"[bold red]Failed to deliver payload to clipboard:[/] {err}")
+            raise typer.Exit(code=1)
+
+    # Culmination / Victory Panel
+    stats_msg = (
+        f"[bold {COLOR_CRIMSON}]COMPLETE // PROMPT CREATED[/]\n\n"
+        f"• [bold {COLOR_BONE}]Clipboard State:[/] [bold {COLOR_EMBER}]Copied[/]\n"
+        f"• [bold {COLOR_BONE}]Files Combined:[/]   [bold {COLOR_GOLD}]{len(filtered_files)}[/]\n"
+        f"• [bold {COLOR_BONE}]Total Lines:[/]      [bold {COLOR_GOLD}]{total_lines:,}[/]\n"
+        f"• [bold {COLOR_BONE}]Context Size:[/]   [bold {COLOR_GOLD}]{total_bytes / 1024:.2f} KB[/]\n\n"
+        f"[{COLOR_ASH}]Press [bold {COLOR_BONE}]⌘ + V[/] to paste the content.[/]"
+    )
+    console.print()
+    console.print(Panel(stats_msg, border_style=COLOR_CRIMSON, box=box.HEAVY))
+
+    # Step 4: Optional terminal preview
+    if preview:
+        console.print(f"\n[bold {COLOR_EMBER}]// PROMPT PREVIEW //[/]\n")
+        syntax = Syntax(
+            final_payload[:2500], "markdown", theme="ansi_dark", line_numbers=True
+        )
+        console.print(
+            Panel(
+                syntax,
+                title=f"[{COLOR_ASH}]Excerpt (First 2500 chars)[/]",
+                border_style=COLOR_IRON,
+            )
+        )
+
+
+if __name__ == "__main__":
+    app()

oni_cli-1.0.0.dist-info/METADATA

@@ -0,0 +1,126 @@
+Metadata-Version: 2.4
+Name: oni-cli
+Version: 1.0.0
+Summary: Single-command context aggregation for LLMs, straight to your macOS clipboard.
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: pyfiglet>=1.0.4
+Requires-Dist: questionary>=2.1.1
+Requires-Dist: rich>=15.0.0
+Requires-Dist: typer>=0.27.2
+
+
+<div align="center">
+
+  <img src="https://rowlinsonmike.com/oni.png" alt="ONI Logo" width="180" />
+
+  # ONI
+
+  **Single-command context aggregation for LLMs, straight to your macOS clipboard.**
+
+  <br />
+
+  <img src="https://rowlinsonmike.com/oni-demo.gif" alt="ONI Demo" width="750" />
+
+</div>
+
+---
+
+CLI that makes a single context widow from your files and sends it directly to your macOS clipboard.
+
+---
+
+## Key Features
+
+- **Direct to Clipboard**: Uses macOS `pbcopy` under the hood to send your aggregated code context straight to your clipboard.
+- **Smart Filtering**: Ignores binary files and junk directories (`.git`, `node_modules`, `.venv`, build caches) automatically.
+- **Syntax Awareness**: Maps file extensions to Markdown code fences (`py`, `ts`, `rs`, `go`, `sql`, etc.) for LLM legibility.
+- **Interactive Prompting**: If run with no arguments, Oni prompts for target glob patterns.
+- **Size Safeguards**: Skips oversized source files with customizable thresholds.
+
+---
+
+## Installation
+
+Ensure you have Python 3.9+ installed:
+
+
+---
+
+## Quick Usage
+
+### Parse everything in the current directory
+
+```bash
+oni
+
+```
+
+*If called without patterns, Oni triggers an interactive prompt defaulting to `**/*`.*
+
+### Parse specific directories or file globs
+
+```bash
+# Ingest entire src directory
+oni src/**/*
+
+# Combine specific file types
+oni "src/**/*.py" "config/*.json"
+
+# Ingest explicit targets
+oni main.py pyproject.toml docs/README.md
+
+```
+
+---
+
+## CLI Options Reference
+
+```text
+Usage: oni [OPTIONS] [PATTERNS]...
+
+```
+
+| Flag / Option | Short | Default | Description |
+| --- | --- | --- | --- |
+| `[PATTERNS]...` | — | `None` (Prompts) | File paths or glob patterns (e.g. `src/**/*.py`, `main.py`). |
+| `--meta / --raw` | — | `--meta` | Includes repository file context header and aggregate statistics. |
+| `--max-kb` | `-m` | `512` | Upper size limit in KB per file. Larger files are spat out. |
+| `--preview` | `-p` | `False` | Prints the first 2,500 characters of the prompt to terminal. |
+| `--help` | — | — | Displays the command-line usage and exits. |
+
+### Examples with Flags
+
+#### Omit Metadata Header
+
+Generate pure code fence blocks with no introductory repository summaries:
+
+```bash
+oni "src/**/*.rs" --raw
+
+```
+
+#### Adjust Max File Size Threshold
+
+Lower file size limit to 200 KB and inspect the payload before pasting:
+
+```bash
+oni "**/*.ts" --max-kb 200 --preview
+
+```
+
+---
+
+## Automatic Exclusions
+
+Oni ignores the following paths during scanning:
+
+```text
+.git          node_modules     __pycache__
+.venv         venv             .pytest_cache
+.mypy_cache   dist             build
+.DS_Store     .idea            .vscode
+
+```
+
+Binary files are detected via null-byte inspection and excluded from the output buffer.

oni_cli-1.0.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+oni/oni.py,sha256=SsGf_olqYCMWrillRqZlXCXN5Okfo0Wy7gNUo8QABbs,11854
+oni_cli-1.0.0.dist-info/METADATA,sha256=T2JQVcVbi4IaOwbyFiAxaaI6LXWCWtj7yiESw_UbcLI,3085
+oni_cli-1.0.0.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+oni_cli-1.0.0.dist-info/entry_points.txt,sha256=yE9SVtyFFlSYMijdNBm5DUtD_vBYg3QWuOgTzQ6mtZo,36
+oni_cli-1.0.0.dist-info/top_level.txt,sha256=y46kSeGeBZ2tP2IRm8OwnYdiwUJbsD4EcvLLbt7D5JU,4
+oni_cli-1.0.0.dist-info/RECORD,,

oni_cli-1.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (84.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

oni_cli-1.0.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+oni = oni.oni:app

oni_cli-1.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+oni