Add chain ID to dApp (#277)

* feat: add chainid column

* feat: create db migrations

* chore: refactor `fromString` to `from` for better object interface

* fix: fake the right event name

* fix: this is not a constructor

* feat: define chain

* feat: add chain id to dapp as a number

* feat: enable chain selection

* fix: broken tests

* fix: broken tests

* fix: chain id type

* chore: fix submodule version

Author: avitali-zama

apps/back/prisma/migrations/20250429154651_add_chain_id_to_dapp/down.sql

@@ -0,0 +1,2 @@
+-- Remove chainId
+ALTER TABLE "Dapp" DROP COLUMN IF EXISTS "chainId";

apps/back/prisma/migrations/20250429154651_add_chain_id_to_dapp/migration.sql

@@ -0,0 +1,2 @@
+-- AlterTable
+ALTER TABLE "Dapp" ADD COLUMN IF NOT EXISTS  "chainId" INTEGER;

apps/back/prisma/migrations/20250430084143_create_chain/down.sql

@@ -0,0 +1,6 @@
+-- Remove ForeignKey
+ALTER TABLE "Dapp" DROP CONSTRAINT IF EXISTS "Dapp_chainId_fkey";
+
+-- Delete table
+DELETE TABLE IF EXISTS "Chain";
+

apps/back/prisma/migrations/20250430084143_create_chain/migration.sql

@@ -0,0 +1,23 @@
+-- CreateTable
+CREATE TABLE IF NOT EXISTS "Chain" (
+    "id" INTEGER NOT NULL,
+    "name" TEXT NOT NULL,
+    "description" TEXT,
+    "enabled" BOOLEAN NOT NULL DEFAULT false,
+
+    CONSTRAINT "Chain_pkey" PRIMARY KEY ("id")
+);
+
+-- Note: there is no `ADD CONSTRAINT IF NOT EXISTS` because there is no way to assume its
+-- properties. So, you should drop it if exists, and add again
+ALTER TABLE "Dapp" DROP CONSTRAINT IF EXISTS "Dapp_chainId_fkey";
+-- AddForeignKey
+ALTER TABLE "Dapp" ADD CONSTRAINT "Dapp_chainId_fkey" FOREIGN KEY ("chainId") REFERENCES "Chain"("id") ON DELETE SET NULL ON UPDATE CASCADE;
+
+-- Seed DB
+INSERT INTO "Chain" ("id", "name", "description")
+VALUES 
+    ('1', 'Mainnet', 'Ethereum Mainnet'),
+    ('11155111', 'Sepolia', 'Ethereum Sepolia'),
+    ('123456', 'Local Testnet', NULL)
+ON CONFLICT("id") DO NOTHING;

apps/back/prisma/migrations/migration_lock.toml

@@ -1,3 +1,3 @@
 # Please do not edit this file manually
 # It should be added in your version-control system (e.g., Git)
-provider = "postgresql"
+provider = "postgresql"

apps/back/prisma/schema.prisma

@@ -66,6 +66,8 @@ model Team {
 model Dapp {
   id        String     @id
   name      String
+  chainId   Int?
+  chain     Chain?     @relation(fields: [chainId], references: [id])
   address   String?
   status    DappStatus @default(DRAFT)
   teamId    String
@@ -77,6 +79,14 @@ model Dapp {
   ApiKey    ApiKey[]
 }
 
+model Chain {
+  id          Int     @id
+  name        String
+  description String?
+  enabled     Boolean @default(false)
+  Dapp        Dapp[]
+}
+
 model DappStat {
   id          String    @id
   name        String

apps/back/src/app.module.ts

@@ -15,6 +15,7 @@ import commonConfig from '#config/common.config.js'
 import { MS_NAME } from '#constants.js'
 import { RestModule } from '#infra/rest/rest.module.js'
 import httpzConfig from '#config/httpz.config.js'
+import { ChainsModule } from '#chains/infra/chains.module.js'
 import { RedisModule } from '#infra/redis/redis.module.js'
 import { FeatureFlagModule } from '#feature-flag/feature-flag.module.js'
 import { EnvFeatureFlagHandler } from '#infra/env-feature-flag.handler.js'
@@ -65,6 +66,7 @@ export const configModule = ConfigModule.forRoot({
     RestModule,
     SqsConsumerModule,
     SqsProducerModule,
+    ChainsModule,
     RedisModule,
     FeatureFlagModule.register({
       global: true,

apps/back/src/auth/use-cases/login.use-case.ts

@@ -19,10 +19,10 @@ export class LogIn
     private readonly jwtService: JwtService,
   ) {}
 
-  execute(input: {
+  execute = (input: {
     email: string
     password: string
-  }): Task<{ user: UserProps; token: string }, AppError> {
+  }): Task<{ user: UserProps; token: string }, AppError> => {
     return (
       this.userRepository
         .findByEmail(input.email)

apps/back/src/auth/use-cases/signup.use-case.ts

@@ -31,9 +31,9 @@ export class SignUp
     private readonly jwtService: JwtService,
   ) {}
 
-  execute(
+  execute = (
     input: SignupInput,
-  ): Task<{ user: UserProps; token: string }, AppError> {
+  ): Task<{ user: UserProps; token: string }, AppError> => {
     return this.uow.exec(
       Task.all<AppError, Invitation, ValidatedPassword>([
         // Note: we check the invitation token and validate the password at
@@ -70,8 +70,8 @@ export class SignUp
   }
 
   private validateInvitation(token: string): Task<Invitation, AppError> {
-    return this.invitationRepository
-      .findByToken(Token.from(token))
+    return Token.from(token)
+      .asyncChain(this.invitationRepository.findByToken)
       .chain<Invitation>(invitation =>
         invitation.isValid
           ? Task.of(invitation)

apps/back/src/chains/domain/entities/chain.ts

@@ -0,0 +1,44 @@
+import {
+  AppError,
+  Entity,
+  fail,
+  ok,
+  Result,
+  Unbrand,
+  validationError,
+} from 'utils'
+import { z } from 'zod'
+import { ChainId } from './value-objects.js'
+
+const schema = z.object({
+  id: ChainId.schema,
+  name: z.string(),
+  description: z.string().nullish(),
+})
+
+export type ChainProps = Unbrand<z.infer<typeof schema>>
+
+export class Chain
+  extends Entity<ChainProps>
+  implements Readonly<Omit<ChainProps, 'id'> & { id: ChainId }>
+{
+  static parse(data: unknown): Result<Chain, AppError> {
+    if (!data) return fail(validationError('data is undefined'))
+    const check = schema.safeParse(data)
+    return check.success
+      ? ok(new Chain(check.data))
+      : fail(validationError(check.error.message))
+  }
+
+  get id() {
+    return new ChainId(this.get('id'))
+  }
+
+  get name() {
+    return this.get('name')
+  }
+
+  get description() {
+    return this.get('description')
+  }
+}