utils.py

from __future__ import annotations

from datetime import datetime
from typing import Iterable, List

from rich.console import Console
from rich.panel import Panel
from rich.table import Table

console = Console()


def format_date(value: str | None) -> str:
    if not value:
        return "—"
    try:
        return datetime.strptime(value, "%Y-%m-%d").strftime("%b %d, %Y")
    except ValueError:
        return value


def validate_date(value: str | None) -> bool:
    if not value:
        return True
    try:
        datetime.strptime(value, "%Y-%m-%d")
        return True
    except ValueError:
        return False


def priority_color(priority: str) -> str:
    return {
        "Low": "green",
        "Medium": "cyan",
        "High": "yellow",
    }.get(priority.title(), "white")


def render_table(tasks: Iterable) -> None:
    table = Table(title="Tasks", header_style="bold magenta")
    table.add_column("ID", style="cyan", width=6)
    table.add_column("Title", style="white")
    table.add_column("Priority", style="magenta")
    table.add_column("Category", style="blue")
    table.add_column("Due", style="yellow")
    table.add_column("Status", style="green")

    for task in tasks:
        status = "✓ completed" if task.completed else "✗ pending"
        table.add_row(
            str(task.id),
            task.title,
            f"[{priority_color(task.priority)}]{task.priority}[/{priority_color(task.priority)}]",
            task.category,
            format_date(task.due_date),
            status,
        )

    console.print(table)


def show_error(message: str) -> None:
    console.print(f"[bold red]Error:[/bold red] {message}")


def show_success(message: str) -> None:
    console.print(f"[bold green]Success:[/bold green] {message}")


def show_info(message: str) -> None:
    console.print(Panel.fit(message, border_style="blue"))