# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: OpenMDW-1.1

"""Render titles explicitly declared by native chart metadata as table captions."""
import re
from html import escape


def transform(text):
    changed = 0

    def chart(m):
        nonlocal changed
        body = m[2]
        if "<chart" in body:
            return m[0]
        title = re.search(r"(?m)^title:[ \t]*([^\n]*)$", body)
        if not title:
            return m[0]
        title_text = title[1].strip()
        if not title_text:
            return m[0]
        count = 0
        pattern = re.compile(
            r"(?m)^\|[^\n]*\|[ \t]*\n\|[ \t:|\-]+\|[ \t]*\n(?:\|[^\n]*\|[ \t]*(?:\n|$))+"
        )

        def table(t):
            nonlocal count
            raw = t[0]
            if r"\|" in raw or "`" in raw:
                return raw
            lines = [a.strip() for a in raw.strip().splitlines()]
            cells = [[c.strip() for c in line.strip("|").split("|")] for line in lines]
            if (
                len(cells) < 3
                or any(len(row) != len(cells[0]) for row in cells)
                or any(
                    not re.fullmatch(r":?-+:?", c.replace(" ", "")) for c in cells[1]
                )
            ):
                return raw
            count += 1
            result = (
                "<table><caption>"
                + escape(title_text)
                + "</caption><thead><tr>"
                + "".join('<th scope="col">' + escape(c) + "</th>" for c in cells[0])
                + "</tr></thead><tbody>"
            )
            result += (
                "".join(
                    "<tr>"
                    + "".join("<td>" + escape(c) + "</td>" for c in row)
                    + "</tr>"
                    for row in cells[2:]
                )
                + "</tbody></table>\n"
            )
            return result

        new = pattern.sub(table, body)
        if not count:
            return m[0]
        # The title is rendered in its associated captions, retaining text and roles.
        new = re.sub(r"(?m)^title:[ \t]*[^\n]*\n?", "", new, count=1)
        changed += count
        return m[1] + new + m[3]

    out = re.sub(r"(<chart\b[^>]*>)(.*?)(</chart>)", chart, text, flags=re.S)
    return out, changed