feat: update revenues task (#21)

* feat: update revenues task

* fix: utc

Author: hmbanan666

apps/web-app/nuxt.config.ts

@@ -38,6 +38,7 @@ export default defineNuxtConfig({
     },
     scheduledTasks: {
       '* * * * *': ['task:auto-create'], // Every minute
+      '0 * * * *': ['kitchen:revenue-update'], // Every hour
       '0 0 * * *': ['kitchen:rating-update'], // Every day
     },
   },

apps/web-app/server/api/kitchen/id/[kitchenId]/revenues-for-period.get.ts

@@ -0,0 +1,23 @@
+import { repository } from '@roll-stack/database'
+
+export default defineEventHandler(async (event) => {
+  const kitchenId = getRouterParam(event, 'kitchenId')
+  if (!kitchenId) {
+    throw createError({
+      statusCode: 400,
+      message: 'Id is required',
+    })
+  }
+
+  const query = getQuery(event)
+  const start = query.start ? new Date(query.start.toString()) : null
+  const end = query.end ? new Date(query.end.toString()) : null
+  if (!start || !end || Number.isNaN(start.getTime()) || Number.isNaN(end.getTime())) {
+    throw createError({
+      statusCode: 400,
+      message: 'Valid start and end dates are required',
+    })
+  }
+
+  return repository.kitchen.listRevenuesByKitchenForPeriod(kitchenId, start, end)
+})

apps/web-app/server/tasks/kitchen/revenue-update.ts

@@ -0,0 +1,41 @@
+import { repository } from '@roll-stack/database'
+import { endOfWeek, startOfWeek } from 'date-fns'
+
+const logger = useLogger('kitchen:revenue-update')
+
+export default defineTask({
+  meta: {
+    name: 'kitchen:revenue-update',
+    description: 'Update weekly revenue of kitchens',
+  },
+  async run() {
+    try {
+      const kitchens = await repository.kitchen.list()
+
+      // From this monday to sunday (use UTC+0 time zone)
+      const now = new Date()
+      const utcNow = new Date(now.getTime() + now.getTimezoneOffset() * 60000)
+
+      const thisMonday = startOfWeek(utcNow, { weekStartsOn: 1 })
+      const thisSunday = endOfWeek(utcNow, { weekStartsOn: 1 })
+
+      logger.log(thisMonday, thisSunday)
+
+      for (const kitchen of kitchens) {
+        const revenues = await repository.kitchen.listRevenuesByKitchenForPeriod(kitchen.id, thisMonday, thisSunday)
+
+        const revenueForThisWeek = revenues.reduce((acc, curr) => acc + curr.total, 0)
+
+        await repository.kitchen.update(kitchen.id, {
+          revenueForThisWeek,
+        })
+
+        // logger.log(`Kitchen ${kitchen.id}: Revenue updated from ${kitchen.revenueForThisWeek} to ${revenueForThisWeek}`)
+      }
+    } catch (error) {
+      errorResolver(error)
+    }
+
+    return { result: true }
+  },
+})

packages/database/src/repository/kitchen.ts

@@ -42,6 +42,18 @@ export class Kitchen {
     })
   }
 
+  static async listRevenuesByKitchenForPeriod(kitchenId: string, start: Date, end: Date) {
+    return useDatabase().query.kitchenRevenues.findMany({
+      where: (revenues, { eq, and }) => and(
+        eq(revenues.kitchenId, kitchenId),
+        sql`date(${revenues.date}) >= date(${start})`,
+        sql`date(${revenues.date}) <= date(${end})`,
+      ),
+      orderBy: (revenues, { desc }) => desc(revenues.date),
+      limit: 1000,
+    })
+  }
+
   static async create(data: KitchenDraft) {
     const [kitchen] = await useDatabase().insert(kitchens).values(data).returning()
     return kitchen