From 46e042907a42ac54662277a52f4f07da5da296f1 Mon Sep 17 00:00:00 2001
From: Hironsan <light.tree.1.13@gmail.com>
Date: Tue, 31 Mar 2020 11:23:50 +0900
Subject: [PATCH] Enable to handle multiline document, resolves #463, resolves
 #577

---
 frontend/components/molecules/EntityItem.vue  |  8 +++-
 .../organisms/annotation/EntityItemBox.vue    | 40 ++++++++++++++-----
 2 files changed, 36 insertions(+), 12 deletions(-)

diff --git a/frontend/components/molecules/EntityItem.vue b/frontend/components/molecules/EntityItem.vue
index 8d823140..cf7e8057 100644
--- a/frontend/components/molecules/EntityItem.vue
+++ b/frontend/components/molecules/EntityItem.vue
@@ -31,7 +31,7 @@
       </v-list-item>
     </v-list>
   </v-menu>
-  <span v-else>{{ content }}</span>
+  <span v-else :class="[newline ? 'newline' : '']">{{ content }}</span>
 </template>
 
 <script>
@@ -56,6 +56,9 @@ export default {
       type: Array,
       default: () => [],
       required: true
+    },
+    newline: {
+      type: Boolean
     }
   },
   data() {
@@ -142,4 +145,7 @@ export default {
   -webkit-font-smoothing: subpixel-antialiased;
   letter-spacing: .1em;
 }
+.newline {
+  width: 100%;
+}
 </style>
diff --git a/frontend/components/organisms/annotation/EntityItemBox.vue b/frontend/components/organisms/annotation/EntityItemBox.vue
index 20dffef0..c6460eb3 100644
--- a/frontend/components/organisms/annotation/EntityItemBox.vue
+++ b/frontend/components/organisms/annotation/EntityItemBox.vue
@@ -4,6 +4,7 @@
       v-for="(chunk, i) in chunks"
       :key="i"
       :content="chunk.text"
+      :newline="chunk.newline"
       :label="chunk.label"
       :color="chunk.color"
       :labels="labels"
@@ -96,16 +97,12 @@ export default {
     },
 
     chunks() {
-      const chunks = []
+      let chunks = []
       const entities = this.sortedEntities
       let startOffset = 0
       for (const entity of entities) {
         // add non-entities to chunks.
-        chunks.push({
-          label: null,
-          color: null,
-          text: this.text.slice(startOffset, entity.start_offset)
-        })
+        chunks = chunks.concat(this.makeChunks(this.text.slice(startOffset, entity.start_offset)))
         startOffset = entity.end_offset
 
         // add entities to chunks.
@@ -118,11 +115,7 @@ export default {
         })
       }
       // add the rest of text.
-      chunks.push({
-        label: null,
-        color: null,
-        text: this.text.slice(startOffset, this.text.length)
-      })
+      chunks = chunks.concat(this.makeChunks(this.text.slice(startOffset, this.text.length)))
       return chunks
     },
 
@@ -135,6 +128,31 @@ export default {
     }
   },
   methods: {
+    makeChunks(text) {
+      const chunks = []
+      const snippets = text.split('\n')
+      for (const snippet of snippets.slice(0, -1)) {
+        chunks.push({
+          label: null,
+          color: null,
+          text: snippet + '\n',
+          newline: false
+        })
+        chunks.push({
+          label: null,
+          color: null,
+          text: '',
+          newline: true
+        })
+      }
+      chunks.push({
+        label: null,
+        color: null,
+        text: snippets.slice(-1)[0],
+        newline: false
+      })
+      return chunks
+    },
     show(e) {
       e.preventDefault()
       this.showMenu = false