Run apollo-compiler in the background (#83)

* Run apollo-compiler in background

* Handle cancellation as much as possible

Author: BoD

plugin/src/main/kotlin/com/apollographql/ijplugin/codegen/ApolloCodegenService.kt

@@ -38,6 +38,7 @@ import com.intellij.openapi.util.CheckedDisposable
 import com.intellij.openapi.vfs.VfsUtil
 import com.intellij.psi.PsiDocumentManager
 import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.Job
 import kotlinx.coroutines.launch
 import org.gradle.tooling.CancellationTokenSource
 import org.gradle.tooling.GradleConnector
@@ -54,6 +55,7 @@ class ApolloCodegenService(
   private var dirtyGqlDocument: Document? = null
 
   private var gradleCodegenCancellation: CancellationTokenSource? = null
+  private var apolloCompilerCodegenJob: Job? = null
 
   init {
     logd("project=${project.name}")
@@ -165,7 +167,10 @@ class ApolloCodegenService(
 
           if (apolloKotlinService?.hasCompilerOptions == true) {
             // We can use the built-in Apollo compiler
-            ApolloCompilerHelper(project).generateSources(apolloKotlinService)
+            apolloCompilerCodegenJob?.cancel()
+            apolloCompilerCodegenJob = coroutineScope.launch {
+              ApolloCompilerHelper(project).generateSources(apolloKotlinService)
+            }
           } else {
             // Fall back to the Gradle codegen task
             startGradleCodegen()
@@ -231,6 +236,8 @@ class ApolloCodegenService(
 
   private fun stopCodegen() {
     logd()
+    apolloCompilerCodegenJob?.cancel()
+    apolloCompilerCodegenJob = null
     gradleCodegenCancellation?.cancel()
     gradleCodegenCancellation = null
   }
@@ -257,7 +264,9 @@ class ApolloCodegenService(
   private fun startCodegen() {
     if (project.apolloKotlinProjectModelService.getApolloKotlinServices().any { it.hasCompilerOptions }) {
       logd("Using Apollo compiler for codegen")
-      ApolloCompilerHelper(project).generateAllSources()
+      apolloCompilerCodegenJob = coroutineScope.launch {
+        ApolloCompilerHelper(project).generateAllSources()
+      }
     } else {
       logd("Using Gradle codegen task")
       startGradleCodegen()

plugin/src/main/kotlin/com/apollographql/ijplugin/codegen/ApolloCompilerHelper.kt

@@ -18,6 +18,8 @@ import com.apollographql.ijplugin.gradle.ApolloKotlinService.Id
 import com.apollographql.ijplugin.gradle.apolloKotlinProjectModelService
 import com.apollographql.ijplugin.util.logd
 import com.apollographql.ijplugin.util.logw
+import com.intellij.openapi.progress.ProcessCanceledException
+import com.intellij.openapi.progress.ProgressManager
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.vfs.VfsUtil
 import java.io.File
@@ -63,7 +65,7 @@ class ApolloCompilerHelper(
           codegenSchemaOptionsFile = schemaService.codegenSchemaOptionsFile!!,
           codegenSchemaFile = codegenSchemaFile,
       )
-
+      ProgressManager.checkCanceled()
 
       val allUpstreamServiceIds = service.allUpstreamServiceIds()
       if (allUpstreamServiceIds == null) {
@@ -91,6 +93,7 @@ class ApolloCompilerHelper(
             irOptionsFile = service.irOptionsFile!!,
             irOperationsFile = irOperationsFile,
         )
+        ProgressManager.checkCanceled()
         irOperationsById[service.id] = irOperationsFile
       }
 
@@ -132,6 +135,7 @@ class ApolloCompilerHelper(
             outputDirectory = service.codegenOutputDir!!,
             metadataOutput = metadataOutput,
         )
+        ProgressManager.checkCanceled()
         upstreamMetadata.add(metadataOutput)
         outputDirs.add(service.codegenOutputDir)
       }
@@ -149,13 +153,18 @@ class ApolloCompilerHelper(
               codegenOptions = service.codegenOptionsFile!!,
               outputDirectory = service.dataBuildersOutputDir!!,
           )
+          ProgressManager.checkCanceled()
           outputDirs.add(service.dataBuildersOutputDir)
         }
       }
 
       logd("Apollo compiler sources generated for service ${service.id} at ${service.codegenOutputDir}")
       return outputDirs
     } catch (e: Exception) {
+      if (e is ProcessCanceledException) {
+        logd("Apollo compiler was canceled")
+        throw e
+      }
       logw(e, "Failed to generate sources for service ${service.id}")
     }
     return emptySet()