bot: Added site info command

discord-bot/bot.py

@@ -69,13 +69,25 @@ async def createsite(ctx, domain: str, licence: str):
     if r.status_code == 200:
         json = r.json()
         if json['success'] == "true":
-            await ctx.response.send_message(f"Site {domain} creating...\nPlease send /siteinfo domain:{domain}")
+            await ctx.response.send_message(f"Site {domain} creating...\nPlease wait a few minutes and then send /siteinfo domain:{domain}")
         else:
             await ctx.response.send_message(f"Error creating site\n" + json['error'])
     else:
         await ctx.response.send_message(f"Error creating site\n" + r.text)
 
 
+@tree.command(name="siteinfo", description="Get info about a WordPress site")
+async def siteinfo(ctx, domain: str):
+    r = requests.get(f"http://{Master_IP}:{Master_Port}/site-info?domain={domain}")
+    if r.status_code == 200:
+        json = r.json()
+        if json['success'] == "true":
+            await ctx.response.send_message(f"Site: {domain}\nStatus: {json['status']}\nWorker: {json['worker']}")
+        else:
+            await ctx.response.send_message(f"Error getting site info\n" + json['error'])
+    else:
+        await ctx.response.send_message(f"Error getting site info\n" + r.text)
+
 # When the bot is ready
 @client.event
 async def on_ready():

master/main.py

@@ -203,13 +203,9 @@ def site_status():
 
     if "tlsa" in json:
         tlsa = json['tlsa']
+        return jsonify({'success': 'true', 'domain': domain, 'ip': ip, 'tlsa': tlsa})
     else:
-        tlsa = "none"
+        return jsonify({'success': 'false', 'domain': domain, 'ip': ip, 'tlsa': 'none','error': 'No TLSA record found'})
-
-    # Return status
-
-
-    return jsonify({'success': 'true', 'domain': domain, 'ip': ip, 'tlsa': tlsa})
 
 
 @app.route('/tlsa', methods=['GET'])