From d8445b3f1a450a9b88abaec6e3a24e97b6908678 Mon Sep 17 00:00:00 2001
From: Emily <adharaultimasheireen@gmail.com>
Date: Fri, 21 Jun 2024 16:56:44 +0200
Subject: [PATCH] .

---
 .../components/referrer/ReferrerBarChart.vue  | 15 ++--
 .../[project_id]/timeline/referrers.post.ts   | 35 +++++---
 shared/package.json                           |  1 +
 shared/pnpm-lock.yaml                         |  8 ++
 shared/services/DateService.ts                | 80 +++++++++++++++++++
 5 files changed, 122 insertions(+), 17 deletions(-)
 create mode 100644 shared/services/DateService.ts

diff --git a/dashboard/components/referrer/ReferrerBarChart.vue b/dashboard/components/referrer/ReferrerBarChart.vue
index 49a81be..5d1f3dc 100644
--- a/dashboard/components/referrer/ReferrerBarChart.vue
+++ b/dashboard/components/referrer/ReferrerBarChart.vue
@@ -1,6 +1,8 @@
 <script lang="ts" setup>
 import { onMounted } from 'vue';
 
+import dayjs from 'dayjs';
+
 const data = ref<number[]>([]);
 const labels = ref<string[]>([]);
 const ready = ref<boolean>(false);
@@ -13,15 +15,18 @@ async function loadData() {
     const response = await $fetch(`/api/metrics/${activeProject.value?._id.toString()}/timeline/referrers`, {
         method: 'POST',
         ...signHeaders({ 'Content-Type': 'application/json' }),
-        body: JSON.stringify({ slice: 'day', referrer: props.referrer })
+        body: JSON.stringify({
+            slice: 'hour',
+            from: Date.now() - 1000 * 60 * 60 * 12,
+            to: Date.now(),
+            referrer: props.referrer
+        })
     });
 
     if (!response) return;
 
-    const fixed = fixMetrics(response, props.slice);
-    console.log(fixed);
-    data.value = fixed.data;
-    labels.value = fixed.labels;
+    data.value = response.map(e => e.count);
+    labels.value = response.map(e => dayjs(e._id).locale(navigator.language));
     ready.value = true;
 
 }
diff --git a/dashboard/server/api/metrics/[project_id]/timeline/referrers.post.ts b/dashboard/server/api/metrics/[project_id]/timeline/referrers.post.ts
index ba770c9..b36998d 100644
--- a/dashboard/server/api/metrics/[project_id]/timeline/referrers.post.ts
+++ b/dashboard/server/api/metrics/[project_id]/timeline/referrers.post.ts
@@ -1,9 +1,9 @@
 import { getTimeline } from "./generic";
 import { VisitModel } from "@schema/metrics/VisitSchema";
+import DateService from "@services/DateService";
 import { Redis, TIMELINE_EXPIRE_TIME } from "~/server/services/CacheService";
 import { getUserProjectFromId } from "~/server/LIVE_DEMO_DATA";
 
-
 export default defineEventHandler(async event => {
     const project_id = getRequestProjectId(event);
     if (!project_id) return;
@@ -12,19 +12,30 @@ export default defineEventHandler(async event => {
     const project = await getUserProjectFromId(project_id, user);
     if (!project) return;
 
-    const { slice, duration, referrer } = await readBody(event);
+    const { slice, referrer, from, to } = await readBody(event);
 
-    // return await Redis.useCache({ key: `timeline:referrers:${project_id}:${slice}`, exp: TIMELINE_EXPIRE_TIME }, async () => {
-    const timelineReferrers = await getTimeline(VisitModel, project_id, slice, duration,
-        {},
-        {},
-        { referrer: "$_id.referrer" },
-        { referrer: "$referrer" },
-        { referrer }
-    );
-    return timelineReferrers;
-    // });
+    const { group, sort, fromParts } = DateService.getQueryDateRange(slice);
 
+    const aggregation = [
+        {
+            $match: {
+                project_id: project._id,
+                created_at: {
+                    $gte: new Date(from),
+                    $lte: new Date(to)
+                },
+                referrer
+            }
+        },
+        { $group: { _id: group, count: { $sum: 1 } } },
+        { $sort: sort },
+        { $project: { _id: { $dateFromParts: fromParts }, count: "$count" } }
+    ]
 
+    const timelineReferrers: { _id: string, count: number }[] = await VisitModel.aggregate(aggregation);
+
+    const filledDates = DateService.fillDates(timelineReferrers.map(e => e._id), slice);
+    const merged = DateService.mergeFilledDates(filledDates, timelineReferrers, '_id', slice, { count: 0 });
+    return merged;
 
 });
\ No newline at end of file
diff --git a/shared/package.json b/shared/package.json
index 2395a90..a3b5319 100644
--- a/shared/package.json
+++ b/shared/package.json
@@ -1,5 +1,6 @@
 {
 	"dependencies": {
+		"dayjs": "^1.11.11",
 		"mongoose": "^8.4.0",
 		"nodemailer": "^6.9.13",
 		"redis": "^4.6.14"
diff --git a/shared/pnpm-lock.yaml b/shared/pnpm-lock.yaml
index cdd2407..3a4efad 100644
--- a/shared/pnpm-lock.yaml
+++ b/shared/pnpm-lock.yaml
@@ -8,6 +8,9 @@ importers:
 
   .:
     dependencies:
+      dayjs:
+        specifier: ^1.11.11
+        version: 1.11.11
       mongoose:
         specifier: ^8.4.0
         version: 8.4.0
@@ -79,6 +82,9 @@ packages:
     resolution: {integrity: sha512-RMr0FhtfXemyinomL4hrWcYJxmX6deFdCxpJzhDttxgO1+bcCnkk+9drydLVDmAMG7NE6aN/fl4F7ucU/90gAA==}
     engines: {node: '>=0.10.0'}
 
+  dayjs@1.11.11:
+    resolution: {integrity: sha512-okzr3f11N6WuqYtZSvm+F776mB41wRZMhKP+hc34YdW+KmtYYK9iqvHSwo2k9FEH3fhGXvOPV6yz2IcSrfRUDg==}
+
   debug@4.3.4:
     resolution: {integrity: sha512-PRWFHuSU3eDtQJPvnNY7Jcket1j0t5OuOsFzPPzsekD52Zl8qUfFIPEiswXqIvHWGVHOgX+7G/vCNNhehwxfkQ==}
     engines: {node: '>=6.0'}
@@ -232,6 +238,8 @@ snapshots:
 
   cluster-key-slot@1.1.2: {}
 
+  dayjs@1.11.11: {}
+
   debug@4.3.4:
     dependencies:
       ms: 2.1.2
diff --git a/shared/services/DateService.ts b/shared/services/DateService.ts
new file mode 100644
index 0000000..7f5afa4
--- /dev/null
+++ b/shared/services/DateService.ts
@@ -0,0 +1,80 @@
+
+import dayjs from 'dayjs';
+
+export type Slice = 'day' | 'hour' | 'month' | 'year';
+
+class DateService {
+
+    getQueryDateRange(slice: Slice) {
+
+        const group: Record<string, any> = {}
+        const sort: Record<string, any> = {}
+        const fromParts: Record<string, any> = {}
+
+        switch (slice) {
+            case 'hour':
+                group.hour = { $hour: '$created_at' }
+                sort['_id.hour'] = 1;
+                fromParts.hour = "$_id.hour";
+            case 'day':
+                group.day = { $dayOfMonth: '$created_at' }
+                sort['_id.day'] = 1;
+                fromParts.day = "$_id.day";
+            case 'month':
+                group.month = { $month: '$created_at' }
+                sort['_id.month'] = 1;
+                fromParts.month = "$_id.month";
+            case 'year':
+                group.year = { $year: '$created_at' }
+                sort['_id.year'] = 1;
+                fromParts.year = "$_id.year";
+        }
+
+        return { group, sort, fromParts }
+    }
+
+    prepareDateRange(from: string, to: string, slice: Slice) {
+
+        let fromDate = dayjs(from).minute(0).second(0).millisecond(0);
+        let toDate = dayjs(to).minute(0).second(0).millisecond(0);
+
+        switch (slice) {
+            case 'day':
+                fromDate = fromDate.hour(0);
+                toDate = toDate.hour(0);
+                break;
+            case 'hour':
+                break;
+        }
+
+        return {
+            from: fromDate.toDate(),
+            to: toDate.toDate()
+        }
+    }
+
+    fillDates(dates: string[], slice: Slice) {
+        const allDates: dayjs.Dayjs[] = [];
+        const firstDate = dayjs(dates.at(0));
+        const lastDate = dayjs(dates.at(-1));
+        let currentDate = firstDate.clone();
+        while (currentDate.isBefore(lastDate)) {
+            currentDate = currentDate.add(1, slice);
+            allDates.push(currentDate);
+        }
+        return allDates;
+    }
+
+    mergeFilledDates<T extends Record<string, any>, K extends keyof T>(dates: dayjs.Dayjs[], items: T[], dateField: K, slice: Slice, fillData: Omit<T, K>) {
+        const result = new Array<T>();
+        for (const date of dates) {
+            const item = items.find(e => dayjs(e[dateField]).isSame(date), slice);
+            result.push(item ?? { ...fillData, [dateField]: date.format() } as T);
+        }
+        return result;
+    }
+
+}
+
+const dateServiceInstance = new DateService();
+export default dateServiceInstance;
\ No newline at end of file