From 965f760da066cb8c35c739088cfb3ac0e296a5c7 Mon Sep 17 00:00:00 2001
From: Hironsan <light.tree.1.13@gmail.com>
Date: Fri, 7 Jan 2022 09:20:38 +0900
Subject: [PATCH] Update useLabelList to pass LabelApplicationService

---
 frontend/composables/useLabelList.ts              | 15 +++++++--------
 .../projects/_id/image-classification/index.vue   |  5 +++--
 .../projects/_id/text-classification/index.vue    |  2 +-
 .../application/label/labelApplicationService.ts  |  3 ++-
 4 files changed, 13 insertions(+), 12 deletions(-)

diff --git a/frontend/composables/useLabelList.ts b/frontend/composables/useLabelList.ts
index b6c770c9..4a84b267 100644
--- a/frontend/composables/useLabelList.ts
+++ b/frontend/composables/useLabelList.ts
@@ -1,27 +1,26 @@
-import { computed, reactive, ref, useContext } from '@nuxtjs/composition-api'
+import { computed, reactive, useContext } from '@nuxtjs/composition-api'
 import { LabelDTO } from '@/services/application/label/labelData'
 import { CreateLabelCommand , UpdateLabelCommand } from '@/services/application/label/labelCommand'
+import { LabelApplicationService } from '@/services/application/label/labelApplicationService'
 
-
-export const useLabelList = () => {
+export const useLabelList = (service: LabelApplicationService) => {
   const state = reactive({
     labels: [] as LabelDTO[]
   })
 
   const { app } = useContext()
-  const $services = app.$services
 
   const getLabelList = async(
     projectId: string
   ) => {
-    state.labels = await $services.label.list(projectId)
+    state.labels = await service.list(projectId)
   }
 
   const createLabel = async(
     projectId: string,
     command: CreateLabelCommand
   ) => {
-    await $services.label.create(projectId, command)
+    await service.create(projectId, command)
     await getLabelList(projectId)
   }
 
@@ -29,14 +28,14 @@ export const useLabelList = () => {
     projectId: string,
     command: UpdateLabelCommand
   ) => {
-    await $services.label.update(projectId, command)
+    await service.update(projectId, command)
   }
 
   const deleteLabelList = async(
     projectId: string,
     items: LabelDTO[]
   ) => {
-    await $services.label.bulkDelete(projectId, items)
+    await service.bulkDelete(projectId, items)
     await getLabelList(projectId)
   }
 
diff --git a/frontend/pages/projects/_id/image-classification/index.vue b/frontend/pages/projects/_id/image-classification/index.vue
index 7a0f90d0..2646aa4a 100644
--- a/frontend/pages/projects/_id/image-classification/index.vue
+++ b/frontend/pages/projects/_id/image-classification/index.vue
@@ -70,7 +70,7 @@
 <script>
 import _ from 'lodash'
 import { mdiText, mdiFormatListBulleted } from '@mdi/js'
-import { toRefs } from '@nuxtjs/composition-api'
+import { toRefs, useContext } from '@nuxtjs/composition-api'
 import LabelGroup from '@/components/tasks/textClassification/LabelGroup'
 import LabelSelect from '@/components/tasks/textClassification/LabelSelect'
 import LayoutText from '@/components/tasks/layout/LayoutText'
@@ -96,7 +96,8 @@ export default {
   },
 
   setup() {
-    const { state, getLabelList, shortKeys } = useLabelList()
+    const { app } = useContext()
+    const { state, getLabelList, shortKeys } = useLabelList(app.$services.categoryType)
 
     return {
       ...toRefs(state),
diff --git a/frontend/pages/projects/_id/text-classification/index.vue b/frontend/pages/projects/_id/text-classification/index.vue
index be405c99..bd502803 100644
--- a/frontend/pages/projects/_id/text-classification/index.vue
+++ b/frontend/pages/projects/_id/text-classification/index.vue
@@ -96,7 +96,7 @@ export default {
       removeTeacher
     } = useTeacherList(app.$services.textClassification)
     const enableAutoLabeling = ref(false)
-    const { state: labelState, getLabelList, shortKeys } = useLabelList()
+    const { state: labelState, getLabelList, shortKeys } = useLabelList(app.$services.categoryType)
     const labelComponent = ref('label-group')
 
     getLabelList(projectId)
diff --git a/frontend/services/application/label/labelApplicationService.ts b/frontend/services/application/label/labelApplicationService.ts
index 781ff72e..c8c87293 100644
--- a/frontend/services/application/label/labelApplicationService.ts
+++ b/frontend/services/application/label/labelApplicationService.ts
@@ -1,4 +1,5 @@
 import { LabelDTO } from './labelData'
+import { CreateLabelCommand } from './labelCommand';
 import { LabelRepository } from '~/domain/models/label/labelRepository'
 import { LabelItem } from '~/domain/models/label/label'
 
@@ -13,7 +14,7 @@ export class LabelApplicationService {
     return items.map(item => new LabelDTO(item))
   }
 
-  public create(projectId: string, item: LabelDTO): void {
+  public create(projectId: string, item: CreateLabelCommand): void {
     const label = new LabelItem(0, item.text, item.prefixKey, item.suffixKey, item.backgroundColor, item.textColor)
     this.repository.create(projectId, label)
   }