ODXStudio - ODX File Management

"""ODX file management: merge, split, and validate."""
import xml.etree.ElementTree as ET
import shutil, os
from pathlib import Path

def split_odx_by_layer(odx_path: str, output_dir: str) -> list:
    """Split a combined ODX file into one file per layer."""
    tree = ET.parse(odx_path)
    root = tree.getroot()
    output_files = []
    container = root.find("DIAG-LAYER-CONTAINER")
    if container is None:
        container = root
    for layer_type in ["BASE-VARIANTS", "ECU-VARIANTS",
                        "FUNCTIONAL-GROUPS", "ECU-SHARED-DATAS"]:
        section = container.find(layer_type)
        if section is None:
            continue
        for layer in section:
            name = layer.findtext("SHORT-NAME", "unknown")
            # Create wrapper ODX file for this layer
            new_root = ET.Element("ODX", attrib=root.attrib)
            new_container = ET.SubElement(new_root, "DIAG-LAYER-CONTAINER")
            new_section  = ET.SubElement(new_container, layer_type)
            new_section.append(layer)
            out_path = os.path.join(output_dir, f"{name}.odx")
            ET.ElementTree(new_root).write(out_path,
                xml_declaration=True, encoding="UTF-8")
            output_files.append(out_path)
    return output_files

def validate_id_refs(odx_path: str) -> list:
    """Check that all ID-REF attributes point to existing IDs."""
    tree = ET.parse(odx_path)
    root = tree.getroot()
    all_ids = {e.get("ID") for e in root.iter() if e.get("ID")}
    broken = []
    for elem in root.iter():
        ref = elem.get("ID-REF")
        if ref and ref not in all_ids:
            broken.append(f"{elem.tag} ID-REF={ref} not found")
    return broken