talezero/integration
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

More refactoring in MCP

Browse Source
This commit is contained in:
jyelon
2026-03-07 03:24:13 -05:00
parent ef0a0cd2cd
commit 9d4a8c99bb
8 changed files with 568 additions and 634 deletions
Download Patch File Download Diff File Expand all files Collapse all files

69
Plugins/BlueprintMCP/Source/BlueprintMCP/Private/MCPUtils.cpp
View File

@@ -1188,53 +1188,54 @@ FString MCPUtils::NodeSpawnerFullName(UBlueprintNodeSpawner* Spawner)
return Category + TEXT("|") + MenuName;
}
TArray<UBlueprintNodeSpawner*> MCPUtils::AllNodeSpawners()
TArray<UBlueprintNodeSpawner*> MCPUtils::SearchNodeSpawners(const FString& Query, int32 MaxResults, bool ExactMatch, UEdGraph* GraphFilter)
{
FString QueryLower = Query.ToLower();
TArray<UBlueprintNodeSpawner*> Result;
for (const auto& Pair : FBlueprintActionDatabase::Get().GetAllActions())
{
for (UBlueprintNodeSpawner* Spawner : Pair.Value)
{
if (!Spawner) continue;
if (Spawner->PrimeDefaultUiSpec().MenuName.IsEmpty()) continue;
// Filter by graph compatibility if a graph was provided
if (GraphFilter && Spawner->NodeClass)
{
UEdGraphNode* NodeCDO = CastChecked<UEdGraphNode>(Spawner->NodeClass->ClassDefaultObject);
if (!NodeCDO->IsCompatibleWithGraph(GraphFilter))
{
continue;
}
}
FString FullName = NodeSpawnerFullName(Spawner);
if (ExactMatch)
{
if (FullName.ToLower() != QueryLower)
{
continue;
}
}
else
{
FString Keywords = Spawner->PrimeDefaultUiSpec().Keywords.ToString();
if (!FullName.ToLower().Contains(QueryLower)
&& !Keywords.ToLower().Contains(QueryLower))
{
continue;
}
}
Result.Add(Spawner);
}
}
return Result;
}
TArray<UBlueprintNodeSpawner*> MCPUtils::SearchNodeSpawners(const FString& Query, int32 MaxResults, bool ExactMatch)
{
FString QueryLower = Query.ToLower();
TArray<UBlueprintNodeSpawner*> Result;
for (UBlueprintNodeSpawner* Spawner : AllNodeSpawners())
{
FString FullName = NodeSpawnerFullName(Spawner);
if (ExactMatch)
{
if (FullName.ToLower() != QueryLower)
if ((MaxResults > 0) && (Result.Num() >= MaxResults))
{
continue;
break;
}
}
else
{
FString Keywords = Spawner->PrimeDefaultUiSpec().Keywords.ToString();
if (!FullName.ToLower().Contains(QueryLower)
&& !Keywords.ToLower().Contains(QueryLower))
{
continue;
}
}
Result.Add(Spawner);
if ((MaxResults > 0) && (Result.Num() >= MaxResults))
{
break;
}
}
return Result;
}