#!/usr/bin/env python3
"""
Validate Mermaid Diagrams
Tests all Mermaid diagrams to ensure they render without syntax errors
"""
import re
import os
import glob
import asyncio
import subprocess
import sys
from pathlib import Path
def install_dependencies():
"""Install required dependencies"""
try:
import playwright
return True
except ImportError:
print("Installing Playwright...")
try:
subprocess.check_call([sys.executable, "-m", "pip", "install", "playwright"])
subprocess.check_call([sys.executable, "-m", "playwright", "install", "chromium"])
return True
except Exception as e:
print(f"Failed to install dependencies: {e}")
return False
def extract_mermaid_diagrams(md_file):
"""Extract all Mermaid diagrams from a Markdown file"""
try:
with open(md_file, 'r', encoding='utf-8') as f:
content = f.read()
# Find all Mermaid code blocks
mermaid_pattern = r'```mermaid\n(.*?)\n```'
diagrams = re.findall(mermaid_pattern, content, re.DOTALL)
return diagrams
except Exception as e:
print(f"Error reading {md_file}: {e}")
return []
def create_test_html(diagrams):
"""Create a test HTML file with all diagrams"""
html_content = """
Mermaid Diagram Validation
🔍 Mermaid Diagram Validation Test
Loading diagrams...
"""
for i, diagram in enumerate(diagrams, 1):
html_content += f"""
Diagram {i}
{diagram}
Rendering...
"""
html_content += """
"""
return html_content
async def validate_diagrams_in_browser(html_file):
"""Validate diagrams using Playwright browser automation"""
try:
from playwright.async_api import async_playwright
async with async_playwright() as p:
browser = await p.chromium.launch()
page = await browser.new_page()
# Enable console logging
page.on("console", lambda msg: print(f"Browser: {msg.text}"))
# Load HTML file
file_path = Path(html_file).resolve()
await page.goto(f"file://{file_path}")
# Wait for validation to complete
await page.wait_for_timeout(10000) # Wait 10 seconds for all diagrams to render
# Get validation results
status_text = await page.text_content('#status')
print(f"Validation Result: {status_text}")
# Count successful renders
svg_count = await page.evaluate("document.querySelectorAll('.mermaid svg').length")
total_diagrams = await page.evaluate("document.querySelectorAll('.mermaid').length")
await browser.close()
return svg_count, total_diagrams
except Exception as e:
print(f"Browser validation error: {e}")
return 0, 0
def main():
"""Validate all Mermaid diagrams in Markdown files"""
if not install_dependencies():
print("❌ Could not install required dependencies")
return False
print("🔍 MERMAID DIAGRAM VALIDATION")
print("=" * 50)
# Find all Markdown files
md_files = glob.glob("*.md")
if not md_files:
print("❌ No Markdown files found in current directory")
return False
all_diagrams = []
file_diagram_counts = {}
# Extract diagrams from all files
for md_file in md_files:
if ('TASK_COMPLETION_SUMMARY' in md_file or md_file.startswith('.')):
continue
print(f"📄 Extracting diagrams from {md_file}...")
diagrams = extract_mermaid_diagrams(md_file)
if diagrams:
print(f" ✅ Found {len(diagrams)} Mermaid diagram(s)")
all_diagrams.extend(diagrams)
file_diagram_counts[md_file] = len(diagrams)
else:
print(f" ℹ️ No Mermaid diagrams found")
if not all_diagrams:
print("❌ No Mermaid diagrams found in any files")
return False
print(f"\n🎯 Total diagrams to validate: {len(all_diagrams)}")
print()
# Create test HTML file
print("📝 Creating validation test file...")
html_content = create_test_html(all_diagrams)
test_file = "mermaid_validation_test.html"
with open(test_file, 'w', encoding='utf-8') as f:
f.write(html_content)
print(f" ✅ Created {test_file}")
# Validate in browser
print("\n🌐 Running browser validation...")
svg_count, total_diagrams = asyncio.run(validate_diagrams_in_browser(test_file))
print("\n" + "=" * 50)
print("🎯 VALIDATION RESULTS")
print("=" * 50)
for md_file, count in file_diagram_counts.items():
print(f"📄 {md_file}: {count} diagram(s)")
print(f"\n📊 Overall Results:")
print(f" • Total diagrams: {total_diagrams}")
print(f" • Successfully rendered: {svg_count}")
print(f" • Failed to render: {total_diagrams - svg_count}")
if svg_count == total_diagrams:
print(f"\n✅ ALL DIAGRAMS VALIDATED SUCCESSFULLY!")
print(f"🎉 No syntax errors found - all Mermaid diagrams are rendering correctly!")
else:
print(f"\n⚠️ Some diagrams failed validation")
print(f"📋 Check {test_file} in your browser for detailed error information")
# Clean up
try:
os.remove(test_file)
print(f"\n🧹 Cleaned up {test_file}")
except:
pass
return svg_count == total_diagrams
if __name__ == "__main__":
success = main()
sys.exit(0 if success else 1)