Gradle Runner PoC Refactor

Author: Stefterv

app/src/processing/app/gradle/ActionGradleJob.kt

@@ -0,0 +1,3 @@
+package processing.app.gradle
+
+class ActionGradleJob : GradleJob()

app/src/processing/app/gradle/BackgroundGradleJob.kt

@@ -0,0 +1,3 @@
+package processing.app.gradle
+
+class BackgroundGradleJob : GradleJob()

app/src/processing/app/gradle/GradleJob.kt

@@ -0,0 +1,118 @@
+package processing.app.gradle
+
+import androidx.compose.runtime.mutableStateListOf
+import androidx.compose.runtime.mutableStateOf
+import com.sun.jdi.Bootstrap
+import com.sun.jdi.VirtualMachine
+import com.sun.jdi.connect.AttachingConnector
+import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.launch
+import org.gradle.tooling.BuildLauncher
+import org.gradle.tooling.GradleConnector
+import org.gradle.tooling.events.ProgressListener
+import org.gradle.tooling.events.problems.ProblemEvent
+import org.gradle.tooling.events.task.TaskFinishEvent
+import org.gradle.tooling.events.task.TaskStartEvent
+import processing.app.Messages
+
+// TODO: Capture and filter gradle output
+open abstract class GradleJob{
+    enum class State{
+        NONE,
+        BUILDING,
+        RUNNING,
+        DONE
+    }
+
+    var service: GradleService? = null
+    var configure: BuildLauncher.() -> Unit = {}
+
+    val state = mutableStateOf(State.NONE)
+    val vm = mutableStateOf<VirtualMachine?>(null)
+    val problems = mutableStateListOf<ProblemEvent>()
+
+    private val scope = CoroutineScope(Dispatchers.IO)
+    private val cancel = GradleConnector.newCancellationTokenSource()
+
+    fun start() {
+        service?.jobs?.add(this)
+        val connection = service?.connection ?: return
+        scope.launch {
+            try {
+                state.value = State.BUILDING
+
+                connection.newBuild()
+                    .apply {
+                        configure()
+                    }
+                    .withCancellationToken(cancel.token())
+                    .addStateListener()
+                    .addDebugging()
+                    .run()
+
+            }catch (e: Exception){
+                if(state.value == State.RUNNING){
+                    Messages.log("Error while running: ${e.message}")
+                }else{
+                    throw e
+                }
+            }finally {
+                state.value = State.DONE
+                vm.value = null
+            }
+
+        }
+    }
+
+    fun cancel(){
+        cancel.cancel()
+    }
+
+    private fun BuildLauncher.addStateListener(): BuildLauncher{
+        this.addProgressListener(ProgressListener { event ->
+            if(event is TaskStartEvent) {
+                when(event.descriptor.name) {
+                    ":run" -> {
+                        state.value = State.RUNNING
+                    }
+                }
+
+            }
+            if(event is TaskFinishEvent) {
+                when(event.descriptor.name){
+                    ":jar"->{
+                        state.value = State.NONE
+                        Messages.log("Jar finished")
+                    }
+                    ":run"->{
+                        state.value = State.NONE
+                    }
+                }
+            }
+            if(event is ProblemEvent) {
+                problems.add(event)
+            }
+        })
+        return this
+    }
+
+    private fun BuildLauncher.addDebugging(): BuildLauncher{
+        this.addProgressListener(ProgressListener { event ->
+            if(event !is TaskStartEvent) return@ProgressListener
+            if(event.descriptor.name != ":run") return@ProgressListener
+
+            Messages.log("Attaching to VM")
+            val connector = Bootstrap.virtualMachineManager().allConnectors()
+                .firstOrNull { it.name() == "com.sun.jdi.SocketAttach" }
+                    as AttachingConnector?
+                ?: return@ProgressListener
+            val args = connector.defaultArguments()
+            args["port"]?.setValue(service?.debugPort.toString())
+            val sketch = connector.attach(args)
+            vm.value = sketch
+            Messages.log("Attached to VM: ${sketch.name()}")
+        })
+        return this
+    }
+}