From 04e50696f679e697c13fe52e75bed345b8882987 Mon Sep 17 00:00:00 2001
From: Artem Babak <artemio.babak@gmail.com>
Date: Thu, 10 Dec 2020 12:36:52 +0200
Subject: [PATCH] Fix for setRootRuleChains button - for edge scope

---
 .../pages/rulechain/rulechain.component.html  |  8 ++++++-
 .../pages/rulechain/rulechain.component.ts    | 24 +++++++++++++++++++
 .../rulechains-table-config.resolver.ts       |  1 +
 3 files changed, 32 insertions(+), 1 deletion(-)

diff --git a/ui-ngx/src/app/modules/home/pages/rulechain/rulechain.component.html b/ui-ngx/src/app/modules/home/pages/rulechain/rulechain.component.html
index b5f57c6107..17a35dcccb 100644
--- a/ui-ngx/src/app/modules/home/pages/rulechain/rulechain.component.html
+++ b/ui-ngx/src/app/modules/home/pages/rulechain/rulechain.component.html
@@ -31,7 +31,13 @@
   <button mat-raised-button color="primary"
           [disabled]="(isLoading$ | async)"
           (click)="onEntityAction($event, 'setRoot')"
-          [fxShow]="!isEdit && !entity?.root">
+          [fxShow]="!isEdit && !entity?.root && !isEdgeRuleChainScope()">
+    {{'rulechain.set-root' | translate }}
+  </button>
+  <button mat-raised-button color="primary"
+          [disabled]="(isLoading$ | async)"
+          (click)="onEntityAction($event, 'setRoot')"
+          [fxShow]="!isEdit && isEdgeRuleChainScope() && !isRootRuleChain()">
     {{'rulechain.set-root' | translate }}
   </button>
   <button mat-raised-button color="primary"
diff --git a/ui-ngx/src/app/modules/home/pages/rulechain/rulechain.component.ts b/ui-ngx/src/app/modules/home/pages/rulechain/rulechain.component.ts
index 91cd9cdae0..f3e86859e4 100644
--- a/ui-ngx/src/app/modules/home/pages/rulechain/rulechain.component.ts
+++ b/ui-ngx/src/app/modules/home/pages/rulechain/rulechain.component.ts
@@ -79,6 +79,14 @@ export class RuleChainComponent extends EntityComponent<RuleChain> {
       }));
   }
 
+  isTenantRuleChainScope() {
+    if (this.entitiesTableConfig) {
+      return this.entitiesTableConfig.componentsData.ruleChainScope == 'tenant';
+    } else {
+      return false;
+    }
+  }
+
   isEdgesRuleChainScope() {
     if (this.entitiesTableConfig) {
       return this.entitiesTableConfig.componentsData.ruleChainScope == 'edges';
@@ -86,4 +94,20 @@ export class RuleChainComponent extends EntityComponent<RuleChain> {
       return false;
     }
   }
+
+  isEdgeRuleChainScope() {
+    if (this.entitiesTableConfig) {
+      return this.entitiesTableConfig.componentsData.ruleChainScope == 'edge';
+    } else {
+      return false;
+    }
+  }
+
+  isRootRuleChain() {
+    if (this.entitiesTableConfig && this.entityValue) {
+      return this.entitiesTableConfig.componentsData.rootRuleChainId == this.entityValue.id.id;
+    } else {
+      return false;
+    }
+  }
 }
diff --git a/ui-ngx/src/app/modules/home/pages/rulechain/rulechains-table-config.resolver.ts b/ui-ngx/src/app/modules/home/pages/rulechain/rulechains-table-config.resolver.ts
index cfd0303cc3..b9a41d65fc 100644
--- a/ui-ngx/src/app/modules/home/pages/rulechain/rulechains-table-config.resolver.ts
+++ b/ui-ngx/src/app/modules/home/pages/rulechain/rulechains-table-config.resolver.ts
@@ -118,6 +118,7 @@ export class RuleChainsTableConfigResolver implements Resolve<EntityTableConfig<
       this.config.entitySelectionEnabled = ruleChain => this.config.componentsData.edge.rootRuleChainId.id != ruleChain.id.id;
       this.edgeService.getEdge(this.config.componentsData.edgeId).subscribe(edge => {
         this.config.componentsData.edge = edge;
+        this.config.componentsData.rootRuleChainId = edge.rootRuleChainId.id;
         this.config.tableTitle = edge.name + ': ' + this.translate.instant('rulechain.edge-rulechains');
       });
       this.config.entitiesDeleteEnabled = false;