Function validate_mxgraphxml_syntax

autofigure/generator.py:954–1052

Validates mxGraph XML syntax for draw.io compatibility. Args: mxgraph_code: The mxGraph XML code to validate. Returns: Tuple of (is_valid, error_message_or_success_message)

(mxgraph_code: str)

def validate_mxgraphxml_syntax(mxgraph_code: str) -> Tuple[bool, str]:
    """
    Validates mxGraph XML syntax for draw.io compatibility.

    Args:
        mxgraph_code: The mxGraph XML code to validate.

    Returns:
        Tuple of (is_valid, error_message_or_success_message)
    """
    try:
        import xml.etree.ElementTree as ET

        code = mxgraph_code.strip()

        # Check for mxfile wrapper
        if not code.startswith('<mxfile'):
            return False, "mxGraph XML must start with <mxfile>"

        if not code.endswith('</mxfile>'):
            return False, "mxGraph XML must end with </mxfile>"

        # Try to parse as XML
        try:
            root = ET.fromstring(code)
        except ET.ParseError as e:
            return False, f"XML parsing error: {e}"

        # Check for required structure: mxfile > diagram > mxGraphModel > root
        diagram = root.find('.//diagram')
        if diagram is None:
            return False, "Missing <diagram> element inside <mxfile>"

        graph_model = diagram.find('.//mxGraphModel')
        if graph_model is None:
            return False, "Missing <mxGraphModel> element inside <diagram>"

        root_elem = graph_model.find('.//root')
        if root_elem is None:
            return False, "Missing <root> element inside <mxGraphModel>"

        # Check for base mxCells (id="0" and id="1")
        mxcells = root_elem.findall('mxCell')
        cell_ids = [cell.get('id') for cell in mxcells]

        if '0' not in cell_ids:
            return False, "Missing required mxCell with id='0'"

        if '1' not in cell_ids:
            return False, "Missing required mxCell with id='1' (default parent)"

        # Check that all mxCells (except id="0") have parent attribute
        for cell in mxcells:
            cell_id = cell.get('id')
            if cell_id != '0' and cell.get('parent') is None:
                return False, f"mxCell id='{cell_id}' is missing required 'parent' attribute"

        # Check for unquoted attributes in XML tags