From 4303f568b3877a0d1c0e869129a9199aa70bf5dd Mon Sep 17 00:00:00 2001
From: Karl <karl@k-world.me.uk>
Date: Sat, 19 Jul 2025 09:16:21 +0100
Subject: [PATCH] modify dns in config

---
 ktvmanager/main.py       |  2 ++
 ktvmanager/routes/dns.py | 47 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 49 insertions(+)
 create mode 100644 ktvmanager/routes/dns.py

diff --git a/ktvmanager/main.py b/ktvmanager/main.py
index 141c234..a39a5a3 100644
--- a/ktvmanager/main.py
+++ b/ktvmanager/main.py
@@ -3,6 +3,7 @@ from flask import Flask, jsonify
 from dotenv import load_dotenv
 from ktvmanager.config import DevelopmentConfig, ProductionConfig
 from routes.api import api_blueprint
+from routes.dns import dns_bp
 from ktvmanager.lib.database import initialize_db_pool
 from ktvmanager.account_checker import send_expiry_notifications
 from apscheduler.schedulers.background import BackgroundScheduler
@@ -27,6 +28,7 @@ def create_app():
 
     # Register blueprints
     app.register_blueprint(api_blueprint)
+    app.register_blueprint(dns_bp)
 
     @app.route('/check-expiry', methods=['POST'])
     def check_expiry():
diff --git a/ktvmanager/routes/dns.py b/ktvmanager/routes/dns.py
new file mode 100644
index 0000000..612c143
--- /dev/null
+++ b/ktvmanager/routes/dns.py
@@ -0,0 +1,47 @@
+from flask import Blueprint, render_template, request, redirect, url_for, flash
+import os
+
+dns_bp = Blueprint('dns', __name__)
+
+DNS_FILE = os.path.join(os.path.dirname(__file__), '..', 'lib', 'DNS_list.txt')
+
+def read_dns_list():
+    if not os.path.exists(DNS_FILE):
+        return []
+    with open(DNS_FILE, 'r') as f:
+        return [line.strip() for line in f.readlines()]
+
+def write_dns_list(dns_list):
+    with open(DNS_FILE, 'w') as f:
+        for item in dns_list:
+            f.write(f"{item}\n")
+
+@dns_bp.route('/add_dns', methods=['POST'])
+def add_dns():
+    dns_entry = request.form.get('dns_entry')
+    if dns_entry:
+        dns_list = read_dns_list()
+        if dns_entry not in dns_list:
+            dns_list.append(dns_entry)
+            write_dns_list(dns_list)
+            flash('DNS entry added successfully.', 'success')
+        else:
+            flash('DNS entry already exists.', 'info')
+    else:
+        flash('DNS entry cannot be empty.', 'danger')
+    return redirect(url_for('config_dashboard'))
+
+@dns_bp.route('/remove_dns', methods=['POST'])
+def remove_dns():
+    dns_entry = request.form.get('dns_entry')
+    if dns_entry:
+        dns_list = read_dns_list()
+        if dns_entry in dns_list:
+            dns_list.remove(dns_entry)
+            write_dns_list(dns_list)
+            flash('DNS entry removed successfully.', 'success')
+        else:
+            flash('DNS entry not found.', 'info')
+    else:
+        flash('DNS entry cannot be empty.', 'danger')
+    return redirect(url_for('config_dashboard'))
\ No newline at end of file