diff --git a/src/core/Elsa.Core/Helpers/TypeNameHelper.cs b/src/core/Elsa.Core/Helpers/TypeNameHelper.cs index 4daaa7052..184c08a5c 100644 --- a/src/core/Elsa.Core/Helpers/TypeNameHelper.cs +++ b/src/core/Elsa.Core/Helpers/TypeNameHelper.cs @@ -1,12 +1,20 @@ +using System.Reflection; +using Elsa.Attributes; + namespace Elsa.Helpers; public static class TypeNameHelper { - public static string? GenerateNamespace(Type activityType) => activityType.Namespace; + public static string? GenerateNamespace(Type activityType) + { + var activityAttr = activityType.GetCustomAttribute(); + return activityAttr?.Namespace ?? activityType.Namespace; + } public static string GenerateTypeName(Type type, string? ns) { - var typeName = type.Name; + var activityAttr = type.GetCustomAttribute(); + var typeName = activityAttr?.TypeName ?? type.Name; return ns != null ? $"{ns}.{typeName}" : typeName; }