From d1273f214e24bfeff08c40e45bb7c2b213ecfaac Mon Sep 17 00:00:00 2001
From: Luca Stefani <luca.stefani.ge1@gmail.com>
Date: Sat, 11 Apr 2020 12:19:06 +0200
Subject: [PATCH] Only cancel notification if update isn't installed

* Even if the package is removed on A/B devices
  we still want to show the reboot notification

Change-Id: I980d36766a1427874b65b11674a8f0ae4ce1db88
---
 src/org/lineageos/updater/controller/UpdaterService.java | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/org/lineageos/updater/controller/UpdaterService.java b/src/org/lineageos/updater/controller/UpdaterService.java
index 61db18ae..33716180 100644
--- a/src/org/lineageos/updater/controller/UpdaterService.java
+++ b/src/org/lineageos/updater/controller/UpdaterService.java
@@ -126,7 +126,10 @@ public class UpdaterService extends Service {
                     if (extras != null && downloadId.equals(
                             extras.getString(UpdaterController.EXTRA_DOWNLOAD_ID))) {
                         mNotificationBuilder.setExtras(null);
-                        mNotificationManager.cancel(NOTIFICATION_ID);
+                        UpdateInfo update = mUpdaterController.getUpdate(downloadId);
+                        if (update.getStatus() != UpdateStatus.INSTALLED) {
+                            mNotificationManager.cancel(NOTIFICATION_ID);
+                        }
                     }
                 }
             }