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 @@ -/// - 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; handle: ( arg: ServerContext, - opt?: { cache_accept?: string } + opt?: { + rewrite?: (arg: { body: Bun.BodyInit; headers: Headers }) => Bun.BodyInit; + } ) => Promise; }; 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 extends ServerContext { export type ServerContext = { req: Request; server: Server; - handle: (req: Request, opt?: { cache_accept?: string }) => Promise; + handle: ( + req: Request, + opt?: { + rewrite?: (arg: { + body: Bun.BodyInit; + headers: Response["headers"]; + }) => Bun.BodyInit; + } + ) => Promise; mode: "dev" | "prod"; url: { raw: URL;