增添 S5 支持

s5e-pressure-anim/tools/s5e_lua_static.py

@@ -0,0 +1,145 @@
+# -*- 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())