import glob
import re

keys = set()
files = glob.glob('/home2/haeyulhome/skhani/public_html/**/*.php', recursive=True) + glob.glob('/home2/haeyulhome/skhani/public_html/**/*.html', recursive=True)
for file in files:
    try:
        with open(file, 'rb') as f:
            c = f.read()
        
        matches = re.findall(br'\$[a-zA-Z0-9_]+\[([a-zA-Z_][a-zA-Z0-9_]*)\]', c)
        for m in matches:
            keys.add(m.decode('ascii', errors='ignore'))
    except Exception as e:
        pass

keys.discard('')
keys.discard('0')

out = b"<?php\n"
for k in sorted(keys):
    # Only alphanumeric and underscore
    if re.match(r'^[a-zA-Z_][a-zA-Z0-9_]*$', k):
        # some constants might be real constants, so check if not defined
        out += f"if(!defined('{k}')) define('{k}', '{k}');\n".encode('ascii')

with open('/home2/haeyulhome/skhani/public_html/lib/constants_polyfill.php', 'wb') as f:
    f.write(out)

# Inject into config.php
with open('/home2/haeyulhome/skhani/public_html/lib/config.php', 'rb') as f:
    c = f.read()

if b'constants_polyfill.php' not in c:
    c = c.replace(b'<?', b"<?\ninclude_once __DIR__ . '/constants_polyfill.php';\n", 1)
    with open('/home2/haeyulhome/skhani/public_html/lib/config.php', 'wb') as f:
        f.write(c)

print("Generated constants_polyfill.php with", len(keys), "keys")