64 lines
2.5 KiB
Python
64 lines
2.5 KiB
Python
import os
|
|
import re
|
|
import json
|
|
|
|
def fix_yaml_quotes(skills_dir):
|
|
print(f"Scanning for YAML quoting errors in {skills_dir}...")
|
|
fixed_count = 0
|
|
|
|
for root, dirs, files in os.walk(skills_dir):
|
|
dirs[:] = [d for d in dirs if not d.startswith('.')]
|
|
if 'SKILL.md' in files:
|
|
file_path = os.path.join(root, 'SKILL.md')
|
|
try:
|
|
with open(file_path, 'r', encoding='utf-8') as f:
|
|
content = f.read()
|
|
except Exception:
|
|
continue
|
|
|
|
fm_match = re.search(r'^---\s*\n(.*?)\n---', content, re.DOTALL)
|
|
if not fm_match:
|
|
continue
|
|
|
|
fm_text = fm_match.group(1)
|
|
new_fm_lines = []
|
|
changed = False
|
|
|
|
for line in fm_text.split('\n'):
|
|
if line.startswith('description:'):
|
|
key, val = line.split(':', 1)
|
|
val = val.strip()
|
|
|
|
# Store original to check if it matches the fixed version
|
|
orig_val = val
|
|
|
|
# Strip matching outer quotes if they exist
|
|
if val.startswith('"') and val.endswith('"') and len(val) >= 2:
|
|
val = val[1:-1]
|
|
elif val.startswith("'") and val.endswith("'") and len(val) >= 2:
|
|
val = val[1:-1]
|
|
|
|
# Now safely encode using JSON to handle internal escapes
|
|
safe_val = json.dumps(val)
|
|
|
|
if safe_val != orig_val:
|
|
new_line = f"description: {safe_val}"
|
|
new_fm_lines.append(new_line)
|
|
changed = True
|
|
continue
|
|
new_fm_lines.append(line)
|
|
|
|
if changed:
|
|
new_fm_text = '\n'.join(new_fm_lines)
|
|
new_content = content[:fm_match.start(1)] + new_fm_text + content[fm_match.end(1):]
|
|
with open(file_path, 'w', encoding='utf-8') as f:
|
|
f.write(new_content)
|
|
print(f"Fixed quotes in {os.path.relpath(file_path, skills_dir)}")
|
|
fixed_count += 1
|
|
|
|
print(f"Total files fixed: {fixed_count}")
|
|
|
|
if __name__ == '__main__':
|
|
base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
fix_yaml_quotes(os.path.join(base_dir, 'skills'))
|