Introduce a new node to data mapper call function

ballerina-platform · Jan 10, 2025 · f6c7034 · f6c7034
1 parent 9b6794b
commit f6c7034
Show file tree

Hide file tree

Showing 22 changed files with 254 additions and 260 deletions.
diff --git a/...ator-core/src/main/java/io/ballerina/flowmodelgenerator/core/AvailableNodesGenerator.java b/...ator-core/src/main/java/io/ballerina/flowmodelgenerator/core/AvailableNodesGenerator.java
@@ -147,7 +147,7 @@ private void setDefaultNodes() {
                 .node(NodeKind.VARIABLE)
                 .node(NodeKind.ASSIGN)
                 .node(function)
-                .node(NodeKind.DATA_MAPPER);
+                .node(NodeKind.DATA_MAPPER_CALL);
 
         this.rootBuilder.stepIn(Category.Name.CONTROL)
                 .node(NodeKind.IF)

diff --git a/...model-generator-core/src/main/java/io/ballerina/flowmodelgenerator/core/CodeAnalyzer.java b/...model-generator-core/src/main/java/io/ballerina/flowmodelgenerator/core/CodeAnalyzer.java
@@ -25,7 +25,6 @@
 import io.ballerina.compiler.api.symbols.FunctionSymbol;
 import io.ballerina.compiler.api.symbols.FunctionTypeSymbol;
 import io.ballerina.compiler.api.symbols.MethodSymbol;
-import io.ballerina.compiler.api.symbols.ParameterKind;
 import io.ballerina.compiler.api.symbols.ParameterSymbol;
 import io.ballerina.compiler.api.symbols.Symbol;
 import io.ballerina.compiler.api.symbols.SymbolKind;
@@ -1075,7 +1074,6 @@ public void visit(FunctionCallExpressionNode functionCallExpressionNode) {
 
         Optional<Documentation> documentation = functionSymbol.documentation();
         String description = documentation.flatMap(Documentation::description).orElse("");
-        SeparatedNodeList<FunctionArgumentNode> arguments = functionCallExpressionNode.arguments();
 
         String functionName = switch (nameReferenceNode.kind()) {
             case QUALIFIED_NAME_REFERENCE -> ((QualifiedNameReferenceNode) nameReferenceNode).identifier().text();
@@ -1089,52 +1087,30 @@ public void visit(FunctionCallExpressionNode functionCallExpressionNode) {
             startNode(NodeKind.FUNCTION_CALL, functionCallExpressionNode.parent());
         }
 
-//        if (dataMappings.containsKey(functionName)) {
-//            startNode(NodeKind.DATA_MAPPER, functionCallExpressionNode.parent()).properties()
-//                    .functionName(functionName)
-//                    .output(this.typedBindingPatternNode);
-//            Optional<List<ParameterSymbol>> funcParams = functionSymbol.typeDescriptor().params();
-//            if (funcParams.isPresent()) {
-//                List<ParameterSymbol> params = funcParams.get().stream()
-//                        .filter(p -> p.paramKind() != ParameterKind.INCLUDED_RECORD)
-//                        .toList();
-//                nodeBuilder.properties().inputs(arguments, params);
-//            }
-//            nodeBuilder.properties().view(dataMappings.get(functionName));
-//        } else {
-//            startNode(NodeKind.FUNCTION_CALL, functionCallExpressionNode.parent());
-            if (CommonUtils.isDefaultPackage(functionSymbol, moduleInfo)) {
-                functionSymbol.getLocation()
-                        .flatMap(location -> CommonUtil.findNode(functionSymbol,
-                                CommonUtils.getDocument(project, location).syntaxTree()))
-                        .ifPresent(node -> nodeBuilder.properties().view(node.lineRange()));
-            }
-            nodeBuilder
-                    .symbolInfo(functionSymbol)
-                    .metadata()
-                    .label(functionName)
-                    .description(description)
-                    .stepOut()
-                    .codedata()
-                    .symbol(functionName);
+        if (CommonUtils.isDefaultPackage(functionSymbol, moduleInfo)) {
+            functionSymbol.getLocation()
+                    .flatMap(location -> CommonUtil.findNode(functionSymbol,
+                            CommonUtils.getDocument(project, location).syntaxTree()))
+                    .ifPresent(node -> nodeBuilder.properties().view(node.lineRange()));
+        }
 
-            DatabaseManager dbManager = DatabaseManager.getInstance();
-            ModuleID id = functionSymbol.getModule().get().id();
-            Optional<FunctionResult> functionResult = dbManager.getAction(id.orgName(), id.moduleName(),
-                    functionSymbol.getName().get(), null, DatabaseManager.FunctionKind.FUNCTION);
+        DatabaseManager dbManager = DatabaseManager.getInstance();
+        ModuleID id = functionSymbol.getModule().get().id();
+        Optional<FunctionResult> functionResult = dbManager.getAction(id.orgName(), id.moduleName(),
+                functionSymbol.getName().get(), null, DatabaseManager.FunctionKind.FUNCTION);
 
-            final Map<String, Node> namedArgValueMap = new HashMap<>();
-            final Queue<Node> positionalArgs = new LinkedList<>();
-            calculateFunctionArgs(namedArgValueMap, positionalArgs, functionCallExpressionNode.arguments());
+        final Map<String, Node> namedArgValueMap = new HashMap<>();
+        final Queue<Node> positionalArgs = new LinkedList<>();
+        SeparatedNodeList<FunctionArgumentNode> arguments = functionCallExpressionNode.arguments();
+        calculateFunctionArgs(namedArgValueMap, positionalArgs, arguments);
 
-            if (functionResult.isPresent()) { // function details are indexed
-                analyzeAndHandleExprArgs(functionCallExpressionNode.arguments(), dbManager, functionResult.get(),
-                        functionSymbol, positionalArgs, namedArgValueMap);
-            } else {
-                handleFunctionCallActionCallsParams(functionCallExpressionNode.arguments(), functionSymbol);
-            }
-            handleCheckFlag(functionCallExpressionNode, SyntaxKind.CHECK_EXPRESSION, functionSymbol.typeDescriptor());
-//        }
+        if (functionResult.isPresent()) { // function details are indexed
+            analyzeAndHandleExprArgs(arguments, dbManager, functionResult.get(), functionSymbol, positionalArgs,
+                    namedArgValueMap);
+        } else {
+            handleFunctionCallActionCallsParams(arguments, functionSymbol);
+        }
+        handleCheckFlag(functionCallExpressionNode, SyntaxKind.CHECK_EXPRESSION, functionSymbol.typeDescriptor());
 
         nodeBuilder
                 .symbolInfo(functionSymbol)

diff --git a/.../src/main/java/io/ballerina/flowmodelgenerator/core/model/node/DataMapperCallBuilder.java b/.../src/main/java/io/ballerina/flowmodelgenerator/core/model/node/DataMapperCallBuilder.java
@@ -1,5 +1,5 @@
 /*
- *  Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com)
+ *  Copyright (c) 2025, WSO2 LLC. (http://www.wso2.com)
  *
  *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
@@ -20,10 +20,18 @@
 
 import io.ballerina.flowmodelgenerator.core.model.NodeKind;
 
+/**
+ * Represents the properties of a data mapper call node.
+ *
+ * @since 2.0.0
+ */
 public class DataMapperCallBuilder extends FunctionCall {
 
+    public static final String LABEL = "Map Data";
+
     @Override
     public void setConcreteConstData() {
+        metadata().label(LABEL);
         codedata().node(NodeKind.DATA_MAPPER_CALL);
     }
 }
diff --git a/...ow-model-generator-ls-extension/src/test/resources/available_nodes/config/connector1.json b/...ow-model-generator-ls-extension/src/test/resources/available_nodes/config/connector1.json
@@ -389,11 +389,10 @@
         },
         {
           "metadata": {
-            "label": "Data Mapper",
-            "description": "Map data from multiple variables to a record type"
+            "label": "Map Data"
           },
           "codedata": {
-            "node": "DATA_MAPPER"
+            "node": "DATA_MAPPER_CALL"
           },
           "enabled": true
         }

diff --git a/...ow-model-generator-ls-extension/src/test/resources/available_nodes/config/connector2.json b/...ow-model-generator-ls-extension/src/test/resources/available_nodes/config/connector2.json
@@ -4079,11 +4079,10 @@
         },
         {
           "metadata": {
-            "label": "Data Mapper",
-            "description": "Map data from multiple variables to a record type"
+            "label": "Map Data"
           },
           "codedata": {
-            "node": "DATA_MAPPER"
+            "node": "DATA_MAPPER_CALL"
           },
           "enabled": true
         }

diff --git a/...-model-generator-ls-extension/src/test/resources/available_nodes/config/force_assign.json b/...-model-generator-ls-extension/src/test/resources/available_nodes/config/force_assign.json
@@ -50,11 +50,10 @@
         },
         {
           "metadata": {
-            "label": "Data Mapper",
-            "description": "Map data from multiple variables to a record type"
+            "label": "Map Data"
           },
           "codedata": {
-            "node": "DATA_MAPPER"
+            "node": "DATA_MAPPER_CALL"
           },
           "enabled": true
         }

diff --git a/...flow-model-generator-ls-extension/src/test/resources/available_nodes/config/foreach1.json b/...flow-model-generator-ls-extension/src/test/resources/available_nodes/config/foreach1.json
@@ -50,11 +50,10 @@
         },
         {
           "metadata": {
-            "label": "Data Mapper",
-            "description": "Map data from multiple variables to a record type"
+            "label": "Map Data"
           },
           "codedata": {
-            "node": "DATA_MAPPER"
+            "node": "DATA_MAPPER_CALL"
           },
           "enabled": true
         }

diff --git a/...low-model-generator-ls-extension/src/test/resources/available_nodes/config/function1.json b/...low-model-generator-ls-extension/src/test/resources/available_nodes/config/function1.json
@@ -50,11 +50,10 @@
         },
         {
           "metadata": {
-            "label": "Data Mapper",
-            "description": "Map data from multiple variables to a record type"
+            "label": "Map Data"
           },
           "codedata": {
-            "node": "DATA_MAPPER"
+            "node": "DATA_MAPPER_CALL"
           },
           "enabled": true
         }

diff --git a/...low-model-generator-ls-extension/src/test/resources/available_nodes/config/function2.json b/...low-model-generator-ls-extension/src/test/resources/available_nodes/config/function2.json
@@ -50,11 +50,10 @@
         },
         {
           "metadata": {
-            "label": "Data Mapper",
-            "description": "Map data from multiple variables to a record type"
+            "label": "Map Data"
           },
           "codedata": {
-            "node": "DATA_MAPPER"
+            "node": "DATA_MAPPER_CALL"
           },
           "enabled": true
         }

diff --git a/...low-model-generator-ls-extension/src/test/resources/available_nodes/config/function3.json b/...low-model-generator-ls-extension/src/test/resources/available_nodes/config/function3.json
@@ -50,11 +50,10 @@
         },
         {
           "metadata": {
-            "label": "Data Mapper",
-            "description": "Map data from multiple variables to a record type"
+            "label": "Map Data"
           },
           "codedata": {
-            "node": "DATA_MAPPER"
+            "node": "DATA_MAPPER_CALL"
           },
           "enabled": true
         }

diff --git a/...es/flow-model-generator-ls-extension/src/test/resources/available_nodes/config/lock1.json b/...es/flow-model-generator-ls-extension/src/test/resources/available_nodes/config/lock1.json
@@ -50,11 +50,10 @@
         },
         {
           "metadata": {
-            "label": "Data Mapper",
-            "description": "Map data from multiple variables to a record type"
+            "label": "Map Data"
           },
           "codedata": {
-            "node": "DATA_MAPPER"
+            "node": "DATA_MAPPER_CALL"
           },
           "enabled": true
         }

diff --git a/...s/flow-model-generator-ls-extension/src/test/resources/available_nodes/config/match1.json b/...s/flow-model-generator-ls-extension/src/test/resources/available_nodes/config/match1.json
@@ -50,11 +50,10 @@
         },
         {
           "metadata": {
-            "label": "Data Mapper",
-            "description": "Map data from multiple variables to a record type"
+            "label": "Map Data"
           },
           "codedata": {
-            "node": "DATA_MAPPER"
+            "node": "DATA_MAPPER_CALL"
           },
           "enabled": true
         }

diff --git a/...s/flow-model-generator-ls-extension/src/test/resources/available_nodes/config/match2.json b/...s/flow-model-generator-ls-extension/src/test/resources/available_nodes/config/match2.json
@@ -50,11 +50,10 @@
         },
         {
           "metadata": {
-            "label": "Data Mapper",
-            "description": "Map data from multiple variables to a record type"
+            "label": "Map Data"
           },
           "codedata": {
-            "node": "DATA_MAPPER"
+            "node": "DATA_MAPPER_CALL"
           },
           "enabled": true
         }

diff --git a/.../flow-model-generator-ls-extension/src/test/resources/available_nodes/config/method1.json b/.../flow-model-generator-ls-extension/src/test/resources/available_nodes/config/method1.json
@@ -50,11 +50,10 @@
         },
         {
           "metadata": {
-            "label": "Data Mapper",
-            "description": "Map data from multiple variables to a record type"
+            "label": "Map Data"
           },
           "codedata": {
-            "node": "DATA_MAPPER"
+            "node": "DATA_MAPPER_CALL"
           },
           "enabled": true
         }

diff --git a/...del-generator-ls-extension/src/test/resources/available_nodes/config/on_fail_clause1.json b/...del-generator-ls-extension/src/test/resources/available_nodes/config/on_fail_clause1.json
@@ -50,11 +50,10 @@
         },
         {
           "metadata": {
-            "label": "Data Mapper",
-            "description": "Map data from multiple variables to a record type"
+            "label": "Map Data"
           },
           "codedata": {
-            "node": "DATA_MAPPER"
+            "node": "DATA_MAPPER_CALL"
           },
           "enabled": true
         }

diff --git a/...-generator-ls-extension/src/test/resources/available_nodes/config/resource_connector.json b/...-generator-ls-extension/src/test/resources/available_nodes/config/resource_connector.json
@@ -17215,11 +17215,10 @@
         },
         {
           "metadata": {
-            "label": "Data Mapper",
-            "description": "Map data from multiple variables to a record type"
+            "label": "Map Data"
           },
           "codedata": {
-            "node": "DATA_MAPPER"
+            "node": "DATA_MAPPER_CALL"
           },
           "enabled": true
         }

diff --git a/...-model-generator-ls-extension/src/test/resources/available_nodes/config/transaction1.json b/...-model-generator-ls-extension/src/test/resources/available_nodes/config/transaction1.json
@@ -50,11 +50,10 @@
         },
         {
           "metadata": {
-            "label": "Data Mapper",
-            "description": "Map data from multiple variables to a record type"
+            "label": "Map Data"
           },
           "codedata": {
-            "node": "DATA_MAPPER"
+            "node": "DATA_MAPPER_CALL"
           },
           "enabled": true
         }

diff --git a/...-model-generator-ls-extension/src/test/resources/available_nodes/config/transaction2.json b/...-model-generator-ls-extension/src/test/resources/available_nodes/config/transaction2.json
@@ -50,11 +50,10 @@
         },
         {
           "metadata": {
-            "label": "Data Mapper",
-            "description": "Map data from multiple variables to a record type"
+            "label": "Map Data"
           },
           "codedata": {
-            "node": "DATA_MAPPER"
+            "node": "DATA_MAPPER_CALL"
           },
           "enabled": true
         }

diff --git a/...s/flow-model-generator-ls-extension/src/test/resources/available_nodes/config/while1.json b/...s/flow-model-generator-ls-extension/src/test/resources/available_nodes/config/while1.json
@@ -50,11 +50,10 @@
         },
         {
           "metadata": {
-            "label": "Data Mapper",
-            "description": "Map data from multiple variables to a record type"
+            "label": "Map Data"
           },
           "codedata": {
-            "node": "DATA_MAPPER"
+            "node": "DATA_MAPPER_CALL"
           },
           "enabled": true
         }