From 7fa99fab8fb0debc57a942de2258550bb65d444a Mon Sep 17 00:00:00 2001
From: Methapon Metanipat <methapon@frappet.com>
Date: Fri, 30 Aug 2024 11:33:08 +0700
Subject: [PATCH] feat: add headoffice as relation

---
 src/controllers/branch-controller.ts      | 10 ++++++++++
 src/controllers/branch-user-controller.ts |  7 +++++++
 2 files changed, 17 insertions(+)

diff --git a/src/controllers/branch-controller.ts b/src/controllers/branch-controller.ts
index d340656..6ee8707 100644
--- a/src/controllers/branch-controller.ts
+++ b/src/controllers/branch-controller.ts
@@ -179,6 +179,7 @@ export class BranchController extends Controller {
     @Query() zipCode?: string,
     @Query() filter?: "head" | "sub",
     @Query() headOfficeId?: string,
+    @Query() includeHead?: boolean,
     @Query() tree?: boolean,
     @Query() query: string = "",
     @Query() page: number = 1,
@@ -206,6 +207,15 @@ export class BranchController extends Controller {
           district: true,
           subDistrict: true,
           contact: true,
+          headOffice: includeHead
+            ? {
+                include: {
+                  province: true,
+                  district: true,
+                  subDistrict: true,
+                },
+              }
+            : false,
           branch: tree
             ? {
                 include: {
diff --git a/src/controllers/branch-user-controller.ts b/src/controllers/branch-user-controller.ts
index 9cfc9cd..57f6c7b 100644
--- a/src/controllers/branch-user-controller.ts
+++ b/src/controllers/branch-user-controller.ts
@@ -244,6 +244,13 @@ export class UserBranchController extends Controller {
               province: true,
               district: true,
               subDistrict: true,
+              headOffice: {
+                include: {
+                  province: true,
+                  district: true,
+                  subDistrict: true,
+                },
+              },
             },
           },
         },