From f80b41cea838f7a2afde8361913590e38941a9c0 Mon Sep 17 00:00:00 2001
From: Florian <45694132+flo-bit@users.noreply.github.com>
Date: Mon, 2 Feb 2026 23:02:05 +0100
Subject: [PATCH] change custom domain stuff

---
 src/routes/+page.server.ts      | 11 +++++------
 src/routes/edit/+page.server.ts |  9 ++++-----
 2 files changed, 9 insertions(+), 11 deletions(-)

diff --git a/src/routes/+page.server.ts b/src/routes/+page.server.ts
index 61d349e..c571bf2 100644
--- a/src/routes/+page.server.ts
+++ b/src/routes/+page.server.ts
@@ -1,20 +1,19 @@
 import { loadData } from '$lib/website/load';
 import { env } from '$env/dynamic/public';
 import type { UserCache } from '$lib/types';
-import type { ActorIdentifier, Handle } from '@atcute/lexicons';
-
-export async function load({ platform, url }) {
-	const hostname = url.hostname;
+import type { ActorIdentifier } from '@atcute/lexicons';
 
+export async function load({ platform, request }) {
 	const handle = env.PUBLIC_HANDLE;
 
 	const kv = platform?.env?.CUSTOM_DOMAINS;
 
 	const cache = platform?.env?.USER_DATA_CACHE as unknown;
+	const customDomain = request.headers.get('X-Custom-Domain')?.toLocaleLowerCase();
 
-	if (kv) {
+	if (kv && customDomain) {
 		try {
-			const did = await kv.get(hostname);
+			const did = await kv.get(customDomain);
 
 			if (did) return await loadData(did as ActorIdentifier, cache as UserCache);
 		} catch (error) {
diff --git a/src/routes/edit/+page.server.ts b/src/routes/edit/+page.server.ts
index 35c5223..c571bf2 100644
--- a/src/routes/edit/+page.server.ts
+++ b/src/routes/edit/+page.server.ts
@@ -3,18 +3,17 @@ import { env } from '$env/dynamic/public';
 import type { UserCache } from '$lib/types';
 import type { ActorIdentifier } from '@atcute/lexicons';
 
-export async function load({ url, platform }) {
-	const hostname = url.hostname;
-
+export async function load({ platform, request }) {
 	const handle = env.PUBLIC_HANDLE;
 
 	const kv = platform?.env?.CUSTOM_DOMAINS;
 
 	const cache = platform?.env?.USER_DATA_CACHE as unknown;
+	const customDomain = request.headers.get('X-Custom-Domain')?.toLocaleLowerCase();
 
-	if (kv) {
+	if (kv && customDomain) {
 		try {
-			const did = await kv.get(hostname);
+			const did = await kv.get(customDomain);
 
 			if (did) return await loadData(did as ActorIdentifier, cache as UserCache);
 		} catch (error) {
-- 
2.51.2