Merge pull request #56 from GeneralMagicio/rf6-staging

Rf6 staging

Author: mmahdigh

.gitignore

@@ -7,6 +7,7 @@
 /src/ai-summary
 /src/rf6-data-import
 src/rf6-data-import
+src/ai-summary-rf5
 
 # Envs
 

package.json

@@ -44,11 +44,14 @@
     "mathjs": "^11.9.1",
     "ml-matrix": "^6.10.5",
     "node-cron": "^3.0.3",
+    "openai": "^4.68.4",
     "prisma": "^5.0.0",
     "reflect-metadata": "^0.1.13",
     "rxjs": "^7.8.1",
     "siwe": "^2.3.2",
-    "viem": "^2.21.5"
+    "thirdweb": "^5.64.2",
+    "viem": "^2.21.5",
+    "zod": "^3.23.8"
   },
   "devDependencies": {
     "@nestjs/cli": "^10.0.0",

prisma/migrations/20241027180330_/migration.sql

@@ -0,0 +1,12 @@
+/*
+  Warnings:
+
+  - A unique constraint covering the columns `[attestation_id]` on the table `UserAttestation` will be added. If there are existing duplicate values, this will fail.
+  - Added the required column `attestation_id` to the `UserAttestation` table without a default value. This is not possible if the table is not empty.
+
+*/
+-- AlterTable
+ALTER TABLE "UserAttestation" ADD COLUMN     "attestation_id" TEXT NOT NULL;
+
+-- CreateIndex
+CREATE UNIQUE INDEX "UserAttestation_attestation_id_key" ON "UserAttestation"("attestation_id");

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[]
 }
@@ -201,16 +202,27 @@ model UserCollectionFinish {
 }
 
 model UserAttestation {
-  userId       Int      @map("user_id")
-  collectionId Int      @map("collection_id")
-  createdAt    DateTime @default(now()) @map("created_at")
-  updatedAt    DateTime @default(now()) @updatedAt @map("updated_at")
-  user         User     @relation(fields: [userId], references: [id])
-  collection   Project  @relation(fields: [collectionId], references: [id])
+  userId        Int      @map("user_id")
+  collectionId  Int      @map("collection_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])
+  collection    Project  @relation(fields: [collectionId], references: [id])
 
   @@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/auth/auth.controller.ts

@@ -16,7 +16,7 @@ import { PrismaService } from 'src/prisma.service';
 import { Response } from 'express';
 import { UsersService } from 'src/user/users.service';
 import { AuthGuard } from './auth.guard';
-import { LoginDTO } from './dto/login.dto';
+import { LoginDTO, ThirdwebLoginDTO } from './dto/login.dto';
 import { ApiResponse } from '@nestjs/swagger';
 import { AuthedReq } from 'src/utils/types/AuthedReq.type';
 import { STAGING_API, generateRandomString } from 'src/utils';
@@ -125,6 +125,46 @@ export class AuthController {
     res.status(200).send({ token, isNewUser });
   }
 
+  @UseGuards(AuthGuard)
+  @ApiResponse({ status: 200, description: 'Gets the sa address of a user' })
+  @Get('/thirdweb/sa-address')
+  async getSaAddress(@Req() { userId }: AuthedReq) {
+    const res = await this.prismaService.user.findUnique({
+      select: { smartaddress: true },
+      where: {
+        id: userId,
+      },
+    });
+
+    return res?.smartaddress || null;
+  }
+
+  @UseGuards(AuthGuard)
+  @ApiResponse({ status: 200, description: 'Sets an auth cookie' })
+  @Post('/thirdweb/login')
+  async loginWithThirdweb(
+    @Req() { userId }: AuthedReq,
+    @Body() { message, signature, address }: ThirdwebLoginDTO,
+  ) {
+    const isAuthentic = await this.authService.verifyThirdwebUser(
+      message,
+      signature,
+      address,
+    );
+    if (!isAuthentic) throw new UnauthorizedException('Invalid signature');
+
+    await this.prismaService.user.update({
+      where: {
+        id: userId,
+      },
+      data: {
+        smartaddress: address,
+      },
+    });
+
+    return 'Success';
+  }
+
   // @ApiResponse({
   //   status: 200,
   //   type: String,

src/auth/auth.service.ts

@@ -3,6 +3,8 @@ import { generateRandomString } from 'src/utils';
 import { PrismaService } from 'src/prisma.service';
 import { SiweMessage } from 'siwe';
 import { verifyMessage } from 'viem';
+import { chain, thirdwebClient } from './thirdweb';
+import { verifySignature } from 'thirdweb/auth';
 // import { chain, thirdwebClient } from 'src/thirdweb';
 
 @Injectable()
@@ -140,6 +142,22 @@ export class AuthService {
     return user;
   };
 
+  verifyThirdwebUser = async (
+    message: string,
+    signature: string,
+    address: string,
+  ) => {
+    const isValid = await verifySignature({
+      message,
+      signature,
+      address,
+      client: thirdwebClient,
+      chain,
+    });
+
+    return isValid;
+  };
+
   verifyUser = async (
     message: string,
     signature: `0x${string}`,

src/auth/dto/login.dto.ts

@@ -1,5 +1,4 @@
 import { ApiProperty } from '@nestjs/swagger';
-import { SiweMessage } from 'siwe';
 
 export class SiweMessageClass {
   @ApiProperty({
@@ -86,3 +85,13 @@ export class LoginDTO {
   @ApiProperty()
   address: string;
 }
+export class ThirdwebLoginDTO {
+  @ApiProperty()
+  message: string;
+
+  @ApiProperty()
+  signature: string;
+
+  @ApiProperty()
+  address: string;
+}

src/auth/thirdweb.ts

@@ -0,0 +1,13 @@
+import { createThirdwebClient } from 'thirdweb';
+import { optimism, optimismSepolia } from 'thirdweb/chains';
+
+const secretKey = process.env.SECRET_KEY;
+
+if (!secretKey) throw new Error('No Third Web Secret Key');
+
+export const chain =
+  process.env.ACTIVE_CHAIN === 'optimism' ? optimism : optimismSepolia;
+
+export const thirdwebClient = createThirdwebClient({
+  secretKey,
+});

src/cronJobs.ts

@@ -1,7 +1,7 @@
 import { schedule } from 'node-cron';
-import { sendCastsFor12Hours } from './neynar/utils';
+import { sendDailyCasts } from './neynar/utils';
 
-const sendCastsCronJobTime = '21 21 0,12 * * *'; // at 00:21 and 12:21 every day
+const sendCastsCronJobTime = '21 1 17 * * *'; // at 17:01 every day
 
 export const initializeCronJobs = () => {
   sendCastsCronJob();
@@ -10,7 +10,7 @@ export const initializeCronJobs = () => {
 const sendCastsCronJob = () => {
   schedule(sendCastsCronJobTime, async () => {
     try {
-      await sendCastsFor12Hours();
+      await sendDailyCasts();
     } catch (e) {
       console.error('sendCastsCronJob error', e);
     }