From f0c75d80d6a3c50070b477ef6f0a42e5a726f3fa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Qu=E1=BB=91c=20Kh=C3=A1nh?= <bkdev98@gmail.com>
Date: Fri, 19 Jul 2024 20:55:38 +0700
Subject: [PATCH] feat(api): [Budget] populate period field and add weekly type
 (#141)

---
 .../20240719123618_add_budget_period_weekly_type/migration.sql | 2 ++
 apps/api/prisma/schema.prisma                                  | 1 +
 apps/api/v1/services/budget.service.ts                         | 3 +++
 packages/validation/src/prisma/index.ts                        | 2 +-
 4 files changed, 7 insertions(+), 1 deletion(-)
 create mode 100644 apps/api/prisma/migrations/20240719123618_add_budget_period_weekly_type/migration.sql

diff --git a/apps/api/prisma/migrations/20240719123618_add_budget_period_weekly_type/migration.sql b/apps/api/prisma/migrations/20240719123618_add_budget_period_weekly_type/migration.sql
new file mode 100644
index 00000000..a17f6bfb
--- /dev/null
+++ b/apps/api/prisma/migrations/20240719123618_add_budget_period_weekly_type/migration.sql
@@ -0,0 +1,2 @@
+-- AlterEnum
+ALTER TYPE "BudgetPeriodType" ADD VALUE 'WEEKLY';
diff --git a/apps/api/prisma/schema.prisma b/apps/api/prisma/schema.prisma
index 66ec8ed0..5c4d541c 100644
--- a/apps/api/prisma/schema.prisma
+++ b/apps/api/prisma/schema.prisma
@@ -86,6 +86,7 @@ model BudgetPeriodConfig {
 }
 
 enum BudgetPeriodType {
+  WEEKLY
   MONTHLY
   QUARTERLY
   YEARLY
diff --git a/apps/api/v1/services/budget.service.ts b/apps/api/v1/services/budget.service.ts
index 753a281c..db41e3d4 100644
--- a/apps/api/v1/services/budget.service.ts
+++ b/apps/api/v1/services/budget.service.ts
@@ -250,6 +250,9 @@ export async function findBudgetsOfUser({
         },
       },
     },
+    include: {
+      periodConfig: true,
+    },
   })
 }
 
diff --git a/packages/validation/src/prisma/index.ts b/packages/validation/src/prisma/index.ts
index 5a5dde16..a0a6af7b 100644
--- a/packages/validation/src/prisma/index.ts
+++ b/packages/validation/src/prisma/index.ts
@@ -68,7 +68,7 @@ export const BudgetTypeSchema = z.enum(['SPENDING','SAVING','INVESTING','DEBT'])
 
 export type BudgetTypeType = `${z.infer<typeof BudgetTypeSchema>}`
 
-export const BudgetPeriodTypeSchema = z.enum(['MONTHLY','QUARTERLY','YEARLY','CUSTOM']);
+export const BudgetPeriodTypeSchema = z.enum(['WEEKLY','MONTHLY','QUARTERLY','YEARLY','CUSTOM']);
 
 export type BudgetPeriodTypeType = `${z.infer<typeof BudgetPeriodTypeSchema>}`