Merge pull request #144 from modelix/feature/model-api-gen-ts-refs

fix(model-api-gen): references where not generated for TypeScript

Author: slisson

model-api-gen/src/main/kotlin/org/modelix/metamodel/generator/TypescriptMMGenerator.kt

@@ -4,6 +4,7 @@ import com.squareup.kotlinpoet.ClassName
 import org.modelix.model.data.LanguageData
 import org.modelix.model.data.Primitive
 import org.modelix.model.data.PrimitivePropertyType
+import java.nio.file.Files
 import java.nio.file.Path
 import kotlin.io.path.writeText
 
@@ -25,6 +26,7 @@ class TypescriptMMGenerator(val outputDir: Path, val nameConfig: NameConfig = Na
     }
 
     internal fun generate(languages: ProcessedLanguageSet) {
+        Files.createDirectories(outputDir)
         for (language in languages.getLanguages()) {
             // TODO delete old files from previous generation
             outputDir
@@ -62,7 +64,8 @@ class TypescriptMMGenerator(val outputDir: Path, val nameConfig: NameConfig = Na
                 INodeJS,
                 ITypedNode, 
                 SingleChildAccessor,
-                TypedNode
+                TypedNode,
+                LanguageRegistry
             } from "@modelix/ts-model-api";
             
             ${language.languageDependencies().joinToString("\n") {
@@ -129,9 +132,19 @@ class TypescriptMMGenerator(val outputDir: Path, val nameConfig: NameConfig = Na
                         }
                     } else defaultPropertyText
                 }
-                is ProcessedReferenceLink -> """
-                    
+                is ProcessedReferenceLink -> {
+                    val typeRef = feature.type.resolved
+                    val languagePrefix = typeRef.languagePrefix(concept.language)
+                    val entityType = "$languagePrefix${typeRef.nodeWrapperInterfaceName()}"
+                    """
+                    public set ${feature.generatedName}(value: $entityType | undefined) {
+                        this.node.setReferenceTargetNode("${feature.originalName}", value.unwrap());
+                    }
+                    public get ${feature.generatedName}: $entityType | undefined {
+                        return LanguageRegistry.INSTANCE.wrapNode(this.node.getReferenceTargetNode("${feature.originalName}"));
+                    }
                 """.trimIndent()
+                }
                 is ProcessedChildLink -> {
                     val accessorClassName = if (feature.multiple) "ChildListAccessor" else "SingleChildAccessor"
                     val typeRef = feature.type.resolved
@@ -167,9 +180,15 @@ class TypescriptMMGenerator(val outputDir: Path, val nameConfig: NameConfig = Na
                         }
                     } else defaultPropertyText
                 }
-                is ProcessedReferenceLink -> """
-                    
-                """.trimIndent()
+                is ProcessedReferenceLink -> {
+                    val typeRef = feature.type.resolved
+                    val languagePrefix = typeRef.languagePrefix(concept.language)
+                    val entityType = "$languagePrefix${typeRef.nodeWrapperInterfaceName()}"
+                        """
+                        set ${feature.generatedName}(value: $entityType | undefined);
+                        get ${feature.generatedName}: $entityType | undefined;
+                    """.trimIndent()
+                }
                 is ProcessedChildLink -> {
                     val accessorClassName = if (feature.multiple) "ChildListAccessor" else "SingleChildAccessor"
                     """

model-api-gen/src/test/kotlin/org/modelix/metamodel/generator/KotlinGeneratorTest.kt

@@ -44,6 +44,41 @@ class KotlinGeneratorTest {
         val outputDir = File("build/test-generator-output").toPath()
         MetaModelGenerator(outputDir).generate(LanguageSet(listOf(language)).process())
     }
+    @Test
+    fun test_ts() {
+        val input = """
+            name: org.modelix.entities
+            concepts:
+            - name: Entity
+              properties:
+              - name: name
+              children:
+              - name: properties
+                type: Property
+                multiple: true
+                optional: true
+            - name: Property
+              children:
+              - name: type
+                type: Type
+                optional: false
+            - name: Type
+            - name: EntityType
+              extends:
+              - Type
+              references:
+              - name: entity
+                type: Entity
+                optional: false
+            enums: []
+        """.trimIndent()
+
+        val language = Yaml.default.decodeFromString<LanguageData>(input)
+        //val outputDir = File(".").toPath().resolve("build").resolve("test-generator-output")
+        val outputDir = File("build/test-generator-output").toPath()
+        TypescriptMMGenerator(outputDir).generate(LanguageSet(listOf(language)).process())
+    }
+
 
     @OptIn(ExperimentalPathApi::class)
     @Test