feat(map): gets type from model file

Signed-off-by: Jonathan Casey <[email protected]>
accordproject · Oct 21, 2023 · 41e2d3f · 41e2d3f
1 parent 8cff2ac
commit 41e2d3f
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 7 deletions.
diff --git a/lib/codegen/fromcto/csharp/csharpvisitor.js b/lib/codegen/fromcto/csharp/csharpvisitor.js
@@ -314,8 +314,8 @@ class CSharpVisitor {
     writeField(field, parameters, externalFieldType, isOptional = false) {
 
         // write Map field
-        if ( Object.keys(field).length > 0 && ModelUtil.isMap?.(field)) {
-            const decl = field.getModelFile().getAllDeclarations().find(d => d.name === field.ast.type.name);
+        if (Object.keys(field).length > 0 && ModelUtil.isMap?.(field)) {
+            const decl = field.getModelFile().getType(field.getType());
             parameters.fileWriter.writeLine(1, `public Dictionary<string, ${this.toCSharpType(decl.getValue().getType())}> ${field.getName()} { get; set; }`);
             return null;
         }

diff --git a/lib/codegen/fromcto/typescript/typescriptvisitor.js b/lib/codegen/fromcto/typescript/typescriptvisitor.js
@@ -288,9 +288,7 @@ class TypescriptVisitor {
         const isPrimitive = ModelUtil.isPrimitiveType(mapDeclaration.getValue().getType());
 
         if (isScalar) {
-            const scalar = mapDeclaration.getModelFile()
-                .getAllDeclarations()
-                .find(decl => decl.name === mapDeclaration.getValue().getType());
+            const scalar = mapDeclaration.getModelFile(mapDeclaration.getValue().getType());
             const scalarType = scalar.getType();
             valueType = this.toTsType(scalarType, false, false);
         } else if (isPrimitive) {

diff --git a/lib/common/graph.js b/lib/common/graph.js
@@ -203,15 +203,15 @@ class ConcertoGraphVisitor extends DiagramVisitor {
         let valueEdge   = mapDeclaration.getValue().getType();
 
         if(!ModelUtil.isPrimitiveType(keyEdge)) {
-            keyEdge = modelFile.getAllDeclarations().find(d => d.name === mapDeclaration.getKey().getType()).getFullyQualifiedName();
+            keyEdge = modelFile.getType(mapDeclaration.getKey().getType()).getFullyQualifiedName();
         }
 
         if(!ModelUtil.isPrimitiveType(valueEdge)) {
             // handle imported types
             if (!modelFile.isLocalType(mapDeclaration.getValue().getType())) {
                 valueEdge = modelFile.resolveImport(mapDeclaration.getValue().getType());
             } else {
-                valueEdge = modelFile.getAllDeclarations().find(d => d.name === mapDeclaration.getValue().getType()).getFullyQualifiedName();
+                valueEdge = modelFile.getType(mapDeclaration.getValue().getType()).getFullyQualifiedName();
             }
         }