From 4a3dbcb2456b805799d1b05fcd055d48b75646b2 Mon Sep 17 00:00:00 2001
From: Sipke Schoorstra <sipkeschoorstra@outlook.com>
Date: Wed, 27 Apr 2022 19:49:25 +0200
Subject: [PATCH] Fix TypeNameHelper

---
 src/core/Elsa.Core/Helpers/TypeNameHelper.cs | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

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<ActivityAttribute>();
+        return activityAttr?.Namespace ?? activityType.Namespace;
+    }
 
     public static string GenerateTypeName(Type type, string? ns)
     {
-        var typeName = type.Name;
+        var activityAttr = type.GetCustomAttribute<ActivityAttribute>();
+        var typeName = activityAttr?.TypeName ?? type.Name;
         return ns != null ? $"{ns}.{typeName}" : typeName;
     }