formatting.py

"""Output formatting — Rich Markdown rendering and JSON display."""

from __future__ import annotations

import json
import sys

from rich.console import Console
from rich.markdown import Markdown
from rich.panel import Panel
from rich.text import Text
from rich.theme import Theme

# ── Custom theme for Talk Python branding ────────────────────────────────────
_TALK_PYTHON_THEME = Theme(
    {
        'tp.title': 'bold cyan',
        'tp.heading': 'bold magenta',
        'tp.id': 'bold yellow',
        'tp.url': 'blue underline',
        'tp.date': 'green',
        'tp.dim': 'dim',
        'tp.error': 'bold red',
        'tp.success': 'bold green',
        'tp.label': 'bold white',
    }
)

console = Console(theme=_TALK_PYTHON_THEME, highlight=False)
error_console = Console(theme=_TALK_PYTHON_THEME, stderr=True, highlight=False)


def is_tty() -> bool:
    """Return True if stdout is connected to a terminal."""
    return sys.stdout.isatty()


def display(content: str, output_format: str) -> None:
    """Route content to the appropriate renderer."""
    if output_format == 'json':
        display_json(content)
    elif output_format == 'markdown':
        display_markdown_raw(content)
    else:
        display_markdown(content)


def display_markdown_raw(content: str) -> None:
    """Print raw Markdown content to stdout without any Rich formatting."""
    print(content)


def display_markdown(content: str) -> None:
    """Render Markdown content with Rich, wrapped in a styled panel."""
    md = Markdown(content, code_theme='monokai')

    panel = Panel(
        md,
        border_style='cyan',
        title='[bold cyan]Talk Python[/bold cyan]',
        title_align='left',
        padding=(1, 2),
    )
    console.print(panel)


def display_json(content: str) -> None:
    """Output JSON content — pretty-printed if on a TTY, raw otherwise."""
    try:
        data = json.loads(content)
    except (json.JSONDecodeError, TypeError):
        # Server may have returned Markdown even though JSON was requested;
        # fall back to printing the raw text.
        console.print(content)
        return

    if is_tty():
        from rich.syntax import Syntax

        formatted = json.dumps(data, indent=2, ensure_ascii=False)
        syntax = Syntax(formatted, 'json', theme='monokai', word_wrap=True)
        console.print(syntax)
    else:
        # Raw compact JSON for piping
        print(json.dumps(data, ensure_ascii=False))


def print_error(message: str) -> None:
    """Print a styled error message to stderr."""
    error_console.print(
        Text.assemble(
            ('ERROR', 'bold red'),
            (' ', ''),
            (message, 'red'),
        )
    )


def print_info(message: str) -> None:
    """Print a styled informational message."""
    console.print(f'[tp.dim]{message}[/tp.dim]')