From 392e4f44651d270660799f42c09183e9e684a787 Mon Sep 17 00:00:00 2001
From: Aidaho <aidaho@roxy-wi.org>
Date: Sat, 24 Feb 2024 21:07:18 +0300
Subject: [PATCH] v7.2.0.0

SMON agent fixes
---
 app/routes/smon/agent_routes.py    | 14 ++++++++++++--
 app/templates/ajax/smon/agent.html |  3 +++
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/app/routes/smon/agent_routes.py b/app/routes/smon/agent_routes.py
index df130e39..81dbfa29 100644
--- a/app/routes/smon/agent_routes.py
+++ b/app/routes/smon/agent_routes.py
@@ -2,8 +2,7 @@ from flask import render_template, request, jsonify, g
 from flask_login import login_required
 
 from app.routes.smon import bp
-
-from middleware import get_user_params
+from app.middleware import get_user_params
 import app.modules.db.smon as smon_sql
 import app.modules.common.common as common
 import app.modules.tools.smon_agent as smon_agent
@@ -74,6 +73,17 @@ def get_free_agents():
     return jsonify(servers)
 
 
+@bp.get('/agent/count')
+@login_required
+def get_agent_count():
+    try:
+        smon_agent.check_agent_limit()
+    except Exception as e:
+        return f'{e}'
+
+    return 'ok'
+
+
 @bp.get('/agent/<int:agent_id>')
 @login_required
 @get_user_params()
diff --git a/app/templates/ajax/smon/agent.html b/app/templates/ajax/smon/agent.html
index 06bbf58b..ef73fe93 100644
--- a/app/templates/ajax/smon/agent.html
+++ b/app/templates/ajax/smon/agent.html
@@ -40,6 +40,9 @@
         <div>
             <b>{{lang.words.uptime|title()}}</b>: <time class="timeago" datetime="" id="agent-uptime-{{ agent.id }}" title="{{ lang.words.agent|title() }} {{ lang.words.uptime }}"></time>
         </div>
+        <div>
+            <b>{{ lang.words.agent|title() }} ID</b>: {{ agent.id }}
+        </div>
     </div>
 </div>
     <script>