Tidy up WingUtils::FormatName

Plugins/UEWingman/Source/UEWingman/Private/WingUtils.cpp

@@ -60,16 +60,27 @@ void WingUtils::SanitizeNameInPlace(FString &Name)
 	for (int32 Src = 0; Src < Name.Len(); Src++)
 	{
 		TCHAR c = Name[Src];
-		if (c <= 0x20 || c == '_' || c == 0x7F) continue;
+		if (c < 0x20 || c == 0x7F) continue;
-		if (c >= 0x21 && c <= 0x7E && !FChar::IsAlnum(c))
+		if ((c == ' ') || (c == ',') || (c == ':')) c = '_';
-			Name[Dst++] = '_';
-		else
 		Name[Dst++] = c;
 	}
+	if (Dst == 0) Name[Dst++] = '_';
 	Name.LeftInline(Dst);
-	if (Name.IsEmpty()) Name = TEXT("_");
 }
 
+FString WingUtils::SanitizeName(const FString &Name)
+{
+	FString Result = Name;
+	SanitizeNameInPlace(Result);
+	return Result;
+}
+
+FString WingUtils::SanitizeName(FName Name)
+{
+	FString Result = Name.ToString();
+	SanitizeNameInPlace(Result);
+	return Result;
+}
 
 FString WingUtils::FormatName(const UWorld *World)
 {
@@ -83,52 +94,43 @@ FString WingUtils::FormatName(const UBlueprint *BP)
 
 FString WingUtils::FormatName(const UActorComponent *C)
 {
-	return C->GetName();
+	return SanitizeName(C->GetName());
 }
 
 FString WingUtils::FormatName(const USCS_Node *Node)
 {
-	return Node->GetVariableName().ToString();
+	return SanitizeName(Node->GetVariableName());
 }
 
 FString WingUtils::FormatName(const UEdGraph *Graph)
 {
-	FString Name = Graph->GetName();
+	
-	SanitizeNameInPlace(Name);
+	return SanitizeName(Graph->GetName());
-	return Name;
 }
 
 FString WingUtils::FormatName(const UEdGraphNode* Node)
 {
-	return Node->GetName();
+	return SanitizeName(Node->GetName());
 }
 
 FString WingUtils::FormatName(const UEdGraphPin *Pin)
 {
-	FString Name = Pin->PinName.ToString();
+	return SanitizeName(Pin->GetName());
-	SanitizeNameInPlace(Name);
-	return Name;
 }
 
 FString WingUtils::FormatName(const FMemberReference &Ref)
 {
-	FString Name = Ref.GetMemberName().ToString();
+	return SanitizeName(Ref.GetMemberName());
-	SanitizeNameInPlace(Name);
-	return Name;
 }
 
 FString WingUtils::FormatName(const FBPVariableDescription &Var)
 {
-	FString Name = Var.VarName.ToString();
+	return SanitizeName(Var.VarName);
-	SanitizeNameInPlace(Name);
-	return Name;
 }
 
 FString WingUtils::FormatName(const UStruct *Struct)
 {
-	FString Name = Struct->GetName();
+	return SanitizeName(Struct->GetName());
-	SanitizeNameInPlace(Name);
-	return Name;
 }
 
 FString WingUtils::FormatName(const UMaterial *Material)
@@ -148,9 +150,7 @@ FString WingUtils::FormatName(const UMaterialFunction *MaterialFunction)
 
 FString WingUtils::FormatName(const UMaterialExpression *Expression)
 {
-	FString Name = Expression->GetName();
+	return SanitizeName(Expression->GetName());
-	SanitizeNameInPlace(Name);
-	return Name;
 }
 
 FString WingUtils::FormatName(const UStaticMesh *Mesh)
@@ -180,21 +180,17 @@ FString WingUtils::FormatName(const UTexture *Texture)
 
 FString WingUtils::FormatName(const UScriptStruct *Struct)
 {
-	FString Name = Struct->GetName();
+	return SanitizeName(Struct->GetName());;
-	SanitizeNameInPlace(Name);
-	return Name;
 }
 
 FString WingUtils::FormatName(const UEnum *Enum)
 {
-	FString Name = Enum->GetName();
+	return SanitizeName(Enum->GetName());
-	SanitizeNameInPlace(Name);
-	return Name;
 }
 
 FString WingUtils::FormatName(const FProperty *Prop)
 {
-	return Prop->GetName();
+	return SanitizeName(Prop->GetName());
 }
 
 // ============================================================

Plugins/UEWingman/Source/UEWingman/Public/WingUtils.h

@@ -95,6 +95,8 @@ public:
 	////////////////////////////////////////////////////////
 
 	static void SanitizeNameInPlace(FString& Name);
+	static FString SanitizeName(const FString& Name);
+	static FString SanitizeName(FName Name);
 	static FString FormatNodeTitle(const UEdGraphNode *Node);
 
 	// ----- Enum helpers -----