|
|
@@ -0,0 +1,160 @@
|
|
|
+/*
|
|
|
+ * Copyright (c) Milan Jurkulák 2024.
|
|
|
+ * Contact:
|
|
|
+ * e: mimoccc@gmail.com
|
|
|
+ * e: mj@mjdev.org
|
|
|
+ * w: https://mjdev.org
|
|
|
+ */
|
|
|
+
|
|
|
+package eu.mjdev.desktop.managers
|
|
|
+
|
|
|
+import androidx.compose.runtime.Composable
|
|
|
+import androidx.compose.runtime.DisposableEffect
|
|
|
+import eu.mjdev.desktop.data.App
|
|
|
+import eu.mjdev.desktop.provider.DesktopProvider
|
|
|
+import eu.mjdev.desktop.provider.DesktopProvider.Companion.LocalDesktop
|
|
|
+import kotlinx.coroutines.*
|
|
|
+import kotlin.jvm.optionals.getOrNull
|
|
|
+
|
|
|
+@Suppress("unused")
|
|
|
+class ProcessManager(
|
|
|
+ val delay: Long = 250L
|
|
|
+) : ArrayList<ProcessManager.ProcessWrapper>(), AutoCloseable {
|
|
|
+ private val scope = CoroutineScope(Dispatchers.IO)
|
|
|
+ private val listeners = mutableListOf<ProcessListener>()
|
|
|
+ private var job: Job? = null
|
|
|
+
|
|
|
+ init {
|
|
|
+ job = scope.launch {
|
|
|
+ while (isActive) {
|
|
|
+ ProcessHandle.allProcesses().toList().let { phList ->
|
|
|
+ phList.forEach { ph ->
|
|
|
+ if (!containsProcess(ph)) {
|
|
|
+ ProcessWrapper(ph).also { pw ->
|
|
|
+ add(pw)
|
|
|
+ }
|
|
|
+ listeners.filterIsInstance<ProcessAddListener>().forEach { l ->
|
|
|
+ l.onProcessAdded(ph)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ cleanup()
|
|
|
+ }
|
|
|
+ delay(delay)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ private fun cleanup() {
|
|
|
+ iterator().apply {
|
|
|
+ while (hasNext()) {
|
|
|
+ val pw = next()
|
|
|
+ if (!pw.processHandle.isAlive) {
|
|
|
+ remove()
|
|
|
+ listeners.filterIsInstance<ProcessRemoveListener>().forEach { l ->
|
|
|
+ l.onProcessRemoved(pw.processHandle)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ }
|
|
|
+
|
|
|
+ fun addOnAddListener(listener: ProcessAddListener) {
|
|
|
+ listeners.add(listener)
|
|
|
+ }
|
|
|
+
|
|
|
+ fun addOnRemoveListener(listener: ProcessRemoveListener) {
|
|
|
+ listeners.add(listener)
|
|
|
+ }
|
|
|
+
|
|
|
+ fun removeOnAddListener(listener: ProcessAddListener) {
|
|
|
+ listeners.remove(listener)
|
|
|
+ }
|
|
|
+
|
|
|
+ fun removeOnRemoveListener(listener: ProcessRemoveListener) {
|
|
|
+ listeners.remove(listener)
|
|
|
+ }
|
|
|
+
|
|
|
+ override fun close() {
|
|
|
+ job?.cancel()
|
|
|
+ }
|
|
|
+
|
|
|
+ fun hasAppProcess(app: App?): Boolean = if (app == null) false
|
|
|
+ else {
|
|
|
+ val appCmd = app.cmd
|
|
|
+ val appName = app.name
|
|
|
+ val appFullName = app.fullAppName
|
|
|
+ any { ph ->
|
|
|
+ ph.command.contains(appName) ||
|
|
|
+ ph.command.contains(appCmd) ||
|
|
|
+ ph.command.contains(appFullName) ||
|
|
|
+ ph.commandLine.contains(appName) ||
|
|
|
+ ph.commandLine.contains(appCmd) ||
|
|
|
+ ph.commandLine.contains(appFullName)
|
|
|
+
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ companion object {
|
|
|
+ fun ArrayList<ProcessWrapper>.containsProcess(ph: ProcessHandle) = ph.pid().let { pid ->
|
|
|
+ any { p -> p.pid == pid }
|
|
|
+ }
|
|
|
+
|
|
|
+ fun ArrayList<ProcessWrapper>.containsProcess(ph: ProcessWrapper) = any { p -> p.pid == ph.pid }
|
|
|
+
|
|
|
+ @Composable
|
|
|
+ fun processManagerListener(
|
|
|
+ onChanged: ProcessManager.(processHandle: ProcessHandle) -> Unit,
|
|
|
+ ) = processManagerListener(onChanged, onChanged)
|
|
|
+
|
|
|
+ @Composable
|
|
|
+ fun processManagerListener(
|
|
|
+ onAdd: ProcessManager.(processHandle: ProcessHandle) -> Unit,
|
|
|
+ onRemove: ProcessManager.(processHandle: ProcessHandle) -> Unit,
|
|
|
+ api: DesktopProvider = LocalDesktop.current,
|
|
|
+ processManger: ProcessManager = api.processManager
|
|
|
+ ) {
|
|
|
+ val addListener = object : ProcessAddListener {
|
|
|
+ override fun onProcessAdded(processHandle: ProcessHandle) {
|
|
|
+ onAdd.invoke(processManger, processHandle)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ val removeListener = object : ProcessRemoveListener {
|
|
|
+ override fun onProcessRemoved(processHandle: ProcessHandle) {
|
|
|
+ onRemove.invoke(processManger, processHandle)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ DisposableEffect(Unit) {
|
|
|
+ processManger.addOnAddListener(addListener)
|
|
|
+ processManger.addOnRemoveListener(removeListener)
|
|
|
+ onDispose {
|
|
|
+ processManger.removeOnAddListener(addListener)
|
|
|
+ processManger.removeOnRemoveListener(removeListener)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ interface ProcessListener
|
|
|
+
|
|
|
+ interface ProcessAddListener : ProcessListener {
|
|
|
+ fun onProcessAdded(processHandle: ProcessHandle)
|
|
|
+ }
|
|
|
+
|
|
|
+ interface ProcessRemoveListener : ProcessListener {
|
|
|
+ fun onProcessRemoved(processHandle: ProcessHandle)
|
|
|
+ }
|
|
|
+
|
|
|
+ class ProcessWrapper(
|
|
|
+ val processHandle: ProcessHandle,
|
|
|
+ val pid: Long = processHandle.pid(),
|
|
|
+ val info: ProcessHandle.Info? = processHandle.info(),
|
|
|
+ val command: String = info?.command()?.getOrNull().orEmpty(),
|
|
|
+ val commandLine: String = info?.commandLine()?.getOrNull().orEmpty()
|
|
|
+ ) {
|
|
|
+ override fun toString(): String {
|
|
|
+ return "$pid | $command | $commandLine"
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
Milan Jurkulak