From 6d83aeeabbe992e873bef5d21c58f829128bac3e Mon Sep 17 00:00:00 2001
From: csf <csf@breakbeat.cn>
Date: Wed, 8 Jun 2022 21:08:05 +0800
Subject: [PATCH] Android screen wakeup

---
 .gitignore                                        |  1 -
 flutter/android/app/src/main/AndroidManifest.xml  |  1 +
 .../com/carriez/flutter_hbb/InputService.kt       |  1 -
 .../kotlin/com/carriez/flutter_hbb/MainService.kt | 15 ++++++++++++++-
 4 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/.gitignore b/.gitignore
index 9ab24b514..74c7fbd0a 100644
--- a/.gitignore
+++ b/.gitignore
@@ -11,7 +11,6 @@ src/version.rs
 *tgz
 *lib
 cert.pfx
-flutter_hbb
 *.bak
 *png
 *svg
diff --git a/flutter/android/app/src/main/AndroidManifest.xml b/flutter/android/app/src/main/AndroidManifest.xml
index c552efae5..ad496edb0 100644
--- a/flutter/android/app/src/main/AndroidManifest.xml
+++ b/flutter/android/app/src/main/AndroidManifest.xml
@@ -9,6 +9,7 @@
     <uses-permission android:name="android.permission.ACCESS_NETWORK_STATE" />
     <uses-permission android:name="android.permission.FOREGROUND_SERVICE" />
     <uses-permission android:name="android.permission.RECORD_AUDIO" />
+    <uses-permission android:name="android.permission.WAKE_LOCK" />
     <!--<uses-permission android:name="android.permission.RECEIVE_BOOT_COMPLETED" />-->
 
     <application
diff --git a/flutter/android/app/src/main/kotlin/com/carriez/flutter_hbb/InputService.kt b/flutter/android/app/src/main/kotlin/com/carriez/flutter_hbb/InputService.kt
index 2db376230..76068eee5 100644
--- a/flutter/android/app/src/main/kotlin/com/carriez/flutter_hbb/InputService.kt
+++ b/flutter/android/app/src/main/kotlin/com/carriez/flutter_hbb/InputService.kt
@@ -206,7 +206,6 @@ class InputService : AccessibilityService() {
         }
     }
 
-    @RequiresApi(Build.VERSION_CODES.O)
     override fun onServiceConnected() {
         super.onServiceConnected()
         ctx = this
diff --git a/flutter/android/app/src/main/kotlin/com/carriez/flutter_hbb/MainService.kt b/flutter/android/app/src/main/kotlin/com/carriez/flutter_hbb/MainService.kt
index a71541cd8..b9f9ff872 100644
--- a/flutter/android/app/src/main/kotlin/com/carriez/flutter_hbb/MainService.kt
+++ b/flutter/android/app/src/main/kotlin/com/carriez/flutter_hbb/MainService.kt
@@ -75,7 +75,17 @@ class MainService : Service() {
     @Keep
     @RequiresApi(Build.VERSION_CODES.N)
     fun rustMouseInput(mask: Int, x: Int, y: Int) {
-        InputService.ctx?.onMouseInput(mask,x,y)
+        // turn on screen with LIFT_DOWN when screen off
+        if (!powerManager.isInteractive && mask == LIFT_DOWN) {
+            if (wakeLock.isHeld) {
+                Log.d(logTag,"Turn on Screen, WakeLock release")
+                wakeLock.release()
+            }
+            Log.d(logTag,"Turn on Screen")
+            wakeLock.acquire(5000)
+        } else {
+            InputService.ctx?.onMouseInput(mask,x,y)
+        }
     }
 
     @Keep
@@ -145,6 +155,9 @@ class MainService : Service() {
     private var serviceLooper: Looper? = null
     private var serviceHandler: Handler? = null
 
+    private val powerManager: PowerManager by lazy { applicationContext.getSystemService(Context.POWER_SERVICE) as PowerManager }
+    private val wakeLock: PowerManager.WakeLock by lazy { powerManager.newWakeLock(PowerManager.ACQUIRE_CAUSES_WAKEUP or PowerManager.SCREEN_BRIGHT_WAKE_LOCK, "rustdesk:wakelock")}
+
     // jvm call rust
     private external fun init(ctx: Context)
     private external fun startServer()