From b224103778f3fa51ea6fd2c27f588a0484f6b3c5 Mon Sep 17 00:00:00 2001
From: amcmanu3 <mcdeweykp@gmail.com>
Date: Sun, 5 Feb 2023 16:47:12 -0500
Subject: [PATCH] Add storage to the dashboard

---
 app/classes/web/panel_handler.py            |  7 +++-
 app/frontend/templates/panel/dashboard.html | 37 +++++++++++++++++++--
 2 files changed, 40 insertions(+), 4 deletions(-)

diff --git a/app/classes/web/panel_handler.py b/app/classes/web/panel_handler.py
index 74c93b88..1f61a702 100644
--- a/app/classes/web/panel_handler.py
+++ b/app/classes/web/panel_handler.py
@@ -332,7 +332,12 @@ class PanelHandler(BaseHandler):
             else None,
             "superuser": superuser,
         }
-
+        try:
+            page_data["hosts_data"]["disk_json"] = json.loads(
+                page_data["hosts_data"]["disk_json"].replace("'", '"')
+            )
+        except:
+            page_data["hosts_data"]["disk_json"] = {}
         if page == "unauthorized":
             template = "panel/denied.html"
 
diff --git a/app/frontend/templates/panel/dashboard.html b/app/frontend/templates/panel/dashboard.html
index adfe7645..49dc6d25 100644
--- a/app/frontend/templates/panel/dashboard.html
+++ b/app/frontend/templates/panel/dashboard.html
@@ -50,7 +50,7 @@
       <div class="card">
         <div class="card-body">
           <div class="row">
-            <div class="col-lg-4 col-md-6">
+            <div class="col-lg-3 col-md-6">
               <div class="d-flex">
                 <div class="wrapper">
                   <h5 class="mb-1 font-weight-medium text-primary"> {{ translate('dashboard', 'host', data['lang']) }}
@@ -72,7 +72,7 @@
                 </div>
               </div>
             </div>
-            <div class="col-lg-4 col-md-6 mt-md-0 mt-4">
+            <div class="col-lg-3 col-md-6 mt-md-0 mt-4">
               <div class="d-flex">
                 <div class="wrapper">
                   <h5 class="mb-1 font-weight-medium text-primary">{{ translate('dashboard', 'servers', data['lang']) }}
@@ -88,7 +88,7 @@
                 </div>
               </div>
             </div>
-            <div class="col-lg-4 col-md-6 mt-md-0 mt-4">
+            <div class="col-lg-2 col-md-6 mt-md-0 mt-4">
               <div class="d-flex">
                 <div class="wrapper">
                   <h5 class="mb-1 font-weight-medium text-primary">{{ translate('dashboard', 'players', data['lang']) }}
@@ -101,6 +101,37 @@
                 </div>
               </div>
             </div>
+            <div class="col-lg-4 col-md-6 mt-md-0 mt-4">
+              <div class="d-flex">
+                <div class="wrapper" style="width: 100%;">
+                  <h5 class="mb-1 font-weight-medium text-primary">Storage
+                  </h5>
+                  <ul style="width: 100%; list-style: none;">
+                    {% for item in data.get('hosts_data').get('disk_json') %}
+                    <li id="{{item['device']}}">
+                      <h6 style="display: inline-block;">{{item["mount"]}}</h6>
+                      <div class="progress"
+                        style="display: inline-block; height: 20%; width: 70%; background-color: rgb(139, 139, 139) !important;">
+                        <div class="progress-bar
+                          {% if item['percent_used'] <= 33 %}
+                          bg-success
+                          {% elif 34 <= item['percent_used'] <= 75 %}
+                          bg-warning
+                          {% else %}
+                          bg-danger
+                          {% end %}
+                          " role="progressbar" style="color: black; height: 100%; width: {{item['percent_used']}}%;"
+                          aria-valuenow="0" aria-valuemin="0" aria-valuemax="100">{{item["used"]}} / {{item["total"]}}
+                        </div>
+                      </div>
+
+                    </li>
+                    {% end %}
+                    </li>
+                  </ul>
+                </div>
+              </div>
+            </div>
           </div>
         </div>
       </div>