diff options
Diffstat (limited to 'src/lib/server/cookiebridge.ts')
-rw-r--r-- | src/lib/server/cookiebridge.ts | 33 |
1 files changed, 33 insertions, 0 deletions
diff --git a/src/lib/server/cookiebridge.ts b/src/lib/server/cookiebridge.ts new file mode 100644 index 0000000..ca4bd44 --- /dev/null +++ b/src/lib/server/cookiebridge.ts @@ -0,0 +1,33 @@ +import { getContextData } from "waku/middleware/context"; +import { + RequestCookies, + ResponseCookies, + type ResponseCookie, +} from "@edge-runtime/cookies"; +import { mergeSetCookies } from "./setcookie"; + +const cookies = () => { + const ctx = getContextData() as { + headers: Record<string, string | string[]>; + cookies?: ResponseCookie[]; + }; + const headerObj = ctx.headers || {}; + headerObj["set-cookie"] = mergeSetCookies( + headerObj["set-cookie"] || [], + (ctx.cookies || []) as ResponseCookie[], + ); + const headers = new Headers(headerObj as Record<string, string>); + const reqCookies = new RequestCookies(headers); + const resCookies = new ResponseCookies(headers); + + const getCookie: ResponseCookies["get"] = (...args) => + resCookies.get(...args) || reqCookies.get(...args); + const setCookie: ResponseCookies["set"] = (...args) => { + const updated = resCookies.set(...args); + ctx.cookies = updated.getAll(); + return updated; + }; + return { getCookie, setCookie }; +}; + +export { cookies }; |