import sys
from dataclasses import dataclass


@dataclass
class ProgressStep:
    label: str
    status: str = "pending"
    detail: str = ""


class ProgressDisplay:
    def __init__(self, stream=None):
        self.stream = stream or sys.stdout
        self.steps: list[ProgressStep] = []
        self._rendered_line_count = 0
        self._supports_rewrite = hasattr(self.stream, "isatty") and self.stream.isatty()

    def add_step(self, label: str, status: str = "pending", detail: str = "") -> int:
        self.steps.append(ProgressStep(label=label, status=status, detail=detail))
        self._render()
        return len(self.steps) - 1

    def update(self, step_index: int, status: str, detail: str = "") -> None:
        step = self.steps[step_index]
        step.status = status
        step.detail = detail
        self._render()

    def insert_step(self, index: int, label: str, status: str = "pending", detail: str = "") -> int:
        self.steps.insert(index, ProgressStep(label=label, status=status, detail=detail))
        self._render()
        return index

    def _render(self) -> None:
        lines = [format_step(step) for step in self.steps]
        output = "\n".join(lines)
        if output:
            output += "\n"

        if self._supports_rewrite and self._rendered_line_count:
            self.stream.write(f"\x1b[{self._rendered_line_count}A")
            for _ in range(self._rendered_line_count):
                self.stream.write("\x1b[2K\x1b[1B")
            self.stream.write(f"\x1b[{self._rendered_line_count}A")

        self.stream.write(output)
        self.stream.flush()
        self._rendered_line_count = len(lines)


class LogProgressDisplay:
    def __init__(self, on_change=None):
        self.steps: list[ProgressStep] = []
        self.on_change = on_change

    def add_step(self, label: str, status: str = "pending", detail: str = "") -> int:
        self.steps.append(ProgressStep(label=label, status=status, detail=detail))
        self._notify()
        return len(self.steps) - 1

    def update(self, step_index: int, status: str, detail: str = "") -> None:
        step = self.steps[step_index]
        step.status = status
        step.detail = detail
        self._notify()

    def get_output(self) -> str:
        return "\n".join(format_step(step) for step in self.steps)

    def _notify(self) -> None:
        if self.on_change is not None:
            self.on_change(self.get_output())


def format_step(step: ProgressStep) -> str:
    prefix_by_status = {
        "pending": "[ ]",
        "in_progress": "[>]",
        "done": "[x]",
        "failed": "[!]",
        "skipped": "[-]",
    }
    prefix = prefix_by_status.get(step.status, "[?]")
    if step.detail:
        return f"{prefix} {step.label} - {step.detail}"
    return f"{prefix} {step.label}"