From c130bfed695b3a79272343203d50e587c03d0fae Mon Sep 17 00:00:00 2001
From: Nathan Woodburn <github@nathan.woodburn.au>
Date: Sun, 17 Sep 2023 08:47:36 +1000
Subject: [PATCH] fix: Set domain to lower in api requests

---
 master/main.py                 | 9 ++++++++-
 master/templates/register.html | 2 +-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/master/main.py b/master/main.py
index 87368fa..6434604 100644
--- a/master/main.py
+++ b/master/main.py
@@ -206,6 +206,7 @@ def list_workers():
 @app.route('/site-info', methods=['GET'])
 def site_status():
     domain = request.args.get('domain')
+    domain = domain.lower()
     if domain == None:
         return jsonify({'error': 'Invalid domain', 'success': 'false'})
     
@@ -236,6 +237,7 @@ def site_status():
 @app.route('/info')
 def site_status_human():
     domain = request.args.get('domain')
+    domain = domain.lower()
     if domain == None:
         return "<h1>Invalid domain</h1>"
     
@@ -265,6 +267,7 @@ def site_status_human():
 @app.route('/tlsa', methods=['GET'])
 def tlsa():
     domain = request.args.get('domain')
+    domain = domain.lower()
     if domain == None:
         return jsonify({'error': 'Invalid domain', 'success': 'false'})
     
@@ -480,6 +483,9 @@ def register_post():
     if domain.startswith("http://") or domain.startswith("https://"):
         return render_template('register.html', buy_licence_link=buy_licence_link, ERROR_MESSAGE="Domain should not contain http:// or https://")
     
+    # Set domain to lowercase
+    domain = domain.lower()
+
     # Check if worker file exists
     workers = None
     try:
@@ -529,6 +535,7 @@ def success():
     if 'domain' not in request.args:
         return redirect('/')
     domain = request.args.get('domain')
+    domain = domain.lower()
     if not site_exists(domain):
         return render_template('success.html', message="Error: Domain does not exist\nPlease contact support")
     
@@ -711,7 +718,7 @@ def addsite():
     if domain.startswith("http://") or domain.startswith("https://"):
         return jsonify({'error': 'Domain should not contain http:// or https://', 'success': 'false'})
     
-
+    domain = domain.lower()
     # Check if worker file exists
     workers = None
     try:
diff --git a/master/templates/register.html b/master/templates/register.html
index 4085806..9dec5a3 100644
--- a/master/templates/register.html
+++ b/master/templates/register.html
@@ -45,7 +45,7 @@
                                 <div class="mb-3"><input type="text" name="domain" placeholder="Domain"></div>
                                 <div class="mb-3"><input type="text" placeholder="Licence Key" name="licence"></div>
                                 <div class="mb-3">
-                                    <p style="color: rgb(255,0,0);">{{ERROR_MESSAGE}}</p><button class="btn btn-primary shadow d-block w-100" type="submit">Sign up</button>
+                                    <p style="color: rgb(255,0,0);">{{ERROR_MESSAGE}}</p><button class="btn btn-primary shadow d-block w-100" type="submit">Create site</button>
                                 </div>
                                 <p class="text-muted">Don't have a licence?<br><a href="{{buy_licence_link}}" target="_blank">Click here to buy one</a></p>
                             </form>