[#64] Add support for getting state as events

permission-flow/src/main/java/dev/shreyaspatil/permissionFlow/watchmen/PermissionWatchmen.kt

@@ -26,10 +26,13 @@ import kotlinx.coroutines.FlowPreview
 import kotlinx.coroutines.Job
 import kotlinx.coroutines.SupervisorJob
 import kotlinx.coroutines.cancelChildren
+import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.MutableSharedFlow
 import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.StateFlow
 import kotlinx.coroutines.flow.asStateFlow
+import kotlinx.coroutines.flow.combine
+import kotlinx.coroutines.flow.filter
 import kotlinx.coroutines.flow.launchIn
 import kotlinx.coroutines.flow.onEach
 import kotlinx.coroutines.launch
@@ -77,6 +80,24 @@ internal class PermissionWatchmen(
         return combineStates(*permissionStates) { MultiplePermissionState(it.toList()) }
     }
 
+    fun watchStateEvents(permission: String): Flow<PermissionState> {
+        // Wakeup watchmen if sleeping
+        wakeUp()
+        return getPermissionEvent(permission)
+    }
+
+    fun watchMultipleStateEvents(permissions: Array<String>): Flow<MultiplePermissionState> {
+        // Wakeup watchmen if sleeping
+        wakeUp()
+
+        val permissionStates = permissions
+            .distinct()
+            .map { permission -> getPermissionEvent(permission) }
+            .toTypedArray()
+
+        return combine(*permissionStates) { MultiplePermissionState(it.toList()) }
+    }
+
     fun notifyPermissionsChanged(permissions: Array<String>) {
         watchmenScope.launch {
             permissions.forEach { permission ->
@@ -108,6 +129,9 @@ internal class PermissionWatchmen(
         }.state
     }
 
+    private fun getPermissionEvent(permission: String) =
+        permissionEvents.filter { it.permission == permission }
+
     /**
      * Watches for the permission events and updates appropriate state holders of permission
      */