From 04e65d80becd61f8724b7393ba19d17f165fe422 Mon Sep 17 00:00:00 2001
From: Jan Lukavsky <je.ik@seznam.cz>
Date: Tue, 9 Feb 2021 16:58:00 +0100
Subject: [PATCH] WIP: add component restart

---
 src/app/backend/api/types.go                  | 22 +++++++++++++++++++
 .../components/list/column/menu/component.ts  |  8 +++++++
 .../components/list/column/menu/template.html |  5 +++++
 .../frontend/common/services/global/verber.ts | 19 ++++++++++++++++
 src/app/frontend/typings/root.api.ts          |  1 +
 5 files changed, 55 insertions(+)

diff --git a/src/app/backend/api/types.go b/src/app/backend/api/types.go
index 6d5272f73f60..623bbab8f05c 100644
--- a/src/app/backend/api/types.go
+++ b/src/app/backend/api/types.go
@@ -78,6 +78,9 @@ type TypeMeta struct {
 
 	// Scalable represents whether or not an object is scalable.
 	Scalable bool `json:"scalable,omitempty"`
+
+        // Restartable represents wheter or not an object is restartable.
+        Restartable bool `json:"restartable,omitempty"`
 }
 
 // ListMeta describes list of objects, i.e. holds information about pagination options set for the list.
@@ -104,6 +107,7 @@ func NewTypeMeta(kind ResourceKind) TypeMeta {
 	return TypeMeta{
 		Kind:     kind,
 		Scalable: kind.Scalable(),
+                Restartable: kind.Restartable(),
 	}
 }
 
@@ -164,6 +168,24 @@ func (k ResourceKind) Scalable() bool {
 	return false
 }
 
+// Restartable method return whether ResourceKind is restartable.
+func (k ResourceKind) Restartable() bool {
+	restartable := []ResourceKind{
+		ResourceKindDeployment,
+		ResourceKindReplicaSet,
+		ResourceKindReplicationController,
+		ResourceKindStatefulSet,
+	}
+
+	for _, kind := range restartable {
+		if k == kind {
+			return true
+		}
+	}
+
+	return false
+}
+
 // ClientType represents type of client that is used to perform generic operations on resources.
 // Different resources belong to different client, i.e. Deployments belongs to extension client
 // and StatefulSets to apps client.
diff --git a/src/app/frontend/common/components/list/column/menu/component.ts b/src/app/frontend/common/components/list/column/menu/component.ts
index 7481176abb43..37076590d8ab 100644
--- a/src/app/frontend/common/components/list/column/menu/component.ts
+++ b/src/app/frontend/common/components/list/column/menu/component.ts
@@ -97,10 +97,18 @@ export class MenuComponent implements ActionColumn {
     return this.typeMeta.scalable;
   }
 
+  isRestartEnabled(): boolean {
+    return this.typeMeta.restartable;
+  }
+
   onScale(): void {
     this.verber_.showScaleDialog(this.typeMeta.kind, this.typeMeta, this.objectMeta);
   }
 
+  onRestart(): void {
+    this.verber_.showRestartDialog(this.typeMeta.kind, this.typeMeta, this.objectMeta);
+  }
+
   isPinEnabled(): boolean {
     return pinnableResources.includes(this.typeMeta.kind);
   }
diff --git a/src/app/frontend/common/components/list/column/menu/template.html b/src/app/frontend/common/components/list/column/menu/template.html
index 52a90aefe284..74cc4d67c674 100644
--- a/src/app/frontend/common/components/list/column/menu/template.html
+++ b/src/app/frontend/common/components/list/column/menu/template.html
@@ -47,6 +47,11 @@
           id="edit"
           (click)="onEdit()"
           i18n>Edit</button>
+  <button mat-menu-item
+          id="restart"
+          *ngIf="isRestartEnabled()"
+          (click)="onRestart()"
+          i18n>Restart</button>
   <button mat-menu-item
           id="delete"
           (click)="onDelete()"
diff --git a/src/app/frontend/common/services/global/verber.ts b/src/app/frontend/common/services/global/verber.ts
index 22be142b3448..7bb6db1ca610 100644
--- a/src/app/frontend/common/services/global/verber.ts
+++ b/src/app/frontend/common/services/global/verber.ts
@@ -84,6 +84,25 @@ export class VerberService {
       .subscribe(_ => this.onScale.emit(true), this.handleErrorResponse_.bind(this));
   }
 
+  showRestartDialog(displayName: string, typeMeta: TypeMeta, objectMeta: ObjectMeta): void {
+    const dialogConfig = this.getDialogConfig_(displayName, typeMeta, objectMeta);
+    this.dialog_
+      //.open(RestartResourceDialog, dialogConfig)
+      .open(ScaleResourceDialog, dialogConfig)
+      .afterClosed()
+      .pipe(filter(result => Number.isInteger(result)))
+      .pipe(
+        switchMap(result => {
+          const url = `api/v1/scale/${typeMeta.kind}${objectMeta.namespace ? `/${objectMeta.namespace}` : ''}/${
+            objectMeta.name
+          }/`;
+
+          return this.http_.put(url, result, {params: {scaleBy: result}});
+        })
+      )
+      .subscribe(_ => this.onScale.emit(true), this.handleErrorResponse_.bind(this));
+  }
+
   showTriggerDialog(displayName: string, typeMeta: TypeMeta, objectMeta: ObjectMeta): void {
     const dialogConfig = this.getDialogConfig_(displayName, typeMeta, objectMeta);
     this.dialog_
diff --git a/src/app/frontend/typings/root.api.ts b/src/app/frontend/typings/root.api.ts
index 5e30d17ea3de..430b441af680 100644
--- a/src/app/frontend/typings/root.api.ts
+++ b/src/app/frontend/typings/root.api.ts
@@ -26,6 +26,7 @@ export enum SupportedResources {
 export interface TypeMeta {
   kind: string;
   scalable?: boolean;
+  restartable?: boolean;
 }
 
 export interface ListMeta {