From 9f9d65a74bb31993c11fdfb206fb3d7aced39bdf Mon Sep 17 00:00:00 2001
From: wangdan-fit2cloud <dan.wang@fit2cloud.com>
Date: Fri, 10 Nov 2023 11:36:04 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BF=AE=E6=94=B9=E6=96=87=E6=A1=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ui/src/api/dataset.ts                         | 14 +++-
 ui/src/api/team.ts                            |  3 +-
 ui/src/components/read-write/index.vue        | 13 ++--
 ui/src/request/index.ts                       |  4 +-
 ui/src/views/dataset/DatasetDocument.vue      | 65 +++++++++++++++++--
 .../dataset/component/SegmentPreview.vue      |  2 +-
 ui/src/views/dataset/index.vue                |  2 +-
 7 files changed, 86 insertions(+), 17 deletions(-)

diff --git a/ui/src/api/dataset.ts b/ui/src/api/dataset.ts
index 1442a77dd..edc6ef4e5 100644
--- a/ui/src/api/dataset.ts
+++ b/ui/src/api/dataset.ts
@@ -99,6 +99,17 @@ const putDocument: (dataset_id: string, document_id: string, data: any) => Promi
   return put(`${prefix}/${dataset_id}/document/${document_id}`, data)
 }
 
+/**
+ * 删除文档
+ * @param 参数 dataset_id, document_id,
+ */
+const delDocument: (dataset_id: string, document_id: string) => Promise<Result<boolean>> = (
+  dataset_id,
+  document_id
+) => {
+  return del(`${prefix}/${dataset_id}/document/${document_id}`)
+}
+
 export default {
   getDateset,
   getAllDateset,
@@ -106,5 +117,6 @@ export default {
   postDateset,
   postSplitDocument,
   getDocument,
-  putDocument
+  putDocument,
+  delDocument
 }
diff --git a/ui/src/api/team.ts b/ui/src/api/team.ts
index 1e08b684b..03c6b18d7 100644
--- a/ui/src/api/team.ts
+++ b/ui/src/api/team.ts
@@ -56,9 +56,10 @@ const putMemberPermissions: (member_id: String, body: any) => Promise<Result<any
   member_id,
   body
 ) => {
-  return put(`${prefix}/${member_id}`, undefined, body)
+  return put(`${prefix}/${member_id}`, body)
 }
 
+
 export default {
   getTeamMember,
   postCreatTeamMember,
diff --git a/ui/src/components/read-write/index.vue b/ui/src/components/read-write/index.vue
index ec63523a3..d0c06433d 100644
--- a/ui/src/components/read-write/index.vue
+++ b/ui/src/components/read-write/index.vue
@@ -12,12 +12,12 @@
       <div class="flex align-center" v-if="isEdit">
         <el-input ref="inputRef" v-model="writeValue" placeholder="请输入"></el-input>
         <span class="ml-4">
-          <el-button type="primary" text @click="submit">
+          <el-button type="primary" text @click="submit" :disabled="loading">
             <el-icon><Select /></el-icon>
           </el-button>
         </span>
         <span>
-          <el-button text @click="isEdit = false">
+          <el-button text @click="isEdit = false" :disabled="loading">
             <el-icon><CloseBold /></el-icon>
           </el-button>
         </span>
@@ -41,6 +41,7 @@ const props = defineProps({
 const emit = defineEmits(['change'])
 const isEdit = ref(false)
 const writeValue = ref('')
+const loading = ref(false)
 
 watch(isEdit, (bool) => {
   if (!bool) {
@@ -48,9 +49,13 @@ watch(isEdit, (bool) => {
   }
 })
 
-function submit() { 
+function submit() {
+  loading.value = true
   emit('change', writeValue.value)
-  isEdit.value = false
+  setTimeout(() => {
+    isEdit.value = false
+    loading.value = false
+  }, 200)
 }
 function editNameHandle(row: any) {
   writeValue.value = props.data
diff --git a/ui/src/request/index.ts b/ui/src/request/index.ts
index 92a39cd51..d195c3e85 100644
--- a/ui/src/request/index.ts
+++ b/ui/src/request/index.ts
@@ -149,8 +149,8 @@ export const put: (
   params?: unknown,
   data?: unknown,
   loading?: NProgress | Ref<boolean>
-) => Promise<Result<any>> = (url, params, data, loading) => {
-  return promise(request({ url: url, method: 'put', params, data }), loading)
+) => Promise<Result<any>> = (url, data, params, loading) => {
+  return promise(request({ url: url, method: 'put', data, params }), loading)
 }
 
 /**
diff --git a/ui/src/views/dataset/DatasetDocument.vue b/ui/src/views/dataset/DatasetDocument.vue
index 49f30814a..0094f840d 100644
--- a/ui/src/views/dataset/DatasetDocument.vue
+++ b/ui/src/views/dataset/DatasetDocument.vue
@@ -1,7 +1,7 @@
 <template>
   <LayoutContainer header="文档">
     <div class="main-calc-height">
-      <div class="p-24" v-loading="loading">
+      <div class="p-24">
         <div class="flex-between">
           <el-button type="primary" @click="router.push({ path: '/dataset/upload' })"
             >上传文档</el-button
@@ -22,6 +22,7 @@
           @changePage="handleCurrentChange"
           @cell-mouse-enter="cellMouseEnter"
           @cell-mouse-leave="cellMouseLeave"
+          v-loading="loading"
         >
           <el-table-column prop="name" label="文件名称" min-width="280">
             <template #default="{ row }">
@@ -53,7 +54,7 @@
           </el-table-column>
           <el-table-column prop="name" label="启动状态">
             <template #default="{ row }">
-              <el-switch v-model="row.is_active" />
+              <el-switch v-model="row.is_active" @change="changeState($event, row)" />
             </template>
           </el-table-column>
           <el-table-column prop="create_time" label="创建时间" width="170">
@@ -77,7 +78,7 @@
               </span>
               <span class="ml-4">
                 <el-tooltip effect="dark" content="删除" placement="top">
-                  <el-button type="primary" text>
+                  <el-button type="primary" text @click="deleteDocument(row)">
                     <el-icon><Delete /></el-icon>
                   </el-button>
                 </el-tooltip>
@@ -95,12 +96,11 @@ import { useRouter, useRoute } from 'vue-router'
 import datasetApi from '@/api/dataset'
 import { toThousands } from '@/utils/utils'
 import { datetimeFormat } from '@/utils/time'
+import { MsgSuccess, MsgConfirm } from '@/utils/message'
 const router = useRouter()
 const route = useRoute()
 const { params } = route
-const { datasetId } = params
-
-const inputRef = ref()
+const { datasetId } = params as any
 
 const loading = ref(false)
 const filterText = ref('')
@@ -113,7 +113,58 @@ const paginationConfig = reactive({
   total: 0
 })
 
-function editName(val: string) {}
+function deleteDocument(row: any) {
+  MsgConfirm(
+    `是否删除文档：${row.name} ?`,
+    `此文档下的 ${row.paragraph_count} 个分段都会被删除，请谨慎操作。`,
+    {
+      confirmButtonText: '删除',
+      confirmButtonClass: 'danger'
+    }
+  )
+    .then(() => {
+      loading.value = true
+      datasetApi
+        .delDocument(datasetId, row.id)
+        .then(() => {
+          MsgSuccess('删除成功')
+          getList()
+        })
+        .catch(() => {
+          loading.value = false
+        })
+    })
+    .catch(() => {})
+}
+
+function updateData(documentId: string, data: any) {
+  loading.value = true
+  datasetApi
+    .putDocument(datasetId, documentId, data)
+    .then((res) => {
+      const index = documentData.value.findIndex((v) => v.id === documentId)
+      documentData.value.splice(index, 1, res.data)
+      MsgSuccess('修改成功')
+      loading.value = false
+    })
+    .catch(() => {
+      loading.value = false
+    })
+}
+
+function changeState(bool: Boolean, row: any) {
+  const obj = {
+    is_active: bool
+  }
+  currentMouseId.value && updateData(row.id, obj)
+}
+
+function editName(val: string) {
+  const obj = {
+    name: val
+  }
+  currentMouseId.value && updateData(currentMouseId.value, obj)
+}
 
 function cellMouseEnter(row: any) {
   currentMouseId.value = row.id
diff --git a/ui/src/views/dataset/component/SegmentPreview.vue b/ui/src/views/dataset/component/SegmentPreview.vue
index 29df31797..d91ea6b83 100644
--- a/ui/src/views/dataset/component/SegmentPreview.vue
+++ b/ui/src/views/dataset/component/SegmentPreview.vue
@@ -74,7 +74,7 @@ function editHandle(item: any, index: number, cIndex: number) {
 }
 
 function deleteHandle(item: any, index: number, cIndex: number) {
-  MsgConfirm(`是否删除分段：${item.title}？`, `删除后将不会存入数据集，对本地文档无影响。`, {
+  MsgConfirm(`是否删除分段：${item.title} ?`, `删除后将不会存入数据集，对本地文档无影响。`, {
     confirmButtonText: '删除',
     confirmButtonClass: 'danger'
   })
diff --git a/ui/src/views/dataset/index.vue b/ui/src/views/dataset/index.vue
index 13352adf5..9dba34141 100644
--- a/ui/src/views/dataset/index.vue
+++ b/ui/src/views/dataset/index.vue
@@ -76,7 +76,7 @@ function loadDataset() {}
 
 function deleteDateset(row: any) {
   MsgConfirm(
-    `是否删除数据集：${row.name}？`,
+    `是否删除数据集：${row.name} ?`,
     `此数据集关联 ${row.char_length} 个应用，删除后无法恢复，请谨慎操作。`,
     {
       confirmButtonText: '删除',