126 lines
4.3 KiB
Python
126 lines
4.3 KiB
Python
#!/usr/bin/env python3
|
|
import json
|
|
import os
|
|
import re
|
|
|
|
def update_readme():
|
|
base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
readme_path = os.path.join(base_dir, "README.md")
|
|
index_path = os.path.join(base_dir, "skills_index.json")
|
|
|
|
print(f"📖 Reading skills index from: {index_path}")
|
|
with open(index_path, 'r', encoding='utf-8') as f:
|
|
skills = json.load(f)
|
|
|
|
total_skills = len(skills)
|
|
print(f"🔢 Total skills found: {total_skills}")
|
|
|
|
print(f"📝 Updating README at: {readme_path}")
|
|
with open(readme_path, 'r', encoding='utf-8') as f:
|
|
content = f.read()
|
|
|
|
# 1. Update Title Count
|
|
content = re.sub(
|
|
r'(# 🌌 Antigravity Awesome Skills: )\d+(\+ Agentic Skills)',
|
|
f'\\g<1>{total_skills}\\g<2>',
|
|
content
|
|
)
|
|
|
|
# 2. Update Blockquote Count
|
|
content = re.sub(
|
|
r'(Collection of )\d+(\+ Universal)',
|
|
f'\\g<1>{total_skills}\\g<2>',
|
|
content
|
|
)
|
|
|
|
# 3. Update Intro Text Count
|
|
content = re.sub(
|
|
r'(library of \*\*)\d+( high-performance skills\*\*)',
|
|
f'\\g<1>{total_skills}\\g<2>',
|
|
content
|
|
)
|
|
|
|
# 4. Update Registry Header Count
|
|
content = re.sub(
|
|
r'(## Full Skill Registry \()\d+/\d+(\))',
|
|
f'\\g<1>{total_skills}/{total_skills}\\g<2>',
|
|
content
|
|
)
|
|
|
|
# 5. Insert Collections / Bundles Section (New in Phase 3)
|
|
# This logic checks if "## 📦 Curated Collections" exists. If not, it creates it before Full Registry.
|
|
collections_header = "## 📦 Curated Collections"
|
|
|
|
if collections_header not in content:
|
|
# Insert before Full Skill Registry
|
|
content = content.replace("## Full Skill Registry", f"{collections_header}\n\n[Check out our Starter Packs in docs/BUNDLES.md](docs/BUNDLES.md) to find the perfect toolkit for your role.\n\n## Full Skill Registry")
|
|
|
|
# 6. Generate New Registry Table
|
|
print("🔄 Generating new registry table...")
|
|
|
|
# Store the Note block to preserve it
|
|
note_pattern = r'(> \[!NOTE\].*?)\n\n\| Skill Name'
|
|
note_match = re.search(note_pattern, content, re.DOTALL)
|
|
note_block = ""
|
|
if note_match:
|
|
note_block = note_match.group(1)
|
|
else:
|
|
note_block = "> [!NOTE] > **Document Skills**: We provide both **community** and **official Anthropic** versions. Locally, the official versions are used by default."
|
|
|
|
table_header = "| Skill Name | Risk | Description | Path |\n| :--- | :--- | :--- | :--- |"
|
|
table_rows = []
|
|
|
|
for skill in skills:
|
|
name = skill.get('name', 'Unknown')
|
|
desc = skill.get('description', '').replace('\n', ' ').strip()
|
|
path = skill.get('path', '')
|
|
risk = skill.get('risk', 'unknown')
|
|
|
|
# Risk Icons
|
|
risk_icon = "⚪"
|
|
if risk == "official": risk_icon = "🟣" # Mapping official to purple
|
|
if risk == "none": risk_icon = "🟢"
|
|
if risk == "safe": risk_icon = "🔵"
|
|
if risk == "critical": risk_icon = "🟠"
|
|
if risk == "offensive": risk_icon = "🔴"
|
|
|
|
# Escape pipes
|
|
desc = desc.replace('|', r'\|')
|
|
|
|
row = f"| **{name}** | {risk_icon} | {desc} | `{path}` |"
|
|
table_rows.append(row)
|
|
|
|
new_table_section = f"{note_block}\n\n{table_header}\n" + "\n".join(table_rows)
|
|
|
|
# Replace the old table section
|
|
header_pattern = r'## Full Skill Registry \(\d+/\d+\)'
|
|
header_match = re.search(header_pattern, content)
|
|
|
|
if not header_match:
|
|
print("❌ Could not find 'Full Skill Registry' header.")
|
|
return
|
|
|
|
start_pos = header_match.end()
|
|
|
|
# Find the next section (## ...) or end of file
|
|
next_section_match = re.search(r'\n## ', content[start_pos:])
|
|
|
|
if next_section_match:
|
|
end_pos = start_pos + next_section_match.start()
|
|
rest_of_file = content[end_pos:]
|
|
else:
|
|
rest_of_file = ""
|
|
|
|
before_header = content[:header_match.start()]
|
|
new_header = f"## Full Skill Registry ({total_skills}/{total_skills})"
|
|
|
|
new_content = f"{before_header}{new_header}\n\n{new_table_section}\n{rest_of_file}"
|
|
|
|
with open(readme_path, 'w', encoding='utf-8') as f:
|
|
f.write(new_content)
|
|
|
|
print("✅ README.md updated successfully with Collections link and Risk columns.")
|
|
|
|
if __name__ == "__main__":
|
|
update_readme()
|