From ec0b048365027fcb733a58cdb0fd3af8e15e1f7a Mon Sep 17 00:00:00 2001
From: yxshv <yashverma30115@gmail.com>
Date: Thu, 11 Apr 2024 17:11:09 +0530
Subject: [PATCH] new db actions

---
 apps/web/src/.gitignore    |  1 -
 apps/web/src/actions/db.ts | 57 ++++++++++++++++++++++++++++++--------
 apps/web/src/app/page.tsx  |  2 +-
 3 files changed, 47 insertions(+), 13 deletions(-)
 delete mode 100644 apps/web/src/.gitignore

diff --git a/apps/web/src/.gitignore b/apps/web/src/.gitignore
deleted file mode 100644
index 43e4c5f7..00000000
--- a/apps/web/src/.gitignore
+++ /dev/null
@@ -1 +0,0 @@
-wrangler.toml
\ No newline at end of file
diff --git a/apps/web/src/actions/db.ts b/apps/web/src/actions/db.ts
index 3b640c96..0b5bbaf6 100644
--- a/apps/web/src/actions/db.ts
+++ b/apps/web/src/actions/db.ts
@@ -1,21 +1,55 @@
 "use server";
+import { cookies, headers } from "next/headers";
 import { db } from "@/server/db";
 import {
   contentToSpace,
+  sessions,
   StoredContent,
   storedContent,
+  users,
 } from "@/server/db/schema";
 import { like, eq, and } from "drizzle-orm";
 import { auth as authOptions } from "@/server/auth";
-import { getSession } from "next-auth/react";
+
+
+async function getUser() {
+  const token =
+    cookies().get("next-auth.session-token")?.value ??
+    cookies().get("__Secure-authjs.session-token")?.value ??
+    cookies().get("authjs.session-token")?.value ??
+    headers().get("Authorization")?.replace("Bearer ", "");
+
+  if (!token) {
+    return null
+  }
+
+  const session = await db
+    .select()
+    .from(sessions)
+    .where(eq(sessions.sessionToken, token!));
+
+  if (!session || session.length === 0) {
+    return null
+  }
+
+  const [userData] = await db
+    .select()
+    .from(users)
+    .where(eq(users.id, session[0].userId))
+    .limit(1);
+
+  if (!userData) {
+    return null
+  }
+
+  return userData
+}
 
 export async function getMemory(title: string) {
-  const session = await getSession();
+  const user = await getUser();
 
-  console.log(session?.user?.name);
-
-  if (!session || !session.user) {
-    return null;
+  if (!user) {
+    return null
   }
 
   return await db
@@ -23,7 +57,7 @@ export async function getMemory(title: string) {
     .from(storedContent)
     .where(
       and(
-        eq(storedContent.user, session.user.id!),
+        eq(storedContent.user, user.id),
         like(storedContent.title, `%${title}%`),
       ),
     );
@@ -33,12 +67,13 @@ export async function addMemory(
   content: typeof storedContent.$inferInsert,
   spaces: number[],
 ) {
-  const session = await getSession();
+  
+  const user = await getUser();
 
-  if (!session || !session.user) {
-    return null;
+  if (!user) {
+    return null
   }
-  content.user = session.user.id;
+  content.user = user.id;
 
   const _content = (
     await db.insert(storedContent).values(content).returning()
diff --git a/apps/web/src/app/page.tsx b/apps/web/src/app/page.tsx
index 6f9ca753..ccceffe0 100644
--- a/apps/web/src/app/page.tsx
+++ b/apps/web/src/app/page.tsx
@@ -77,7 +77,7 @@ export default async function Home() {
   // });
 
   return (
-    <MemoryProvider spaces={collectedSpaces}>
+    <MemoryProvider spaces={collectedSpaces} freeMemories={[]}>
       <Content jwt={token} />
       {/* <MessagePoster jwt={token} /> */}
     </MemoryProvider>