Add budget attestations

Author: mmahdigh

prisma/migrations/20241029125304_/migration.sql

@@ -0,0 +1,15 @@
+-- CreateTable
+CREATE TABLE "UserBudgetAttestation" (
+    "user_id" INTEGER NOT NULL,
+    "attestation_id" TEXT NOT NULL,
+    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+
+    CONSTRAINT "UserBudgetAttestation_pkey" PRIMARY KEY ("user_id")
+);
+
+-- CreateIndex
+CREATE UNIQUE INDEX "UserBudgetAttestation_attestation_id_key" ON "UserBudgetAttestation"("attestation_id");
+
+-- AddForeignKey
+ALTER TABLE "UserBudgetAttestation" ADD CONSTRAINT "UserBudgetAttestation_user_id_fkey" FOREIGN KEY ("user_id") REFERENCES "User"("id") ON DELETE RESTRICT ON UPDATE CASCADE;

prisma/schema.prisma

@@ -9,14 +9,14 @@ generator client {
 }
 
 model User {
-  id                  Int                    @id @default(autoincrement())
-  budget              Int                    @default(2000000) // 2M In OP
-  address             String                 @unique()
-  smartaddress        String?                @unique()
-  ballotSuccess       Int?                   @map("ballot_success")
-  opAddress           String?                @unique() @map("op_address")
-  createdAt           DateTime               @default(now()) @map("created_at")
-  updatedAt           DateTime               @default(now()) @updatedAt @map("updated_at")
+  id                  Int                     @id @default(autoincrement())
+  budget              Int                     @default(2000000) // 2M In OP
+  address             String                  @unique()
+  smartaddress        String?                 @unique()
+  ballotSuccess       Int?                    @map("ballot_success")
+  opAddress           String?                 @unique() @map("op_address")
+  createdAt           DateTime                @default(now()) @map("created_at")
+  updatedAt           DateTime                @default(now()) @updatedAt @map("updated_at")
   identity            Json? // Ideally should be unique except {} and null values but Prisma doesn't support partial
   // unique constraints
   badges              Json?
@@ -29,6 +29,7 @@ model User {
   finishedCollection  UserCollectionFinish[]
   shares              Share[]
   attestations        UserAttestation[]
+  budgetAttestations  UserBudgetAttestation[]
   projectStars        ProjectStar[]
   cois                ProjectCoI[]
 }
@@ -212,6 +213,16 @@ model UserAttestation {
   @@id([userId, collectionId])
 }
 
+model UserBudgetAttestation {
+  userId        Int      @map("user_id")
+  attestationId String   @unique() @map("attestation_id")
+  createdAt     DateTime @default(now()) @map("created_at")
+  updatedAt     DateTime @default(now()) @updatedAt @map("updated_at")
+  user          User     @relation(fields: [userId], references: [id])
+
+  @@id([userId])
+}
+
 enum DelegationPlatform {
   FARCASTER
   TWITTER

src/flow/flow.controller.ts

@@ -1106,11 +1106,27 @@ export class FlowController {
     @Req() { userId }: AuthedReq,
     @Body() { collectionId, attestationId }: AttestationDto,
   ) {
+    if (collectionId === -1) {
+      await this.prismaService.userBudgetAttestation.upsert({
+        where: {
+          userId: userId,
+        },
+        create: {
+          userId: userId,
+          attestationId,
+        },
+        update: {
+          attestationId,
+        },
+      });
+
+      return 'Success';
+    }
     // collectionId = -1 is for the budget attestation
-    const isFinished =
-      collectionId > 0
-        ? await this.flowService.isCollectionFinished(userId, collectionId)
-        : true;
+    const isFinished = await this.flowService.isCollectionFinished(
+      userId,
+      collectionId,
+    );
 
     if (!isFinished)
       throw new ForbiddenException(