Add optional form kind as parameter

UK-SBCoA · Jul 3, 2024 · c0d273b · c0d273b
1 parent 24de48b
commit c0d273b
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/UDS.Net.API/Extensions/EntityToDtoMapper.cs b/src/UDS.Net.API/Extensions/EntityToDtoMapper.cs
@@ -134,13 +134,16 @@ public static VisitDto ToDto(this Visit visit, string formKind)
             return dto;
         }
 
-        private static void SetBaseFormProperties(this FormDto dto, Form form)
+        private static void SetBaseFormProperties(this FormDto dto, Form form, string? formKind = "")
         {
             if (form != null)
             {
                 dto.Id = form.Id;
                 dto.VisitId = form.VisitId;
-                dto.Kind = form.GetType().Name;
+                if (String.IsNullOrWhiteSpace(formKind))
+                    dto.Kind = form.GetType().Name;
+                else
+                    dto.Kind = formKind;
                 dto.CreatedAt = form.CreatedAt;
                 dto.CreatedBy = form.CreatedBy;
                 dto.ModifiedBy = form.ModifiedBy;
@@ -160,7 +163,7 @@ private static void SetBaseFormProperties(this FormDto dto, Form form)
         public static FormDto ToSummaryDto(this Form form, string formKind)
         {
             var dto = new FormDto();
-            dto.SetBaseFormProperties(form);
+            dto.SetBaseFormProperties(form, formKind);
             return dto;
         }