From 383194cf66fefe9a15b0eced1c7fd4d3104d5cec Mon Sep 17 00:00:00 2001
From: amcmanu3 <mcdeweykp@gmail.com>
Date: Mon, 18 Jul 2022 17:12:43 -0400
Subject: [PATCH] Fix servers being out of order on creation page.

---
 app/classes/web/panel_handler.py  |  2 --
 app/classes/web/server_handler.py | 21 ++++++++++++++++++++-
 2 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/app/classes/web/panel_handler.py b/app/classes/web/panel_handler.py
index f161ab47..fd4c2367 100644
--- a/app/classes/web/panel_handler.py
+++ b/app/classes/web/panel_handler.py
@@ -36,8 +36,6 @@ class PanelHandler(BaseHandler):
         user_roles = {}
         for user_id in self.controller.users.get_all_user_ids():
             user_roles_list = self.controller.users.get_user_roles_names(user_id)
-            # user_servers =
-            # self.controller.servers.get_authorized_servers(user.user_id)
             user_roles[user_id] = user_roles_list
         return user_roles
 
diff --git a/app/classes/web/server_handler.py b/app/classes/web/server_handler.py
index a649f4ac..e4e10a04 100644
--- a/app/classes/web/server_handler.py
+++ b/app/classes/web/server_handler.py
@@ -62,12 +62,31 @@ class ServerHandler(BaseHandler):
                 exec_user_role.add(role["role_name"])
                 list_roles.append(self.controller.roles.get_role(role["role_id"]))
 
+        user_order = self.controller.users.get_user_by_id(exec_user["user_id"])
+        user_order = user_order["server_order"].split(",")
         page_servers = []
+        server_ids = []
+
+        for server_id in user_order[:]:
+            for server in defined_servers[:]:
+                if str(server.server_id) == str(server_id):
+                    page_servers.append(
+                        DatabaseShortcuts.get_data_obj(server.server_object)
+                    )
+                    user_order.remove(server_id)
+                    defined_servers.remove(server)
+
         for server in defined_servers:
+            server_ids.append(str(server.server_id))
             if server not in page_servers:
                 page_servers.append(
                     DatabaseShortcuts.get_data_obj(server.server_object)
                 )
+
+        for server_id in user_order[:]:
+            # remove IDs in list that user no longer has access to
+            if str(server_id) not in server_ids:
+                user_order.remove(server_id)
         defined_servers = page_servers
 
         template = "public/404.html"
@@ -92,7 +111,7 @@ class ServerHandler(BaseHandler):
                 ),
             },
             "hosts_data": self.controller.management.get_latest_hosts_stats(),
-            "menu_servers": defined_servers,
+            "menu_servers": page_servers,
             "show_contribute": self.helper.get_setting("show_contribute_link", True),
             "lang": self.controller.users.get_user_lang_by_id(exec_user["user_id"]),
             "lang_page": Helpers.get_lang_page(