Initial Commit

2026-10-02 21:34:56 +08:00
parent 5a3867ff25
commit 79bf4cdbf2
915 changed files with 20552 additions and 2 deletions
@@ -0,0 +1,72 @@
+"""Crude but effective Lua block-balance checker: strips comments and strings,
+then counts block openers vs `end`."""
+import sys, re
+
+def strip(src):
+    out = []
+    i, n = 0, len(src)
+    while i < n:
+        c = src[i]
+        # long bracket comment / string  --[[ ]] , --[==[ ]==]
+        m = re.match(r'--\[(=*)\[', src[i:])
+        if m:
+            close = ']' + m.group(1) + ']'
+            j = src.find(close, i)
+            i = n if j < 0 else j + len(close)
+            out.append(' ')
+            continue
+        if src.startswith('--', i):
+            j = src.find('\n', i)
+            i = n if j < 0 else j
+            out.append(' ')
+            continue
+        m = re.match(r'\[(=*)\[', src[i:])
+        if m:
+            close = ']' + m.group(1) + ']'
+            j = src.find(close, i)
+            i = n if j < 0 else j + len(close)
+            out.append('""')
+            continue
+        if c in '"\'':
+            j = i + 1
+            while j < n:
+                if src[j] == '\\':
+                    j += 2
+                    continue
+                if src[j] == c:
+                    break
+                j += 1
+            i = j + 1
+            out.append('""')
+            continue
+        out.append(c)
+        i += 1
+    return ''.join(out)
+
+
+def check(path):
+    src = open(path, encoding='utf-8').read()
+    code = strip(src)
+    def cnt(k):
+        return len(re.findall(r'(?<![\w.])' + k + r'(?![\w])', code))
+    # every while/for consumes exactly one `do`, wherever it sits (may wrap lines)
+    do_total = cnt('do')
+    do_standalone = do_total - cnt('while') - cnt('for')
+    openers = cnt('function') + cnt('if') + cnt('while') + cnt('for') + do_standalone
+    enders = cnt('end')
+    repeat_n, until_n = cnt('repeat'), cnt('until')
+    print(f'{path}')
+    print(f'  function={cnt("function")} if={cnt("if")} while={cnt("while")} '
+          f'for={cnt("for")} do(standalone)={do_standalone} repeat={repeat_n}')
+    print(f'  openers={openers}  end={enders}  until={until_n}')
+    ok = (openers == enders) and (repeat_n == until_n)
+    print(f'  BALANCED: {ok}')
+    if not ok:
+        print('  ^^^ 括号/块不匹配，需要检查')
+    return ok
+
+
+allok = True
+for p in sys.argv[1:]:
+    allok &= check(p)
+print('\nALL OK' if allok else '\nPROBLEM')