"""Parse unified diff format into structured models."""

from __future__ import annotations

from collections.abc import Callable, Generator, Iterable, Iterator
from types import TracebackType

from . import binary_diff as _binary_diff
from . import diff_headers as _diff_headers
from . import empty_file_diff as _empty_file_diff
from . import file_metadata_diff as _file_metadata_diff
from . import gitlink_diff as _gitlink_diff
from . import hunk_headers as _hunk_headers
from . import line_change_body as _line_change_body
from . import patch_headers as _patch_headers
from .buffer import LineBuffer
from .models import (
    BinaryFileChange,
    FileModeChange,
    GitlinkChange,
    LineLevelChange,
    HunkHeader,
    RenameChange,
    SingleHunkPatch,
    TextFileDeletionChange,
)
from ..exceptions import CommandError
from ..i18n import _
from ..git_paths import encode_path, quote_path_token


# Type for annotator hooks that enrich LineLevelChange with additional metadata
LineLevelChangeAnnotator = Callable[[str, LineLevelChange], LineLevelChange]
UnifiedDiffItem = (
    SingleHunkPatch
    | BinaryFileChange
    | FileModeChange
    | GitlinkChange
    | RenameChange
    | TextFileDeletionChange
)


def patch_is_file_deletion(patch_lines: Iterable[bytes]) -> bool:
    """Return whether patch lines target a deleted file path."""
    return _patch_headers.patch_targets_file_deletion(patch_lines)


def patch_is_new_file(patch_lines: Iterable[bytes]) -> bool:
    """Return whether patch lines target a newly added file path."""
    return _patch_headers.patch_targets_new_file(patch_lines)


def patch_is_empty_file_change(patch_lines: Iterable[bytes]) -> bool:
    """Return whether patch lines describe a synthetic empty-file change."""
    return any(
        line.rstrip(b"\n") == _empty_file_diff.SYNTHETIC_EMPTY_HUNK_HEADER
        for line in patch_lines
    )


class _UnifiedDiffParserBuildContext:
    """Own parser-created hunk buffers for a scoped unified diff parse."""

    def __init__(self, lines: Iterable[bytes]) -> None:
        self._lines = lines
        self._buffers: list[LineBuffer] = []
        self._parser: Generator[UnifiedDiffItem, None, None] | None = None
        self._closed = False

    def close(self) -> None:
        if self._closed:
            return
        self._closed = True

        if self._parser is not None:
            self._parser.close()
            self._parser = None

        for buffer in self._buffers:
            buffer.close()
        self._buffers.clear()

    def __enter__(self) -> Iterator[UnifiedDiffItem]:
        if self._closed:
            raise ValueError("parser context is closed")
        if self._parser is not None:
            raise RuntimeError("parser context can only be entered once")
        self._parser = self._iter_owned()
        return self._parser

    def __exit__(
        self,
        exc_type: type[BaseException] | None,
        exc: BaseException | None,
        traceback: TracebackType | None,
    ) -> None:
        self.close()

    def _own_buffer(self, buffer: LineBuffer) -> LineBuffer:
        self._buffers.append(buffer)
        return buffer

    def _release_buffer(self, buffer: LineBuffer) -> None:
        try:
            self._buffers.remove(buffer)
        except ValueError:
            return
        buffer.close()

    def _release_item(self, item: UnifiedDiffItem | None) -> None:
        if isinstance(item, SingleHunkPatch) and isinstance(item.lines, LineBuffer):
            self._release_buffer(item.lines)

    def _iter_owned(self) -> Generator[UnifiedDiffItem, None, None]:
        current_item: UnifiedDiffItem | None = None
        parser = self._parse()

        try:
            while True:
                self._release_item(current_item)
                current_item = None
                current_item = next(parser)
                yield current_item
        except StopIteration:
            return
        finally:
            self._release_item(current_item)
            parser.close()

    def _build_single_hunk_patch(
        self,
        *,
        old_path: str,
        new_path: str,
        lines: Iterable[bytes],
    ) -> SingleHunkPatch:
        return SingleHunkPatch(
            old_path=old_path,
            new_path=new_path,
            lines=self._own_buffer(LineBuffer.from_chunks(lines)),
        )

    def _parse(self) -> Generator[UnifiedDiffItem, None, None]:
        line_iter = iter(self._lines)
        lookahead: bytes | None = None  # One-line lookahead buffer

        def next_line() -> bytes | None:
            """Get next line, using lookahead if available."""
            nonlocal lookahead
            if lookahead is not None:
                line = lookahead
                lookahead = None
                return line
            try:
                return next(line_iter)
            except StopIteration:
                return None

        def peek_line() -> bytes | None:
            """Peek at next line without consuming it."""
            nonlocal lookahead
            if lookahead is None:
                try:
                    lookahead = next(line_iter)
                except StopIteration:
                    lookahead = None
            return lookahead

        def hunk_line_chunks(
            old_file_line: bytes,
            new_file_line: bytes,
            hunk_header_line: bytes,
        ) -> Iterator[bytes]:
            yield old_file_line + b'\n'
            yield new_file_line + b'\n'
            yield hunk_header_line + b'\n'

            header = _hunk_headers.parse_hunk_header_line(hunk_header_line)
            old_consumed = 0
            new_consumed = 0

            while True:
                body_line = peek_line()
                old_remaining, new_remaining = header.remaining_body_counts(
                    old_consumed,
                    new_consumed,
                )

                if body_line is None:
                    if old_remaining or new_remaining:
                        raise CommandError(
                            _("Diff ended before the hunk body was complete")
                        )
                    return

                body_line_stripped = body_line.rstrip(b'\n')

                if body_line_stripped.startswith(b"\\"):
                    if body_line_stripped != b"\\ No newline at end of file":
                        raise CommandError(_("Invalid marker in diff hunk body"))
                    next_line()
                    yield body_line
                    continue

                if not old_remaining and not new_remaining:
                    if (
                        _diff_headers.line_is_diff_git_header(body_line_stripped)
                        or _hunk_headers.line_is_hunk_header(body_line_stripped)
                    ):
                        return
                    if body_line_stripped.startswith((b" ", b"+", b"-")):
                        raise CommandError(_("Diff hunk body exceeds declared counts"))
                    raise CommandError(_("Invalid line prefix after diff hunk body"))

                prefix = body_line_stripped[:1]
                if prefix == b" ":
                    if not old_remaining or not new_remaining:
                        raise CommandError(_("Diff hunk body exceeds declared counts"))
                    old_consumed += 1
                    new_consumed += 1
                elif prefix == b"-":
                    if not old_remaining:
                        raise CommandError(_("Diff hunk body exceeds declared old count"))
                    old_consumed += 1
                elif prefix == b"+":
                    if not new_remaining:
                        raise CommandError(_("Diff hunk body exceeds declared new count"))
                    new_consumed += 1
                else:
                    raise CommandError(_("Invalid line prefix in diff hunk body"))

                next_line()
                yield body_line

        try:
            while True:
                line = next_line()
                if line is None:
                    break

                # Strip only the diff format's \n terminator (preserve \r in content)
                line = line.rstrip(b'\n')

                # Look for start of a file diff
                if _diff_headers.line_is_diff_git_header(line):
                    file_paths = _diff_headers.diff_git_paths(line)
                    if file_paths is None:
                        raise CommandError(_("Malformed diff --git header"))
                    old_path, new_path = file_paths

                    # Collect metadata lines until we hit the --- line (start of unified diff)
                    # Files with no hunks (binary, mode-only, rename-only, empty) won't have --- line
                    metadata_lines: list[bytes] = []
                    old_file_line: bytes | None = None
                    while True:
                        next_l = next_line()
                        if next_l is None:
                            # End of input - check for empty file before returning
                            break
                        next_l = next_l.rstrip(b'\n')
                        if next_l.startswith(b"---"):
                            old_file_line = next_l
                            break
                        # Collect metadata lines
                        metadata_lines.append(next_l)
                        # If we hit another diff header, this file has no hunks - check if it's an empty new file
                        if _diff_headers.line_is_diff_git_header(next_l):
                            # Put the line back for next iteration (with \n restored for consistency)
                            lookahead = next_l + b'\n'
                            break

                    is_gitlink = _gitlink_diff.metadata_indicates_gitlink(
                        metadata_lines
                    )
                    is_rename = _file_metadata_diff.metadata_indicates_rename(
                        metadata_lines
                    )
                    if is_rename:
                        renamed_paths = _file_metadata_diff.rename_paths(
                            metadata_lines
                        )
                        if renamed_paths is not None:
                            old_path, new_path = renamed_paths
                    mode_transition = _file_metadata_diff.executable_mode_change(
                        metadata_lines
                    )
                    type_transition = _file_metadata_diff.file_type_change(
                        metadata_lines
                    )
                    if type_transition is not None and not is_gitlink:
                        raise CommandError(
                            _(
                                "File type changes are atomic and are not supported yet: "
                                "{file} ({old} -> {new})"
                            ).format(
                                file=old_path,
                                old=type_transition[0],
                                new=type_transition[1],
                            )
                        )
                    mode_change = (
                        FileModeChange(
                            new_path,
                            *mode_transition,
                            index_path=old_path if is_rename else None,
                        )
                        if mode_transition is not None
                        else None
                    )
                    is_deleted_file = (
                        _file_metadata_diff.metadata_indicates_deleted_file(
                            metadata_lines
                        )
                    )
                    index_old_oid, index_new_oid = (
                        _gitlink_diff.gitlink_oids_from_index(metadata_lines)
                    )

                    # Handle files without unified diff hunks
                    if old_file_line is None:
                        if is_rename:
                            yield RenameChange(old_path=old_path, new_path=new_path)

                        if is_gitlink:
                            yield GitlinkChange(
                                old_path=_gitlink_diff.gitlink_old_path(
                                    old_path,
                                    index_old_oid,
                                ),
                                new_path=_gitlink_diff.gitlink_new_path(
                                    new_path,
                                    index_new_oid,
                                ),
                                old_oid=_gitlink_diff.non_null_git_oid(index_old_oid),
                                new_oid=_gitlink_diff.non_null_git_oid(index_new_oid),
                                change_type=_gitlink_diff.gitlink_change_type(
                                    metadata_lines,
                                    index_old_oid,
                                    index_new_oid,
                                ),
                            )
                            continue

                        if _binary_diff.metadata_indicates_binary_file(metadata_lines):
                            yield BinaryFileChange(
                                old_path=old_path,
                                new_path=new_path,
                                change_type=_binary_diff.binary_change_type(
                                    metadata_lines
                                ),
                            )
                            if mode_change is not None:
                                yield mode_change
                            continue

                        if is_rename:
                            if mode_change is not None:
                                yield mode_change
                            continue

                        if is_deleted_file:
                            yield TextFileDeletionChange(old_path=old_path)
                            continue

                        if _empty_file_diff.metadata_indicates_new_empty_file(
                            metadata_lines
                        ):
                            yield self._build_single_hunk_patch(
                                old_path="/dev/null",
                                new_path=new_path,
                                lines=_empty_file_diff.synthetic_empty_file_patch_lines(
                                    b"--- /dev/null",
                                    b"+++ "
                                    + quote_path_token(b"b/" + encode_path(new_path)),
                                ),
                            )
                        if mode_change is not None:
                            yield mode_change
                        # Skip other files without hunks (mode-only, rename-only, etc.)
                        continue

                    # Get +++ line
                    plus_line = next_line()
                    if plus_line is None:
                        raise CommandError(
                            _("Malformed unified diff: missing +++ file header.")
                        )
                    plus_line_stripped = plus_line.rstrip(b'\n')
                    if not _patch_headers.line_is_new_file_header(plus_line_stripped):
                        raise CommandError(
                            _("Malformed unified diff: expected +++ file header.")
                        )
                    new_file_line = plus_line_stripped

                    patch_old_path = _patch_headers.old_file_path_from_header(
                        old_file_line
                    )
                    patch_new_path = _patch_headers.new_file_path_from_header(
                        new_file_line
                    )
                    if _patch_headers.path_names_repository_file(patch_old_path):
                        old_path = patch_old_path
                    if _patch_headers.path_names_repository_file(patch_new_path):
                        new_path = patch_new_path

                    if is_rename:
                        yield RenameChange(old_path=old_path, new_path=new_path)

                    if is_gitlink:
                        hunk_old_oid, hunk_new_oid = (
                            _gitlink_diff.consume_gitlink_hunks(
                                next_line,
                                peek_line,
                            )
                        )
                        old_oid = hunk_old_oid or _gitlink_diff.non_null_git_oid(
                            index_old_oid
                        )
                        new_oid = hunk_new_oid or _gitlink_diff.non_null_git_oid(
                            index_new_oid
                        )
                        if old_oid is not None and old_oid == new_oid:
                            continue
                        yield GitlinkChange(
                            old_path=_gitlink_diff.gitlink_old_path(
                                old_path,
                                old_oid or index_old_oid,
                            ),
                            new_path=_gitlink_diff.gitlink_new_path(
                                new_path,
                                new_oid or index_new_oid,
                            ),
                            old_oid=old_oid,
                            new_oid=new_oid,
                            change_type=_gitlink_diff.gitlink_change_type(
                                metadata_lines,
                                old_oid or index_old_oid,
                                new_oid or index_new_oid,
                            ),
                        )
                        continue

                    # Process all hunks for this file
                    has_hunks = False
                    while True:
                        # Check if next line is a hunk header
                        hunk_header_line = peek_line()
                        if hunk_header_line is None:
                            break
                        hunk_header_stripped = hunk_header_line.rstrip(b'\n')
                        if not _hunk_headers.line_is_hunk_header(
                            hunk_header_stripped
                        ):
                            # No more hunks for this file
                            break

                        has_hunks = True

                        # Consume the hunk header
                        next_line()

                        patch_lines: Iterable[bytes] = hunk_line_chunks(
                            old_file_line,
                            new_file_line,
                            hunk_header_stripped,
                        )
                        subproject_oids = None
                        if not is_gitlink and not metadata_lines:
                            patch_lines = list(patch_lines)
                            subproject_oids = (
                                _gitlink_diff.gitlink_oids_from_subproject_commit_patch(
                                    patch_lines
                                )
                            )
                        if subproject_oids is not None:
                            old_oid, new_oid = subproject_oids
                            if old_oid is not None and old_oid == new_oid:
                                continue
                            yield GitlinkChange(
                                old_path=_gitlink_diff.gitlink_old_path(
                                    old_path,
                                    old_oid,
                                ),
                                new_path=_gitlink_diff.gitlink_new_path(
                                    new_path,
                                    new_oid,
                                ),
                                old_oid=old_oid,
                                new_oid=new_oid,
                                change_type=_gitlink_diff.gitlink_change_type(
                                    metadata_lines,
                                    old_oid,
                                    new_oid,
                                ),
                            )
                            continue

                        # Yield this hunk immediately
                        yield self._build_single_hunk_patch(
                            old_path=old_path,
                            new_path=new_path,
                            lines=patch_lines,
                        )

                    if not has_hunks:
                        if _empty_file_diff.metadata_indicates_new_empty_file(
                            metadata_lines
                        ):
                            yield self._build_single_hunk_patch(
                                old_path=old_path,
                                new_path=new_path,
                                lines=_empty_file_diff.synthetic_empty_file_patch_lines(
                                    old_file_line,
                                    new_file_line,
                                ),
                            )
                        elif is_deleted_file:
                            yield TextFileDeletionChange(old_path=old_path)
                    if mode_change is not None:
                        yield mode_change
        finally:
            close = getattr(line_iter, "close", None)
            if close is not None:
                close()


def acquire_unified_diff(lines: Iterable[bytes]) -> _UnifiedDiffParserBuildContext:
    """Acquire a scoped unified diff parser with parser-owned hunk buffers."""
    return _UnifiedDiffParserBuildContext(lines)


def build_line_changes_from_patch_lines(
    patch_lines: Iterable[bytes],
    *,
    annotator: LineLevelChangeAnnotator | None = None,
) -> LineLevelChange:
    """Parse single-hunk patch lines into a LineLevelChange structure.

    Args:
        patch_lines: Unified diff patch lines for a single hunk
        annotator: Optional function to enrich LineLevelChange with provenance metadata
                   (e.g., batch source alignment, 3-way merge base). If None, source_line
                   fields remain None (no provenance).

    Returns:
        LineLevelChange object with parsed line entries and IDs
    """
    path_value = ""
    old_path_value = ""
    new_path_value = ""
    hunk_header: HunkHeader | None = None
    body_builder = _line_change_body.LineChangeBodyBuilder()

    # Preserve line endings so a parsed hunk can be emitted unchanged.
    for line_with_ending in patch_lines:
        # Strip only \n for comparison (preserve \r in content)
        line = line_with_ending.rstrip(b'\n')

        if hunk_header is not None:
            body_builder.append_patch_line(line)
        elif _patch_headers.line_is_old_file_header(line):
            old_path_value = _patch_headers.old_file_path_from_header(line)
        elif _patch_headers.line_is_new_file_header(line):
            new_path_value = _patch_headers.new_file_path_from_header(line)
        elif _hunk_headers.line_is_hunk_header(line):
            hunk_header = _hunk_headers.parse_hunk_header_line(line)
            body_builder.reset_for_hunk_header(hunk_header)

    path_value = _patch_headers.line_change_path(old_path_value, new_path_value)

    if hunk_header is None:
        raise CommandError(_("Failed to parse hunk header."))

    line_changes = LineLevelChange(
        path=path_value,
        header=hunk_header,
        lines=body_builder.line_entries,
    )

    # Apply annotator hook if provided
    if annotator is not None:
        line_changes = annotator(path_value, line_changes)

    return line_changes
