From 0f0d760d47348a7eda3b5c512b071912136c0d1e Mon Sep 17 00:00:00 2001
From: Hironsan <light.tree.1.13@gmail.com>
Date: Fri, 12 Mar 2021 15:30:26 +0900
Subject: [PATCH] Add TextClassification.vue component

---
 .../TextClassification/TextClassification.vue | 85 +++++++++++++++++++
 1 file changed, 85 insertions(+)
 create mode 100644 frontend/components/tasks/TextClassification/TextClassification.vue

diff --git a/frontend/components/tasks/TextClassification/TextClassification.vue b/frontend/components/tasks/TextClassification/TextClassification.vue
new file mode 100644
index 00000000..d8a12b04
--- /dev/null
+++ b/frontend/components/tasks/TextClassification/TextClassification.vue
@@ -0,0 +1,85 @@
+<template>
+  <v-combobox
+    v-model="annotatedLabels"
+    :items="labels"
+    item-text="text"
+    :label="$t('labels.labels')"
+    hide-selected
+    chips
+    multiple
+    :search-input.sync="search"
+    @change="search=''"
+  >
+    <template v-slot:selection="{ attrs, item, select, selected }">
+      <v-chip
+        v-bind="attrs"
+        :input-value="selected"
+        :color="item.backgroundColor"
+        :text-color="$contrastColor(item.backgroundColor)"
+        close
+        @click="select"
+        @click:close="remove(item)"
+      >
+        {{ item.text }}
+      </v-chip>
+    </template>
+  </v-combobox>
+</template>
+
+<script>
+export default {
+  props: {
+    labels: {
+      type: Array,
+      default: () => [],
+      required: true
+    },
+    annotations: {
+      type: Array,
+      default: () => ([]),
+      required: true
+    }
+  },
+
+  data() {
+    return {
+      search: ''
+    }
+  },
+
+  computed: {
+    annotatedLabels: {
+      get() {
+        const labelIds = this.annotations.map(item => item.label)
+        return this.labels.filter(item => labelIds.includes(item.id))
+      },
+      set(newValue) {
+        if (newValue.length > this.annotations.length) {
+          const label = newValue[newValue.length - 1]
+          if (typeof label === 'object') {
+            this.add(label)
+          } else {
+            newValue.pop()
+          }
+        } else {
+          const label = this.annotatedLabels.find(x => !newValue.some(y => y.id === x.id))
+          if (typeof label === 'object') {
+            this.remove(label)
+          }
+        }
+      }
+    }
+  },
+
+  methods: {
+    add(label) {
+      this.$emit('add', label.id)
+    },
+
+    remove(label) {
+      const annotation = this.annotations.find(item => item.label === label.id)
+      this.$emit('remove', annotation.id)
+    }
+  }
+}
+</script>