smileos

src/dynamic_routes.py

@@ -1,7 +1,12 @@
 # Imports
-from flask import Blueprint, render_template, abort
+from flask import Blueprint, render_template, abort, request
 import os, markdown
 
+try:
+    from src.name import get_name
+except ImportError:
+    from name import get_name
+
 
 # Create blueprint
 bp = Blueprint('dynamic_routes', __name__)
@@ -33,20 +38,22 @@ def catch_all(filename):
             return render_template(
                 'bases/directory.html',
                 directory=filename + "/" if not filename.endswith('/') else filename,
-                pages=ListFiles(filename)
+                pages=ListFiles(filename), 
+                name=get_name(request)
             )
         
-        return render_template(f'pages/{filename}')
+        return render_template(f'pages/{filename}', name=get_name(request))
 
     elif os.path.exists(get_path(filename + '.html')):
-        return render_template(f'pages/{filename}.html')
+        return render_template(f'pages/{filename}.html', name=get_name(request))
 
     elif os.path.exists(get_path(filename + '.md')):
         output = markdown.markdown(open(get_path(filename + '.md'), "r").read())
         return render_template(
             f'bases/md.html',
             title = filename.split("/")[-1],
-            markdown = output
+            markdown = output,
+            name=get_name(request)
         )
 
     else:

src/errors.py

@@ -1,6 +1,10 @@
-from flask import Blueprint, render_template
+from flask import Blueprint, render_template, request
 from werkzeug.exceptions import HTTPException
-from os import getenv as env
+
+try:
+    from src.name import get_name
+except ImportError:
+    from name import get_name
 
 
 bp = Blueprint("errors", __name__)
@@ -9,19 +13,19 @@ bp = Blueprint("errors", __name__)
 @bp.route('/500')
 @bp.app_errorhandler(500)
 def internal_server_error(error:HTTPException=None):
-    return render_template('errors/500.html', error=error), 500
+    return render_template('errors/500.html', error=error, name=get_name(request)), 500
 
 
 @bp.route('/404')
 @bp.app_errorhandler(404)
 def not_found(error:HTTPException=None):
-    return render_template('errors/404.html', error=error), 404
+    return render_template('errors/404.html', error=error, name=get_name(request)), 404
 
 
 @bp.route('/400')
 @bp.app_errorhandler(400)
 def bad_request(error:HTTPException=None):
-    return render_template('errors/400.html', error=error), 400
+    return render_template('errors/400.html', error=error, name=get_name(request)), 400
 
 
 @bp.route('/idk')
@@ -35,5 +39,6 @@ def idk(error:HTTPException=None):
         'errors/error.html',
         code = error.code,
         description = error.description,
-        name = error.name
+        err_name = error.name,
+        name=get_name(request)
     ), error.code

src/main.py

@@ -1,5 +1,5 @@
 # IMPORTS
-from flask import Flask, render_template
+from flask import Flask, render_template, request
 from os import getenv as env
 import logging
 
@@ -7,8 +7,10 @@ try:
     import src.dynamic_routes as dynamic_routes
     import src.errors as errors
     import src.pg_log as pg_log
+    from src.name import get_name
 except ImportError:
     import dynamic_routes, errors, pg_log
+    from name import get_name
     from dotenv import load_dotenv
     load_dotenv()
 
@@ -60,11 +62,16 @@ log.info("Blueprints registered.")
 # ROUTES
 @app.route("/")
 def index():
-    return render_template("index.html")
+    return render_template("index.html", name=get_name(request))
 
 @app.route("/toaster")
 def toaster():
-    return render_template("toaster.html")
+    return render_template("toaster.html", name=get_name(request))
+
+@app.route("/terminal")
+def terminal():
+    return render_template("terminal.html", name=get_name(request))
+
 
 
 # DEBUG (DONT RUN LIKE THIS IN PROD)

src/name.py

@@ -0,0 +1,5 @@
+def get_name(req):
+    if req.headers.get("Host") == "proot.uk":
+        return "Toaster"
+    else:
+        return "Alfie King"