forward ip to backend

3 files changed, 28 insertions, 2 deletions

diff --git a/src/hooks.server.ts b/src/hooks.server.ts
new file mode 100644
index 00000000..3dd9913a
--- /dev/null
+++ b/src/hooks.server.ts
@@ -0,0 +1,11 @@
+import { ipStore } from "$lib/server/ip";
+import type { Handle } from "@sveltejs/kit";
+
+export const handle: Handle = async ({ event, resolve }) => {
+	const ip =
+		event.request.headers.get("cf-connecting-ip") ||
+		event.request.headers.get("x-forwarded-for")?.split(",")[0]?.trim() ||
+		event.getClientAddress();
+
+	return ipStore.run(ip, () => resolve(event));
+};
diff --git a/src/lib/server/ip.ts b/src/lib/server/ip.ts
new file mode 100644
index 00000000..cfd933ac
--- /dev/null
+++ b/src/lib/server/ip.ts
@@ -0,0 +1,2 @@
+import { AsyncLocalStorage } from "node:async_hooks";
+export const ipStore = new AsyncLocalStorage<string>();
diff --git a/src/lib/utils.ts b/src/lib/utils.ts
index f220ed45..1b2bdfba 100644
--- a/src/lib/utils.ts
+++ b/src/lib/utils.ts
@@ -21,11 +21,23 @@ export function scroll(section: any) {
 
 const base = browser ? "" : PUBLIC_COINOS_URL;
 
+let ipStore: any;
+if (!browser) {
+	import("$lib/server/ip").then((m) => {
+		ipStore = m.ipStore;
+	});
+}
+
+const ipHeaders = (): Record<string, string> => {
+	const ip = ipStore?.getStore();
+	return ip ? { "cf-connecting-ip": ip } : {};
+};
+
 export const punk = (k: string) =>
 	`${Math.floor((parseInt(k.slice(-2), 16) / 256) * 64) + 1}.webp`;
 
 export const g = (url: string, fetch: any, headers: object) =>
-	fetch(base + url, { headers })
+	fetch(base + url, { headers: { ...ipHeaders(), ...headers } })
 		.then((r: Response) => r.text())
 		.then((body: string) => {
 			try {
@@ -36,7 +48,7 @@ export const g = (url: string, fetch: any, headers: object) =>
 		});
 
 export const get = (url: string, headers = {}) => {
-	return fetch(base + url, { headers })
+	return fetch(base + url, { headers: { ...ipHeaders(), ...headers } })
 		.then(async (r) => {
 			if (r.ok) return r.text();
 			throw new Error(await r.text());
@@ -58,6 +70,7 @@ export const post = async (
 	headers = {
 		"content-type": "application/json",
 		accept: "application/json",
+		...ipHeaders(),
 		...headers,
 	};