From 99e5baeefb4c2274c85d3e9a876f5dd05b8442f5 Mon Sep 17 00:00:00 2001
From: Rizky <rizky@avolut.com>
Date: Wed, 11 Dec 2024 09:27:26 +0000
Subject: [PATCH] fix

---
 session/server-session.ts |  8 ++++----
 session/type.ts           | 10 +++++++++-
 2 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/session/server-session.ts b/session/server-session.ts
index 46d7cab..5f036b4 100755
--- a/session/server-session.ts
+++ b/session/server-session.ts
@@ -1,16 +1,16 @@
-/// <reference types="bun-types" />
-
 import { ServerWebSocket } from "bun";
 import { useServerRouter } from "../server/server-route";
 import { newSessionStore } from "./store/session-store";
 import { ServerContext } from "./type";
 
 type WS = ServerWebSocket<{ url: string }>;
-type SessionServerHandler = {
+export type SessionServerHandler = {
   cleanup: () => Promise<void>;
   handle: (
     arg: ServerContext,
-    opt?: { cache_accept?: string }
+    opt?: {
+      rewrite?: (arg: { body: Bun.BodyInit; headers: Headers }) => Bun.BodyInit;
+    }
   ) => Promise<Response>;
 };
 
diff --git a/session/type.ts b/session/type.ts
index d2ccbb6..bca46f1 100755
--- a/session/type.ts
+++ b/session/type.ts
@@ -71,7 +71,15 @@ export interface SessionContext<T> extends ServerContext {
 export type ServerContext = {
   req: Request;
   server: Server;
-  handle: (req: Request, opt?: { cache_accept?: string }) => Promise<Response>;
+  handle: (
+    req: Request,
+    opt?: {
+      rewrite?: (arg: {
+        body: Bun.BodyInit;
+        headers: Response["headers"];
+      }) => Bun.BodyInit;
+    }
+  ) => Promise<Response>;
   mode: "dev" | "prod";
   url: {
     raw: URL;