Cancel the progress notification here

Author: Wavesonics

app/src/main/kotlin/com/darkrockstudios/app/securecamera/import/ImportPhotosViewModel.kt

@@ -1,17 +1,21 @@
 package com.darkrockstudios.app.securecamera.import
 
+import android.app.NotificationManager
+import android.content.Context
 import android.net.Uri
 import androidx.core.net.toUri
 import androidx.lifecycle.viewModelScope
 import androidx.work.*
 import com.darkrockstudios.app.securecamera.BaseViewModel
+import com.darkrockstudios.app.securecamera.import.ImportWorker.Companion.NOTIFICATION_ID
 import dev.whyoleg.cryptography.CryptographyProvider
 import dev.whyoleg.cryptography.algorithms.SHA512
 import dev.whyoleg.cryptography.operations.Hasher
 import kotlinx.coroutines.launch
 import timber.log.Timber
 
 class ImportPhotosViewModel(
+	private val appContext: Context,
 	private val workManager: WorkManager,
 ) : BaseViewModel<ImportPhotosState>() {
 
@@ -128,6 +132,10 @@ class ImportPhotosViewModel(
 		currentImportWorkName?.let { workName ->
 			Timber.d("Cancelling import work: $workName")
 			workManager.cancelUniqueWork(workName)
+
+			val notificationManager =
+				appContext.getSystemService(Context.NOTIFICATION_SERVICE) as NotificationManager
+			notificationManager.cancel(NOTIFICATION_ID)
 		}
 	}
 }