From ed2341832b7ad51e253f80fd0129ac273aadb13a Mon Sep 17 00:00:00 2001
From: Hironsan <light.tree.1.13@gmail.com>
Date: Fri, 7 Jan 2022 11:42:48 +0900
Subject: [PATCH] Change bar color in statistics page

---
 frontend/components/statistics/CategoryDistribution.vue | 8 ++++++--
 frontend/components/statistics/LabelDistribution.vue    | 7 ++++++-
 frontend/components/statistics/SpanDistribution.vue     | 8 ++++++--
 3 files changed, 18 insertions(+), 5 deletions(-)

diff --git a/frontend/components/statistics/CategoryDistribution.vue b/frontend/components/statistics/CategoryDistribution.vue
index 0b1f2847..781c622e 100644
--- a/frontend/components/statistics/CategoryDistribution.vue
+++ b/frontend/components/statistics/CategoryDistribution.vue
@@ -2,6 +2,7 @@
   <label-distribution
     title="Category Distribution"
     :distribution="distribution"
+    :color-mapping="colorMapping"
   />
 </template>
 
@@ -16,12 +17,15 @@ export default Vue.extend({
 
   data() {
     return {
-      distribution: {}
+      distribution: {},
+      colorMapping: {},
     }
   },
 
   async created() {
     this.distribution = await this.$services.statistics.fetchCategoryDistribution(this.$route.params.id)
+    const labels = await this.$services.categoryType.list(this.$route.params.id)
+    this.colorMapping = Object.fromEntries(labels.map((label) => [label.text, label.backgroundColor]))
   }
 })
-</script>
\ No newline at end of file
+</script>
diff --git a/frontend/components/statistics/LabelDistribution.vue b/frontend/components/statistics/LabelDistribution.vue
index bd951e5d..13b8c28a 100644
--- a/frontend/components/statistics/LabelDistribution.vue
+++ b/frontend/components/statistics/LabelDistribution.vue
@@ -43,6 +43,10 @@ export default Vue.extend({
     distribution: {
       type: Object as PropType<Distribution>,
       required: true
+    },
+    colorMapping: {
+      type: Object,
+      default: () => {},
     }
   },
 
@@ -53,11 +57,12 @@ export default Vue.extend({
         const labels = Object.keys(this.distribution[user])
         labels.sort()
         const counts = labels.map((label) => this.distribution[user][label])
+        const colors = labels.map((label) => label in this.colorMapping ? this.colorMapping[label] : '#00d1b2')
         data[user] = {
           labels,
           datasets: [{
             title: this.title,
-            backgroundColor: '#00d1b2',
+            backgroundColor: colors,
             data: counts
           }]
         }
diff --git a/frontend/components/statistics/SpanDistribution.vue b/frontend/components/statistics/SpanDistribution.vue
index d67e1ee7..07082cc8 100644
--- a/frontend/components/statistics/SpanDistribution.vue
+++ b/frontend/components/statistics/SpanDistribution.vue
@@ -2,6 +2,7 @@
   <label-distribution
     title="Span Distribution"
     :distribution="distribution"
+    :color-mapping="colorMapping"
   />
 </template>
 
@@ -16,12 +17,15 @@ export default Vue.extend({
 
   data() {
     return {
-      distribution: {}
+      distribution: {},
+      colorMapping: {},
     }
   },
 
   async created() {
     this.distribution = await this.$services.statistics.fetchSpanDistribution(this.$route.params.id)
+    const labels = await this.$services.spanType.list(this.$route.params.id)
+    this.colorMapping = Object.fromEntries(labels.map((label) => [label.text, label.backgroundColor]))
   }
 })
-</script>
\ No newline at end of file
+</script>