fix: add lost super classes

unit-mesh · Dec 19, 2023 · ebcd6d4 · ebcd6d4
1 parent aa49873
commit ebcd6d4
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/ext/CodeDataStructUtil.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/ext/CodeDataStructUtil.kt
@@ -6,7 +6,14 @@ import chapi.domain.core.CodePosition
 fun CodeDataStruct.toUml(): String {
     val output = StringBuilder()
 
-    output.append("class $NodeName {\n")
+    val superClass = Implements + Extend
+    val superClasses = if (superClass.isNotEmpty()) {
+        ":" + superClass.joinToString(", ") + " "
+    } else {
+        ""
+    }
+
+    output.append("class $NodeName $superClasses{\n")
     Fields.forEach {
         output.append("   ${it.TypeKey}: ${it.TypeType}\n")
     }