From c27791e52a4da175c492cb71d675eaf45ef78884 Mon Sep 17 00:00:00 2001
From: mehmet-erim <bilgi@mehmeterim.com>
Date: Fri, 19 Jun 2020 14:22:40 +0300
Subject: [PATCH] feat: add findRequiredPolicy method to routes.component

---
 .../components/routes/routes.component.html   |  5 ++-
 .../lib/components/routes/routes.component.ts | 35 +++++++++++++++++--
 2 files changed, 35 insertions(+), 5 deletions(-)

diff --git a/npm/ng-packs/packages/theme-basic/src/lib/components/routes/routes.component.html b/npm/ng-packs/packages/theme-basic/src/lib/components/routes/routes.component.html
index 4d22ddc8d2..60f2de6c1c 100644
--- a/npm/ng-packs/packages/theme-basic/src/lib/components/routes/routes.component.html
+++ b/npm/ng-packs/packages/theme-basic/src/lib/components/routes/routes.component.html
@@ -18,8 +18,8 @@
   <ng-template #dropdownLink let-route>
     <li
       #navbarRootDropdown
-      *abpPermission="route.requiredPolicy"
-      [abpVisibility]="routeContainer"
+      *ngIf="findRequiredPolicy(route)"
+      [abpPermission]="route.requiredPolicy"
       class="nav-item dropdown"
       display="static"
       (click)="
@@ -69,7 +69,6 @@
 
   <ng-template #dropdownChild let-child>
     <div
-      [abpVisibility]="childrenContainer"
       class="dropdown-submenu"
       ngbDropdown
       #dropdownSubmenu="ngbDropdown"
diff --git a/npm/ng-packs/packages/theme-basic/src/lib/components/routes/routes.component.ts b/npm/ng-packs/packages/theme-basic/src/lib/components/routes/routes.component.ts
index 8b73c8ee29..05d5740efd 100644
--- a/npm/ng-packs/packages/theme-basic/src/lib/components/routes/routes.component.ts
+++ b/npm/ng-packs/packages/theme-basic/src/lib/components/routes/routes.component.ts
@@ -1,5 +1,6 @@
-import { ABP, RoutesService, TreeNode } from '@abp/ng.core';
+import { ABP, RoutesService, TreeNode, LocalizationService, ConfigState } from '@abp/ng.core';
 import { Component, Input, TrackByFunction } from '@angular/core';
+import { Store } from '@ngxs/store';
 
 @Component({
   selector: 'abp-routes',
@@ -10,5 +11,35 @@ export class RoutesComponent {
 
   trackByFn: TrackByFunction<TreeNode<ABP.Route>> = (_, item) => item.name;
 
-  constructor(public readonly routes: RoutesService) {}
+  constructor(public readonly routes: RoutesService, private store: Store) {}
+
+  findRequiredPolicy(route: TreeNode<ABP.Route>): boolean {
+    if (route.requiredPolicy || !route.children?.length) return true;
+
+    const policies = getPolicies(route.children);
+
+    let isVisible = true;
+    for (let i = 0; i < policies.length; i++) {
+      if (!this.store.selectSnapshot(ConfigState.getGrantedPolicy(policies[i]))) {
+        isVisible = false;
+        break;
+      }
+    }
+
+    return isVisible;
+  }
+}
+
+function getPolicies(routes: TreeNode<ABP.Route>[]): string[] {
+  return routes.reduce((acc, val) => {
+    if (val.requiredPolicy) acc.push(val.requiredPolicy);
+
+    if (val.children?.length) {
+      const childPolicies = getPolicies(val.children);
+
+      if (childPolicies?.length) acc.push(...childPolicies);
+    }
+
+    return acc;
+  }, []);
 }