From 277ee7bdefbb098e2e4184d220bd480b2f81e664 Mon Sep 17 00:00:00 2001
From: Hironsan <hiroki.nakayama.py@gmail.com>
Date: Tue, 8 Aug 2023 17:42:05 +0900
Subject: [PATCH] Enable to reset assignment

---
 frontend/components/example/ActionMenu.vue    |  8 +++++++-
 .../components/example/FormAssignment.vue     |  1 -
 .../example/FormResetAssignment.vue           | 19 +++++++++++++++++++
 frontend/pages/projects/_id/dataset/index.vue | 15 ++++++++++++++-
 .../example/apiAssignmentRepository.ts        |  5 +++++
 5 files changed, 45 insertions(+), 3 deletions(-)
 create mode 100644 frontend/components/example/FormResetAssignment.vue

diff --git a/frontend/components/example/ActionMenu.vue b/frontend/components/example/ActionMenu.vue
index 47d39b8e..1b0465c3 100644
--- a/frontend/components/example/ActionMenu.vue
+++ b/frontend/components/example/ActionMenu.vue
@@ -6,12 +6,13 @@
     @upload="$emit('upload')"
     @download="$emit('download')"
     @assign="$emit('assign')"
+    @reset="$emit('reset')"
   />
 </template>
 
 <script lang="ts">
 import Vue from 'vue'
-import { mdiAccountCheck, mdiUpload, mdiDownload } from '@mdi/js'
+import { mdiAccountCheck, mdiUpload, mdiDownload, mdiUpdate } from '@mdi/js'
 import ActionMenu from '~/components/utils/ActionMenu.vue'
 
 export default Vue.extend({
@@ -36,6 +37,11 @@ export default Vue.extend({
           title: 'Assign to member',
           icon: mdiAccountCheck,
           event: 'assign'
+        },
+        {
+          title: 'Reset Assignment',
+          icon: mdiUpdate,
+          event: 'reset'
         }
       ]
     }
diff --git a/frontend/components/example/FormAssignment.vue b/frontend/components/example/FormAssignment.vue
index f249e7d2..f21f56f5 100644
--- a/frontend/components/example/FormAssignment.vue
+++ b/frontend/components/example/FormAssignment.vue
@@ -117,7 +117,6 @@ export default Vue.extend({
         weight,
         member_id: this.members[i].id
       }))
-      console.log(workloads)
       this.$repositories.assignment.bulkAssign(this.projectId, {
         strategy_name: this.selectedStrategy,
         workloads
diff --git a/frontend/components/example/FormResetAssignment.vue b/frontend/components/example/FormResetAssignment.vue
new file mode 100644
index 00000000..e7c43ee3
--- /dev/null
+++ b/frontend/components/example/FormResetAssignment.vue
@@ -0,0 +1,19 @@
+<template>
+  <confirm-form
+    title="Reset assignment"
+    message="Are you sure you want to reset all the assignments?"
+    @ok="$emit('reset')"
+    @cancel="$emit('cancel')"
+  />
+</template>
+
+<script lang="ts">
+import Vue from 'vue'
+import ConfirmForm from '@/components/utils/ConfirmForm.vue'
+
+export default Vue.extend({
+  components: {
+    ConfirmForm
+  }
+})
+</script>
diff --git a/frontend/pages/projects/_id/dataset/index.vue b/frontend/pages/projects/_id/dataset/index.vue
index a66c2f04..b01bfd9c 100644
--- a/frontend/pages/projects/_id/dataset/index.vue
+++ b/frontend/pages/projects/_id/dataset/index.vue
@@ -5,6 +5,7 @@
         @upload="$router.push('dataset/import')"
         @download="$router.push('dataset/export')"
         @assign="dialogAssignment = true"
+        @reset="dialogReset = true"
       />
       <v-btn
         class="text-capitalize ms-2"
@@ -37,6 +38,9 @@
       <v-dialog v-model="dialogAssignment">
         <form-assignment @assigned="assigned" @cancel="dialogAssignment = false" />
       </v-dialog>
+      <v-dialog v-model="dialogReset">
+        <form-reset-assignment @cancel="dialogReset = false" @reset="resetAssignment" />
+      </v-dialog>
     </v-card-title>
     <image-list
       v-if="project.isImageProject"
@@ -90,6 +94,7 @@ import DocumentList from '@/components/example/DocumentList.vue'
 import FormAssignment from '~/components/example/FormAssignment.vue'
 import FormDelete from '@/components/example/FormDelete.vue'
 import FormDeleteBulk from '@/components/example/FormDeleteBulk.vue'
+import FormResetAssignment from '~/components/example/FormResetAssignment.vue'
 import ActionMenu from '~/components/example/ActionMenu.vue'
 import AudioList from '~/components/example/AudioList.vue'
 import ImageList from '~/components/example/ImageList.vue'
@@ -105,7 +110,8 @@ export default Vue.extend({
     ImageList,
     FormAssignment,
     FormDelete,
-    FormDeleteBulk
+    FormDeleteBulk,
+    FormResetAssignment
   },
 
   layout: 'project',
@@ -121,6 +127,7 @@ export default Vue.extend({
       dialogDelete: false,
       dialogDeleteAll: false,
       dialogAssignment: false,
+      dialogReset: false,
       item: {} as ExampleListDTO,
       selected: [] as ExampleDTO[],
       members: [] as MemberItem[],
@@ -216,6 +223,12 @@ export default Vue.extend({
     async assigned() {
       this.dialogAssignment = false
       this.item = await this.$services.example.list(this.projectId, this.$route.query)
+    },
+
+    async resetAssignment() {
+      this.dialogReset = false
+      await this.$repositories.assignment.reset(this.projectId)
+      this.item = await this.$services.example.list(this.projectId, this.$route.query)
     }
   }
 })
diff --git a/frontend/repositories/example/apiAssignmentRepository.ts b/frontend/repositories/example/apiAssignmentRepository.ts
index 762924b5..d15388ee 100644
--- a/frontend/repositories/example/apiAssignmentRepository.ts
+++ b/frontend/repositories/example/apiAssignmentRepository.ts
@@ -20,4 +20,9 @@ export class APIAssignmentRepository {
     const url = `/projects/${projectId}/assignments/bulk_assign`
     await this.request.post(url, workloadAllocation)
   }
+
+  async reset(projectId: string): Promise<void> {
+    const url = `/projects/${projectId}/assignments/reset`
+    await this.request.delete(url)
+  }
 }