From da01339caea4b659fc7034deed7418f82ab61424 Mon Sep 17 00:00:00 2001
From: rashmibharambe <93034034+rashmibharambe@users.noreply.github.com>
Date: Sun, 8 Oct 2023 11:13:13 +0530
Subject: [PATCH] Improve client tool constant generation code

---
 .../utils/ClientSampleSyntaxTreeUtils.java    | 43 ++++---------------
 1 file changed, 8 insertions(+), 35 deletions(-)

diff --git a/protoc-cli/src/main/java/io/ballerina/protoc/builder/syntaxtree/utils/ClientSampleSyntaxTreeUtils.java b/protoc-cli/src/main/java/io/ballerina/protoc/builder/syntaxtree/utils/ClientSampleSyntaxTreeUtils.java
index 73a4c08..8541ca5 100644
--- a/protoc-cli/src/main/java/io/ballerina/protoc/builder/syntaxtree/utils/ClientSampleSyntaxTreeUtils.java
+++ b/protoc-cli/src/main/java/io/ballerina/protoc/builder/syntaxtree/utils/ClientSampleSyntaxTreeUtils.java
@@ -304,41 +304,14 @@ private static VariableDeclarationNode getInputDeclarationStatement(Method metho
                         method.getInputType() + " ")),
                 getCaptureBindingPatternNode(getRequestName(method.getMethodName())));
         ExpressionNode node = null;
-        switch (method.getInputType()) {
-            case "int":
-            case "float":
-            case "decimal":
-                node = getNumericLiteralNode(1);
-                break;
-            case "boolean":
-                node = getBooleanLiteralNode(true);
-                break;
-            case "string":
-                node = getStringLiteralNode("ballerina");
-                break;
-            case "byte[]":
-                node = getByteArrayLiteralNode("[72,101,108,108,111]");
-                break;
-            case "time:Utc":
-                node = getTupleLiteralNode("[1659688553,0.310073000d]");
-                break;
-            case "time:Seconds":
-                node = getDecimalLiteralNode("0.310073000d");
-                break;
-            case "map<anydata>":
-                node = createBasicLiteralNode(SyntaxKind.MAP_TYPE_DESC, "{message: \"Hello Ballerina\"}");
-                break;
-            case "'any:Any":
-                node = getCheckExpressionNode(getFunctionCallExpressionNode("'any", "pack", "\"ballerina\""));
-                break;
-            default:
-                if (msgMap.containsKey(method.getInputType())) {
-                    Message msg = msgMap.get(method.getInputType());
-                    node = NodeFactory.createMappingConstructorExpressionNode(
-                            SyntaxTreeConstants.SYNTAX_TREE_OPEN_BRACE,
-                            NodeFactory.createSeparatedNodeList(getFieldNodes(msg, msgMap)),
-                            SyntaxTreeConstants.SYNTAX_TREE_CLOSE_BRACE);
-                }
+        if(INPUT_TYPE_EXPR_NODES.containsKey(method.getInputType())) {
+            node = INPUT_TYPE_EXPR_NODES.get(method.getInputType());
+        } else if (msgMap.containsKey(method.getInputType())) {
+            Message msg = msgMap.get(method.getInputType());
+            node = NodeFactory.createMappingConstructorExpressionNode(
+                    SyntaxTreeConstants.SYNTAX_TREE_OPEN_BRACE,
+                    NodeFactory.createSeparatedNodeList(getFieldNodes(msg, msgMap)),
+                    SyntaxTreeConstants.SYNTAX_TREE_CLOSE_BRACE);
         }
         VariableDeclaration valueVariable = new VariableDeclaration(bindingPatternNode, node);
         return valueVariable.getVariableDeclarationNode();