From c6ade79d6cf3ac3f7c700b8b00487a1541e827e5 Mon Sep 17 00:00:00 2001
From: Hironsan <light.tree.1.13@gmail.com>
Date: Tue, 8 Oct 2019 15:44:44 +0900
Subject: [PATCH] Update text classification components

---
 .../containers/TextClassification.vue         | 41 ++++++++++++--
 .../organisms/MultiClassClassification.vue    | 53 +++++++++++++++----
 2 files changed, 78 insertions(+), 16 deletions(-)

diff --git a/frontend/components/containers/TextClassification.vue b/frontend/components/containers/TextClassification.vue
index 00e6bf36..5b3aa49b 100644
--- a/frontend/components/containers/TextClassification.vue
+++ b/frontend/components/containers/TextClassification.vue
@@ -2,18 +2,23 @@
   <base-text-area>
     <template #title>
       <multi-class-classification
+        v-if="currentDoc"
         :labels="items"
+        :annotations="currentDoc.annotations"
+        :add-label="addLabel"
+        :delete-label="removeLabel"
       />
     </template>
     <template #content>
-      <div class="title">
-        Text area
+      <div v-if="currentDoc" class="title">
+        {{ currentDoc.text }}<!-- {{currentDoc}}-->
       </div>
     </template>
   </base-text-area>
 </template>
 
 <script>
+import { mapActions, mapGetters, mapState } from 'vuex'
 import BaseTextArea from '@/components/molecules/BaseTextArea'
 import MultiClassClassification from '@/components/organisms/MultiClassClassification'
 
@@ -24,15 +29,41 @@ export default {
   },
 
   computed: {
-    items() {
-      return ['foo', 'bar']
-    }
+    ...mapState('labels', ['items']),
+    ...mapGetters('documents', ['currentDoc'])
   },
 
   created() {
+    this.getLabelList()
+    this.getDocumentList()
   },
 
   methods: {
+    ...mapActions('labels', ['getLabelList']),
+    ...mapActions('documents', ['getDocumentList', 'deleteAnnotation', 'updateAnnotation', 'addAnnotation']),
+    removeLabel(annotationId) {
+      const payload = {
+        annotationId,
+        projectId: this.$route.params.id
+      }
+      this.deleteAnnotation(payload)
+    },
+    updateLabel(labelId, annotationId) {
+      const payload = {
+        annotationId,
+        label: labelId,
+        projectId: this.$route.params.id
+      }
+      this.updateAnnotation(payload)
+    },
+    addLabel(labelId) {
+      alert(labelId)
+      const payload = {
+        label: labelId,
+        projectId: this.$route.params.id
+      }
+      this.addAnnotation(payload)
+    }
   }
 }
 </script>
diff --git a/frontend/components/organisms/MultiClassClassification.vue b/frontend/components/organisms/MultiClassClassification.vue
index 31207f1f..6f9fc23c 100644
--- a/frontend/components/organisms/MultiClassClassification.vue
+++ b/frontend/components/organisms/MultiClassClassification.vue
@@ -1,21 +1,25 @@
 <template>
   <v-combobox
-    v-model="chips"
+    :value="annotatedLabels"
     :items="labels"
-    chips
-    clearable
+    item-text="text"
     label="Label"
+    hide-selected
+    chips
     multiple
+    @input="add"
   >
     <template v-slot:selection="{ attrs, item, select, selected }">
       <v-chip
         v-bind="attrs"
         :input-value="selected"
+        :color="item.background_color"
+        text-color="white"
         close
         @click="select"
-        @click:close="remove(item)"
+        @click:close="remove(item.id)"
       >
-        {{ item }}
+        {{ item.text }}
       </v-chip>
     </template>
   </v-combobox>
@@ -28,19 +32,46 @@ export default {
       type: Array,
       default: () => [],
       required: true
+    },
+    annotations: {
+      type: Array,
+      default: () => ([]),
+      required: true
+    },
+    addLabel: {
+      type: Function,
+      default: () => ([]),
+      required: true
+    },
+    deleteLabel: {
+      type: Function,
+      default: () => ([]),
+      required: true
     }
   },
 
-  data() {
-    return {
-      chips: []
+  computed: {
+    annotatedLabels() {
+      const labelIds = this.annotations.map(item => item.label)
+      return this.labels.filter(item => labelIds.includes(item.id))
+    },
+    labelObject() {
+      const obj = {}
+      for (const label of this.labels) {
+        obj[label.id] = label
+      }
+      return obj
     }
   },
 
   methods: {
-    remove(item) {
-      this.chips.splice(this.chips.indexOf(item), 1)
-      this.chips = [...this.chips]
+    add(labels) {
+      const label = labels[labels.length - 1]
+      this.addLabel(label.id)
+    },
+    remove(labelId) {
+      const annotation = this.annotations.find(item => item.label === labelId)
+      this.deleteLabel(annotation.id)
     }
   }
 }