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, + }, + }, }, }, },