Definition層

moorestech · Nov 28, 2024 · 7976661 · 7976661
1 parent fea200c
commit 7976661
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/mooresmaster.Generator/Definitions/Definition.cs b/mooresmaster.Generator/Definitions/Definition.cs
@@ -18,9 +18,10 @@ public record InterfaceDefinition(string FileName, TypeName TypeName, Dictionary
     public TypeName TypeName = TypeName;
 }
 
-public record TypeDefinition(string FileName, TypeName TypeName, TypeName[] InheritList, Dictionary<string, PropertyDefinition> PropertyTable)
+public record TypeDefinition(string FileName, TypeName TypeName, TypeName[] InheritList, Dictionary<string, PropertyDefinition> PropertyTable, List<TypeName> ImplementationList)
 {
     public string FileName = FileName;
+    public List<TypeName> ImplementationList = ImplementationList;
 
     public TypeName[] InheritList = InheritList;
     public Dictionary<string, PropertyDefinition> PropertyTable = PropertyTable;

diff --git a/mooresmaster.Generator/Definitions/DefinitionGenerator.cs b/mooresmaster.Generator/Definitions/DefinitionGenerator.cs
@@ -96,7 +96,16 @@ public static Definition Generate(Semantics semantics, NameTable nameTable, Sche
                 fileName = $"mooresmaster.{firstInterface.Name}.g.cs";
             }
 
-            definition.TypeDefinitions.Add(new TypeDefinition(fileName, typeName, inheritList, propertyTable));
+            // implementation
+            var implementations = new List<TypeName>();
+            if (semantics.ClassInterfaceImplementationTable.TryGetValue(kvp.Key, out var interfaceIds))
+                foreach (var interfaceId in interfaceIds)
+                {
+                    var interfaceName = nameTable.TypeNames[interfaceId];
+                    implementations.Add(interfaceName);
+                }
+
+            definition.TypeDefinitions.Add(new TypeDefinition(fileName, typeName, inheritList, propertyTable, implementations));
         }
 
         return definition;