From de8df46aef11170982d72a5a2482b1b0a651e7c4 Mon Sep 17 00:00:00 2001
From: Thanaphon Frappet <thanaphon@frappet.com>
Date: Tue, 10 Dec 2024 16:03:26 +0700
Subject: [PATCH] refactor: check role

---
 src/components/TreeComponent.vue            | 7 ++++++-
 src/pages/01_branch-management/MainPage.vue | 2 ++
 src/router/routes.ts                        | 1 +
 3 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/components/TreeComponent.vue b/src/components/TreeComponent.vue
index 68ce72a4..ee5130ba 100644
--- a/src/components/TreeComponent.vue
+++ b/src/components/TreeComponent.vue
@@ -1,6 +1,7 @@
 <script setup lang="ts">
 import { BranchWithChildren } from 'stores/branch/types';
 import KebabAction from './shared/KebabAction.vue';
+import { isRoleInclude } from 'stores/utils';
 
 const nodes = defineModel<(any | BranchWithChildren)[]>('nodes', {
   default: [],
@@ -119,7 +120,11 @@ defineEmits<{
             />
 
             <q-btn
-              v-if="node.isHeadOffice && typeTree === 'branch'"
+              v-if="
+                node.isHeadOffice &&
+                typeTree === 'branch' &&
+                isRoleInclude(['head_of_admin', 'admin', 'system'])
+              "
               :id="`create-sub-branch-btn-${node.name}`"
               @click.stop="$emit('create', node)"
               icon="mdi-file-plus-outline"
diff --git a/src/pages/01_branch-management/MainPage.vue b/src/pages/01_branch-management/MainPage.vue
index 4d58b1d7..24f1260d 100644
--- a/src/pages/01_branch-management/MainPage.vue
+++ b/src/pages/01_branch-management/MainPage.vue
@@ -10,6 +10,7 @@ import type { QTableProps } from 'quasar';
 import { resetScrollBar } from 'src/stores/utils';
 import useBranchStore from 'stores/branch';
 import useFlowStore from 'stores/flow';
+import { isRoleInclude } from 'stores/utils';
 import {
   BranchWithChildren,
   BranchCreate,
@@ -1040,6 +1041,7 @@ watch(currentHq, () => {
                     {{ $t('branch.allBranch') }}
                   </div>
                   <q-btn
+                    v-if="isRoleInclude(['head_of_admin', 'admin', 'system'])"
                     round
                     flat
                     size="md"
diff --git a/src/router/routes.ts b/src/router/routes.ts
index 767830cc..b2382b7f 100644
--- a/src/router/routes.ts
+++ b/src/router/routes.ts
@@ -24,6 +24,7 @@ const routes: RouteRecordRaw[] = [
               'head_of_account',
               'system',
               'owner',
+              'branch_manager',
             ])
           ) {
             next();