Function handle_get_processing_order

codewiki/mcp/tools/module_tree.py:109–146

Write the leaf-first processing order to a workspace file and return its path.

(
    arguments: Dict[str, Any],
    store: SessionStore,
)

    return unmatched, leftover


def handle_save_module_tree(
    arguments: Dict[str, Any],
    store: SessionStore,
) -> str:
    """Persist the IDE agent's clustering result as the module tree."""
    session_id = arguments["session_id"]
    session = store.get(session_id)
    if session is None:
        return json.dumps({"error": f"Session {session_id} not found or expired."})

    module_tree = arguments["module_tree"]
    output_dir = session.output_dir
    known_ids = set(session.components.keys())
    candidate_ids = set(session.leaf_nodes)

    # Save both immutable snapshot and mutable working copy
    first_path = os.path.join(output_dir, FIRST_MODULE_TREE_FILENAME)
    working_path = os.path.join(output_dir, MODULE_TREE_FILENAME)

    os.makedirs(output_dir, exist_ok=True)

    with open(first_path, "w", encoding="utf-8") as f:
        json.dump(module_tree, f, indent=2, ensure_ascii=False)
    with open(working_path, "w", encoding="utf-8") as f:
        json.dump(module_tree, f, indent=2, ensure_ascii=False)

    # Cache in session
    session.module_tree = module_tree

    # Validate the tree so orphaned / stale ids surface instead of being
    # silently dropped from docs. Unmatched ids are checked against the full
    # index; leftovers only against the clustering candidate set (leaf nodes),
    # since the cluster prompt deliberately excludes non-essential components.
    unmatched_ids, leftover_ids = _validate_module_tree(
        module_tree, known_ids, candidate_ids
    )
    full_validation = {
        "unmatched_ids": unmatched_ids,
        "unmatched_count": len(unmatched_ids),
        "leftover_component_ids": leftover_ids,