app/src/main/java/com/draco/buoy/viewmodels/PermissionActivityViewModel.kt


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

package com.draco.buoy.viewmodels

import android.app.Application
import androidx.lifecycle.AndroidViewModel
import androidx.lifecycle.LiveData
import androidx.lifecycle.MutableLiveData
import androidx.lifecycle.viewModelScope
import com.draco.buoy.BuildConfig
import com.draco.buoy.utils.PermissionUtils
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.delay
import kotlinx.coroutines.launch

class PermissionActivityViewModel(application: Application) : AndroidViewModel(application) {
    private val _permissionGranted = MutableLiveData(false)
    val permissionGranted: LiveData<Boolean> = _permissionGranted

    private fun askRootPermission() {
        try {
            ProcessBuilder(
                "su",
                "-c",
                "pm grant ${BuildConfig.APPLICATION_ID} android.permission.WRITE_SECURE_SETTINGS"
            ).start()
        } catch (_: Exception) {}
    }

    private fun isWriteSecureSettingsPermissionGranted(): Boolean {
        val context = getApplication<Application>().applicationContext
        return PermissionUtils.isPermissionsGranted(context, android.Manifest.permission.WRITE_SECURE_SETTINGS)
    }

    private fun startPermissionCheckLoop() {
        viewModelScope.launch(Dispatchers.IO) {
            while (!isWriteSecureSettingsPermissionGranted())
                delay(100)
            _permissionGranted.postValue(true)
        }
    }

    init {
        if (!isWriteSecureSettingsPermissionGranted()) {
            viewModelScope.launch(Dispatchers.IO) {
                askRootPermission()
            }
            startPermissionCheckLoop()
        }
    }
}