#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""PBL-RTX-0005 生成器补丁：修 pbl_assert_append_only 过程体 + _with_delimiter 行尾注释加固。

每处替换都断言命中，未命中即 rc=1（禁止静默漏改）。
"""
import io
import os
import re
import sys

HERE = os.path.dirname(os.path.abspath(__file__))
MODULE_DIR = os.path.dirname(HERE)
TARGET = os.path.join(MODULE_DIR, "scripts", "pbl_runtime_event_ddl.py")

src = io.open(TARGET, encoding="utf-8").read()
orig = src
applied = []


def rep(old, new, tag):
    global src
    n = src.count(old)
    if n != 1:
        print("FAIL 定位不唯一(%d)：%s" % (n, tag))
        sys.exit(1)
    src = src.replace(old, new)
    applied.append(tag)


# ---------------------------------------------------------------- 1) 过程体 RETURN; → 正向条件 SIGNAL
rep(
    """CREATE PROCEDURE `pbl_assert_append_only`(
    IN p_table VARCHAR(64), IN p_priv VARCHAR(16), IN p_detail VARCHAR(255))
    READS SQL DATA
BEGIN
    IF p_table <> '%(table)s' THEN
        RETURN;                       -- 只守护事件表，其他表不受影响
    END IF;
    IF IFNULL(@pbl_guard_ctx, '') = '%(ctx)s' THEN
        RETURN;                       -- break-glass：一次性历史数据回填
    END IF;
    SIGNAL SQLSTATE '45000'
        SET MESSAGE_TEXT = 'PBL_APPEND_ONLY_DENIED',
            MYSQL_ERRNO = 45001;
END;""",
    """CREATE PROCEDURE `pbl_assert_append_only`(
    IN p_table VARCHAR(64), IN p_priv VARCHAR(16), IN p_detail VARCHAR(255))
    READS SQL DATA
BEGIN
    -- PBL-RTX-0005：MariaDB 存储过程体禁用 RETURN（RETURN 只在 FUNCTION 中合法），
    -- 旧写法 `IF ... THEN RETURN; END IF;` 在建过程阶段即 ERROR 1064，导致本过程
    -- 连同其后的触发器/维护过程全部未创建。改为等价正向条件单语句 SIGNAL：
    --   仅「是本表」且「非 break-glass 上下文」时才拒绝，其余情形不执行任何语句，
    --   与旧的两段提前返回语义完全一致。
    IF p_table = '%(table)s'
       AND IFNULL(@pbl_guard_ctx, '') <> '%(ctx)s' THEN
        SIGNAL SQLSTATE '45000'
            SET MESSAGE_TEXT = 'PBL_APPEND_ONLY_DENIED',
                MYSQL_ERRNO = 45001;
    END IF;
END;""",
    "append_only_sql: RETURN; -> 正向条件 SIGNAL",
)

# ---------------------------------------------------------------- 2) _with_delimiter 行尾注释加固
rep(
    '''_BARE_TOKEN = re.compile(r"^(CREATE|DROP|ALTER|GRANT|REVOKE)\\s*\\$\\$\\s*$", re.I)''',
    '''_BARE_TOKEN = re.compile(r"^(CREATE|DROP|ALTER|GRANT|REVOKE)\\s*\\$\\$\\s*$", re.I)
# 语句末行的「代码 + 行尾注释」形态：`$$` 若与 `-- 注释` 同行，会被行注释吞掉，
# 客户端永远等不到语句结束符（PBL-RTX-0005 加固项：注释上移，`$$` 独占行尾）。
_TRAILING_COMMENT = re.compile(r"^(?P<code>\\S.*?\\s)(?P<cmt>(?:--|#)\\s.*|/\\*.*\\*/)$")
# 过程/函数/触发器体内非法的裸 RETURN（MariaDB 仅 FUNCTION 允许 RETURN；PROCEDURE 报 1064）
_ILLEGAL_RETURN = re.compile(r"^RETURN\\s*;?\\s*$", re.I)''',
    "_with_delimiter: 新增 _TRAILING_COMMENT / _ILLEGAL_RETURN 常量",
)

rep(
    """    def flush():
        if not buf:
            if cmts:
                out.append("\\n".join(cmts))
                del cmts[:]
            return
        stmt = re.sub(r";\\s*$", "", "\\n".join(buf).strip())
        head = ("\\n".join(cmts) + "\\n") if cmts else ""
        out.append(head + stmt + delim)
        del buf[:]
        del cmts[:]""",
    """    def flush():
        if not buf:
            if cmts:
                out.append("\\n".join(cmts))
                del cmts[:]
            return
        lines = list(buf)
        # 末行行尾注释上移到语句之前：否则产出 `... -- note$$`，`$$` 落在注释内
        # → 客户端认为语句未结束，与下一条语句黏连（同类 1064 根因）。
        tail_cmts = []
        m = _TRAILING_COMMENT.match(lines[-1].rstrip())
        if m and not m.group("code").lstrip().startswith(("--", "#")):
            tail_cmts.append(m.group("cmt"))
            lines[-1] = m.group("code")
        stmt = re.sub(r";\\s*$", "", "\\n".join(lines).strip())
        head_lines = cmts + tail_cmts
        head = ("\\n".join(head_lines) + "\\n") if head_lines else ""
        out.append(head + stmt + delim)
        del buf[:]
        del cmts[:]""",
    "_with_delimiter.flush(): 末行行尾注释上移",
)

# ---------------------------------------------------------------- 3) 静态断言：禁止过程体裸 RETURN
rep(
    '''def _delimiter_integrity_errors(text, label=""):''',
    '''def _procedure_body_return_errors(text, label=""):
    """PBL-RTX-0005 回归门禁：`CREATE PROCEDURE` 体内不得出现裸 `RETURN`。

    MariaDB/MySQL 的 RETURN 语句只允许出现在 FUNCTION（及触发器外的函数语义）中；
    存储过程里写 `RETURN;` 会在 CREATE PROCEDURE 阶段直接 ERROR 1064，且因为整段
    在 `DELIMITER $$` 块内，失败会连带后续触发器/维护过程一起未创建（现场表现为
    `SHOW PROCEDURE STATUS` 0 个 + CALL 报 1305 does not exist）。
    返回缺陷描述列表；空列表 = 通过。
    """
    errs = []
    pfx = ("[%s] " % label) if label else ""
    in_proc = False
    for i, ln in enumerate(text.split("\\n"), 1):
        s = ln.strip()
        if re.match(r"^CREATE\\s+(OR\\s+REPLACE\\s+)?PROCEDURE\\b", s, re.I):
            in_proc = True
            continue
        if in_proc and re.match(r"^CREATE\\s+\\w", s, re.I):
            in_proc = False
        if in_proc and _ILLEGAL_RETURN.match(s):
            errs.append(pfx + "L%d 存储过程体内出现裸 RETURN（MariaDB 过程体非法 → ERROR 1064）: %r"
                        % (i, s))
    return errs


def _delimiter_integrity_errors(text, label=""):''',
    "新增 _procedure_body_return_errors() 回归门禁函数",
)

rep(
    """    struct_errs = []
    for label, text in artifacts:
        struct_errs.extend(_delimiter_integrity_errors(text, label))""",
    """    struct_errs = []
    for label, text in artifacts:
        struct_errs.extend(_delimiter_integrity_errors(text, label))
        # PBL-RTX-0005：过程体裸 RETURN 静态门禁（实库 1064 的根因形态）
        struct_errs.extend(_procedure_body_return_errors(text, label))""",
    "verify(): 接入过程体 RETURN 静态门禁",
)

if "PBL_APPEND_ONLY_DENIED" not in src or src.count("RETURN;") > 0:
    print("FAIL 校验：src 中仍残留 `RETURN;`")
    sys.exit(1)

io.open(TARGET, "w", encoding="utf-8").write(src)
print("OK 已应用 %d 处替换：" % len(applied))
for a in applied:
    print("   -", a)
print("bytes %d -> %d" % (len(orig), len(src)))