Add wallet to user

src/datasources/users/entities/wallets.entity.db.ts

@@ -1,4 +1,3 @@
-import { RowSchema } from '@/datasources/db/v1/entities/row.entity';
 import {
   Column,
   Entity,
@@ -9,14 +8,8 @@ import {
 } from 'typeorm';
 import { z } from 'zod';
 import { getAddress } from 'viem';
-import { AddressSchema } from '@/validation/entities/schemas/address.schema';
 import { User } from '@/datasources/users/entities/users.entity.db';
-import { UserSchema } from '@/domain/users/entities/user.entity';
-
-export const WalletSchema = RowSchema.extend({
-  address: AddressSchema,
-  user: UserSchema,
-});
+import { WalletSchema } from '@/domain/users/entities/wallet.entity';
 
 @Entity('wallets')
 @Unique('UQ_wallet_address', ['address'])

src/domain/users/entities/wallet.entity.ts

@@ -0,0 +1,11 @@
+import type { z } from 'zod';
+import { RowSchema } from '@/datasources/db/v1/entities/row.entity';
+import { UserSchema } from '@/domain/users/entities/user.entity';
+import { AddressSchema } from '@/validation/entities/schemas/address.schema';
+
+export type Wallet = z.infer<typeof WalletSchema>;
+
+export const WalletSchema = RowSchema.extend({
+  address: AddressSchema,
+  user: UserSchema,
+});

src/domain/users/users.repository.interface.ts

@@ -1,5 +1,6 @@
 import type { AuthPayload } from '@/domain/auth/entities/auth-payload.entity';
 import type { User, UserStatus } from '@/domain/users/entities/user.entity';
+import type { Wallet } from '@/domain/users/entities/wallet.entity';
 
 export const IUsersRepository = Symbol('IUsersRepository');
 
@@ -8,4 +9,9 @@ export interface IUsersRepository {
     status: UserStatus;
     authPayload: AuthPayload;
   }): Promise<Pick<User, 'id'>>;
+
+  addWalletToUser(args: {
+    newSignerAddress: `0x${string}`;
+    authPayload: AuthPayload;
+  }): Promise<Pick<Wallet, 'id'>>;
 }

src/domain/users/users.repository.ts

@@ -46,4 +46,43 @@ export class UsersRepository implements IUsersRepository {
       },
     );
   }
+
+  async addWalletToUser(args: {
+    newSignerAddress: `0x${string}`;
+    authPayload: AuthPayload;
+  }): Promise<Pick<Wallet, 'id'>> {
+    return await this.postgresDatabaseService.transaction(
+      async (entityManager: EntityManager) => {
+        const walletRepository = entityManager.getRepository(Wallet);
+
+        const authenticatedWallet = await walletRepository.findOne({
+          where: { address: args.authPayload.signer_address },
+          relations: { user: true },
+        });
+
+        if (!authenticatedWallet?.user) {
+          throw new Error('User not found');
+        }
+
+        const walletAlreadyExists = Boolean(
+          await walletRepository.findOne({
+            where: { address: args.newSignerAddress },
+          }),
+        );
+
+        if (walletAlreadyExists) {
+          throw new ConflictException(
+            'A wallet with the same address already exists',
+          );
+        }
+
+        const walletInsertResult = await walletRepository.insert({
+          user: authenticatedWallet.user,
+          address: args.newSignerAddress,
+        });
+
+        return { id: walletInsertResult.identifiers[0].id };
+      },
+    );
+  }
 }