Refresh shortcuts after removing one

appshortcuts/src/main/java/de/mm20/launcher2/appshortcuts/AppShortcutRepository.kt

@@ -4,7 +4,11 @@ import android.content.Context
 import android.content.pm.LauncherActivityInfo
 import android.content.pm.LauncherApps
 import android.content.pm.PackageManager
+import android.content.pm.ShortcutInfo
+import android.os.Handler
+import android.os.Looper
 import android.os.Process
+import android.os.UserHandle
 import android.util.Log
 import androidx.core.content.getSystemService
 import de.mm20.launcher2.ktx.normalize
@@ -12,11 +16,11 @@ import de.mm20.launcher2.permissions.PermissionGroup
 import de.mm20.launcher2.permissions.PermissionsManager
 import de.mm20.launcher2.preferences.LauncherDataStore
 import de.mm20.launcher2.search.data.AppShortcut
+import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.Dispatchers
-import kotlinx.coroutines.flow.Flow
+import kotlinx.coroutines.Job
-import kotlinx.coroutines.flow.channelFlow
+import kotlinx.coroutines.channels.awaitClose
-import kotlinx.coroutines.flow.collectLatest
+import kotlinx.coroutines.flow.*
-import kotlinx.coroutines.flow.map
 import kotlinx.coroutines.withContext
 import org.apache.commons.text.similarity.FuzzyScore
 import java.util.*
@@ -38,6 +42,8 @@ internal class AppShortcutRepositoryImpl(
     private val dataStore: LauncherDataStore,
 ) : AppShortcutRepository {
 
+    private val scope = CoroutineScope(Dispatchers.Default + Job())
+
     override suspend fun getShortcutsForActivity(
         launcherActivityInfo: LauncherActivityInfo,
         count: Int,
@@ -84,6 +90,7 @@ internal class AppShortcutRepositoryImpl(
                     return@collectLatest
                 }
 
+                shortcutChangeEmitter.collectLatest {
                     val launcherApps =
                         context.getSystemService<LauncherApps>() ?: return@collectLatest send(
                             emptyList()
@@ -128,6 +135,53 @@ internal class AppShortcutRepositoryImpl(
                 }
             }
         }
+    }
+
+    private val shortcutChangeEmitter = callbackFlow {
+        send(Unit)
+        val launcherApps = context.getSystemService(Context.LAUNCHER_APPS_SERVICE) as LauncherApps
+
+        val callback = object : LauncherApps.Callback() {
+            override fun onPackageRemoved(packageName: String?, user: UserHandle?) {
+            }
+
+            override fun onPackageAdded(packageName: String?, user: UserHandle?) {
+            }
+
+            override fun onPackageChanged(packageName: String?, user: UserHandle?) {
+            }
+
+            override fun onPackagesAvailable(
+                packageNames: Array<out String>?,
+                user: UserHandle?,
+                replacing: Boolean
+            ) {
+            }
+
+            override fun onPackagesUnavailable(
+                packageNames: Array<out String>?,
+                user: UserHandle?,
+                replacing: Boolean
+            ) {
+            }
+
+            override fun onShortcutsChanged(
+                packageName: String,
+                shortcuts: MutableList<ShortcutInfo>,
+                user: UserHandle
+            ) {
+                super.onShortcutsChanged(packageName, shortcuts, user)
+                trySend(Unit)
+            }
+
+        }
+
+        launcherApps.registerCallback(callback, Handler(Looper.getMainLooper()))
+
+        awaitClose {
+            launcherApps.unregisterCallback(callback)
+        }
+    }.shareIn(scope, SharingStarted.WhileSubscribed(500), 1)
 
     override fun removePinnedShortcut(shortcut: AppShortcut) {
         val launcherApps = context.getSystemService(Context.LAUNCHER_APPS_SERVICE) as LauncherApps

ui/src/main/java/de/mm20/launcher2/ui/launcher/search/shortcut/ShortcutItemVM.kt

@@ -29,5 +29,6 @@ class ShortcutItemVM(private val shortcut: AppShortcut) : SearchableItemVM(short
 
     fun deleteShortcut() {
         shortcutRepository.removePinnedShortcut(shortcut)
+        favoritesRepository.unpinItem(shortcut)
     }
 }