Add LS API support for function, type discovery for workspaces

architecture-model-generator/modules/architecture-model-generator-ls-extension/src/main/java/io/ballerina/designmodelgenerator/extension/ProjectInfoBuilder.java

@@ -18,8 +18,15 @@
 
 package io.ballerina.designmodelgenerator.extension;
 
+import io.ballerina.compiler.syntax.tree.ImportDeclarationNode;
+import io.ballerina.compiler.syntax.tree.ModulePartNode;
+import io.ballerina.compiler.syntax.tree.SyntaxTree;
+import io.ballerina.compiler.syntax.tree.Token;
 import io.ballerina.designmodelgenerator.extension.response.ProjectInfoResponse;
 import io.ballerina.projects.BallerinaToml;
+import io.ballerina.projects.Document;
+import io.ballerina.projects.DocumentId;
+import io.ballerina.projects.Module;
 import io.ballerina.projects.Package;
 import io.ballerina.projects.Project;
 import io.ballerina.projects.ProjectKind;
@@ -105,6 +112,7 @@ public void populate() {
             // Set org and version for build projects
             response.setOrg(currentPackage.packageOrg().value());
             response.setVersion(currentPackage.packageVersion().value().toString());
+            response.setIsLibrary(isLibraryProject(project));
         }
     }
 
@@ -141,6 +149,42 @@ private List<ProjectInfoResponse> extractChildProjects() {
         return children;
     }
 
+    private static final String LIB_BAL = "lib.bal";
+    private static final String BI_VALIDATOR_ORG = "wso2";
+    private static final String BI_VALIDATOR_MODULE = "bi.validator";
+
+    /**
+     * Checks if the given project is a library project by looking for a lib.bal file containing the
+     * {@code import wso2/bi.validator as _;} import declaration.
+     *
+     * @param project The project to check
+     * @return true if the project is a library project
+     */
+    private static boolean isLibraryProject(Project project) {
+        Module defaultModule = project.currentPackage().getDefaultModule();
+        for (DocumentId documentId : defaultModule.documentIds()) {
+            Document document = defaultModule.document(documentId);
+            if (!LIB_BAL.equals(document.name())) {
+                continue;
+            }
+            SyntaxTree syntaxTree = document.syntaxTree();
+            ModulePartNode modulePartNode = syntaxTree.rootNode();
+            for (ImportDeclarationNode importDecl : modulePartNode.imports()) {
+                if (importDecl.orgName().isEmpty()) {
+                    continue;
+                }
+                String orgName = importDecl.orgName().get().orgName().text();
+                String moduleName = importDecl.moduleName().stream()
+                        .map(Token::text)
+                        .collect(Collectors.joining("."));
+                if (BI_VALIDATOR_ORG.equals(orgName) && BI_VALIDATOR_MODULE.equals(moduleName)) {
+                    return true;
+                }
+            }
+        }
+        return false;
+    }
+
     /**
      * Formats a package/project name to title case for display purposes. Converts snake_case and camelCase to Title
      * Case.

architecture-model-generator/modules/architecture-model-generator-ls-extension/src/main/java/io/ballerina/designmodelgenerator/extension/response/ProjectInfoResponse.java

@@ -34,6 +34,7 @@ public class ProjectInfoResponse extends AbstractResponse {
     private String org;            // Package organization
     private String version;        // Package version
     private List<ProjectInfoResponse> children;  // Only populated for workspace projects (recursive)
+    private Boolean isLibrary;                     // Whether the project is a library project
 
     public String getProjectKind() {
         return projectKind;
@@ -90,4 +91,12 @@ public List<ProjectInfoResponse> getChildren() {
     public void setChildren(List<ProjectInfoResponse> children) {
         this.children = children;
     }
+
+    public Boolean getIsLibrary() {
+        return isLibrary;
+    }
+
+    public void setIsLibrary(Boolean isLibrary) {
+        this.isLibrary = isLibrary;
+    }
 }

architecture-model-generator/modules/architecture-model-generator-ls-extension/src/test/resources/project_info/config/build_project.json

@@ -7,6 +7,7 @@
     "title": "Simple Service",
     "projectPath": "proj",
     "org": "test",
-    "version": "0.1.0"
+    "version": "0.1.0",
+    "isLibrary": false
   }
 }

architecture-model-generator/modules/architecture-model-generator-ls-extension/src/test/resources/project_info/config/workspace_project.json

@@ -13,23 +13,26 @@
         "title": "Common",
         "projectPath": "workspace_proj/common",
         "org": "wso2",
-        "version": "0.1.0"
+        "version": "0.1.0",
+        "isLibrary": true
       },
       {
         "projectKind": "BUILD_PROJECT",
         "name": "api_gateway",
         "title": "API Gateway",
         "projectPath": "workspace_proj/api-gateway",
         "org": "wso2",
-        "version": "0.1.0"
+        "version": "0.1.0",
+        "isLibrary": false
       },
       {
         "projectKind": "BUILD_PROJECT",
         "name": "user_service",
         "title": "User Service",
         "projectPath": "workspace_proj/user-service",
         "org": "wso2",
-        "version": "0.1.0"
+        "version": "0.1.0",
+        "isLibrary": false
       }
     ]
   }

architecture-model-generator/modules/architecture-model-generator-ls-extension/src/test/resources/project_info/source/workspace_proj/common/lib.bal

@@ -0,0 +1 @@
+import wso2/bi.validator as _;

flow-model-generator/modules/flow-model-generator-core/src/main/java/io/ballerina/flowmodelgenerator/core/model/Category.java

@@ -51,8 +51,10 @@ public enum Name {
         CONCURRENCY("Concurrency", "Concurrency nodes", null),
         ERROR_HANDLING("Error Handling", "Handle errors that occur during execution", null),
         DATA("Data", "Data nodes are used to create, read, update, delete, and transform data", null),
-        CURRENT_INTEGRATION("Current Integration", "Functions defined within the current integration",
-                List.of("Project", "Local", "Function")),
+        CURRENT_INTEGRATION("Current Integration", "Components defined within the current integration",
+                List.of("Project", "Local", "Function", "Types", "Data Mapper")),
+        CURRENT_WORKSPACE("Current Workspace", "Components available in the current workspace",
+                List.of("Workspace", "Local", "Function", "Types", "Data Mapper")),
         AGENT_TOOLS("Agent Tools", "Functions used as agent tools", List.of("Project", "Local", "Function")),
         CURRENT_ORGANIZATION("Current Organization", "Components in the current organization",
                 List.of("Organization", "Function", "Library")),

flow-model-generator/modules/flow-model-generator-core/src/main/java/io/ballerina/flowmodelgenerator/core/model/FormBuilder.java

@@ -1087,7 +1087,7 @@ public FormBuilder<T> isPublic(boolean value, boolean optional, boolean editable
                 .editable(editable)
                 .optional(optional)
                 .advanced(advanced)
-                .value(String.valueOf(value))
+                .value(value)
                 .type()
                     .fieldType(Property.ValueType.FLAG)
                     .selected(true)

flow-model-generator/modules/flow-model-generator-core/src/main/java/io/ballerina/flowmodelgenerator/core/model/node/FunctionDefinitionBuilder.java

@@ -82,6 +82,7 @@ public static void setMandatoryProperties(NodeBuilder nodeBuilder, String return
                 .functionDescription(description)
                 .returnType(returnType, null, true)
                 .returnDescription(returnDescription)
+                .isPublic(false, false, true, false)
                 .nestedProperty();
     }
 
@@ -164,6 +165,12 @@ public Map<Path, List<TextEdit>> toSource(SourceBuilder sourceBuilder) {
         if (annotationsProperty.isPresent()) {
             sourceBuilder.token().name(annotationsProperty.get().toSourceCode());
         }
+
+        Optional<Property> visibilityProperty = sourceBuilder.getProperty(Property.IS_PUBLIC_KEY);
+        if (visibilityProperty.isPresent() && Boolean.parseBoolean(visibilityProperty.get().value().toString())) {
+            sourceBuilder.token().keyword(SyntaxKind.PUBLIC_KEYWORD);
+        }
+
         Optional<Property> isolatedProperty = sourceBuilder.getProperty(Property.IS_ISOLATED_KEY);
         if (isolatedProperty.isPresent()) {
             sourceBuilder.token().keyword(SyntaxKind.ISOLATED_KEYWORD);