#489 attempt sleep prevention linux

siliconeagle · siliconeagle · commit 25814e7fb0b7 · 2024-11-14T09:36:55.000+01:00
diff --git a/hub/src/main/kotlin/uk/co/sentinelweb/cuer/hub/di/Modules.kt b/hub/src/main/kotlin/uk/co/sentinelweb/cuer/hub/di/Modules.kt
@@ -1,6 +1,7 @@
 package uk.co.sentinelweb.cuer.hub.di
 
 import PlatformWifiInfo
+import SleepPreventerMac
 import com.arkivanov.essenty.lifecycle.LifecycleRegistry
 import com.russhwolf.settings.JvmPreferencesSettings
 import com.russhwolf.settings.Settings
@@ -38,6 +39,7 @@ import uk.co.sentinelweb.cuer.db.di.JvmDatabaseModule
 import uk.co.sentinelweb.cuer.di.DomainModule
 import uk.co.sentinelweb.cuer.di.JvmDomainModule
 import uk.co.sentinelweb.cuer.domain.BuildConfigDomain
+import uk.co.sentinelweb.cuer.domain.NodeDomain
 import uk.co.sentinelweb.cuer.hub.BuildConfigInject
 import uk.co.sentinelweb.cuer.hub.service.remote.RemoteServerService
 import uk.co.sentinelweb.cuer.hub.service.remote.RemoteServerServiceManager
@@ -57,6 +59,9 @@ import uk.co.sentinelweb.cuer.hub.util.remote.FileEncryption
 import uk.co.sentinelweb.cuer.hub.util.remote.KeyStoreManager
 import uk.co.sentinelweb.cuer.hub.util.remote.RemoteConfigFileInitialiseer
 import uk.co.sentinelweb.cuer.hub.util.share.scan.TodoLinkScanner
+import uk.co.sentinelweb.cuer.hub.util.sleep.SleepPreventer
+import uk.co.sentinelweb.cuer.hub.util.sleep.SleepPreventerEmpty
+import uk.co.sentinelweb.cuer.hub.util.sleep.SleepPreventerLinuxDBus
 import uk.co.sentinelweb.cuer.hub.util.system_tray.SystemTrayComposePopup
 import uk.co.sentinelweb.cuer.hub.util.system_tray.SystemTrayIcon
 import uk.co.sentinelweb.cuer.hub.util.wrapper.EmptyVibrateWrapper
@@ -108,6 +113,18 @@ object Modules {
         }
         single { SystemTrayIcon() }
         factory { SystemTrayComposePopup() }
+        single<SleepPreventer> {
+            when (getOS()) {
+                NodeDomain.DeviceType.MAC -> SleepPreventerMac()
+                NodeDomain.DeviceType.LINUX -> SleepPreventerLinuxDBus()
+                else -> SleepPreventerEmpty()
+//                NodeDomain.DeviceType.ANDROID -> TODO()
+//                NodeDomain.DeviceType.IOS -> TODO()
+//                NodeDomain.DeviceType.WEB -> TODO()
+//                NodeDomain.DeviceType.WINDOWS -> TODO()
+//                NodeDomain.DeviceType.OTHER -> TODO()
+            }
+        }
     }
 
     private val connectivityModule = module {
diff --git a/hub/src/main/kotlin/uk/co/sentinelweb/cuer/hub/ui/player/vlc/VlcPlayerSwingWindow.kt b/hub/src/main/kotlin/uk/co/sentinelweb/cuer/hub/ui/player/vlc/VlcPlayerSwingWindow.kt
@@ -3,13 +3,11 @@ package uk.co.sentinelweb.cuer.hub.ui.player.vlc
 import androidx.compose.ui.graphics.Color.Companion.Black
 import com.sun.jna.StringArray
 import httpLocalNetworkUrl
-import kotlinx.coroutines.ExperimentalCoroutinesApi
 import loadSVG
 import org.koin.core.component.KoinComponent
 import org.koin.core.component.inject
 import toImageIcon
 import uk.co.caprica.vlcj.binding.internal.libvlc_instance_t
-import uk.co.caprica.vlcj.binding.lib.LibVlc
 import uk.co.caprica.vlcj.binding.lib.LibVlc.libvlc_new
 import uk.co.caprica.vlcj.binding.lib.LibVlc.libvlc_release
 import uk.co.caprica.vlcj.factory.discovery.NativeDiscovery
@@ -22,7 +20,6 @@ import uk.co.caprica.vlcj.player.base.MediaPlayerEventAdapter
 import uk.co.caprica.vlcj.player.base.State
 import uk.co.caprica.vlcj.player.component.CallbackMediaPlayerComponent
 import uk.co.caprica.vlcj.support.version.LibVlcVersion
-import uk.co.sentinelweb.cuer.app.orchestrator.OrchestratorContract.Source.LOCAL_NETWORK
 import uk.co.sentinelweb.cuer.app.ui.player.PlayerContract
 import uk.co.sentinelweb.cuer.app.ui.player.PlayerContract.PlayerCommand.*
 import uk.co.sentinelweb.cuer.app.ui.player.PlayerContract.View.Event.*
@@ -32,16 +29,11 @@ import uk.co.sentinelweb.cuer.core.providers.CoroutineContextProvider
 import uk.co.sentinelweb.cuer.core.providers.PlayerConfigProvider
 import uk.co.sentinelweb.cuer.core.providers.TimeProvider
 import uk.co.sentinelweb.cuer.core.wrapper.LogWrapper
-import uk.co.sentinelweb.cuer.core.wrapper.URLEncoder
-import uk.co.sentinelweb.cuer.domain.PlatformDomain
 import uk.co.sentinelweb.cuer.domain.PlayerNodeDomain
 import uk.co.sentinelweb.cuer.domain.PlayerStateDomain
 import uk.co.sentinelweb.cuer.domain.PlayerStateDomain.*
 import uk.co.sentinelweb.cuer.domain.PlaylistItemDomain
 import uk.co.sentinelweb.cuer.remote.server.LocalRepository
-import uk.co.sentinelweb.cuer.remote.server.RemoteWebServerContract.Companion.VIDEO_STREAM_API
-import uk.co.sentinelweb.cuer.remote.server.http
-import uk.co.sentinelweb.cuer.remote.server.locator
 import java.awt.BorderLayout
 import java.awt.BorderLayout.*
 import java.awt.Color
@@ -51,14 +43,16 @@ import javax.swing.*
 import javax.swing.JOptionPane.ERROR_MESSAGE
 import javax.swing.event.ChangeEvent
 
+const val WINDOW_NAME = "CuerVlcSwingVideo"
+
 class VlcPlayerSwingWindow(
     private val coordinator: VlcPlayerUiCoordinator,
     private val folderListUseCase: GetFolderListUseCase,
     private val showHideControls: VlcPlayerShowHideControls,
     private val keyMap: VlcPlayerKeyMap,
     private val localRepository: LocalRepository,
     private val coroutineContextProvider: CoroutineContextProvider
-) : JFrame(), KoinComponent {
+) : JFrame(WINDOW_NAME), KoinComponent {
 
     lateinit var mediaPlayerComponent: CallbackMediaPlayerComponent
     private val log: LogWrapper by inject()
@@ -272,8 +266,8 @@ class VlcPlayerSwingWindow(
 
         is Pause -> mediaPlayerComponent.mediaPlayer().controls().pause()
         is Play -> mediaPlayerComponent.mediaPlayer().controls().play()
-        is SkipFwd -> mediaPlayerComponent.mediaPlayer().controls().skipTime(command.ms.toLong())
-        is SkipBack -> mediaPlayerComponent.mediaPlayer().controls().skipTime(-command.ms.toLong())
+        is SkipFwd -> mediaPlayerComponent.mediaPlayer().controls().skipTime(command.ms)
+        is SkipBack -> mediaPlayerComponent.mediaPlayer().controls().skipTime(-command.ms)
         is SeekTo -> mediaPlayerComponent.mediaPlayer().controls().setTime(command.ms)
     }.also { log.d("command:${command::class.java.simpleName}") }
 
@@ -497,10 +491,10 @@ class VlcPlayerSwingWindow(
             fun tryload(): Boolean {
                 try {
                     val instance: libvlc_instance_t = libvlc_new(0, StringArray(arrayOf<String>()))
-                    libvlc_release(instance);
-                    val version: LibVlcVersion = LibVlcVersion();
+                    libvlc_release(instance)
+                    val version = LibVlcVersion()
                     if (version.isSupported()) {
-                        return true;
+                        return true
                     }
                 } catch (e: UnsatisfiedLinkError) {
                     e.printStackTrace()
diff --git a/hub/src/main/kotlin/uk/co/sentinelweb/cuer/hub/ui/player/vlc/VlcPlayerUiCoordinator.kt b/hub/src/main/kotlin/uk/co/sentinelweb/cuer/hub/ui/player/vlc/VlcPlayerUiCoordinator.kt
@@ -1,6 +1,5 @@
 package uk.co.sentinelweb.cuer.hub.ui.player.vlc
 
-import SleepPreventer
 import androidx.compose.foundation.layout.height
 import androidx.compose.material3.MaterialTheme
 import androidx.compose.runtime.Composable
@@ -13,7 +12,6 @@ import com.arkivanov.mvikotlin.core.utils.diff
 import com.arkivanov.mvikotlin.core.view.BaseMviView
 import com.arkivanov.mvikotlin.core.view.ViewRenderer
 import com.arkivanov.mvikotlin.main.store.DefaultStoreFactory
-import kotlinx.coroutines.ExperimentalCoroutinesApi
 import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.launch
 import org.koin.core.component.KoinComponent
@@ -43,6 +41,7 @@ import uk.co.sentinelweb.cuer.domain.ext.serialise
 import uk.co.sentinelweb.cuer.hub.ui.home.HomeUiCoordinator
 import uk.co.sentinelweb.cuer.hub.util.extension.DesktopScopeComponent
 import uk.co.sentinelweb.cuer.hub.util.extension.desktopScopeWithSource
+import uk.co.sentinelweb.cuer.hub.util.sleep.SleepPreventer
 import uk.co.sentinelweb.cuer.hub.util.view.UiCoordinator
 
 class VlcPlayerUiCoordinator(
@@ -66,6 +65,7 @@ class VlcPlayerUiCoordinator(
     private val playlistOrchestrator: PlaylistOrchestrator by inject()
     private val coroutines: CoroutineContextProvider by inject()
     private val queueProducer: QueueMediatorContract.Producer by inject()
+    private val sleepPreventer: SleepPreventer by inject()
 
     private lateinit var playerWindow: VlcPlayerSwingWindow
 
@@ -106,7 +106,7 @@ class VlcPlayerUiCoordinator(
         playerWindow = if (VlcPlayerSwingWindow.checkShowWindow()) {
             scope.get<VlcPlayerSwingWindow>(VlcPlayerSwingWindow::class)
                 .apply { assemble(screen) }
-                .also { SleepPreventer.preventSleep() }
+                .also { sleepPreventer.preventSleep() }
         } else error("Can't find VLC")
     }
 
@@ -116,7 +116,7 @@ class VlcPlayerUiCoordinator(
                 ?.takeIf { it.source == MEMORY }
                 ?.also { playlistOrchestrator.delete(it.id, it.deepOptions()) }
             playerWindow.destroy()
-            SleepPreventer.allowSleep()
+            sleepPreventer.allowSleep()
             lifecycle.onPause()
             lifecycle.onStop()
             controller.onViewDestroyed()
diff --git a/hub/src/main/kotlin/uk/co/sentinelweb/cuer/hub/util/sleep/SleepPreventer.kt b/hub/src/main/kotlin/uk/co/sentinelweb/cuer/hub/util/sleep/SleepPreventer.kt
@@ -1,83 +1,7 @@
-import com.sun.jna.Library
-import com.sun.jna.Native
-import com.sun.jna.Pointer
-import com.sun.jna.PointerType
-import com.sun.jna.ptr.PointerByReference
-import uk.co.sentinelweb.cuer.core.providers.getOS
-import uk.co.sentinelweb.cuer.domain.NodeDomain
-import uk.co.sentinelweb.cuer.domain.NodeDomain.DeviceType.MAC
+package uk.co.sentinelweb.cuer.hub.util.sleep
 
-// uses JNA to prevent sleep
-class SleepPreventer {
+interface SleepPreventer {
+    fun preventSleep()
 
-    interface IOKit : Library {
-        fun IOPMAssertionCreateWithName(
-            assertionType: Pointer,
-            assertionLevel: Int,
-            assertionName: Pointer,
-            assertionID: PointerByReference
-        ): Int
-
-        fun IOPMAssertionRelease(assertionID: Pointer): Int
-    }
-
-    interface CoreFoundation : Library {
-        fun CFStringCreateWithCString(
-            alloc: Pointer?,
-            cStr: String,
-            encoding: Int
-        ): Pointer
-    }
-
-    class CFStringRef(s: String) : PointerType() {
-        init {
-            pointer = CoreFoundationLib.CFStringCreateWithCString(null, s, kCFStringEncodingUTF8)
-        }
-    }
-
-    companion object {
-        private const val kCFStringEncodingUTF8 = 0x08000100
-        private const val ASSERTION_TYPE_NO_DISPLAY_SLEEP = "NoDisplaySleepAssertion"
-        private const val ASSERTION_LEVEL_ON = 255
-        private var assertionIDKeeper: PointerByReference? = null
-        private val CoreFoundationLib: CoreFoundation by lazy { Native.load("CoreFoundation", CoreFoundation::class.java)}
-        private val iokit: IOKit by lazy {Native.load("IOKit", IOKit::class.java)}
-
-        @JvmStatic
-        fun preventSleep() {
-            if (getOS() == MAC) {
-                if (assertionIDKeeper == null) {
-                    val assertionID = PointerByReference()
-                    val assertionType = CFStringRef(ASSERTION_TYPE_NO_DISPLAY_SLEEP).pointer
-                    val assertionName = CFStringRef("Prevent sleep while playing video").pointer
-                    val iokit: IOKit = Native.load("IOKit", IOKit::class.java)
-
-                    val result =
-                        iokit.IOPMAssertionCreateWithName(assertionType, ASSERTION_LEVEL_ON, assertionName, assertionID)
-                    if (result != 0) {
-                        assertionIDKeeper = null
-                        println("Failed to create assertion: $result")
-                    } else {
-                        assertionIDKeeper = assertionID
-                        println("Sleep prevention activated.")
-                    }
-                }
-            }
-        }
-
-        @JvmStatic
-        fun allowSleep() {
-            if (getOS() == MAC) {
-                assertionIDKeeper?.let {
-                    val result = iokit.IOPMAssertionRelease(it.value)
-                    if (result == 0) {
-                        println("Sleep prevention deactivated.")
-                        assertionIDKeeper = null
-                    } else {
-                        println("Failed to release assertion: $result")
-                    }
-                }
-            }
-        }
-    }
-}
+    fun allowSleep()
+}
diff --git a/hub/src/main/kotlin/uk/co/sentinelweb/cuer/hub/util/sleep/SleepPreventerEmpty.kt b/hub/src/main/kotlin/uk/co/sentinelweb/cuer/hub/util/sleep/SleepPreventerEmpty.kt
@@ -0,0 +1,7 @@
+package uk.co.sentinelweb.cuer.hub.util.sleep
+
+class SleepPreventerEmpty : SleepPreventer {
+    override fun preventSleep() = Unit
+
+    override fun allowSleep() = Unit
+}
diff --git a/hub/src/main/kotlin/uk/co/sentinelweb/cuer/hub/util/sleep/SleepPreventerLinuxDBus.kt b/hub/src/main/kotlin/uk/co/sentinelweb/cuer/hub/util/sleep/SleepPreventerLinuxDBus.kt
@@ -0,0 +1,68 @@
+package uk.co.sentinelweb.cuer.hub.util.sleep
+
+import java.io.BufferedReader
+import java.io.InputStreamReader
+
+class SleepPreventerLinuxDBus : SleepPreventer {
+    private var inhibitCookie: String? = null
+
+    //dbus-send --session --print-reply=literal --dest=org.freedesktop.PowerManager /org/freedesktop/PowerManager org.freedesktop.PowerManager.Inhibit string:MyApp string:Inhibit screensaver
+    override fun preventSleep() {
+        try {
+            // Inhibit screensaver via dbus-send
+            val pb = ProcessBuilder(
+                "dbus-send", "--session",
+                "--print-reply",
+                "--dest=org.freedesktop.ScreenSaver",
+                "/ScreenSaver",
+                "org.freedesktop.ScreenSaver.Inhibit",
+                "string:Cuer",
+                "string:Inhibiting for demonstration"
+            )
+            pb.redirectErrorStream(true)
+            val process = pb.start()
+
+            BufferedReader(InputStreamReader(process.inputStream)).use { reader ->
+                // Read the response from dbus-send which contains the inhibit cookie
+                var response: String
+                while ((reader.readLine().also { response = it }) != null) {
+                    if (response.contains("uint32")) {
+                        inhibitCookie =
+                            response.split("uint32\\s+".toRegex()).dropLastWhile { it.isEmpty() }
+                                .toTypedArray()[1].trim { it <= ' ' }
+                        println("Inhibit Cookie: " + inhibitCookie)
+                    }
+                }
+            }
+            // Optionally wait for the process to complete or handle as needed
+            process.waitFor()
+        } catch (e: java.lang.Exception) {
+            e.printStackTrace()
+        }
+    }
+
+    override fun allowSleep() {
+        if (inhibitCookie.isNullOrEmpty()) {
+            System.err.println("Inhibit cookie is null or empty. Cannot allow screen sleep.")
+            return
+        }
+
+        try {
+            // Uninhibit screensaver via dbus-send using the stored inhibit cookie
+            val pb = ProcessBuilder(
+                "dbus-send", "--session",
+                "--dest=org.freedesktop.ScreenSaver",
+                "/ScreenSaver",
+                "org.freedesktop.ScreenSaver.UnInhibit",
+                "uint32:" + inhibitCookie
+            )
+            pb.inheritIO()
+            val process = pb.start()
+
+            // Optionally wait for the process to complete or handle as needed
+            process.waitFor()
+        } catch (e: Exception) {
+            e.printStackTrace()
+        }
+    }
+}
diff --git a/hub/src/main/kotlin/uk/co/sentinelweb/cuer/hub/util/sleep/SleepPreventerLinuxXdg.kt b/hub/src/main/kotlin/uk/co/sentinelweb/cuer/hub/util/sleep/SleepPreventerLinuxXdg.kt
diff --git a/hub/src/main/kotlin/uk/co/sentinelweb/cuer/hub/util/sleep/SleepPreventerMac.kt b/hub/src/main/kotlin/uk/co/sentinelweb/cuer/hub/util/sleep/SleepPreventerMac.kt
