Merge pull request #54 from cipherstash/generic-next-middleware

feat: add support for optional middleware function

Author: calvinbrewer

.changeset/gentle-singers-occur.md

@@ -0,0 +1,5 @@
+---
+"@cipherstash/nextjs": minor
+---
+
+Added support for optional next response in generic jseql middleware functions.

packages/nextjs/src/index.ts

@@ -24,13 +24,22 @@ export const getCtsToken = async () => {
   return cts_token
 }
 
-export const resetCtsToken = () => {
+export const resetCtsToken = (res?: NextResponse) => {
+  if (res) {
+    res.cookies.delete(CS_COOKIE_NAME)
+    return res
+  }
+
   const response = NextResponse.next()
   response.cookies.delete(CS_COOKIE_NAME)
   return response
 }
 
-export const jseqlMiddleware = async (oidcToken: string, req: NextRequest) => {
+export const jseqlMiddleware = async (
+  oidcToken: string,
+  req: NextRequest,
+  res?: NextResponse,
+) => {
   const ctsSession = req.cookies.has(CS_COOKIE_NAME)
 
   if (oidcToken && !ctsSession) {
@@ -49,5 +58,9 @@ export const jseqlMiddleware = async (oidcToken: string, req: NextRequest) => {
     'The JWT token was undefined, so the CipherStash session was not set.',
   )
 
+  if (res) {
+    return res
+  }
+
   return NextResponse.next()
 }