# -*- coding: utf-8 -*-
"""main.lua 的静态体检 —— 编译前把「真机上才会炸」的几类错抓出来。

S4e 那版的真机事故是：`segs/ddWrite/ddRead` 被放在 `slurp/sh` **之前**，
于是函数体里引用的 `sh` 不是那个 local，而是**全局 nil**，一调用就
`attempt to call a nil value (global 'sh')` —— 而且当时整条流水线全绿
（Compiler.exe 只打包 Lua、不解析它）。所以本文件专门查：

  ① 块配平（function/if/for/while/do/end/repeat/until）
  ② **先用后声明**：某个名字在文件里是 `local`，但第一次调用发生在它声明之前
  ③ 调用了既不是 local、也不在 Lua 标准库白名单里的名字（拼写错 / 忘了 local）
  ④ 模板约定：4 个占位符各 1 次、`--[[` 只在文件头、以 `return ui\\n` 结尾
     （tools/s5e_verify.py 会从 .face 里反解 Lua，这几条是它定位 Lua 的前提）

用法：python tools/s5e_lua_static.py [main.lua]
"""
import os, re, sys
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from paths import PROJ
from lua_balance import strip

LUAC = os.path.join(PROJ, 'main.lua')

# Lua 5.4 标准库 + 本项目用到的宿主全局（都不是拼写错）
WHITELIST = set("""
require pcall xpcall type pairs ipairs next select tostring tonumber rawget rawset
rawequal rawlen setmetatable getmetatable error assert print unpack load
string table math io os coroutine utf8 debug collectgarbage
""".split())
KEYWORDS = set("""
and break do else elseif end false for function goto if in local nil not or
repeat return then true until while
""".split())
REQUIRED_PH = ('__UIIMG__', '__PAY__', '__RB__', '__ROLE__')

fail = 0


def bad(msg):
    global fail
    fail += 1
    print('  FAIL ' + msg)


def ck(cond, msg):
    print(('  OK   ' if cond else '  FAIL ') + msg)
    if not cond:
        global fail
        fail += 1


def lineno(src, idx):
    return src.count('\n', 0, idx) + 1


def decls(src, code):
    """名字 -> 它在 code 里第一次被 local 声明（或作为函数参数出现）的位置。"""
    out = {}
    for m in re.finditer(r'\blocal\s+function\s+([A-Za-z_]\w*)', code):
        out.setdefault(m.group(1), m.start(1))
    for m in re.finditer(r'\blocal\s+([A-Za-z_][\w,\s]*?)\s*(?:=|$)', code, re.M):
        base = m.start(1)
        for piece in m.group(1).split(','):
            nm = piece.strip()
            if re.fullmatch(r'[A-Za-z_]\w*', nm):
                out.setdefault(nm, base)
    # 函数参数也是 local（`function(a, b)` / `function f(a)`），否则会误报
    for m in re.finditer(r'\bfunction\b\s*[A-Za-z_.:]*\s*\(([^)]*)\)', code):
        for piece in m.group(1).split(','):
            nm = piece.strip()
            if re.fullmatch(r'[A-Za-z_]\w*', nm):
                out.setdefault(nm, m.start())
    return out


def calls(code):
    """所有函数调用点（排除 `function NAME(` 这种定义）。"""
    out = []
    for m in re.finditer(r'(?<![\w.:])([A-Za-z_]\w*)\s*\(', code):
        head = code[max(0, m.start(1) - 12):m.start(1)]
        if re.search(r'\bfunction\s+$', head):
            continue                      # 定义，不是调用
        out.append((m.group(1), m.start(1)))
    return out


def main():
    src = open(LUAC, encoding='utf-8').read()
    code = strip(src)
    print('=== tools/s5e_lua_static.py  %s  (%d B) ===' % (LUAC, len(src.encode('utf-8'))))

    print('\n[1] 块配平')
    def cnt(k):
        return len(re.findall(r'(?<![\w.])' + k + r'(?![\w])', code))
    do_standalone = cnt('do') - cnt('while') - cnt('for')
    openers = cnt('function') + cnt('if') + cnt('while') + cnt('for') + do_standalone
    ck(openers == cnt('end') and cnt('repeat') == cnt('until'),
       'openers=%d end=%d（function=%d if=%d for=%d while=%d do=%d）repeat=%d until=%d'
       % (openers, cnt('end'), cnt('function'), cnt('if'), cnt('for'), cnt('while'),
          do_standalone, cnt('repeat'), cnt('until')))

    print('\n[2] 先用后声明（真机上会变成 "call a nil value"）')
    d = decls(src, code)
    early, unknown = [], []
    for name, pos in calls(code):
        if name in KEYWORDS or name in WHITELIST:
            continue
        if name in d:
            if pos < d[name]:
                early.append((name, lineno(src, pos), lineno(src, d[name])))
        else:
            unknown.append((name, lineno(src, pos)))
    ck(not early, '没有"先调用后 local"：%s'
       % ('' if not early else ' / '.join('%s 调用@%d 声明@%d' % e for e in early)))
    ck(not unknown, '没有未声明的被调用名：%s'
       % ('' if not unknown else ' / '.join('%s@%d' % u for u in unknown)))

    print('\n[3] 模板约定（s5e_verify.py 靠这些从 .face 里反解 Lua）')
    for ph in REQUIRED_PH:
        ck(src.count(ph) == 1, '占位符 %s 出现 %d 次（要 1）' % (ph, src.count(ph)))
    ck(src.startswith('--[[--'), '文件以 --[[-- 开头')
    ck(src.count('--[[') == 1, '--[[ 只出现 %d 次（要 1，否则 verify 会从中间截 Lua）'
       % src.count('--[['))
    ck(src.endswith('return ui\n'), '以 return ui\\n 结尾')
    ck(src.count('return ui\n') == 1, 'return ui\\n 只出现 %d 次（要 1）' % src.count('return ui\n'))
    ck('local UIIMG = {' in src, '有 local UIIMG = {')

    print('\n[4] 与 s5e_verify.py 对齐的字面量')
    ck(re.search(r"local ROLE = '__ROLE__'", src) is not None, "local ROLE = '__ROLE__'")
    ck(re.search(r'local W, H = 480, 480', src) is not None, 'local W, H = 480, 480')
    m = re.search(r'local DEVS = \{([^}]*)\}', src)
    ck(m is not None and '/dev/health' in m.group(1),
       'DEVS 里有 health -> /dev/health：%s' % (m.group(1).strip() if m else None))
    ck("local DEV = DEVS.health" in src,
       'DEV 来自 DEVS.health（表里的键名真的被用于取节点，不是摆设）')
    paytbl = re.search(r'local PAYTBL = \{\n__PAY__\n__RB__\n\}', src)
    ck(paytbl is not None, 'PAYTBL 里只有 __PAY__ / __RB__ 两处占位符（没有硬编码节点名）')
    ck('466' not in src, '没有残留的 466（旧 S4e 尺寸）')

    print('\n=== %s ===' % ('全部通过' if not fail else '%d 项失败' % fail))
    return 1 if fail else 0


if __name__ == '__main__':
    sys.exit(main())