From d56594d3289002566f4653d607f0837befd65109 Mon Sep 17 00:00:00 2001
From: polwex <polwex@sortug.com>
Date: Thu, 15 May 2025 10:13:00 +0700
Subject: wtf man

---
 src/lib/server/setcookie.ts | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 src/lib/server/setcookie.ts

(limited to 'src/lib/server/setcookie.ts')

diff --git a/src/lib/server/setcookie.ts b/src/lib/server/setcookie.ts
new file mode 100644
index 0000000..f64b380
--- /dev/null
+++ b/src/lib/server/setcookie.ts
@@ -0,0 +1,25 @@
+import type { Middleware } from "waku/config";
+import { type ResponseCookie, stringifyCookie } from "@edge-runtime/cookies";
+
+export const mergeSetCookies = (
+  resSetCookies: string | string[],
+  cookiesInContext: ResponseCookie[],
+) => {
+  if (typeof resSetCookies === "string") {
+    resSetCookies = [resSetCookies];
+  }
+  return [...resSetCookies, ...cookiesInContext.map(stringifyCookie)];
+};
+
+const setCookieMiddleware: Middleware = () => {
+  return async (ctx, next) => {
+    await next();
+    ctx.res.headers ||= {};
+    ctx.res.headers["set-cookie"] = mergeSetCookies(
+      ctx.res.headers["set-cookie"] || [],
+      (ctx.data.cookies || []) as ResponseCookie[],
+    );
+  };
+};
+
+export default setCookieMiddleware;
-- 
cgit v1.2.3