Android native code for wakelock implementation for Flutter project

wakelock.kt

  lateinit var wakeLock: PowerManager.WakeLock
   private fun deactivateAutoScreenOffOnProximity() {
        if(wakeLock.isHeld){
            wakeLock.release()
        }
    }

    @SuppressLint("WakelockTimeout")
    private fun activateAutoScreenOffOnProximity() {
        if(!wakeLock.isHeld) {
            wakeLock.acquire()
        }
    }

   override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
       wakeLock= getSystemService(PowerManager::class.java).newWakeLock(
            PowerManager.PROXIMITY_SCREEN_OFF_WAKE_LOCK,
            "socialupp:wake_lock_for_proximity",
        )
    }

    override fun configureFlutterEngine(flutterEngine: FlutterEngine) {
        super.configureFlutterEngine(flutterEngine)
        MethodChannel(flutterEngine.dartExecutor, "channel_name").setMethodCallHandler { call, result ->
            when (call.method) {
                "activateAutoScreenOffOnProximity" -> {
                    try {
                        activateAutoScreenOffOnProximity()
                        result.success("Activated auto screen off on proximity")
                    } catch (e: Error) {
                        result.error("",
                            "Couldn't activate auto screen off on proximity ${e.message}",
                            null)
                    }
                }
                "deactivateAutoScreenOffOnProximity" -> {
                    try {
                        deactivateAutoScreenOffOnProximity()
                        result.success("Deactivated auto screen off on proximity")
                    } catch (e: Error) {
                        result.error("",
                            "Couldn't deactivate auto screen off on proximity ${e.message}",
                            null)
                    }
                }
                else -> result.notImplemented()
            }
        }
    }