Merge pull request #4354 from vector-im/feature/bma/shortcut_fixes

bmarty · web-flow · commit 2ce4d8d84f26 · 2021-10-28T10:13:02.000+02:00
Shortcut fixes
diff --git a/changelog.d/4170.bugfix b/changelog.d/4170.bugfix
@@ -0,0 +1 @@
+Do not show shortcuts if a PIN code is set
diff --git a/matrix-sdk-android/src/main/java/org/matrix/android/sdk/api/session/room/RoomService.kt b/matrix-sdk-android/src/main/java/org/matrix/android/sdk/api/session/room/RoomService.kt
@@ -94,13 +94,15 @@ interface RoomService {
      * Get a snapshot list of room summaries.
      * @return the immutable list of [RoomSummary]
      */
-    fun getRoomSummaries(queryParams: RoomSummaryQueryParams): List<RoomSummary>
+    fun getRoomSummaries(queryParams: RoomSummaryQueryParams,
+                         sortOrder: RoomSortOrder = RoomSortOrder.NONE): List<RoomSummary>
 
     /**
      * Get a live list of room summaries. This list is refreshed as soon as the data changes.
      * @return the [LiveData] of List[RoomSummary]
      */
-    fun getRoomSummariesLive(queryParams: RoomSummaryQueryParams): LiveData<List<RoomSummary>>
+    fun getRoomSummariesLive(queryParams: RoomSummaryQueryParams,
+                             sortOrder: RoomSortOrder = RoomSortOrder.ACTIVITY): LiveData<List<RoomSummary>>
 
     /**
      * Get a snapshot list of Breadcrumbs
diff --git a/matrix-sdk-android/src/main/java/org/matrix/android/sdk/api/session/space/SpaceService.kt b/matrix-sdk-android/src/main/java/org/matrix/android/sdk/api/session/space/SpaceService.kt
@@ -19,6 +19,7 @@ package org.matrix.android.sdk.api.session.space
 import android.net.Uri
 import androidx.lifecycle.LiveData
 import org.matrix.android.sdk.api.session.events.model.Event
+import org.matrix.android.sdk.api.session.room.RoomSortOrder
 import org.matrix.android.sdk.api.session.room.RoomSummaryQueryParams
 import org.matrix.android.sdk.api.session.room.model.RoomSummary
 import org.matrix.android.sdk.internal.session.space.peeking.SpacePeekResult
@@ -74,9 +75,11 @@ interface SpaceService {
      * Get a live list of space summaries. This list is refreshed as soon as the data changes.
      * @return the [LiveData] of List[SpaceSummary]
      */
-    fun getSpaceSummariesLive(queryParams: SpaceSummaryQueryParams): LiveData<List<RoomSummary>>
+    fun getSpaceSummariesLive(queryParams: SpaceSummaryQueryParams,
+                              sortOrder: RoomSortOrder = RoomSortOrder.NONE): LiveData<List<RoomSummary>>
 
-    fun getSpaceSummaries(spaceSummaryQueryParams: SpaceSummaryQueryParams): List<RoomSummary>
+    fun getSpaceSummaries(spaceSummaryQueryParams: SpaceSummaryQueryParams,
+                          sortOrder: RoomSortOrder = RoomSortOrder.NONE): List<RoomSummary>
 
     suspend fun joinSpace(spaceIdOrAlias: String,
                           reason: String? = null,
diff --git a/matrix-sdk-android/src/main/java/org/matrix/android/sdk/internal/session/room/DefaultRoomService.kt b/matrix-sdk-android/src/main/java/org/matrix/android/sdk/internal/session/room/DefaultRoomService.kt
@@ -85,12 +85,14 @@ internal class DefaultRoomService @Inject constructor(
         return roomSummaryDataSource.getRoomSummary(roomIdOrAlias)
     }
 
-    override fun getRoomSummaries(queryParams: RoomSummaryQueryParams): List<RoomSummary> {
-        return roomSummaryDataSource.getRoomSummaries(queryParams)
+    override fun getRoomSummaries(queryParams: RoomSummaryQueryParams,
+                                  sortOrder: RoomSortOrder): List<RoomSummary> {
+        return roomSummaryDataSource.getRoomSummaries(queryParams, sortOrder)
     }
 
-    override fun getRoomSummariesLive(queryParams: RoomSummaryQueryParams): LiveData<List<RoomSummary>> {
-        return roomSummaryDataSource.getRoomSummariesLive(queryParams)
+    override fun getRoomSummariesLive(queryParams: RoomSummaryQueryParams,
+                                      sortOrder: RoomSortOrder): LiveData<List<RoomSummary>> {
+        return roomSummaryDataSource.getRoomSummariesLive(queryParams, sortOrder)
     }
 
     override fun getPagedRoomSummariesLive(queryParams: RoomSummaryQueryParams,
diff --git a/matrix-sdk-android/src/main/java/org/matrix/android/sdk/internal/session/room/summary/RoomSummaryDataSource.kt b/matrix-sdk-android/src/main/java/org/matrix/android/sdk/internal/session/room/summary/RoomSummaryDataSource.kt
@@ -25,7 +25,6 @@ import androidx.paging.PagedList
 import com.zhuinden.monarchy.Monarchy
 import io.realm.Realm
 import io.realm.RealmQuery
-import io.realm.Sort
 import io.realm.kotlin.where
 import org.matrix.android.sdk.api.query.ActiveSpaceFilter
 import org.matrix.android.sdk.api.query.RoomCategoryFilter
@@ -80,25 +79,27 @@ internal class RoomSummaryDataSource @Inject constructor(@SessionDatabase privat
         }
     }
 
-    fun getRoomSummaries(queryParams: RoomSummaryQueryParams): List<RoomSummary> {
+    fun getRoomSummaries(queryParams: RoomSummaryQueryParams,
+                         sortOrder: RoomSortOrder = RoomSortOrder.NONE): List<RoomSummary> {
         return monarchy.fetchAllMappedSync(
-                { roomSummariesQuery(it, queryParams) },
+                { roomSummariesQuery(it, queryParams).process(sortOrder) },
                 { roomSummaryMapper.map(it) }
         )
     }
 
-    fun getRoomSummariesLive(queryParams: RoomSummaryQueryParams): LiveData<List<RoomSummary>> {
+    fun getRoomSummariesLive(queryParams: RoomSummaryQueryParams,
+                             sortOrder: RoomSortOrder = RoomSortOrder.NONE): LiveData<List<RoomSummary>> {
         return monarchy.findAllMappedWithChanges(
                 {
-                    roomSummariesQuery(it, queryParams)
-                            .sort(RoomSummaryEntityFields.LAST_ACTIVITY_TIME, Sort.DESCENDING)
+                    roomSummariesQuery(it, queryParams).process(sortOrder)
                 },
                 { roomSummaryMapper.map(it) }
         )
     }
 
-    fun getSpaceSummariesLive(queryParams: SpaceSummaryQueryParams): LiveData<List<RoomSummary>> {
-        return getRoomSummariesLive(queryParams)
+    fun getSpaceSummariesLive(queryParams: SpaceSummaryQueryParams,
+                              sortOrder: RoomSortOrder = RoomSortOrder.NONE): LiveData<List<RoomSummary>> {
+        return getRoomSummariesLive(queryParams, sortOrder)
     }
 
     fun getSpaceSummary(roomIdOrAlias: String): RoomSummary? {
@@ -122,8 +123,9 @@ internal class RoomSummaryDataSource @Inject constructor(@SessionDatabase privat
         }
     }
 
-    fun getSpaceSummaries(spaceSummaryQueryParams: SpaceSummaryQueryParams): List<RoomSummary> {
-        return getRoomSummaries(spaceSummaryQueryParams)
+    fun getSpaceSummaries(spaceSummaryQueryParams: SpaceSummaryQueryParams,
+                          sortOrder: RoomSortOrder = RoomSortOrder.NONE): List<RoomSummary> {
+        return getRoomSummaries(spaceSummaryQueryParams, sortOrder)
     }
 
     fun getRootSpaceSummaries(): List<RoomSummary> {
diff --git a/matrix-sdk-android/src/main/java/org/matrix/android/sdk/internal/session/space/DefaultSpaceService.kt b/matrix-sdk-android/src/main/java/org/matrix/android/sdk/internal/session/space/DefaultSpaceService.kt
@@ -23,6 +23,7 @@ import org.matrix.android.sdk.api.session.events.model.Event
 import org.matrix.android.sdk.api.session.events.model.EventType
 import org.matrix.android.sdk.api.session.events.model.toContent
 import org.matrix.android.sdk.api.session.events.model.toModel
+import org.matrix.android.sdk.api.session.room.RoomSortOrder
 import org.matrix.android.sdk.api.session.room.model.GuestAccess
 import org.matrix.android.sdk.api.session.room.model.Membership
 import org.matrix.android.sdk.api.session.room.model.PowerLevelsContent
@@ -94,12 +95,14 @@ internal class DefaultSpaceService @Inject constructor(
         return spaceGetter.get(spaceId)
     }
 
-    override fun getSpaceSummariesLive(queryParams: SpaceSummaryQueryParams): LiveData<List<RoomSummary>> {
-        return roomSummaryDataSource.getSpaceSummariesLive(queryParams)
+    override fun getSpaceSummariesLive(queryParams: SpaceSummaryQueryParams,
+                                       sortOrder: RoomSortOrder): LiveData<List<RoomSummary>> {
+        return roomSummaryDataSource.getSpaceSummariesLive(queryParams, sortOrder)
     }
 
-    override fun getSpaceSummaries(spaceSummaryQueryParams: SpaceSummaryQueryParams): List<RoomSummary> {
-        return roomSummaryDataSource.getSpaceSummaries(spaceSummaryQueryParams)
+    override fun getSpaceSummaries(spaceSummaryQueryParams: SpaceSummaryQueryParams,
+                                   sortOrder: RoomSortOrder): List<RoomSummary> {
+        return roomSummaryDataSource.getSpaceSummaries(spaceSummaryQueryParams, sortOrder)
     }
 
     override fun getRootSpaceSummaries(): List<RoomSummary> {
diff --git a/vector/src/main/java/im/vector/app/features/home/ShortcutsHandler.kt b/vector/src/main/java/im/vector/app/features/home/ShortcutsHandler.kt
@@ -22,51 +22,85 @@ import android.os.Build
 import androidx.core.content.getSystemService
 import androidx.core.content.pm.ShortcutManagerCompat
 import im.vector.app.core.di.ActiveSessionHolder
+import im.vector.app.features.pin.PinCodeStore
+import im.vector.app.features.pin.PinCodeStoreListener
 import io.reactivex.disposables.Disposable
 import io.reactivex.disposables.Disposables
 import org.matrix.android.sdk.api.session.room.RoomSortOrder
 import org.matrix.android.sdk.api.session.room.model.Membership
+import org.matrix.android.sdk.api.session.room.model.RoomSummary
 import org.matrix.android.sdk.api.session.room.roomSummaryQueryParams
 import org.matrix.android.sdk.rx.asObservable
+import timber.log.Timber
 import javax.inject.Inject
 
 class ShortcutsHandler @Inject constructor(
         private val context: Context,
         private val shortcutCreator: ShortcutCreator,
-        private val activeSessionHolder: ActiveSessionHolder
-) {
+        private val activeSessionHolder: ActiveSessionHolder,
+        private val pinCodeStore: PinCodeStore
+) : PinCodeStoreListener {
+    private val isRequestPinShortcutSupported = ShortcutManagerCompat.isRequestPinShortcutSupported(context)
+
+    // Value will be set correctly if necessary
+    private var hasPinCode = true
 
     fun observeRoomsAndBuildShortcuts(): Disposable {
         if (Build.VERSION.SDK_INT < Build.VERSION_CODES.N_MR1) {
             // No op
             return Disposables.empty()
         }
 
-        return activeSessionHolder.getSafeActiveSession()
-                ?.getPagedRoomSummariesLive(
-                        roomSummaryQueryParams {
-                            memberships = listOf(Membership.JOIN)
-                        },
-                        sortOrder = RoomSortOrder.PRIORITY_AND_ACTIVITY
-                )
-                ?.asObservable()
-                ?.subscribe { rooms ->
+        hasPinCode = pinCodeStore.getEncodedPin() != null
+
+        val session = activeSessionHolder.getSafeActiveSession() ?: return Disposables.empty()
+        return session.getRoomSummariesLive(
+                roomSummaryQueryParams {
+                    memberships = listOf(Membership.JOIN)
+                },
+                sortOrder = RoomSortOrder.PRIORITY_AND_ACTIVITY
+        )
+                .asObservable()
+                .doOnSubscribe { pinCodeStore.addListener(this) }
+                .doFinally { pinCodeStore.removeListener(this) }
+                .subscribe { rooms ->
                     // Remove dead shortcuts (i.e. deleted rooms)
-                    val roomIds = rooms.map { it.roomId }
-                    val deadShortcutIds = ShortcutManagerCompat.getShortcuts(context, ShortcutManagerCompat.FLAG_MATCH_DYNAMIC)
-                            .map { it.id }
-                            .filter { !roomIds.contains(it) }
-                    ShortcutManagerCompat.removeLongLivedShortcuts(context, deadShortcutIds)
-
-                    val shortcuts = rooms.mapIndexed { index, room ->
-                        shortcutCreator.create(room, index)
-                    }
-
-                    shortcuts.forEach { shortcut ->
-                        ShortcutManagerCompat.pushDynamicShortcut(context, shortcut)
-                    }
+                    removeDeadShortcut(rooms.map { it.roomId })
+
+                    // Create shortcuts
+                    createShortcuts(rooms)
                 }
-                ?: Disposables.empty()
+    }
+
+    private fun removeDeadShortcut(roomIds: List<String>) {
+        val deadShortcutIds = ShortcutManagerCompat.getShortcuts(context, ShortcutManagerCompat.FLAG_MATCH_DYNAMIC)
+                .map { it.id }
+                .filter { !roomIds.contains(it) }
+
+        if (deadShortcutIds.isNotEmpty()) {
+            Timber.d("Removing shortcut(s) $deadShortcutIds")
+            ShortcutManagerCompat.removeLongLivedShortcuts(context, deadShortcutIds)
+            if (isRequestPinShortcutSupported) {
+                if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.N_MR1) {
+                    context.getSystemService<ShortcutManager>()?.disableShortcuts(deadShortcutIds)
+                }
+            }
+        }
+    }
+
+    private fun createShortcuts(rooms: List<RoomSummary>) {
+        if (hasPinCode) {
+            // No shortcut in this case (privacy)
+            ShortcutManagerCompat.removeAllDynamicShortcuts(context)
+        } else {
+            val shortcuts = rooms.mapIndexed { index, room ->
+                shortcutCreator.create(room, index)
+            }
+
+            shortcuts.forEach { shortcut ->
+                ShortcutManagerCompat.pushDynamicShortcut(context, shortcut)
+            }
+        }
     }
 
     fun clearShortcuts() {
@@ -82,7 +116,7 @@ class ShortcutsHandler @Inject constructor(
         ShortcutManagerCompat.removeLongLivedShortcuts(context, shortcuts)
 
         // We can only disabled pinned shortcuts with the API, but at least it will prevent the crash
-        if (ShortcutManagerCompat.isRequestPinShortcutSupported(context)) {
+        if (isRequestPinShortcutSupported) {
             if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.N_MR1) {
                 context.getSystemService<ShortcutManager>()
                         ?.let {
@@ -91,4 +125,14 @@ class ShortcutsHandler @Inject constructor(
             }
         }
     }
+
+    override fun onPinSetUpChange(isConfigured: Boolean) {
+        hasPinCode = isConfigured
+        if (isConfigured) {
+            // Remove shortcuts immediately
+            ShortcutManagerCompat.removeAllDynamicShortcuts(context)
+        }
+        // Else shortcut will be created next time any room summary is updated, or
+        // next time the app is started which is acceptable
+    }
 }
diff --git a/vector/src/main/java/im/vector/app/features/pin/PinCodeStore.kt b/vector/src/main/java/im/vector/app/features/pin/PinCodeStore.kt
@@ -25,6 +25,7 @@ import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.withContext
 import org.matrix.android.sdk.api.extensions.orFalse
 import javax.inject.Inject
+import javax.inject.Singleton
 import kotlin.coroutines.resume
 import kotlin.coroutines.suspendCoroutine
 
@@ -56,26 +57,40 @@ interface PinCodeStore {
      * Will reset the counters
      */
     fun resetCounters()
+
+    fun addListener(listener: PinCodeStoreListener)
+    fun removeListener(listener: PinCodeStoreListener)
+}
+
+interface PinCodeStoreListener {
+    fun onPinSetUpChange(isConfigured: Boolean)
 }
 
+@Singleton
 class SharedPrefPinCodeStore @Inject constructor(private val sharedPreferences: SharedPreferences) : PinCodeStore {
+    private val listeners = mutableSetOf<PinCodeStoreListener>()
 
-    override suspend fun storeEncodedPin(encodePin: String) = withContext(Dispatchers.IO) {
-        sharedPreferences.edit {
-            putString(ENCODED_PIN_CODE_KEY, encodePin)
+    override suspend fun storeEncodedPin(encodePin: String) {
+        withContext(Dispatchers.IO) {
+            sharedPreferences.edit {
+                putString(ENCODED_PIN_CODE_KEY, encodePin)
+            }
         }
+        listeners.forEach { it.onPinSetUpChange(isConfigured = true) }
     }
 
-    override suspend fun deleteEncodedPin() = withContext(Dispatchers.IO) {
-        // Also reset the counters
-        resetCounters()
-        sharedPreferences.edit {
-            remove(ENCODED_PIN_CODE_KEY)
+    override suspend fun deleteEncodedPin() {
+        withContext(Dispatchers.IO) {
+            // Also reset the counters
+            resetCounters()
+            sharedPreferences.edit {
+                remove(ENCODED_PIN_CODE_KEY)
+            }
+            awaitPinCodeCallback<Boolean> {
+                PFSecurityManager.getInstance().pinCodeHelper.delete(it)
+            }
         }
-        awaitPinCodeCallback<Boolean> {
-            PFSecurityManager.getInstance().pinCodeHelper.delete(it)
-        }
-        return@withContext
+        listeners.forEach { it.onPinSetUpChange(isConfigured = false) }
     }
 
     override fun getEncodedPin(): String? {
@@ -124,6 +139,14 @@ class SharedPrefPinCodeStore @Inject constructor(private val sharedPreferences:
         }
     }
 
+    override fun addListener(listener: PinCodeStoreListener) {
+        listeners.add(listener)
+    }
+
+    override fun removeListener(listener: PinCodeStoreListener) {
+        listeners.remove(listener)
+    }
+
     private suspend inline fun <T> awaitPinCodeCallback(crossinline callback: (PFPinCodeHelperCallback<T>) -> Unit) = suspendCoroutine<PFResult<T>> { cont ->
         callback(PFPinCodeHelperCallback<T> { result -> cont.resume(result) })
     }

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+Do not show shortcuts if a PIN code is set`
Original file line number	Diff line number	Diff line change
`@@ -85,12 +85,14 @@ internal class DefaultRoomService @Inject constructor(`
`85`	`85`	`return roomSummaryDataSource.getRoomSummary(roomIdOrAlias)`
`86`	`86`	`}`
`87`	`87`
`88`		`- override fun getRoomSummaries(queryParams: RoomSummaryQueryParams): List<RoomSummary> {`
`89`		`- return roomSummaryDataSource.getRoomSummaries(queryParams)`
	`88`	`+ override fun getRoomSummaries(queryParams: RoomSummaryQueryParams,`
	`89`	`+ sortOrder: RoomSortOrder): List<RoomSummary> {`
	`90`	`+ return roomSummaryDataSource.getRoomSummaries(queryParams, sortOrder)`
`90`	`91`	`}`
`91`	`92`
`92`		`- override fun getRoomSummariesLive(queryParams: RoomSummaryQueryParams): LiveData<List<RoomSummary>> {`
`93`		`- return roomSummaryDataSource.getRoomSummariesLive(queryParams)`
	`93`	`+ override fun getRoomSummariesLive(queryParams: RoomSummaryQueryParams,`
	`94`	`+ sortOrder: RoomSortOrder): LiveData<List<RoomSummary>> {`
	`95`	`+ return roomSummaryDataSource.getRoomSummariesLive(queryParams, sortOrder)`
`94`	`96`	`}`
`95`	`97`
`96`	`98`	`override fun getPagedRoomSummariesLive(queryParams: RoomSummaryQueryParams,`
Original file line number	Diff line number	Diff line change
`@@ -25,7 +25,6 @@ import androidx.paging.PagedList`
`25`	`25`	`import com.zhuinden.monarchy.Monarchy`
`26`	`26`	`import io.realm.Realm`
`27`	`27`	`import io.realm.RealmQuery`
`28`		`-import io.realm.Sort`
`29`	`28`	`import io.realm.kotlin.where`
`30`	`29`	`import org.matrix.android.sdk.api.query.ActiveSpaceFilter`
`31`	`30`	`import org.matrix.android.sdk.api.query.RoomCategoryFilter`
`@@ -80,25 +79,27 @@ internal class RoomSummaryDataSource @Inject constructor(@SessionDatabase privat`
`80`	`79`	`}`
`81`	`80`	`}`
`82`	`81`
`83`		`- fun getRoomSummaries(queryParams: RoomSummaryQueryParams): List<RoomSummary> {`
	`82`	`+ fun getRoomSummaries(queryParams: RoomSummaryQueryParams,`
	`83`	`+ sortOrder: RoomSortOrder = RoomSortOrder.NONE): List<RoomSummary> {`
`84`	`84`	`return monarchy.fetchAllMappedSync(`
`85`		`- { roomSummariesQuery(it, queryParams) },`
	`85`	`+ { roomSummariesQuery(it, queryParams).process(sortOrder) },`
`86`	`86`	`{ roomSummaryMapper.map(it) }`
`87`	`87`	`)`
`88`	`88`	`}`
`89`	`89`
`90`		`- fun getRoomSummariesLive(queryParams: RoomSummaryQueryParams): LiveData<List<RoomSummary>> {`
	`90`	`+ fun getRoomSummariesLive(queryParams: RoomSummaryQueryParams,`
	`91`	`+ sortOrder: RoomSortOrder = RoomSortOrder.NONE): LiveData<List<RoomSummary>> {`
`91`	`92`	`return monarchy.findAllMappedWithChanges(`
`92`	`93`	`{`
`93`		`- roomSummariesQuery(it, queryParams)`
`94`		`- .sort(RoomSummaryEntityFields.LAST_ACTIVITY_TIME, Sort.DESCENDING)`
	`94`	`+ roomSummariesQuery(it, queryParams).process(sortOrder)`
`95`	`95`	`},`
`96`	`96`	`{ roomSummaryMapper.map(it) }`
`97`	`97`	`)`
`98`	`98`	`}`
`99`	`99`
`100`		`- fun getSpaceSummariesLive(queryParams: SpaceSummaryQueryParams): LiveData<List<RoomSummary>> {`
`101`		`- return getRoomSummariesLive(queryParams)`
	`100`	`+ fun getSpaceSummariesLive(queryParams: SpaceSummaryQueryParams,`
	`101`	`+ sortOrder: RoomSortOrder = RoomSortOrder.NONE): LiveData<List<RoomSummary>> {`
	`102`	`+ return getRoomSummariesLive(queryParams, sortOrder)`
`102`	`103`	`}`
`103`	`104`
`104`	`105`	`fun getSpaceSummary(roomIdOrAlias: String): RoomSummary? {`
`@@ -122,8 +123,9 @@ internal class RoomSummaryDataSource @Inject constructor(@SessionDatabase privat`
`122`	`123`	`}`
`123`	`124`	`}`
`124`	`125`
`125`		`- fun getSpaceSummaries(spaceSummaryQueryParams: SpaceSummaryQueryParams): List<RoomSummary> {`
`126`		`- return getRoomSummaries(spaceSummaryQueryParams)`
	`126`	`+ fun getSpaceSummaries(spaceSummaryQueryParams: SpaceSummaryQueryParams,`
	`127`	`+ sortOrder: RoomSortOrder = RoomSortOrder.NONE): List<RoomSummary> {`
	`128`	`+ return getRoomSummaries(spaceSummaryQueryParams, sortOrder)`
`127`	`129`	`}`
`128`	`130`
`129`	`131`	`fun getRootSpaceSummaries(): List<RoomSummary> {`