From 3f44808332fd221e064a27e2ea8d5005b1098dfd Mon Sep 17 00:00:00 2001
From: Shivam Mishra <scm.mymail@gmail.com>
Date: Wed, 5 Jul 2023 17:33:07 +0530
Subject: [PATCH] feat: fetch label suggestions from the hook

---
 .../widgets/conversation/MessagesView.vue     |   8 ++
 .../conversation/LabelSuggestion.vue          | 101 ++++++++++++++++--
 2 files changed, 99 insertions(+), 10 deletions(-)

diff --git a/app/javascript/dashboard/components/widgets/conversation/MessagesView.vue b/app/javascript/dashboard/components/widgets/conversation/MessagesView.vue
index b201c91e7..b778e2a6c 100644
--- a/app/javascript/dashboard/components/widgets/conversation/MessagesView.vue
+++ b/app/javascript/dashboard/components/widgets/conversation/MessagesView.vue
@@ -72,6 +72,8 @@
           'bugs',
           'authority',
         ]"
+        :chat-labels="currentChat.labels"
+        :conversation-id="currentChat.id"
       />
     </ul>
     <div
@@ -303,6 +305,7 @@ export default {
   mounted() {
     this.addScrollListener();
     this.fetchAllAttachmentsFromCurrentChat();
+    this.fetchIntegrations();
   },
 
   beforeDestroy() {
@@ -314,6 +317,11 @@ export default {
     fetchAllAttachmentsFromCurrentChat() {
       this.$store.dispatch('fetchAllAttachments', this.currentChat.id);
     },
+    fetchIntegrations() {
+      if (!this.appIntegrations.length) {
+        this.$store.dispatch('integrations/get');
+      }
+    },
     removeBusListeners() {
       bus.$off(BUS_EVENTS.SCROLL_TO_MESSAGE, this.onScrollToMessage);
       bus.$off(BUS_EVENTS.SET_TWEET_REPLY, this.setSelectedTweet);
diff --git a/app/javascript/dashboard/components/widgets/conversation/conversation/LabelSuggestion.vue b/app/javascript/dashboard/components/widgets/conversation/conversation/LabelSuggestion.vue
index 2c2e2d237..1891b8e7f 100644
--- a/app/javascript/dashboard/components/widgets/conversation/conversation/LabelSuggestion.vue
+++ b/app/javascript/dashboard/components/widgets/conversation/conversation/LabelSuggestion.vue
@@ -1,25 +1,30 @@
 <template>
-  <li v-if="preparedLabels.length" class="label-suggestion right">
+  <li v-if="shouldShowSuggestions" class="label-suggestion right">
     <div class="wrap">
       <div class="label-suggestion--container">
         <h6 class="label-suggestion--title">Suggested labels</h6>
         <div class="label-suggestion--options">
-          <woot-label
+          <button
             v-for="label in preparedLabels"
             :key="label.title"
-            variant="smooth"
-            :dashed="true"
-            v-bind="label"
-          />
+            @click="addLabel(label.title)"
+          >
+            <woot-label variant="smooth" :dashed="true" v-bind="label" />
+          </button>
         </div>
         <woot-button
           variant="smooth"
           class="label--add"
           icon="add"
           size="tiny"
-          @click="addLabel"
+          @click="addAllLabels"
         >
-          Add all labls
+          <template v-if="selectedLabels.length === 0">
+            Add all labels
+          </template>
+          <template v-else>
+            Add selected labels
+          </template>
         </woot-button>
       </div>
       <div class="sender--info has-tooltip" data-original-title="null">
@@ -57,7 +62,9 @@
 <script>
 import WootButton from '../../../ui/WootButton.vue';
 import Avatar from '../../Avatar.vue';
+import OpenAPI from 'dashboard/api/integrations/openapi';
 import { mapGetters } from 'vuex';
+import aiMixin from 'dashboard/mixins/aiMixin';
 
 export default {
   name: 'LabelSuggestion',
@@ -65,18 +72,92 @@ export default {
     Avatar,
     WootButton,
   },
+  mixins: [aiMixin],
   props: {
-    labels: {
+    chatLabels: {
       type: Array,
+      required: false,
+      default: () => [],
+    },
+    conversationId: {
+      type: Number,
       required: true,
     },
   },
+  data() {
+    return {
+      suggestedLabels: [],
+      selectedLabels: [],
+    };
+  },
   computed: {
     ...mapGetters({
       allLabels: 'labels/getLabels',
     }),
     preparedLabels() {
-      return this.allLabels.filter(label => this.labels.includes(label.title));
+      return this.allLabels.filter(label =>
+        this.suggestedLabels.includes(label.title)
+      );
+    },
+    shouldShowSuggestions() {
+      return this.preparedLabels.length && this.chatLabels.length === 0;
+    },
+  },
+  mounted() {
+    if (this.chatLabels.length !== 0) {
+      return;
+    }
+
+    this.fetchIntegrationsIfRequired().then(() => {
+      if (this.isAIIntegrationEnabled) {
+        this.fetchLabelSuggestion().then(labels => {
+          this.suggestedLabels = labels;
+        });
+      }
+    });
+  },
+  methods: {
+    async fetchLabelSuggestion() {
+      try {
+        const result = await OpenAPI.processEvent({
+          type: 'label_suggestion',
+          hookId: this.hookId,
+          conversationId: this.conversationId,
+        });
+
+        const {
+          data: { message: labels },
+        } = result;
+
+        return this.cleanLabels(labels);
+      } catch (error) {
+        return [];
+      }
+    },
+    cleanLabels(labels) {
+      return labels
+        .toLowerCase() // Set it to lowercase
+        .split(',') // split the string into an array
+        .filter(label => label.trim()) // remove any empty strings
+        .filter((label, index, self) => self.indexOf(label) === index) // remove any duplicates
+        .map(label => label.trim()); // trim the words
+    },
+    addLabel(label) {
+      if (!this.selectedLabels.includes(label)) {
+        this.selectedLabels.push(label);
+      } else {
+        this.selectedLabels = this.selectedLabels.filter(l => l !== label);
+      }
+    },
+    addAllLabels() {
+      if (this.selectedLabels.length) {
+        this.$emit('add-labels', this.selectedLabels);
+      } else {
+        this.$emit(
+          'add-labels',
+          this.preparedLabels.map(label => label.title)
+        );
+      }
     },
   },
 };