From 4c245f93e4101a6a467403fd3fb45c0323bc792e Mon Sep 17 00:00:00 2001
From: wyang14 <william.1022@gmail.com>
Date: Tue, 9 Apr 2024 01:09:44 +0100
Subject: [PATCH] confirm before deleting knowledge base

---
 components/KnowledgeBaseDeletePrompt.vue | 28 +++++++++++++++++++
 pages/knowledgebases/index.vue           | 34 ++++++++++++++++++------
 2 files changed, 54 insertions(+), 8 deletions(-)
 create mode 100644 components/KnowledgeBaseDeletePrompt.vue

diff --git a/components/KnowledgeBaseDeletePrompt.vue b/components/KnowledgeBaseDeletePrompt.vue
new file mode 100644
index 0000000..dbeae76
--- /dev/null
+++ b/components/KnowledgeBaseDeletePrompt.vue
@@ -0,0 +1,28 @@
+<script setup lang="ts">
+import { KnowledgeBase } from '@prisma/client'
+
+const props = defineProps<{
+  knowledgeBase: KnowledgeBase,
+  onConfirm: () => void
+}>()
+
+const onSubmit = () => {
+  props.onConfirm()
+}
+
+</script>
+
+<template>
+  <UModal>
+    <div class="p-6 w-full">
+      <h2 class="font-bold text-xl mb-4">Confirm</h2>
+      <form class="space-y-4" @submit="onSubmit">
+        <div>Are you sure deleting knowledge base <span class="font-bold">{{ knowledgeBase.name }}</span>?</div>
+
+        <UButton type="submit">
+          Yes
+        </UButton>
+      </form>
+    </div>
+  </UModal>
+</template>
diff --git a/pages/knowledgebases/index.vue b/pages/knowledgebases/index.vue
index d915d0b..48f8287 100644
--- a/pages/knowledgebases/index.vue
+++ b/pages/knowledgebases/index.vue
@@ -1,8 +1,9 @@
 <script setup lang="ts">
 import { useStorage } from '@vueuse/core'
+import { useConfirmDialog } from '@vueuse/core'
 import { fetchHeadersOllama, fetchHeadersThirdApi } from '@/utils/settings'
 import { type KnowledgeBase } from '@prisma/client'
-import { KnowledgeBaseUpdateForm } from '#components'
+import { KnowledgeBaseUpdateForm, KnowledgeBaseDeletePrompt } from '#components'
 
 const router = useRouter()
 const toast = useToast()
@@ -14,6 +15,7 @@ const state = reactive({
   urls: ''
 })
 const modal = useModal()
+const { isRevealed, reveal, confirm, cancel, onReveal, onConfirm, onCancel } = useConfirmDialog()
 const currentSessionId = useStorage<number>('currentSessionId', 0)
 
 const validate = (data: typeof state) => {
@@ -89,7 +91,7 @@ const actionsItems = (row: KnowledgeBase) => {
   return [[{
     label: 'Delete',
     icon: 'i-heroicons-trash-20-solid',
-    click: () => onDelete(row.id)
+    click: () => onDelete(row)
   }],
   [{
     label: 'Update',
@@ -98,16 +100,19 @@ const actionsItems = (row: KnowledgeBase) => {
   }]]
 }
 
-const onDelete = async (id: number) => {
-  await $fetch(`/api/knowledgebases/${id}`, {
-    method: 'DELETE',
-    body: { id },
+const onDelete = async (row: KnowledgeBase) => {
+  modal.open(KnowledgeBaseDeletePrompt, {
+    knowledgeBase: row,
+    onConfirm: async () => {
+      await $fetch(`/api/knowledgebases/${row.id}`, {
+        method: 'DELETE'
+      })
+      refresh()
+    }
   })
-  refresh()
 }
 
 const onAppendFiles = async (row: KnowledgeBase) => {
-  console.log(row)
   modal.open(KnowledgeBaseUpdateForm, {
     knowledgeBase: row,
     onClose: () => modal.close(),
@@ -126,6 +131,19 @@ function reset() {
 
 <template>
   <div class="flex flex-row w-full">
+    <teleport to="body">
+      <div v-if="isRevealed" class="modal-bg">
+        <div class="modal">
+          <h2>Confirm?</h2>
+          <button @click="confirm">
+            Yes
+          </button>
+          <button @click="cancel">
+            Cancel
+          </button>
+        </div>
+      </div>
+    </teleport>
     <div class="px-6 w-[400px]">
       <h2 class="font-bold text-xl mb-4">Create a New Knowledge Base</h2>
       <UForm :state="state" :validate="validate" class="space-y-4" @submit="onSubmit">