From 241e9022cf5b990245501845d6d226f2be3fc544 Mon Sep 17 00:00:00 2001
From: mauro <mauro.cappelli@celi.it>
Date: Wed, 26 May 2021 19:02:07 +0200
Subject: [PATCH] fix post merge with master

---
 .../tasks/sequenceLabeling/EntityItemBox.vue  |  6 --
 .../projects/_id/sequence-labeling/index.vue  | 77 ++++++++++++++++++-
 2 files changed, 75 insertions(+), 8 deletions(-)

diff --git a/frontend/components/tasks/sequenceLabeling/EntityItemBox.vue b/frontend/components/tasks/sequenceLabeling/EntityItemBox.vue
index 2becd6ae..143fef99 100644
--- a/frontend/components/tasks/sequenceLabeling/EntityItemBox.vue
+++ b/frontend/components/tasks/sequenceLabeling/EntityItemBox.vue
@@ -19,7 +19,6 @@
           @selectTarget="selectTarget(chunk)"
           @deleteLink="deleteLink($event.id, $event.ndx)"
           @selectNewLinkType="selectNewLinkType($event)"
-          @changeLinkType="changeLinkType($event)"
           @hideAllLinkMenus="hideAllLinkMenus()"
       />
       <v-menu
@@ -134,11 +133,6 @@ export default {
       default: () => ([]),
       required: true
     },
-    changeLinkType: {
-      type: Function,
-      default: () => ([]),
-      required: true
-    },
     hideAllLinkMenus: {
       type: Function,
       default: () => ([]),
diff --git a/frontend/pages/projects/_id/sequence-labeling/index.vue b/frontend/pages/projects/_id/sequence-labeling/index.vue
index 943440bd..cf85ddaa 100644
--- a/frontend/pages/projects/_id/sequence-labeling/index.vue
+++ b/frontend/pages/projects/_id/sequence-labeling/index.vue
@@ -22,11 +22,19 @@
         <v-card-text class="title">
           <entity-item-box
             :labels="labels"
+            :link-types="linkTypes"
             :text="doc.text"
             :entities="annotations"
             :delete-annotation="remove"
             :update-entity="update"
             :add-entity="add"
+            :source-chunk="sourceChunk"
+            :source-link-type="sourceLinkType"
+            :select-source="selectSource"
+            :select-target="selectTarget"
+            :delete-link="deleteLink"
+            :select-new-link-type="selectNewLinkType"
+            :hide-all-link-menus="hideAllLinkMenus"
           />
         </v-card-text>
       </v-card>
@@ -39,12 +47,18 @@
 
 <script>
 import _ from 'lodash'
+import {mapGetters} from 'vuex'
 import LayoutText from '@/components/tasks/layout/LayoutText'
 import ListMetadata from '@/components/tasks/metadata/ListMetadata'
 import ToolbarLaptop from '@/components/tasks/toolbar/ToolbarLaptop'
 import ToolbarMobile from '@/components/tasks/toolbar/ToolbarMobile'
 import EntityItemBox from '~/components/tasks/sequenceLabeling/EntityItemBox'
 
+const NONE = {
+  id: -1,
+  none: true
+};
+
 export default {
   layout: 'workspace',
 
@@ -76,12 +90,19 @@ export default {
       annotations: [],
       docs: [],
       labels: [],
+      links: [],
+      linkTypes: [],
       project: {},
-      enableAutoLabeling: false
+      enableAutoLabeling: false,
+      sourceChunk: NONE,
+      sourceLink: NONE,
+      sourceLinkType: NONE
     }
   },
 
   computed: {
+    ...mapGetters('auth', ['isAuthenticated', 'getUsername', 'getUserId']),
+
     shortKeys() {
       return Object.fromEntries(this.labels.map(item => [item.id, [item.suffixKey]]))
     },
@@ -108,25 +129,48 @@ export default {
 
   async created() {
     this.labels = await this.$services.label.list(this.projectId)
+
+    this.linkTypes = await this.$services.linkTypes.list(this.projectId)
+
     this.project = await this.$services.project.findById(this.projectId)
   },
 
   methods: {
     async list(docId) {
-      this.annotations = await this.$services.sequenceLabeling.list(this.projectId, docId)
+      this.hideAllLinkMenus();
+
+      const annotations = await this.$services.sequenceLabeling.list(this.projectId, docId);
+      const links = await this.$services.sequenceLabeling.listLinks(this.projectId);
+
+      annotations.forEach(function(annotation) {
+        annotation.links = links.filter(link => link.annotation_id_1 === annotation.id);
+      });
+
+      this.annotations = annotations;
+      this.links = links;
+    },
+
+    populateLinks() {
+      const links = this.links;
+      this.annotations.forEach(function(annotation) {
+        annotation.links = links.filter(link => link.annotation_id_1 === annotation.id);
+      });
     },
 
     async remove(id) {
+      this.hideAllLinkMenus();
       await this.$services.sequenceLabeling.delete(this.projectId, this.doc.id, id)
       await this.list(this.doc.id)
     },
 
     async add(startOffset, endOffset, labelId) {
+      this.hideAllLinkMenus();
       await this.$services.sequenceLabeling.create(this.projectId, this.doc.id, labelId, startOffset, endOffset)
       await this.list(this.doc.id)
     },
 
     async update(labelId, annotationId) {
+      this.hideAllLinkMenus();
       await this.$services.sequenceLabeling.changeLabel(this.projectId, this.doc.id, annotationId, labelId)
       await this.list(this.doc.id)
     },
@@ -148,6 +192,35 @@ export default {
       const approved = !this.doc.isApproved
       await this.$services.example.approve(this.projectId, this.doc.id, approved)
       await this.$fetch()
+    },
+
+    selectSource(chunk) {
+      this.sourceChunk = chunk;
+    },
+
+    async selectTarget(chunk) {
+      // to avoid duplicated links:
+      if (!chunk.links.find(ch => ch.id === this.sourceChunk.id)) {
+        await this.$services.sequenceLabeling.createLink(this.projectId, this.sourceChunk.id, chunk.id, this.sourceLinkType.id, this.getUserId);
+        await this.list(this.doc.id);
+      }
+      this.hideAllLinkMenus();
+    },
+
+    async deleteLink(id, ndx) {
+      await this.$services.sequenceLabeling.deleteLink(this.projectId, this.sourceChunk.links[ndx].id)
+      await this.list(this.doc.id)
+
+      this.hideAllLinkMenus();
+    },
+
+    selectNewLinkType(type) {
+      this.sourceLinkType = type;
+    },
+
+    hideAllLinkMenus() {
+      this.sourceChunk = NONE;
+      this.sourceLinkType = NONE;
     }
   },