From 602a63f2c988dae6b36910a04cd238ad951f01f0 Mon Sep 17 00:00:00 2001
From: nameless <nameless@noreply@foxo.me>
Date: Sat, 11 Nov 2023 00:11:35 +0000
Subject: [PATCH] feat: user daily stats

---
 app/userDailyStatsStore.ts | 34 +++++++++++++++++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

diff --git a/app/userDailyStatsStore.ts b/app/userDailyStatsStore.ts
index 63e1751..485fdeb 100644
--- a/app/userDailyStatsStore.ts
+++ b/app/userDailyStatsStore.ts
@@ -2,6 +2,8 @@ import { JsonSchema, jsonType } from "t_rest/server";
 import { kvMemoize } from "./kvMemoize.ts";
 import { db } from "./db.ts";
 import { generationStore } from "./generationStore.ts";
+import { hoursToMilliseconds, isSameDay, minutesToMilliseconds } from "date-fns";
+import { UTCDateMini } from "date-fns/utc";
 
 export const userDailyStatsSchema = {
   type: "object",
@@ -19,6 +21,36 @@ export const getUserDailyStats = kvMemoize(
   db,
   ["userDailyStats"],
   async (userId: number, year: number, month: number, day: number): Promise<UserDailyStats> => {
-    throw new Error("Not implemented");
+    let imageCount = 0;
+    let pixelCount = 0;
+
+    for await (
+      const generation of generationStore.listBy("fromId", {
+        before: new Date(new Date(year, month - 1, day).getTime() + 24 * 60 * 60 * 1000),
+        after: new Date(year, month - 1, day),
+        value: userId,
+      })
+    ) {
+      imageCount++;
+      pixelCount += (generation.value.info?.width ?? 0) * (generation.value.info?.height ?? 0);
+    }
+
+    return {
+      imageCount,
+      pixelCount,
+      timestamp: Date.now(),
+    };
+  },
+  {
+    // expire in 1 minute if was calculated on the same day, otherwise 7-14 days.
+    expireIn: (result, year, month, day) => {
+      const requestDate = new UTCDateMini(year, month - 1, day);
+      const calculatedDate = new UTCDateMini(result.timestamp);
+      return isSameDay(requestDate, calculatedDate)
+        ? minutesToMilliseconds(1)
+        : hoursToMilliseconds(24 * 7 + Math.random() * 24 * 7);
+    },
+    // should cache if the stats are non-zero
+    shouldCache: (result) => result.imageCount > 0 || result.pixelCount > 0,
   },
 );