"""S19: Final bridge test — extract_structure with new procedure parser"""
import sys, os, re, time
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))

ROOT = "D:/cobol-java/cobol-test-programs/"
COPYBOOKS = os.path.join(ROOT, "common", "copybooks")
from cobol_testgen import extract_structure, generate_data
from cobol_testgen.read import preprocess, resolve_copybooks
from cobol_testgen.flatfile import analyze_fd_layout, write_all_files

def find_main(d):
    cbls = [f for f in os.listdir(d) if f.endswith('.cbl')]
    ws = [f for f in cbls if re.match(r'main-\d{2}-', f, re.IGNORECASE)]
    if ws: return max(ws, key=lambda f: os.path.getsize(os.path.join(d, f)))
    return max(cbls, key=lambda f: os.path.getsize(os.path.join(d, f))) if cbls else None

fmt = "{:<28} {:>4} {:>7} {:>5} {:>5}  {}"
print(fmt.format("Program", "Br", "Time", "Recs", "Flats", "Parser"))
print("-"*78)

with_br = 0; gen_ok = 0; old_cnt = 0; new_cnt = 0; total_t = 0
for d in sorted(os.listdir(ROOT)):
    dp = os.path.join(ROOT, d)
    if not os.path.isdir(dp) or d in ('common','docs','cross-cutting'): continue
    fn = find_main(dp)
    if not fn: continue

    t0 = time.time()
    try:
        src = open(os.path.join(dp, fn), encoding='utf-8').read()
        st = extract_structure(src)
        branches = st.get('total_branches', 0)
        t = (time.time()-t0)*1000
        total_t += t

        # Also generate data + flat files
        pp = resolve_copybooks(src, dp, extra_search_paths=[COPYBOOKS])
        pp = preprocess(pp)
        recs = generate_data(pp, st)
        layouts = analyze_fd_layout(pp)
        flats = write_all_files(recs, pp, dp) if layouts else []

        parser = st.get("parser", "old")
        if branches > 0: with_br += 1
        gen_ok += 1
        if parser == "new": new_cnt += 1
        else: old_cnt += 1
        print(fmt.format(d, branches, f"{t:.0f}ms", len(recs), len(flats), parser))
    except Exception as e:
        print(fmt.format(d, "ERR", f"{(time.time()-t0)*1000:.0f}ms", "", "", str(e)[:40]))

print(f"\nWith branches: {with_br}/{gen_ok}")
print(f"Parser: new={new_cnt} old={old_cnt}")
print(f"Total time: {total_t/1000:.1f}s")