File size: 3,804 Bytes
bed5cc5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 |
"""
Cursor Rules Generator - MDC Formatter
This module handles the formatting of rules in MDC format.
"""
from typing import Dict, Any, Optional
class MDCFormatter:
"""Formatter for MDC (Markdown with Context) format."""
def format_to_mdc(
self,
rule_type: str,
content: str,
description: str,
parameters: Optional[Dict[str, Any]] = None
) -> str:
"""Format rule content to MDC format.
Args:
rule_type: The type of rule (Always, Auto Attached, Agent Requested, Manual)
content: The main content of the rule
description: A short description of the rule's purpose
parameters: Additional parameters for formatting
Returns:
str: The formatted rule in MDC format
"""
# Set default parameters if not provided
if parameters is None:
parameters = {}
# Extract parameters
globs = parameters.get('globs', '')
referenced_files = parameters.get('referenced_files', '')
# Create MDC format
mdc = '---\n'
mdc += f'description: {description}\n'
if rule_type == 'Auto Attached' and globs:
mdc += f'globs: {globs}\n'
if rule_type == 'Always':
mdc += 'alwaysApply: true\n'
else:
mdc += 'alwaysApply: false\n'
mdc += '---\n\n'
mdc += content + '\n'
# Add referenced files
if referenced_files:
mdc += '\n' + referenced_files
return mdc
def parse_mdc(self, mdc_content: str) -> Dict[str, Any]:
"""Parse MDC content into a structured format.
Args:
mdc_content: The MDC content to parse
Returns:
Dict[str, Any]: A dictionary containing the parsed MDC content
"""
result = {
'metadata': {},
'content': '',
'referenced_files': []
}
# Check if the content has metadata section
if mdc_content.startswith('---'):
# Split the content into metadata and body
parts = mdc_content.split('---', 2)
if len(parts) >= 3:
# Parse metadata
metadata_lines = parts[1].strip().split('\n')
for line in metadata_lines:
if ':' in line:
key, value = line.split(':', 1)
result['metadata'][key.strip()] = value.strip()
# Get content
content = parts[2].strip()
else:
# No proper metadata section, treat everything as content
content = mdc_content
else:
# No metadata section, treat everything as content
content = mdc_content
# Split content and referenced files
lines = content.split('\n')
content_lines = []
referenced_files = []
for line in lines:
if line.strip().startswith('@'):
referenced_files.append(line.strip())
else:
content_lines.append(line)
result['content'] = '\n'.join(content_lines).strip()
result['referenced_files'] = referenced_files
# Determine rule type
if result['metadata'].get('alwaysApply') == 'true':
result['rule_type'] = 'Always'
elif 'globs' in result['metadata']:
result['rule_type'] = 'Auto Attached'
else:
# Default to Manual if not specified
result['rule_type'] = 'Manual'
return result
|