From a2d64dd854f80e0b74ddd3c26aa8c020eef1292d Mon Sep 17 00:00:00 2001
From: Will Sheldon <114631109+wssheldon@users.noreply.github.com>
Date: Wed, 29 Nov 2023 13:37:51 -0800
Subject: [PATCH] Generalize LockButton

---
 .../static/dispatch/src/case/PageHeader.vue   |  6 +++-
 .../dispatch/src/components/LockButton.vue    | 30 +++++++++++--------
 2 files changed, 23 insertions(+), 13 deletions(-)

diff --git a/src/dispatch/static/dispatch/src/case/PageHeader.vue b/src/dispatch/static/dispatch/src/case/PageHeader.vue
index e0d28a7b1dc3..ec64a2c6a20b 100644
--- a/src/dispatch/static/dispatch/src/case/PageHeader.vue
+++ b/src/dispatch/static/dispatch/src/case/PageHeader.vue
@@ -11,7 +11,11 @@
           <ParticipantAvatarGroup :participants="caseParticipants" class="pl-3" v-bind="tooltip" />
         </template>
       </FancyTooltip>
-      <LockButton :case-visibility="caseVisibility" class="ml-n2 mr-n2" />
+      <LockButton
+        :subject-visibility="caseVisibility"
+        subject-type="case_management"
+        class="ml-n2 mr-n2"
+      />
       <EscalateButton class="ml-n4" />
       <v-divider vertical inset />
       <FancyTooltip text="Case details" :hotkeys="['⌘', '⇧', 'I']">
diff --git a/src/dispatch/static/dispatch/src/components/LockButton.vue b/src/dispatch/static/dispatch/src/components/LockButton.vue
index c31d0a50315b..203daecbe1bf 100644
--- a/src/dispatch/static/dispatch/src/components/LockButton.vue
+++ b/src/dispatch/static/dispatch/src/components/LockButton.vue
@@ -1,4 +1,3 @@
-<!-- LockButton.vue -->
 <template>
   <div>
     <v-btn variant="plain" :ripple="false" @click="dialog = true">
@@ -7,10 +6,10 @@
 
     <v-dialog v-model="dialog" max-width="600px">
       <v-card>
-        <v-card-title>Change Case Visibility</v-card-title>
+        <v-card-title>Change {{ subjectTitle }} Visibility</v-card-title>
         <v-card-text>
-          You are about to change the case visibility from <b>{{ visibility }}</b> to
-          <b>{{ toggleVisibility }}</b
+          You are about to change the {{ subjectTitle.toLowerCase() }} visibility from
+          <b>{{ visibility }}</b> to <b>{{ toggleVisibility }}</b
           >.
         </v-card-text>
         <v-card-actions>
@@ -26,9 +25,14 @@
 import { ref, watch, computed } from "vue"
 import { useStore } from "vuex"
 import CaseApi from "@/case/api"
+import IncidentApi from "@/incident/api"
 
 const props = defineProps({
-  caseVisibility: {
+  subjectVisibility: {
+    type: String,
+    required: true,
+  },
+  subjectType: {
     type: String,
     required: true,
   },
@@ -36,14 +40,16 @@ const props = defineProps({
 
 const store = useStore()
 const dialog = ref(false)
-const visibility = ref(props.caseVisibility)
+const visibility = ref(props.subjectVisibility)
 
 const lockIcon = computed(() => (visibility.value === "Open" ? "mdi-lock-open" : "mdi-lock"))
 
 const toggleVisibility = computed(() => (visibility.value === "Open" ? "Restricted" : "Open"))
 
+const subjectTitle = computed(() => (props.subjectType === "incident" ? "Incident" : "Case"))
+
 watch(
-  () => props.caseVisibility,
+  () => props.subjectVisibility,
   (newVal) => {
     visibility.value = newVal
   },
@@ -51,25 +57,25 @@ watch(
 )
 
 async function updateVisibility() {
-  const caseDetails = store.state.case_management.selected
+  const subject = store.state[props.subjectType].selected
   const previousVisibility = visibility.value // Store the previous visibility value
 
   // Optimistically update the UI
   visibility.value = toggleVisibility.value
-  caseDetails.visibility = visibility.value
+  subject.visibility = visibility.value
   dialog.value = false
 
   try {
-    await CaseApi.update(caseDetails.id, caseDetails)
+    await (props.subjectType === "incident" ? IncidentApi : CaseApi).update(subject.id, subject)
   } catch (e) {
-    console.error("Failed to update case visibility", e)
+    console.error(`Failed to update ${props.subjectType} visibility`, e)
 
     // If the API call fails, revert the visibility change and show a toast
     visibility.value = previousVisibility
     store.commit(
       "notification_backend/addBeNotification",
       {
-        text: "Failed to update case visibility",
+        text: `Failed to update ${props.subjectType} visibility`,
         type: "error",
       },
       { root: true }