Merge pull request #33 from Robunnin/feature-component-types

Added component types
8T4 · Jun 30, 2022 · ebbbdc6 · ebbbdc6
2 parents 3036ef6 + 9141edd
commit ebbbdc6
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 1 deletion.
diff --git a/src/C4Sharp/Models/Component.cs b/src/C4Sharp/Models/Component.cs
@@ -16,23 +16,37 @@ public record Component : Structure
 {
     public string? Technology { get; init; }
 
+    public ComponentType ComponentType { get; init; }
+
     public Component(string alias, string label): base(alias, label)
     {
+        ComponentType = ComponentType.None;
     }
 
     public Component(StructureIdentity alias, string label): base(alias, label)
     {
+        ComponentType = ComponentType.None;
     }    
 
     public Component(string alias, string label, string technology): this(alias, label)
     {
         Technology = technology;
+    }    
+
+    public Component(string alias, string label, ComponentType componentType, string technology): this(alias, label, technology)
+    {
+        ComponentType = componentType;
     }
 
     public Component(string alias, string label, string technology, string description): this(alias, label)
     {
         Technology = technology;
         Description = description;
+    }
+
+    public Component(string alias, string label, ComponentType componentType, string technology, string description): this(alias, label, technology, description)
+    {
+        ComponentType = componentType;
     }    
 }
 

diff --git a/src/C4Sharp/Models/ComponentType.cs b/src/C4Sharp/Models/ComponentType.cs
@@ -0,0 +1,18 @@
+using System.ComponentModel;
+
+namespace C4Sharp.Models;
+
+/// <summary>
+/// Component types
+/// </summary>
+public enum ComponentType
+{
+    [Description("Database")]
+    Database,
+
+    [Description("Queue")]
+    Queue,
+
+    [Description("")]
+    None
+}
diff --git a/src/C4Sharp/Models/Plantuml/Extensions/PlantumlStructure.cs b/src/C4Sharp/Models/Plantuml/Extensions/PlantumlStructure.cs
@@ -83,7 +83,13 @@ private static string ToPumlString(this EnterpriseBoundary boundary)
 
     private static string ToPumlString(this Component component)
     {
-        var procedureName = $"Component{GetExternalSuffix(component)}";
+        var externalSuffix = GetExternalSuffix(component);
+        var procedureName = component.ComponentType switch
+        {
+            ComponentType.Database => $"ComponentDb{externalSuffix}",
+            ComponentType.Queue => $"ComponentQueue{externalSuffix}",
+            _ => $"Component{externalSuffix}"
+        };
 
         return
             $"{procedureName}({component.Alias}, \"{component.Label}\", \"{component.Technology}\", \"{component.Description}\""