import re

path = 'tools/build/export_skills_to_ts.py'
with open(path, 'r', encoding='utf-8') as f:
    content = f.read()

# Fix window block for canonical skills
content = re.sub(r"if \(typeof window !== 'undefined'\) \{\n.*?\n\}\n", "", content, flags=re.DOTALL)

# Add any casting for return SKILL_TAGS[id]
content = re.sub(r'return SKILL_TAGS\[id\] \|\| null;', 'return (SKILL_TAGS as any)[id] || null;', content)
content = re.sub(r'return CANONICAL_SKILLS\[id\] \|\| null;', 'return (CANONICAL_SKILLS as any)[id] || null;', content)
content = re.sub(r'return SUPPORT_SKILLS\[id\] \|\| null;', 'return (SUPPORT_SKILLS as any)[id] || null;', content)

with open(path, 'w', encoding='utf-8') as f:
    f.write(content)