shift to subdomains

Makefile

@@ -17,3 +17,7 @@ logs:
 
 clean: stop
 	rm -f output.log app.pid
+
+reset: clean
+	rm -rf instance
+	rm -rf sites

app.py

@@ -25,15 +25,51 @@ import os
 import shutil
 from datetime import datetime, timedelta
 from waitress import serve
-
+import re
+from secrets import token_hex
 
 app = Flask(__name__)
-app.config["SECRET_KEY"] = "your-secret-key"
+
+try:
+	with open("instance/secret.key", "rb") as f:
+		app.config["SECRET_KEY"] = bytes.hex(f.readline())
+except FileNotFoundError as e:
+	os.mkdir('instance')
+	with open("instance/secret.key", "wb") as f:
+		newKey = token_hex(64)
+		f.write(bytearray.fromhex(newKey))
+		app.config["SECRET_KEY"] = newKey
+
 app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///db.sqlite"
 app.config["UPLOAD_FOLDER"] = "sites"
+app.config["SERVER_NAME"] = "tinysite.cloud"
+app.config["SESSION_COOKIE_DOMAIN"] = ".tinysite.cloud"
 db = SQLAlchemy(app)
 
 
+RESERVED_SUBDOMAINS = {
+	"",
+	"www",
+	"api",
+	"admin",
+	"support",
+	"docs",
+	"blog",
+	"cdn",
+	"test",
+	"dev",
+	"staging",
+	"secure",
+	"mail",
+	"status",
+	"gateway"
+}
+
+
+def isDefaultRoute(subdomain):
+	return not subdomain or subdomain in RESERVED_SUBDOMAINS
+
+
 # Models
 class User(UserMixin, db.Model):
 	id = db.Column(db.Integer, primary_key=True)
@@ -51,11 +87,16 @@ class Site(db.Model):
 	id = db.Column(db.Integer, primary_key=True)
 	user_id = db.Column(db.Integer)
 	name = db.Column(db.String(100))
-    slug = db.Column(db.String(100), unique=True)
+	subdomain = db.Column(db.String(100), unique=True)
 	last_accessed = db.Column(db.DateTime)
 	created_at = db.Column(db.DateTime, default=datetime.utcnow)
 
 
+@app.errorhandler(404)
+def page_not_found(_):
+	return render_template("404.html", domain=request.host), 404
+
+
 # Auth setup
 login_manager = LoginManager()
 login_manager.init_app(app)
@@ -138,21 +179,28 @@ def register():
 @login_required
 def dashboard():
 	sites = Site.query.filter_by(user_id=current_user.id).all()
-    return render_template("dashboard.html", sites=sites)
+	return render_template("dashboard.html", sites=sites, subdomain=request.host.split('.')[0], hostname=app.config["SERVER_NAME"])
 
 
 @app.route("/upload", methods=["POST"])
 @login_required
 def upload_site():
 	site_name = request.form.get("name")
-    slug = request.form.get("slug").strip()
+	subdomain = request.form.get("subdomain").strip().lower()  # normalize to lowercase
 
-    if not site_name or not slug:
-        flash("Site name and URL slug are required", "error")
+	# Subdomain validation
+	if not re.match(r"^[a-z0-9](?:[a-z0-9-]{0,61}[a-z0-9])?$", subdomain):
+		flash(
+			"Invalid subdomain. Use lowercase letters, numbers, and hyphens.", "error"
+		)
 		return redirect(url_for("dashboard"))
 
-    # Check slug availability
-    if Site.query.filter_by(slug=slug).first():
+	if not site_name or not subdomain:
+		flash("Site name and URL subdomain are required", "error")
+		return redirect(url_for("dashboard"))
+
+	# Check subdomain availability
+	if Site.query.filter_by(subdomain=subdomain).first():
 		flash("This URL is already taken", "error")
 		return redirect(url_for("dashboard"))
 
@@ -163,7 +211,7 @@ def upload_site():
 		return redirect(url_for("dashboard"))
 
 	# Create site directory
-    site = Site(user_id=current_user.id, name=site_name, slug=slug)
+	site = Site(user_id=current_user.id, name=site_name, subdomain=subdomain)
 	db.session.add(site)
 	db.session.commit()
 
@@ -259,10 +307,13 @@ def delete_file(site_id, filename):
 	return redirect(url_for("dashboard"))
 
 
-@app.route("/site/<slug>/", defaults={"filename": "index.html"})
-@app.route("/site/<slug>/<path:filename>")
-def serve_site_content(slug, filename):
-    site = Site.query.filter_by(slug=slug).first_or_404()
+# use subdomains
+@app.route("/", subdomain="<subdomain>", defaults={"filename": "index.html"})
+@app.route("/<path:filename>")
+def serve_site_content(filename):
+	subdomain = request.host.split('.')[0]
+    
+	site = Site.query.filter_by(subdomain=subdomain).first_or_404()
 	site.last_accessed = datetime.utcnow()
 	db.session.commit()
   
@@ -270,21 +321,22 @@ def serve_site_content(slug, filename):
 		app.config["UPLOAD_FOLDER"], str(site.user_id), str(site.id)
 	)
 
+	# TODO: add specific page redirects here as they're added
+	if isDefaultRoute(subdomain):
+		return send_from_directory("index.html")
+
 	# Security check
 	if ".." in filename or filename.startswith("/") or not os.path.exists(site_dir):
-        abort(404)
+		return render_template("404.html"), 404
 
 	try:
-        # First try to serve requested file
 		return send_from_directory(site_dir, filename)
 	except NotFound:
-        # Handle extensionless URLs and SPA-style routing
 		if "." not in filename:
-            # Try with .html extension
 			try:
 				return send_from_directory(site_dir, f"{filename}.html")
 			except NotFound:
-                # Fallback to index.html for client-side routing
+				redirect(app.config["SERVER_NAME"])
 				return send_from_directory(site_dir, "index.html")
 		abort(404)
 
@@ -303,7 +355,10 @@ def inject_utilities():
 
 @app.route("/")
 def home():
+	if isDefaultRoute(request.host):
 		return render_template("home.html")
+	else:
+		return serve_site_content('index.html')
 
 
 if __name__ == "__main__":

templates/404.html

@@ -0,0 +1,83 @@
+{% extends "base.html" %}
+
+{% block content %}
+<style>
+	.error-container {
+		text-align: center;
+		padding: 100px 20px;
+		min-height: 60vh;
+	}
+
+	.error-emoji {
+		font-size: 4rem;
+		margin-bottom: 1rem;
+	}
+
+	.error-message {
+		max-width: 600px;
+		margin: 0 auto 2rem;
+	}
+
+	.cta-button {
+		margin-top: 2rem;
+	}
+
+	.alien {
+		font-size: 5rem;
+		margin: 2rem 0;
+		animation: float 3s ease-in-out infinite;
+	}
+
+	@keyframes float {
+		0% {
+			transform: translateY(0px);
+		}
+
+		50% {
+			transform: translateY(-20px);
+		}
+
+		100% {
+			transform: translateY(0px);
+		}
+	}
+</style>
+
+<script>
+	document.addEventListener("DOMContentLoaded", () => {
+		const emojis = ["👽", "🚀", "🛸", "🌌", "🌠", "🔭", "🪐", "🌍", "✨", "👾"];
+		const randomEmoji = emojis[Math.floor(Math.random() * emojis.length)];
+
+		const alienElement = document.querySelector(".alien");
+		if (alienElement) {
+			alienElement.textContent = randomEmoji;
+		}
+	});
+</script>
+
+<div class="error-container">
+	<div class="alien">👽</div>
+	<h1>Houston, We Have a Problem!</h1>
+
+	<div class="error-message">
+		<p>The site you're looking for doesn't exist... yet!</p>
+		<p>But don't worry, this corner of the internet is just waiting for your creativity.</p>
+	</div>
+
+	{% if current_user.is_authenticated %}
+	<a href="{{ url_for('dashboard') }}" class="btn cta-button">
+		🚀 Create {{ domain }}
+	</a>
+	{% else %}
+	<div class="auth-buttons">
+		<p>Start your web hosting journey today!</p>
+		<a href="{{ url_for('register') }}" class="btn">Sign Up</a>
+		<a href="{{ url_for('login') }}" class="btn">Login</a>
+	</div>
+	{% endif %}
+
+	<div style="margin-top: 3rem;">
+		<small>PS: If you were looking for someone else's site, maybe they forgot to launch it! 🚀</small>
+	</div>
+</div>
+{% endblock %}

templates/dashboard.html

@@ -12,11 +12,14 @@
 		<input type="text" name="name" required>
 	</div>
 
-	<!-- Add slug input -->
 	<div class="form-group">
-		<label>Custom URL Slug:</label>
-		<input type="text" name="slug" required pattern="[a-zA-Z0-9\-_]+"
-			title="Letters, numbers, hyphens, and underscores only">
+		<label>Site URL:</label>
+		<div style="display: flex; align-items: center;">
+			<input type="text" name="subdomain" required pattern="[a-zA-Z0-9\-_]+"
+				title="Letters, numbers, hyphens, and underscores only" value="{{ subdomain }}"
+				style="flex: 1;">
+			<span style="margin-left: 5px;">.{{ hostname }}</span>
+		</div>
 	</div>
 
 	<div class="form-group" onclick="document.querySelector('#file-input').click()">
@@ -27,7 +30,6 @@
 		</label>
 	</div>
 
-	<!-- File Preview Section -->
 	<div id="file-preview" class="file-preview">
 		<h5>Selected Files:</h5>
 		<div id="file-grid" class="file-grid"></div>
@@ -69,7 +71,9 @@
 		<!-- Site Actions -->
 		<div class="site-actions">
 			<a href="{{ url_for('edit_site', site_id=site.id) }}" class="btn">Edit</a>
-			<a href="{{ url_for('serve_site_content', slug=site.slug) }}" class="btn" target="_blank">View Site</a>
+			<a href="{{ url_for('serve_site_content', filename='index.html', _external=True, subdomain=site.subdomain) }}">Visit
+				{{ site.name }}
+			</a>
 			<form method="POST" action="{{ url_for('delete_site', site_id=site.id) }}" style="display: inline;">
 				<button type="submit" class="btn btn-danger">Delete Site</button>
 			</form>

templates/edit.html

@@ -4,7 +4,7 @@
 
 {% block content %}
 <h2>Edit {{ site.name }}</h2>
-<a href="{{ url_for('serve_site_content', slug=site.slug) }}" class="btn" target="_blank">View Site</a>
+<a href="{{ url_for('serve_site_content', subdomain=site.subdomain) }}" class="btn" target="_blank">View Site</a>
 
 <div style="margin-bottom: 30px;"></div>