From da0be5cd4a185f1cc17854b98017a13569643cbf Mon Sep 17 00:00:00 2001
From: knguyen100000010 <63071572+knguyen100000010@users.noreply.github.com>
Date: Wed, 21 Feb 2024 08:28:57 -0500
Subject: [PATCH] Pie (#1457)

* add more host dist stat

* resume suspend process support

* avoid hardcoding process name

* remove Preview status
---
 .../clusters/cluster-replacements.tmpl        |  2 +-
 .../templates/groups/asg_config.tmpl          | 10 ++++
 .../templates/groups/asg_processes.tmpl       | 52 +++++++++++++++++++
 .../templates/groups/host_az_dist.tmpl        | 27 +++++++++-
 .../deploy_board/webapp/arcee_urls.py         |  6 +++
 .../deploy_board/webapp/group_view.py         | 43 ++++++++++++++-
 .../helpers/autoscaling_groups_helper.py      |  6 +++
 7 files changed, 142 insertions(+), 4 deletions(-)
 create mode 100644 deploy-board/deploy_board/templates/groups/asg_processes.tmpl

diff --git a/deploy-board/deploy_board/templates/clusters/cluster-replacements.tmpl b/deploy-board/deploy_board/templates/clusters/cluster-replacements.tmpl
index 6082ab250f..74ec350e61 100644
--- a/deploy-board/deploy_board/templates/clusters/cluster-replacements.tmpl
+++ b/deploy-board/deploy_board/templates/clusters/cluster-replacements.tmpl
@@ -48,7 +48,7 @@
     <div class="row">
         <a href="/env/{{ env_name }}/{{ env_stage }}/cluster_replacements/auto_refresh"
             type="button" class="btn btn-default btn-block">
-            <span class="glyphicon glyphicon-leaf" style="color:green;"></span> Set up Auto Refresh <b style="color:blue;">(Preview)</b>
+            <span class="glyphicon glyphicon-leaf" style="color:green;"></span> Set up Auto Refresh
         </a>
     </div>
     <div class="row">
diff --git a/deploy-board/deploy_board/templates/groups/asg_config.tmpl b/deploy-board/deploy_board/templates/groups/asg_config.tmpl
index 9656425aed..0522868bb7 100644
--- a/deploy-board/deploy_board/templates/groups/asg_config.tmpl
+++ b/deploy-board/deploy_board/templates/groups/asg_config.tmpl
@@ -70,6 +70,7 @@
     </div>
 </div>
 
+<div id="suspendedProcessesPid"></div>
 <div id="scheduledActionsPid"></div>
 <div id="scalingPolicyPid"></div>
 <div id="alarmMetricPid"></div>
@@ -102,6 +103,14 @@ function loadAsgScheduledActions() {
     });
 }
 
+function loadAsgSuspendedProcesses() {
+    $("#suspendedProcessesPid").addClass("panel panel-default");
+    var asg_suspended_processes_url = "/groups/{{ group_name }}/autoscaling/get_suspended_processes/";
+    $.get(asg_suspended_processes_url, function(response) {
+        $("#suspendedProcessesPid").html(response);
+    });
+}
+
 function loadAsgPolicy() {
     $("#scalingPolicyPid").addClass("panel panel-default");
     var asg_policy_url = "/groups/{{ group_name }}/autoscaling/get_asg_policy/";
@@ -121,6 +130,7 @@ function loadAsgAlarms() {
 function getAdvancedSetting() {
     loadAsgPolicy();
     loadAsgScheduledActions();
+    loadAsgSuspendedProcesses();
     loadAsgAlarms();
 }
 
diff --git a/deploy-board/deploy_board/templates/groups/asg_processes.tmpl b/deploy-board/deploy_board/templates/groups/asg_processes.tmpl
new file mode 100644
index 0000000000..30f4fe527a
--- /dev/null
+++ b/deploy-board/deploy_board/templates/groups/asg_processes.tmpl
@@ -0,0 +1,52 @@
+{% load utils %}
+{% include "panel_heading.tmpl" with panel_title="Suspended Processes" panel_body_id="asSuspendedProcessId" direction="down" %}
+<div id="asSuspendedProcessId" class="collapse in panel-body">
+
+<div class="container-fluid">
+    <table id="suspendedProcessStatus" class="table">
+        <tr>
+            <th>Process Name</th>
+            <th>State</th>
+            <th>Click to ...</th>
+        </tr>
+        {% for process in process_suspended_status %}
+        <tr>
+            <td>
+               {{ process.name }}
+            </td>
+            <td>
+                {% if process.suspended == True %} <p style="color:red;">Suspended</p> {% else %} <p style="color:green;">Active</p>{% endif %}
+            </td>
+            <td>
+                {% if process.suspended == True %} <input type="button" value="Resume" onclick="updateProcess(this, '{{ process.name }}', false)"> {% else %} <input type="button" value="Suspend" onclick="updateProcess(this, '{{ process.name }}', true)"> {% endif %}
+            </td>
+        </tr>
+        {% endfor%}
+      </table>
+</div>
+
+<script>
+function updateProcess(element, name, shouldSuspend) {
+    var url = ""
+
+    if (shouldSuspend == true) {
+        url = '/groups/{{ group_name }}/autoscaling/suspend_process/' + name;
+    }
+    else {
+        url = '/groups/{{ group_name }}/autoscaling/resume_process/' + name
+    }
+
+    $.ajax({
+        type: 'POST',
+        url: url,
+        data: {'csrfmiddlewaretoken': '{{csrf_token}}'},
+        success: function (data) {
+            $("#asSuspendedProcessId").parent().html(data);
+        }
+    });
+
+    element.value = "Updating ...";
+    element.disabled = true;
+}
+</script>
+</div>
diff --git a/deploy-board/deploy_board/templates/groups/host_az_dist.tmpl b/deploy-board/deploy_board/templates/groups/host_az_dist.tmpl
index a5c4fe866f..b554fdc613 100644
--- a/deploy-board/deploy_board/templates/groups/host_az_dist.tmpl
+++ b/deploy-board/deploy_board/templates/groups/host_az_dist.tmpl
@@ -1,11 +1,36 @@
 {% load utils %}
 
 {% block content %}
-  <div id="container" style="width: 75%;">
+  <div id="container" style="width: 100%;">
     <h2>Host Distribution Among Availability Zones</h2>
     <canvas id="pie-chart"></canvas>
   </div>
 
+  <div>
+    <table id="host-dist-stat" class="table">
+      <tr>
+          <th>Availability Zone</th>
+          <th>Host Count</th>
+          <th>Percentage</th>
+      </tr>
+      {% for l, d, p in label_data_percentage %}
+      <tr>
+          <td>
+             {{ l }}
+          </td>
+          <td>
+             {{ d }}
+          </td>
+          <td>
+             {{ p }}%
+          </td>
+      </tr>
+      {% endfor%}
+    </table>
+    <p>==============================</p>
+  <b>Total:</b> {{ total }} hosts
+  </div>
+
   <script src="https://cdn.jsdelivr.net/npm/chart.js"></script>
   <script>
     var config = {
diff --git a/deploy-board/deploy_board/webapp/arcee_urls.py b/deploy-board/deploy_board/webapp/arcee_urls.py
index e4d2b49b3d..ed954b611b 100644
--- a/deploy-board/deploy_board/webapp/arcee_urls.py
+++ b/deploy-board/deploy_board/webapp/arcee_urls.py
@@ -105,6 +105,12 @@
         group_view.get_pas_config),
     url(r'^groups/(?P<group_name>[a-zA-Z0-9\-_]+)/autoscaling/get_scheduled_actions/$',
         group_view.get_scheduled_actions),
+    url(r'^groups/(?P<group_name>[a-zA-Z0-9\-_]+)/autoscaling/get_suspended_processes/$',
+        group_view.get_suspended_processes),
+    url(r'^groups/(?P<group_name>[a-zA-Z0-9\-_]+)/autoscaling/suspend_process/(?P<process_name>[a-zA-Z\-_]+)$',
+        group_view.suspend_process),
+    url(r'^groups/(?P<group_name>[a-zA-Z0-9\-_]+)/autoscaling/resume_process/(?P<process_name>[a-zA-Z\-_]+)$',
+        group_view.resume_process),
     url(r'^groups/(?P<group_name>[a-zA-Z0-9\-_]+)/autoscaling/delete_scheduled_actions/$',
         group_view.delete_scheduled_actions),
     url(r'^groups/(?P<group_name>[a-zA-Z0-9\-_]+)/autoscaling/add_scheduled_actions/$',
diff --git a/deploy-board/deploy_board/webapp/group_view.py b/deploy-board/deploy_board/webapp/group_view.py
index a284ee0dad..a2a5bb160f 100644
--- a/deploy-board/deploy_board/webapp/group_view.py
+++ b/deploy-board/deploy_board/webapp/group_view.py
@@ -1474,10 +1474,16 @@ def get_host_az_dist(request, group_name):
     )
 
     counter = Counter([x['location'] for x in host_az_dist.json()])
+    labels = list(counter.keys())
+    data = list(counter.values())
+    total = sum(data)
+    percentages = map(lambda x: round((x / total) * 100, 1), data)
 
     return render(request, 'groups/host_az_dist.tmpl', {
-        'labels': list(counter.keys()),
-        'data': list(counter.values()),
+        'labels': labels,
+        'data': data,
+        'label_data_percentage': list(zip(labels, data, percentages)),
+        'total': total
     })
 
 def get_health_check_details(request, id):
@@ -1558,6 +1564,39 @@ def get_scheduled_actions(request, group_name):
     })
     return HttpResponse(json.dumps(content), content_type="application/json")
 
+def get_suspended_processes(request, group_name):
+    suspended_processes = autoscaling_groups_helper.get_disabled_asg_actions(request, group_name)
+    all_processes = autoscaling_groups_helper.get_available_scaling_process(request, group_name)
+    all_processes.sort()
+
+    process_suspended_status = []
+
+    for p in all_processes:
+        if p in suspended_processes:
+            process_suspended_status.append({"name": p, "suspended": True})
+        else:
+            process_suspended_status.append({"name": p, "suspended": False})
+
+    content = render_to_string("groups/asg_processes.tmpl", {
+        'group_name': group_name,
+        'process_suspended_status': process_suspended_status,
+        'csrf_token': get_token(request),
+    })
+    return HttpResponse(json.dumps(content), content_type="application/json")
+
+def suspend_process(request, group_name, process_name):
+    update_request = {}
+    update_request["suspend"] = [process_name]
+    autoscaling_groups_helper.update_scaling_process(request, group_name, update_request)
+
+    return get_suspended_processes(request, group_name)
+
+def resume_process(request, group_name, process_name):
+    update_request = {}
+    update_request["resume"] = [process_name]
+    autoscaling_groups_helper.update_scaling_process(request, group_name, update_request)
+
+    return get_suspended_processes(request, group_name)
 
 def delete_scheduled_actions(request, group_name):
     params = request.POST
diff --git a/deploy-board/deploy_board/webapp/helpers/autoscaling_groups_helper.py b/deploy-board/deploy_board/webapp/helpers/autoscaling_groups_helper.py
index bdede847a2..72ae8689e8 100644
--- a/deploy-board/deploy_board/webapp/helpers/autoscaling_groups_helper.py
+++ b/deploy-board/deploy_board/webapp/helpers/autoscaling_groups_helper.py
@@ -159,6 +159,12 @@ def disable_scaling_down_event(request, cluster_name):
 def get_disabled_asg_actions(request, cluster_name):
     return rodimus_client.get("/clusters/%s/autoscaling/action" % cluster_name, request.teletraan_user_id.token)
 
+def update_scaling_process(request, cluster_name, update_request):
+    return rodimus_client.post("/clusters/%s/autoscaling/processes" % cluster_name, request.teletraan_user_id.token,
+                               data=update_request)
+
+def get_available_scaling_process(request, cluster_name):
+    return rodimus_client.get("/clusters/%s/autoscaling/available-processes" % cluster_name, request.teletraan_user_id.token)
 
 # Asg Alarms
 def put_scaling_policies(request, cluster_name, policies_info):