73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
|
"""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')
|