revamped extention

3 files changed, 28 insertions, 15 deletions

diff --git a/apps/web/app/api/ensureAuth.ts b/apps/web/app/api/ensureAuth.ts
index d2fbac0b..d2c14b3b 100644
--- a/apps/web/app/api/ensureAuth.ts
+++ b/apps/web/app/api/ensureAuth.ts
@@ -22,7 +22,7 @@ export async function ensureAuth(req: NextRequest) {
     .innerJoin(users, eq(users.id, sessions.userId))
     .where(eq(sessions.sessionToken, token!));
 
-  if (!sessionData || sessionData.length < 0) {
+  if (!sessionData || sessionData.length === 0) {
     return undefined;
   }
 
diff --git a/apps/web/app/api/store/route.ts b/apps/web/app/api/store/route.ts
index d9f99277..26636c5c 100644
--- a/apps/web/app/api/store/route.ts
+++ b/apps/web/app/api/store/route.ts
@@ -84,11 +84,7 @@ const createMemoryFromAPI = async (input: {
     const error = e as Error;
     console.log("Error: ", error.message);
 
-    if (
-      error.message.includes(
-        "D1_ERROR: UNIQUE constraint failed: storedContent.baseUrl",
-      )
-    ) {
+    if (error.message.includes("D1_ERROR: UNIQUE constraint failed:")) {
       return {
         success: false,
         data: 0,
@@ -204,9 +200,15 @@ export async function POST(req: NextRequest) {
         message: "Failed to save document",
         error: result.error,
       }),
-      { status: 500 },
+      { status: 501 },
     );
   }
 
-  return new Response("ok", { status: 200 });
+  return new Response(
+    JSON.stringify({
+      message: "Document saved",
+      data: result.data,
+    }),
+    { status: 200 },
+  );
 }
diff --git a/apps/web/middleware.ts b/apps/web/middleware.ts
index b6cc4548..58ac535c 100644
--- a/apps/web/middleware.ts
+++ b/apps/web/middleware.ts
@@ -1,13 +1,24 @@
-// middleware.js
 import { NextRequest, NextResponse } from "next/server";
 
+const corsHeaders = {
+  "Access-Control-Allow-Origin": "*",
+  "Access-Control-Allow-Methods": "GET, POST, PUT, DELETE, OPTIONS",
+  "Access-Control-Allow-Headers": "Content-Type, Authorization",
+};
+
 export function middleware(request: NextRequest) {
-  const requestHeaders = new Headers(request.headers);
-  requestHeaders.set("x-pathname", request.nextUrl.pathname);
+  if (request.method === "OPTIONS") {
+    return new NextResponse(null, { headers: corsHeaders });
+  }
 
-  return NextResponse.next({
-    request: {
-      headers: requestHeaders,
-    },
+  const response = NextResponse.next();
+  Object.entries(corsHeaders).forEach(([key, value]) => {
+    response.headers.set(key, value);
   });
+
+  return response;
 }
+
+export const config = {
+  matcher: "/api/:path*",
+};