Add flag to indicate a drft has been deleted

Author: Rkareko

android/quest/src/main/java/org/smartregister/fhircore/quest/ui/dialog/AlertDialogViewModel.kt

@@ -17,15 +17,20 @@
 package org.smartregister.fhircore.quest.ui.dialog
 
 import androidx.lifecycle.ViewModel
-import androidx.lifecycle.viewModelScope
 import dagger.hilt.android.lifecycle.HiltViewModel
+import java.util.Date
 import javax.inject.Inject
-import kotlinx.coroutines.launch
-import kotlinx.coroutines.withContext
+import org.hl7.fhir.r4.model.CodeableConcept
+import org.hl7.fhir.r4.model.Coding
+import org.hl7.fhir.r4.model.Flag
+import org.hl7.fhir.r4.model.Identifier
+import org.hl7.fhir.r4.model.Period
+import org.hl7.fhir.r4.model.QuestionnaireResponse
 import org.hl7.fhir.r4.model.QuestionnaireResponse.QuestionnaireResponseStatus
 import org.smartregister.fhircore.engine.configuration.QuestionnaireConfig
 import org.smartregister.fhircore.engine.data.local.DefaultRepository
 import org.smartregister.fhircore.engine.util.DispatcherProvider
+import org.smartregister.fhircore.engine.util.extension.asReference
 
 @HiltViewModel
 class AlertDialogViewModel
@@ -43,22 +48,64 @@ constructor(
       return
     }
 
-    viewModelScope.launch {
-      withContext(dispatcherProvider.io()) {
-        val questionnaireResponse =
-          defaultRepository.searchQuestionnaireResponse(
-            resourceId = questionnaireConfig.resourceIdentifier!!,
-            resourceType = questionnaireConfig.resourceType!!,
-            questionnaireId = questionnaireConfig.id,
-            encounterId = null,
-            questionnaireResponseStatus = QuestionnaireResponseStatus.INPROGRESS.toCode(),
-          )
+    val questionnaireResponse =
+      defaultRepository.searchQuestionnaireResponse(
+        resourceId = questionnaireConfig.resourceIdentifier!!,
+        resourceType = questionnaireConfig.resourceType!!,
+        questionnaireId = questionnaireConfig.id,
+        encounterId = null,
+        questionnaireResponseStatus = QuestionnaireResponseStatus.INPROGRESS.toCode(),
+      )
 
-        if (questionnaireResponse != null) {
-          questionnaireResponse.status = QuestionnaireResponseStatus.STOPPED
-          defaultRepository.update(questionnaireResponse)
+    if (questionnaireResponse != null) {
+      questionnaireResponse.status = QuestionnaireResponseStatus.STOPPED
+      defaultRepository.update(questionnaireResponse)
+      defaultRepository.addOrUpdate(
+        resource = createDeleteDraftFlag(questionnaireConfig, questionnaireResponse),
+      )
+    }
+  }
+
+  fun createDeleteDraftFlag(
+    questionnaireConfig: QuestionnaireConfig,
+    questionnaireResponse: QuestionnaireResponse,
+  ): Flag {
+    return Flag().apply {
+      subject =
+        questionnaireConfig.resourceType?.let {
+          questionnaireConfig.resourceIdentifier?.asReference(
+            it,
+          )
+        }
+      identifier =
+        listOf(
+          Identifier().apply { value = questionnaireResponse.id },
+        )
+      status = Flag.FlagStatus.ACTIVE
+      code =
+        CodeableConcept().apply {
+          coding =
+            listOf(
+              Coding().apply {
+                system = FLAG_SYSTEM
+                code = FLAG_CODE
+                display = FLAG_DISPLAY
+              },
+            )
+          text = FLAG_TEXT
+        }
+      period =
+        Period().apply {
+          start = Date()
+          end = Date()
         }
-      }
     }
   }
+
+  companion object {
+    const val FLAG_SYSTEM = "http://smartregister.org/"
+    const val FLAG_CODE = "delete_draft"
+    const val FLAG_DISPLAY = "Delete Draft"
+    const val FLAG_TEXT = "QR Draft has been deleted"
+  }
 }