feat: Add discord notifications and some speed improvments

server.py

@@ -25,6 +25,9 @@ import dnslib.dns
 import socket
 from datetime import datetime
 from dateutil import relativedelta
+import dotenv
+
+dotenv.load_dotenv()
 
 app = Flask(__name__)
 
@@ -45,6 +48,8 @@ node_locations = {
 nodes = []
 last_log = datetime.now() - relativedelta.relativedelta(years=1)
 
+sent_notifications = {}
+
 log_dir = "/data"
 if not os.path.exists(log_dir):
     if not os.path.exists("./logs"):
@@ -116,6 +121,7 @@ def get_node_list() -> list:
         ips.append(ipval.to_text())
     return ips
 
+
 def check_plain_dns(ip: str) -> bool:
     resolver = dns.resolver.Resolver()
     resolver.nameservers = [ip]
@@ -130,16 +136,18 @@ def check_plain_dns(ip: str) -> bool:
         print(e)
         return False
 
-def build_dns_query(domain: str, qtype: str = 'A'):
+
+def build_dns_query(domain: str, qtype: str = "A"):
     """
     Constructs a DNS query in binary wire format using dnslib.
     """
     q = dnslib.DNSRecord.question(domain, qtype)
     return q.pack()
 
+
 def check_doh(ip: str) -> bool:
     status = False
-    dns_query = build_dns_query("2.wdbrn", 'TXT')
+    dns_query = build_dns_query("2.wdbrn", "TXT")
     request = (
         f"POST /dns-query HTTP/1.1\r\n"
         f"Host: hnsdoh.com\r\n"
@@ -161,10 +169,10 @@ def check_doh(ip: str) -> bool:
                 break
             response_data += data
 
-        response_str = response_data.decode('latin-1')
+        response_str = response_data.decode("latin-1")
         headers, body = response_str.split("\r\n\r\n", 1)
 
-        dns_response:dnslib.DNSRecord = dnslib.DNSRecord.parse(body.encode('latin-1'))
+        dns_response: dnslib.DNSRecord = dnslib.DNSRecord.parse(body.encode("latin-1"))
         for rr in dns_response.rr:
             if "Test 2" in str(rr):
                 status = True
@@ -177,11 +185,14 @@ def check_doh(ip: str) -> bool:
         ssock.close()
     return status
 
+
 def check_dot(ip: str) -> bool:
     qname = dns.name.from_text("3.wdbrn")
     q = dns.message.make_query(qname, dns.rdatatype.TXT)
     try:
-        response = dns.query.tls(q, ip, timeout=5, port=853, server_hostname="hnsdoh.com")
+        response = dns.query.tls(
+            q, ip, timeout=5, port=853, server_hostname="hnsdoh.com"
+        )
         if response.rcode() == dns.rcode.NOERROR:
             for rrset in response.answer:
                 for rr in rrset:
@@ -192,6 +203,7 @@ def check_dot(ip: str) -> bool:
         print(e)
         return False
 
+
 def verify_cert(ip: str, port: int) -> bool:
     sock = socket.create_connection((ip, port))
 
@@ -204,20 +216,19 @@ def verify_cert(ip: str,port:int) -> bool:
         cert = ssock.getpeercert()
 
         # Extract the expiry date from the certificate
-        expiry_date_str = cert['notAfter']
+        expiry_date_str = cert["notAfter"]
 
         # Convert the expiry date string to a datetime object
-        expiry_date = datetime.strptime(expiry_date_str, '%b %d %H:%M:%S %Y GMT')
+        expiry_date = datetime.strptime(expiry_date_str, "%b %d %H:%M:%S %Y GMT")
         expires = format_relative_time(expiry_date)
         valid = expiry_date > datetime.now()
 
-        
-    
     finally:
         # Close the SSL and socket connection
         ssock.close()
     return {"valid": valid, "expires": expires, "expiry_date": expiry_date_str}
 
+
 def format_relative_time(expiry_date: datetime) -> str:
     now = datetime.now()
     delta = expiry_date - now
@@ -235,13 +246,17 @@ def format_relative_time(expiry_date: datetime) -> str:
     else:
         return f"in {delta.seconds} seconds" if delta.seconds > 1 else "in 1 second"
 
+
 def check_nodes() -> list:
     global nodes
     if last_log > datetime.now() - relativedelta.relativedelta(minutes=1):
         # Load the last log
         with open(f"{log_dir}/node_status.json", "r") as file:
             data = json.load(file)
-        newest = {"date": datetime.now()-relativedelta.relativedelta(years=1), "nodes": []}
+        newest = {
+            "date": datetime.now() - relativedelta.relativedelta(years=1),
+            "nodes": [],
+        }
         for entry in data:
             if datetime.strptime(entry["date"], "%Y-%m-%d %H:%M:%S") > newest["date"]:
                 newest = entry
@@ -252,38 +267,160 @@ def check_nodes() -> list:
             nodes = get_node_list()
             node_status = []
             for ip in nodes:
-                node_status.append({
+                node_status.append(
+                    {
                         "ip": ip,
                         "name": node_names[ip] if ip in node_names else ip,
-                    "location": node_locations[ip] if ip in node_locations else "Unknown",
+                        "location": (
+                            node_locations[ip] if ip in node_locations else "Unknown"
+                        ),
                         "plain_dns": check_plain_dns(ip),
                         "doh": check_doh(ip),
                         "dot": check_dot(ip),
                         "cert": verify_cert(ip, 443),
-                    "cert_853": verify_cert(ip,853)
-                    })
+                        "cert_853": verify_cert(ip, 853),
+                    }
+                )
         else:
             node_status = []
             for ip in nodes:
-                node_status.append({
+                node_status.append(
+                    {
                         "ip": ip,
                         "name": node_names[ip] if ip in node_names else ip,
-                    "location": node_locations[ip] if ip in node_locations else "Unknown",
+                        "location": (
+                            node_locations[ip] if ip in node_locations else "Unknown"
+                        ),
                         "plain_dns": check_plain_dns(ip),
                         "doh": check_doh(ip),
                         "dot": check_dot(ip),
                         "cert": verify_cert(ip, 443),
-                    "cert_853": verify_cert(ip,853)
-                    })
+                        "cert_853": verify_cert(ip, 853),
+                    }
+                )
         # Save the node status to a file
         log_status(node_status)
     print("Finished checking nodes", flush=True)
+
+    # Send notifications if any nodes are down
+    for node in node_status:
+        if not node["plain_dns"] or not node["doh"] or not node["dot"] or not node["cert"]["valid"] or not node["cert_853"]["valid"]:
+            send_down_notification(node)
+            break
+        # Check if cert is expiring in 7 days
+        cert_expiry = datetime.strptime(
+            node["cert"]["expiry_date"], "%b %d %H:%M:%S %Y GMT"
+        )
+        if cert_expiry < datetime.now() + relativedelta.relativedelta(days=7):
+            if node["ip"] not in sent_notifications:
+                sent_notifications[node["ip"]] = datetime.now()
+                send_down_notification(node)
+                break
+            if sent_notifications[node["ip"]] < datetime.now() - relativedelta.relativedelta(days=1):   
+                send_down_notification(node)
+                break
+        cert_853_expiry = datetime.strptime(
+            node["cert_853"]["expiry_date"], "%b %d %H:%M:%S %Y GMT"
+        )
+        if cert_853_expiry < datetime.now() + relativedelta.relativedelta(days=7):
+            if node["ip"] not in sent_notifications:
+                sent_notifications[node["ip"]] = datetime.now()
+                send_down_notification(node)
+                break
+            if sent_notifications[node["ip"]] < datetime.now() - relativedelta.relativedelta(days=1):   
+                send_down_notification(node)
+                break
     return node_status
 
+def check_nodes_from_log() -> list:
+    global last_log
+    if last_log > datetime.now() - relativedelta.relativedelta(minutes=10):
+        # Load the last log
+        with open(f"{log_dir}/node_status.json", "r") as file:
+            data = json.load(file)
+        newest = {
+            "date": datetime.now() - relativedelta.relativedelta(years=1),
+            "nodes": [],
+        }
+        for entry in data:
+            if datetime.strptime(entry["date"], "%Y-%m-%d %H:%M:%S") > newest["date"]:
+                newest = entry
+                newest["date"] = datetime.strptime(newest["date"], "%Y-%m-%d %H:%M:%S")
+        node_status = newest["nodes"]
+    elif last_log < datetime.now() - relativedelta.relativedelta(days=7):
+        # Load the log file
+        with open(f"{log_dir}/node_status.json", "r") as file:
+            data = json.load(file)
+        newest = {
+            "date": datetime.now() - relativedelta.relativedelta(years=1),
+            "nodes": [],
+        }
+        for entry in data:
+            if datetime.strptime(entry["date"], "%Y-%m-%d %H:%M:%S") > newest["date"]:
+                newest = entry
+                newest["date"] = datetime.strptime(newest["date"], "%Y-%m-%d %H:%M:%S")
+        node_status = newest["nodes"]
+        if newest["date"] > datetime.now() - relativedelta.relativedelta(minutes=10):
+            last_log = newest["date"]
+            print(f"Loaded log from {last_log}", flush=True)
+            return node_status
+    else:
+        node_status = check_nodes()
+    return node_status
+
+
+
+def send_notification(title, description,author):
+    discord_hook = os.getenv("DISCORD_HOOK")
+    if discord_hook:
+        data = {
+            "content": "",
+            "embeds": [
+                {
+                    "title": title,
+                    "description": description,
+                    "url": "https://hnsdoh.c.woodburn.au",
+                    "color": 5814783,
+                    "author": {
+                        "name": author,
+                        "icon_url": "https://hnsdoh.c.woodburn.au/favicon.png",
+                    },
+                }
+            ],
+            "username": "HNSDoH",
+            "avatar_url": "https://hnsdoh.c.woodburn.au/favicon.png",
+            "attachments": [],
+        }
+        response = requests.post(discord_hook, json=data)
+        print(response.text, flush=True)
+
+def send_down_notification(node):
+    title = f"{node['name']} is down"
+
+    description = f"{node['name']} ({node['ip']}) is down with the following issues:\n"
+    if not node["plain_dns"]:
+        description += "- Plain DNS is down\n"
+    if not node["doh"]:
+        description += "- DoH is down\n"
+    if not node["dot"]:
+        description += "- DoT is down\n"
+    if not node["cert"]["valid"]:
+        description += "- Certificate on port 443 is invalid\n"
+    if not node["cert_853"]["valid"]:
+        description += "- Certificate on port 853 is invalid\n"
+    
+    # Also add the expiry date of the certificates
+    description += "\nCertificate expiry dates:\n"
+    description += f"- Certificate on port 443 expires {node['cert']['expires']}\n"
+    description += f"- Certificate on port 853 expires {node['cert_853']['expires']}\n"
+    send_notification(title, description, node['name'])
+
+
 # endregion
 
 # region File logs
 
+
 def log_status(node_status: list):
     global last_log
     last_log = datetime.now()
@@ -313,23 +450,27 @@ def log_status(node_status: list):
         data = []
 
     # Add the new entry
-    data.append({
-        "date": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
-        "nodes": node_status
-    })
+    data.append(
+        {"date": datetime.now().strftime("%Y-%m-%d %H:%M:%S"), "nodes": node_status}
+    )
 
     with open(filename, "w") as file:
         json.dump(data, file, indent=4)
 
+
 # endregion
 # region History functions
 def get_history(days: int) -> list:
-    log_files = [f for f in os.listdir(log_dir) if f.endswith('.json') and f.startswith('node_status')]
+    log_files = [
+        f
+        for f in os.listdir(log_dir)
+        if f.endswith(".json") and f.startswith("node_status")
+    ]
     history = []
 
     for log_file in log_files:
         file_path = os.path.join(log_dir, log_file)
-        with open(file_path, 'r') as file:
+        with open(file_path, "r") as file:
             data = json.load(file)
             for entry in data:
                 entry_date = datetime.strptime(entry["date"], "%Y-%m-%d %H:%M:%S")
@@ -337,26 +478,31 @@ def get_history(days: int) -> list:
                     history.append(entry)
     return history
 
+
 def summarize_history(history: list) -> dict:
-    nodes_status = defaultdict(lambda: {
+    nodes_status = defaultdict(
+        lambda: {
             "name": "",
             "location": "",
             "plain_dns": {"last_down": "Never", "percentage": 0},
             "doh": {"last_down": "Never", "percentage": 0},
-        "dot": {"last_down": "Never", "percentage": 0}
-    })
+            "dot": {"last_down": "Never", "percentage": 0},
+        }
+    )
     overall_status = {
         "plain_dns": {"last_down": "Never", "percentage": 0},
         "doh": {"last_down": "Never", "percentage": 0},
-        "dot": {"last_down": "Never", "percentage": 0}
+        "dot": {"last_down": "Never", "percentage": 0},
     }
 
     # Collect data
-    total_counts = defaultdict(lambda: {
+    total_counts = defaultdict(
+        lambda: {
             "plain_dns": {"down": 0, "total": 0},
             "doh": {"down": 0, "total": 0},
-        "dot": {"down": 0, "total": 0}
-    })
+            "dot": {"down": 0, "total": 0},
+        }
+    )
 
     for entry in history:
         date = datetime.strptime(entry["date"], "%Y-%m-%d %H:%M:%S")
@@ -377,7 +523,9 @@ def summarize_history(history: list) -> dict:
             # Update last downtime for each key
             for key in ["plain_dns", "doh", "dot"]:
                 if node.get(key) == "down":
-                    nodes_status[ip][key]["last_down"] = date.strftime("%Y-%m-%d %H:%M:%S")
+                    nodes_status[ip][key]["last_down"] = date.strftime(
+                        "%Y-%m-%d %H:%M:%S"
+                    )
 
     # Calculate percentages and prepare final summary
     node_list = []
@@ -393,7 +541,11 @@ def summarize_history(history: list) -> dict:
         node_list.append(node_data)
 
     # Aggregate overall status
-    overall_counts = {"plain_dns": {"down": 0, "total": 0}, "doh": {"down": 0, "total": 0}, "dot": {"down": 0, "total": 0}}
+    overall_counts = {
+        "plain_dns": {"down": 0, "total": 0},
+        "doh": {"down": 0, "total": 0},
+        "dot": {"down": 0, "total": 0},
+    }
     for ip, counts in total_counts.items():
         for key in ["plain_dns", "doh", "dot"]:
             overall_counts[key]["total"] += counts[key]["total"]
@@ -404,25 +556,28 @@ def summarize_history(history: list) -> dict:
         down = overall_counts[key]["down"]
         if total > 0:
             overall_status[key]["percentage"] = ((total - down) / total) * 100
-            last_downs = [nodes_status[ip][key]["last_down"] for ip in nodes_status if nodes_status[ip][key]["last_down"] != "Never"]
+            last_downs = [
+                nodes_status[ip][key]["last_down"]
+                for ip in nodes_status
+                if nodes_status[ip][key]["last_down"] != "Never"
+            ]
             if last_downs:
                 overall_status[key]["last_down"] = max(last_downs)
         else:
             overall_status[key]["percentage"] = 100
 
-    return {
-        "nodes": node_list,
-        "overall": overall_status,
-        "check_counts": total_counts
-    }
+    return {"nodes": node_list, "overall": overall_status, "check_counts": total_counts}
+
+
 def convert_nodes_to_dict(nodes):
     nodes_dict = {}
     for node in nodes:
-        name = node.get('name')
+        name = node.get("name")
         if name:
             nodes_dict[name] = node
     return nodes_dict
 
+
 # endregion
 
 
@@ -432,6 +587,7 @@ def api_nodes():
     node_status = check_nodes()
     return jsonify(node_status)
 
+
 @app.route("/api/history")
 def api_history():
     history_days = 7
@@ -444,33 +600,71 @@ def api_history():
     history_summary = summarize_history(history)
     return jsonify(history_summary)
 
+
+@app.route("/api/history/<int:days>")
+def api_history_days(days: int):
+    history = get_history(days)
+    history_summary = summarize_history(history)
+    return jsonify(history_summary)
+
+
+@app.route("/api/full")
+def api_all():
+    history_days = 7
+    if "history" in request.args:
+        try:
+            history_days = int(request.args["history"])
+        except:
+            pass
+    if "days" in request.args:
+        try:
+            history_days = int(request.args["days"])
+        except:
+            pass
+    history = get_history(history_days)
+    return jsonify(history)
+
+
 # endregion
 
 
 # region Main routes
 @app.route("/")
 def index():
-    node_status = check_nodes()
-    
+    node_status = check_nodes_from_log()
 
     warnings = []
     for node in node_status:
         if not node["plain_dns"]:
             warnings.append(f"{node['name']} does not support plain DNS")
+
         if not node["doh"]:
             warnings.append(f"{node['name']} does not support DoH")
+
         if not node["dot"]:
             warnings.append(f"{node['name']} does not support DoT")
+
         if not node["cert"]["valid"]:
             warnings.append(f"{node['name']} has an invalid certificate")
+
         if not node["cert_853"]["valid"]:
             warnings.append(f"{node['name']} has an invalid certificate on port 853")
-        cert_expiry = datetime.strptime(node["cert"]["expiry_date"], '%b %d %H:%M:%S %Y GMT')
+
+        cert_expiry = datetime.strptime(
+            node["cert"]["expiry_date"], "%b %d %H:%M:%S %Y GMT"
+        )
         if cert_expiry < datetime.now() + relativedelta.relativedelta(days=7):
-            warnings.append(f"{node['name']} has a certificate expiring in less than 7 days on port 443")
-        cert_853_expiry = datetime.strptime(node["cert_853"]["expiry_date"], '%b %d %H:%M:%S %Y GMT')
+            warnings.append(
+                f"{node['name']} has a certificate expiring in less than 7 days on port 443"
+            )
+        cert_853_expiry = datetime.strptime(
+            node["cert_853"]["expiry_date"], "%b %d %H:%M:%S %Y GMT"
+        )
         if cert_853_expiry < datetime.now() + relativedelta.relativedelta(days=7):
-            warnings.append(f"{node['name']} has a certificate expiring in less than 7 days on port 853")
+            warnings.append(
+                f"{node['name']} has a certificate expiring in less than 7 days on port 853"
+            )
+        
 
     history_days = 7
     if "history" in request.args:
@@ -483,7 +677,13 @@ def index():
     history_summary["nodes"] = convert_nodes_to_dict(history_summary["nodes"])
     last_check = last_log.strftime("%Y-%m-%d %H:%M:%S")
     print(f"Last check: {last_check}", flush=True)
-    return render_template("index.html",nodes=node_status,warnings=warnings,history=history_summary,last_check=last_check)
+    return render_template(
+        "index.html",
+        nodes=node_status,
+        warnings=warnings,
+        history=history_summary,
+        last_check=last_check,
+    )
 
 
 @app.route("/<path:path>")