From cacec3ad6e154b86ebbedbfde2a86ed833326d14 Mon Sep 17 00:00:00 2001
From: dragongoose <19649813+dragongoose@users.noreply.github.com>
Date: Thu, 30 Mar 2023 10:30:21 -0400
Subject: [PATCH] Add chat badges

---
 src/components/Badge.vue      | 29 +++++++++++++++++
 src/components/TwitchChat.vue | 59 +++++++++++++++++++++++++++++------
 2 files changed, 79 insertions(+), 9 deletions(-)
 create mode 100644 src/components/Badge.vue

diff --git a/src/components/Badge.vue b/src/components/Badge.vue
new file mode 100644
index 0000000..4319430
--- /dev/null
+++ b/src/components/Badge.vue
@@ -0,0 +1,29 @@
+<script lang="ts">
+interface Badge {
+    id: string,
+    title: string,
+    setId: string,
+    version: string,
+    images: { [k:string]: string }
+}
+
+export default {
+  props: {
+    badgeInfo: {
+      type: Object,
+    }
+  },
+
+  setup(props) {
+    const data = props.badgeInfo as Badge
+    return {
+      src: data.images.image1x,
+      name: data.title
+    }
+  }
+}
+</script>
+
+<template>
+<img :src="src" class="w-5 h-5">
+</template>
diff --git a/src/components/TwitchChat.vue b/src/components/TwitchChat.vue
index d9c993f..d0daa97 100644
--- a/src/components/TwitchChat.vue
+++ b/src/components/TwitchChat.vue
@@ -1,5 +1,14 @@
 <script lang="ts">
 import { ref, type Ref } from 'vue'
+import BadgeVue from './Badge.vue'
+
+interface Badge {
+    id: string,
+    title: string,
+    setId: string,
+    version: string,
+    images: { [k:string]: string }
+}
 
 export default {
   props: {
@@ -13,7 +22,7 @@ export default {
       type: String
     }
   },
-  setup(props) {
+  async setup(props) {
     let messages: Ref<
       {
         username: string
@@ -23,12 +32,15 @@ export default {
         tags: Record<string, string>
       }[]
     > = ref([])
-    let ws = new WebSocket('ws://localhost:7000')
+    
+    const badgesFetch = await fetch(`${import.meta.env.VITE_BACKEND_URL}/api/badges?channelName=${props.channelName}`)
+    let badges: Badge[] =  await badgesFetch.json()
 
     return {
-      ws,
+      ws: new WebSocket('ws://localhost:7000'),
       messages,
-      props
+      badges,
+      props,
     }
   },
   mounted() {
@@ -40,13 +52,13 @@ export default {
         chatStatusMessage.textContent = `Connected to ${this.channelName}`
       } else {
         this.messages.push(JSON.parse(message.data))
+        this.getBadges(JSON.parse(message.data))
         this.clearMessages()
         this.scrollToBottom(chatList)
       }
     }
 
     this.ws.onopen = (data) => {
-      console.log(data)
       this.ws.send('JOIN ' + this.props.channelName?.toLowerCase())
     }
   },
@@ -61,7 +73,30 @@ export default {
       if (this.messages.length > 50) {
         this.messages.shift
       }
+    },
+    getBadges(message: { username: string, channel: string, message: string, messageType: string, tags: Record<string, string> }) {
+      let badges = message.tags.badges.split(',')
+      let formatedBadges = badges.map((badgeWithVersion) => {
+        const [setId, version] = badgeWithVersion.split('/')
+        return { setId, version }
+      })
+
+
+      const foundBadges = []
+      for(let badgeToFind of formatedBadges) {
+        const badge = this.badges
+          .filter((badge) => badge.setId === badgeToFind.setId)
+          .find((badge) => badge.version === badgeToFind.version);
+
+          if(badge)
+            foundBadges.push(badge)
+      }
+
+      return foundBadges
     }
+  },
+  components: {
+    BadgeVue
   }
 }
 </script>
@@ -78,12 +113,18 @@ export default {
       </li>
       <li v-for="message in getChat()" :key="messages.indexOf(message)">
         <div class="text-white inline-flex">
-          <p class="text-sm">
+          <p class="text-sm items-center">
+
+            <ul class="inline-flex space-x-1 pr-1">
+              <li v-for="badge in getBadges(message)">
+                <badge-vue :badge-info="badge"></badge-vue>
+              </li>
+            </ul>
+            
             <strong
               :style="message.tags.color ? `color: ${message.tags.color};` : ``"
-              class="text-ctp-pink font-bold text-sm"
-              >{{ message.username }}</strong
-            >: {{ message.message }}
+              class="text-ctp-pink font-bold">
+              {{ message.username }}</strong>: {{ message.message }}
           </p>
         </div>
       </li>