feat(extensions): add OnInstall and OnUninstall lifecycle nodes

Author: itsvks19

feature/extension/src/commonMain/kotlin/com/klyx/extension/nodegraph/MetadataNodes.kt

@@ -2,6 +2,7 @@ package com.klyx.extension.nodegraph
 
 import androidx.compose.runtime.Immutable
 import androidx.compose.ui.graphics.Color
+import com.klyx.core.logging.log
 import com.klyx.nodegraph.EvaluateScope
 import com.klyx.nodegraph.InputPin
 import com.klyx.nodegraph.NodeRegistry
@@ -13,6 +14,7 @@ import com.klyx.nodegraph.SequentialEventNode
 import com.klyx.nodegraph.customPinType
 import com.klyx.nodegraph.extension.GraphExtension
 import com.klyx.nodegraph.extension.Variable
+import kotlinx.coroutines.CancellationException
 import kotlinx.serialization.Serializable
 
 internal const val EXTENSION_SCHEMA_VERSION = "1.0.0"
@@ -119,6 +121,52 @@ internal object OnInitializeMetadataNode : SequentialEventNode() {
     override val pins = listOf(OutputHeaderPin(defaultNextLabel))
 }
 
+private object OnInstall : SequentialEventNode() {
+    override val key = "klyx.extension.on_install"
+    override val title = "On Install"
+    override val category = "Extension"
+    override val description = "Runs once when the extension is first installed."
+    override val headerColor = Color(0xFF4CAF50)
+    override val triggerName = key
+
+    override val pins = listOf(OutputHeaderPin(defaultNextLabel))
+}
+
+private object OnUninstall : SequentialEventNode() {
+    override val key = "klyx.extension.on_uninstall"
+    override val title = "On Uninstall"
+    override val category = "Extension"
+    override val description = "Runs once when the extension is being uninstalled."
+    override val headerColor = Color(0xFFD32F2F)
+    override val triggerName = key
+
+    override val pins = listOf(OutputHeaderPin(defaultNextLabel))
+}
+
+suspend fun ExtensionManager.onInstall(extensionId: String) {
+    val graph = getById(extensionId)?.graph ?: return
+
+    try {
+        graph.trigger(OnInstall.triggerName, emptyMap(), listener)
+    } catch (e: CancellationException) {
+        throw e
+    } catch (e: Exception) {
+        log.error { "Extension crash on 'OnInstall': ${e.message}" }
+    }
+}
+
+suspend fun ExtensionManager.onUninstall(extensionId: String) {
+    val graph = getById(extensionId)?.graph ?: return
+
+    try {
+        graph.trigger(OnUninstall.triggerName, emptyMap(), listener)
+    } catch (e: CancellationException) {
+        throw e
+    } catch (e: Exception) {
+        log.error { "Extension crash on 'OnUninstall': ${e.message}" }
+    }
+}
+
 object MetadataExtension : GraphExtension {
 
     override val name = "klyx-extension-metadata"
@@ -144,7 +192,8 @@ object MetadataExtension : GraphExtension {
         registry.register(
             MakeMetadataNode,
             BreakMetadataNode,
-            OnInitializeMetadataNode
+            OnInitializeMetadataNode,
+            OnInstall, OnUninstall
         )
     }
 }

klyx/src/commonMain/kotlin/com/klyx/ui/page/extension/ExtensionPage.kt

@@ -85,6 +85,8 @@ import com.klyx.core.util.join
 import com.klyx.extension.nodegraph.Extension
 import com.klyx.extension.nodegraph.ExtensionManager
 import com.klyx.extension.nodegraph.ExtensionMetadata
+import com.klyx.extension.nodegraph.onInstall
+import com.klyx.extension.nodegraph.onUninstall
 import com.klyx.icons.Add
 import com.klyx.icons.Edit
 import com.klyx.icons.Icons
@@ -95,8 +97,10 @@ import kotlinx.coroutines.IO
 import kotlinx.coroutines.async
 import kotlinx.coroutines.coroutineScope
 import kotlinx.coroutines.launch
+import kotlinx.coroutines.withTimeoutOrNull
 import kotlinx.io.files.Path
 import kotlin.time.Clock
+import kotlin.time.Duration.Companion.minutes
 
 @Composable
 context(navigationScope: NavigationScope)
@@ -305,6 +309,10 @@ fun ExtensionPage(modifier: Modifier = Modifier) {
                                 onDelete = {
                                     scope.launch(Dispatchers.IO) {
                                         try {
+                                            withTimeoutOrNull(1.minutes) {
+                                                extensionManager.onUninstall(metadata.id)
+                                            }
+
                                             extensionManager.removeExtension(extension)
                                             snackbarHostState.showSnackbar("${metadata.name} deleted.")
                                         } catch (_: Exception) {
@@ -354,7 +362,7 @@ fun ExtensionPage(modifier: Modifier = Modifier) {
                                 val localExt = extensions.find { it.metadata.id == storeExt.id }
                                 val isInstalled = localExt != null
 
-                                val hasUpdate = isInstalled && localExt!!.metadata.version != storeExt.version
+                                val hasUpdate = isInstalled && localExt.metadata.version != storeExt.version
 
                                 StoreExtensionItem(
                                     storeExt = storeExt,
@@ -365,6 +373,7 @@ fun ExtensionPage(modifier: Modifier = Modifier) {
                                     onInstall = {
                                         val destDir = Paths.extensionsDir
                                         val installedPath = store.downloadExtension(storeExt, destDir)
+
                                         if (installedPath != null) {
                                             extensionManager.addOrReplaceExtension(installedPath, isLocal = false)
                                             snackbarHostState.showSnackbar("${storeExt.name} Installed!")
@@ -374,6 +383,12 @@ fun ExtensionPage(modifier: Modifier = Modifier) {
                                                 storeExtensions[index] =
                                                     storeExt.copy(downloadCount = storeExt.downloadCount + 1)
                                             }
+
+                                            scope.launch(Dispatchers.IO) {
+                                                withTimeoutOrNull(5.minutes) {
+                                                    extensionManager.onInstall(storeExt.id)
+                                                }
+                                            }
                                         } else {
                                             snackbarHostState.showSnackbar("Failed to install ${storeExt.name}.")
                                         }