Fix the CompilerPluginManager for LS

ballerina-platform · Jul 10, 2024 · fb33eaa · fb33eaa
1 parent 6c996f9
commit fb33eaa
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/compiler/ballerina-lang/src/main/java/io/ballerina/projects/CompilerPluginManager.java b/compiler/ballerina-lang/src/main/java/io/ballerina/projects/CompilerPluginManager.java
@@ -51,6 +51,12 @@ private CompilerPluginManager(PackageCompilation compilation,
     }
 
     static CompilerPluginManager from(PackageCompilation compilation) {
+        // Skip initialization if the compiler plugins are already initialized for the project
+        if (!compilation.packageContext().project().compilerPluginContexts().isEmpty()) {
+            List<CompilerPluginContextIml> compilerPluginContexts =
+                    compilation.packageContext().project().compilerPluginContexts();
+            return new CompilerPluginManager(compilation, compilerPluginContexts);
+        }
         // TODO We need to update the DependencyGraph API. Right now it is a mess
         PackageResolution packageResolution = compilation.getResolution();
         ResolvedPackageDependency rootPkgNode = new ResolvedPackageDependency(
@@ -191,10 +197,6 @@ private static List<Package> getDirectDependencies(ResolvedPackageDependency roo
 
     private static List<CompilerPluginContextIml> initializePlugins(List<CompilerPluginInfo> compilerPlugins,
                                                                     PackageCompilation compilation) {
-        // Skip initialization if the compiler plugins are already initialized for the project
-        if (!compilation.packageContext().project().compilerPluginContexts().isEmpty()) {
-            return compilation.packageContext().project().compilerPluginContexts();
-        }
         List<CompilerPluginContextIml> compilerPluginContexts = new ArrayList<>(compilerPlugins.size());
         for (CompilerPluginInfo compilerPluginInfo : compilerPlugins) {
             CompilerPluginCache pluginCache =

diff --git a/compiler/ballerina-lang/src/main/java/io/ballerina/projects/Package.java b/compiler/ballerina-lang/src/main/java/io/ballerina/projects/Package.java
@@ -635,6 +635,8 @@ private Package createNewPackage() {
             this.project.setCurrentPackage(new Package(newPackageContext, this.project));
             if (isOldDependencyGraphValid(oldPackage, this.project.currentPackage())) {
                 this.project.currentPackage().packageContext().getResolution(oldResolution);
+            } else {
+                this.project.compilerPluginContexts().clear();
             }
             CompilationOptions offlineCompOptions = CompilationOptions.builder().setOffline(true).build();
             offlineCompOptions = offlineCompOptions.acceptTheirs(project.currentPackage().compilationOptions());