integration/Plugins/BlueprintMCP/Source/BlueprintMCP/Private/MCPNotifier.cpp

#include "MCPNotifier.h"
#include "EdGraph/EdGraphNode.h"
#include "EdGraph/EdGraph.h"
#include "Engine/Blueprint.h"
#include "Materials/Material.h"
#include "Kismet2/BlueprintEditorUtils.h"
#include "MaterialEditingLibrary.h"

void MCPNotifier::AddTouchedObject(UObject* Obj)
{
	if (!Obj) return;
	bool bAlreadyInSet = false;
	TouchedSet.Add(Obj, &bAlreadyInSet);
	if (bAlreadyInSet) return;
	TouchedArray.Add(Obj);
	Obj->PreEditChange(nullptr);
}

void MCPNotifier::SendNotifications()
{
	TSet<UEdGraphNode*> Nodes;
	TSet<UEdGraph*> Graphs;
	TSet<UMaterial*> Materials;
	TSet<UBlueprint*> Blueprints;
	for (int32 i = TouchedArray.Num() - 1; i >= 0; --i)
	{
		UObject* Obj = TouchedArray[i];
		Obj->PostEditChange();
		Obj->MarkPackageDirty();

		if (UEdGraphNode* Node = ::Cast<UEdGraphNode>(Obj))
			Nodes.Add(Node);

		if (UEdGraph* Graph = ::Cast<UEdGraph>(Obj))
			Graphs.Add(Graph);

		if (UBlueprint* BP = ::Cast<UBlueprint>(Obj))
			Blueprints.Add(BP);

		if (UMaterialInterface* MatIface = ::Cast<UMaterialInterface>(Obj))
			if (UMaterial* BaseMat = MatIface->GetMaterial())
				Materials.Add(BaseMat);
	}
	for (UEdGraphNode* Node : Nodes)
		Node->ReconstructNode();
	for (UEdGraph* Graph : Graphs)
		Graph->NotifyGraphChanged();
	for (UMaterial *Material : Materials)
		UMaterialEditingLibrary::RebuildMaterialInstanceEditors(Material);
	for (UBlueprint *Blueprint : Blueprints)
		FBlueprintEditorUtils::MarkBlueprintAsStructurallyModified(Blueprint);

	if (GEditor)
		GEditor->RedrawAllViewports();

	TouchedSet.Empty();
	TouchedArray.Empty();
}