# -*- coding: utf-8 -*-
import io, os, re, glob

base = r'C:\Users\jcxiong\Documents\Codex\MyProject\trafficAudit'
with io.open(os.path.join(base, 'docs', 'init.sql'), encoding='utf-8') as f:
    sql = f.read()

def snake(name):
    return re.sub(r'(?<!^)(?=[A-Z])', '_', name).lower()

tables = {}
for m in re.finditer(r'CREATE TABLE\s+(\w+)\s*\((.*?)\)\s*ENGINE', sql, re.S):
    name, body = m.group(1), m.group(2)
    cols = set()
    for cm in re.finditer(r'^\s*(\w+)\s', body, re.M):
        cols.add(cm.group(1))
    tables[name] = cols

entity_files = glob.glob(os.path.join(base, 'traffic-audit-server', 'src', 'main', 'java', 'com', 'trafficaudit', '**', 'entity', '*.java'), recursive=True)
for f in entity_files:
    with io.open(f, encoding='utf-8') as fh:
        src = fh.read()
    tm = re.search(r'@TableName\("(\w+)"\)', src)
    if not tm:
        print(os.path.basename(f), '-> NO @TableName'); continue
    tname = tm.group(1)
    fields = re.findall(r'private\s+[\w<>]+\s+(\w+);', src)
    fields = [x for x in fields if x != 'serialVersionUID']
    col_fields = {snake(fld) for fld in fields}
    missing = sorted(col_fields - tables.get(tname, set()))
    extra = sorted(tables.get(tname, set()) - col_fields)
    status = 'OK' if not missing else 'MISSING: ' + ', '.join(missing)
    print(f'{os.path.basename(f)} -> {tname}: {status}')
    if extra:
        print('   (table-only cols):', ', '.join(extra))