From 0e3ca90cef2786f24950431ca25bae9a6db3568d Mon Sep 17 00:00:00 2001
From: IshaVenikar <ishavenikar7@gmail.com>
Date: Wed, 5 Feb 2025 21:11:32 +0530
Subject: [PATCH] Store tweet URL after verification

---
 src/app/api/tweet/route.ts | 4 ++--
 src/entity/Tweet.ts        | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/app/api/tweet/route.ts b/src/app/api/tweet/route.ts
index dcc902f..bf187c6 100644
--- a/src/app/api/tweet/route.ts
+++ b/src/app/api/tweet/route.ts
@@ -14,7 +14,7 @@ export async function POST(req: NextRequest): Promise<NextResponse> {
 
     const response = await fetch(url);
     const data = await response.json();
-    const { handle, txSignature } = extractData(data.html);
+    const { handle, txSignature, memeUrl } = extractData(data.html);
     if (!handle || !txSignature) {
       return NextResponse.json(
         { error: 'Could not extract tweet data' },
@@ -31,7 +31,7 @@ export async function POST(req: NextRequest): Promise<NextResponse> {
       throw new Error('Tweet is not valid');
     }
 
-    const { isFourthUser } = await saveTweet({ transactionSignature: txSignature });
+    const { isFourthUser } = await saveTweet({ transactionSignature: txSignature, url: memeUrl });
     if (isFourthUser) {
       createTokenLockForRecipient();
     }
diff --git a/src/entity/Tweet.ts b/src/entity/Tweet.ts
index c16b241..0f0d9be 100644
--- a/src/entity/Tweet.ts
+++ b/src/entity/Tweet.ts
@@ -5,8 +5,8 @@ export class Tweet {
   @PrimaryGeneratedColumn()
   id!: number;
 
-  @Column({ unique: true })
-  url!: string;
+  @Column({ unique: true, nullable: true })
+  url!: string | null;
 
   @Column({ unique: true })
   transactionSignature!: string;