AddressBooks and AddressBookItems deletion API endpoints (#2148)

safe-global · Nov 21, 2024 · a4cc91f · a4cc91f
1 parent 0d8eeea
commit a4cc91f
Show file tree

Hide file tree

Showing 5 changed files with 499 additions and 1 deletion.
diff --git a/src/domain/accounts/address-books/address-books.repository.interface.ts b/src/domain/accounts/address-books/address-books.repository.interface.ts
@@ -24,6 +24,19 @@ export interface IAddressBooksRepository {
     chainId: string;
     createAddressBookItemDto: CreateAddressBookItemDto;
   }): Promise<AddressBookItem>;
+
+  deleteAddressBook(args: {
+    authPayload: AuthPayload;
+    address: `0x${string}`;
+    chainId: string;
+  }): Promise<void>;
+
+  deleteAddressBookItem(args: {
+    authPayload: AuthPayload;
+    address: `0x${string}`;
+    chainId: string;
+    addressBookItemId: number;
+  }): Promise<void>;
 }
 
 @Module({

diff --git a/src/domain/accounts/address-books/address-books.repository.ts b/src/domain/accounts/address-books/address-books.repository.ts
@@ -86,6 +86,56 @@ export class AddressBooksRepository implements IAddressBooksRepository {
     });
   }
 
+  async deleteAddressBook(args: {
+    authPayload: AuthPayload;
+    address: `0x${string}`;
+    chainId: string;
+  }): Promise<void> {
+    if (!args.authPayload.isForSigner(args.address)) {
+      throw new UnauthorizedException();
+    }
+    await this.checkAddressBooksIsEnabled({
+      authPayload: args.authPayload,
+      address: args.address,
+    });
+    const account = await this.accountsRepository.getAccount({
+      authPayload: args.authPayload,
+      address: args.address,
+    });
+    const addressBook = await this.datasource.getAddressBook({
+      account,
+      chainId: args.chainId,
+    });
+    return this.datasource.deleteAddressBook(addressBook);
+  }
+
+  async deleteAddressBookItem(args: {
+    authPayload: AuthPayload;
+    address: `0x${string}`;
+    chainId: string;
+    addressBookItemId: number;
+  }): Promise<void> {
+    if (!args.authPayload.isForSigner(args.address)) {
+      throw new UnauthorizedException();
+    }
+    await this.checkAddressBooksIsEnabled({
+      authPayload: args.authPayload,
+      address: args.address,
+    });
+    const account = await this.accountsRepository.getAccount({
+      authPayload: args.authPayload,
+      address: args.address,
+    });
+    const addressBook = await this.datasource.getAddressBook({
+      account,
+      chainId: args.chainId,
+    });
+    return this.datasource.deleteAddressBookItem({
+      addressBook,
+      id: args.addressBookItemId,
+    });
+  }
+
   // TODO: Extract this functionality in AccountsRepository['checkIsEnabled(DataType, Account)']
   private async checkAddressBooksIsEnabled(args: {
     authPayload: AuthPayload;