Implement custom LS API for CodeMap generation in BI Copilot

architecture-model-generator/modules/architecture-model-generator-core/build.gradle

@@ -37,8 +37,10 @@ dependencies {
     implementation "org.ballerinalang:ballerina-tools-api:${ballerinaLangVersion}"
     implementation "org.ballerinalang:ballerina-runtime:${ballerinaLangVersion}"
     implementation "com.google.code.gson:gson:${gsonVersion}"
+    compileOnly "org.eclipse.lsp4j:org.eclipse.lsp4j:${eclipseLsp4jVersion}"
 
     testImplementation "org.testng:testng:${testngVersion}"
+    testImplementation "org.eclipse.lsp4j:org.eclipse.lsp4j:${eclipseLsp4jVersion}"
 
     balTools("org.ballerinalang:jballerina-tools:${ballerinaLangVersion}") {
         transitive = false

architecture-model-generator/modules/architecture-model-generator-core/src/main/java/io/ballerina/artifactsgenerator/codemap/CodeMapArtifact.java

@@ -0,0 +1,142 @@
+/*
+ *  Copyright (c) 2026, 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
+ *  in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing,
+ *  software distributed under the License is distributed on an
+ *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ *  KIND, either express or implied.  See the License for the
+ *  specific language governing permissions and limitations
+ *  under the License.
+ */
+
+package io.ballerina.artifactsgenerator.codemap;
+
+import io.ballerina.compiler.syntax.tree.Node;
+import io.ballerina.tools.text.LinePosition;
+import io.ballerina.tools.text.LineRange;
+import org.eclipse.lsp4j.Position;
+import org.eclipse.lsp4j.Range;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * Represents a code artifact extracted from Ballerina source code for the code map.
+ *
+ * @param name       the name of the artifact
+ * @param type       the type of the artifact (e.g., function, service, class)
+ * @param range      the range in source code where this artifact is located
+ * @param properties additional properties of the artifact
+ * @param children   nested artifacts contained within this artifact
+ * @since 1.6.0
+ */
+public record CodeMapArtifact(String name, String type, Range range,
+                              Map<String, Object> properties, List<CodeMapArtifact> children) {
+
+    // Property key constants
+    private static final String MODIFIERS = "modifiers";
+    private static final String DOCUMENTATION = "documentation";
+    private static final String COMMENT = "comment";
+
+
+    /**
+     * Converts a Ballerina LineRange to an LSP4J Range.
+     *
+     * @param lineRange the Ballerina line range
+     * @return the corresponding LSP4J Range
+     */
+    public static Range toRange(LineRange lineRange) {
+        return new Range(toPosition(lineRange.startLine()), toPosition(lineRange.endLine()));
+    }
+
+    /**
+     * Converts a Ballerina LinePosition to an LSP4J Position.
+     *
+     * @param linePosition the Ballerina line position
+     * @return the corresponding LSP4J Position
+     */
+    public static Position toPosition(LinePosition linePosition) {
+        return new Position(linePosition.line(), linePosition.offset());
+    }
+
+    public CodeMapArtifact {
+        properties = Collections.unmodifiableMap(properties);
+        children = Collections.unmodifiableList(children);
+    }
+
+    /**
+     * Builder class for constructing {@link CodeMapArtifact} instances.
+     */
+    public static class Builder {
+        private String name;
+        private String type;
+        private Range range;
+        private final Map<String, Object> properties = new HashMap<>();
+        private final List<CodeMapArtifact> children = new ArrayList<>();
+
+        /**
+         * Creates a new Builder initialized with the line range from the given syntax node.
+         *
+         * @param node the syntax node to extract line range from
+         */
+        public Builder(Node node) {
+            this.range = toRange(node.lineRange());
+        }
+
+        public Builder name(String name) {
+            this.name = name;
+            return this;
+        }
+
+        public Builder type(String type) {
+            this.type = type;
+            return this;
+        }
+
+        public Builder modifiers(List<String> modifiers) {
+            if (!modifiers.isEmpty()) {
+                this.properties.put(MODIFIERS, new ArrayList<>(modifiers));
+            }
+            return this;
+        }
+
+        public Builder addProperty(String key, Object value) {
+            this.properties.put(key, value);
+            return this;
+        }
+
+        public Builder addChild(CodeMapArtifact child) {
+            this.children.add(child);
+            return this;
+        }
+
+        public Builder documentation(String documentation) {
+            return addProperty(DOCUMENTATION, documentation);
+        }
+
+        public Builder comment(String comment) {
+            return addProperty(COMMENT, comment);
+        }
+
+
+        /**
+         * Builds and returns the {@link CodeMapArtifact} instance.
+         *
+         * @return the constructed CodeMapArtifact
+         */
+        public CodeMapArtifact build() {
+            return new CodeMapArtifact(name, type, range,
+                    new HashMap<>(properties), new ArrayList<>(children));
+        }
+    }
+}

architecture-model-generator/modules/architecture-model-generator-core/src/main/java/io/ballerina/artifactsgenerator/codemap/CodeMapFile.java

@@ -0,0 +1,35 @@
+/*
+ *  Copyright (c) 2026, 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
+ *  in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing,
+ *  software distributed under the License is distributed on an
+ *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ *  KIND, either express or implied.  See the License for the
+ *  specific language governing permissions and limitations
+ *  under the License.
+ */
+
+package io.ballerina.artifactsgenerator.codemap;
+
+import java.util.Collections;
+import java.util.List;
+
+/**
+ * Represents a Ballerina source file with its extracted code map artifacts.
+ *
+ * @param artifacts the list of code map artifacts extracted from this file
+ * @since 1.6.0
+ */
+public record CodeMapFile(List<CodeMapArtifact> artifacts) {
+
+    public CodeMapFile {
+        artifacts = artifacts == null ? Collections.emptyList() : Collections.unmodifiableList(artifacts);
+    }
+}

architecture-model-generator/modules/architecture-model-generator-core/src/main/java/io/ballerina/artifactsgenerator/codemap/CodeMapFilesTracker.java

@@ -0,0 +1,91 @@
+/*
+ *  Copyright (c) 2026, 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
+ *  in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing,
+ *  software distributed under the License is distributed on an
+ *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ *  KIND, either express or implied.  See the License for the
+ *  specific language governing permissions and limitations
+ *  under the License.
+ */
+
+package io.ballerina.artifactsgenerator.codemap;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+
+/**
+ * Tracks modified (changed or added) files per project for incremental code map generation.
+ * This singleton maintains a thread-safe record of file modifications between API calls.
+ *
+ * @since 1.6.0
+ */
+public class CodeMapFilesTracker {
+
+    // Map: projectKey (URI) -> Set of modified (changed or added) file relative paths
+    private final Map<String, Set<String>> modifiedFilesMap;
+
+    private CodeMapFilesTracker() {
+        this.modifiedFilesMap = new ConcurrentHashMap<>();
+    }
+
+    private static class Holder {
+        private static final CodeMapFilesTracker INSTANCE = new CodeMapFilesTracker();
+    }
+
+    /**
+     * Returns the singleton instance of ChangedFilesTracker.
+     *
+     * @return the ChangedFilesTracker instance
+     */
+    public static CodeMapFilesTracker getInstance() {
+        return Holder.INSTANCE;
+    }
+
+    /**
+     * Track a changed file for a given project.
+     *
+     * @param projectKey   the project identifier
+     * @param relativePath the relative path of the changed file from project root
+     */
+    public void trackFile(String projectKey, String relativePath) {
+        modifiedFilesMap
+                .computeIfAbsent(projectKey, k -> ConcurrentHashMap.newKeySet())
+                .add(relativePath);
+    }
+
+    /**
+     * Retrieves all tracked modified files for the given project.
+     *
+     * @param projectKey the project URI key
+     * @return list of modified file relative paths, or empty list if none tracked
+     */
+    public List<String> getModifiedFiles(String projectKey) {
+        Set<String> files = modifiedFilesMap.get(projectKey);
+        if (files == null || files.isEmpty()) {
+            return Collections.emptyList();
+        }
+        return new ArrayList<>(files);
+    }
+
+    /**
+     * Clears all tracked modified files for the given project.
+     *
+     * @param projectKey the project URI key
+     */
+    public void clearModifiedFiles(String projectKey) {
+        modifiedFilesMap.remove(projectKey);
+    }
+
+}