chore: Add transfers by domain route

sprintertech · Sep 10, 2024 · 233a204 · 233a204
1 parent 5ef7c52
commit 233a204
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 14 deletions.
diff --git a/src/controllers/schemas.ts b/src/controllers/schemas.ts
@@ -280,14 +280,16 @@ export const transfersByResourceBetweenDomainsSchema = {
 }
 
 export const transfersByDomainSchema = {
-  summary: "Get all transfers with a specific domain as source or destination",
+  summary:
+    "Get all transfers with a specific domain as source or destination. Possible to get transfers by domain only as source and only as destination.",
   querystring: {
     type: "object",
     properties: {
       ...paginationSchema,
       domain: {
         type: "string",
-        default: "source",
+        default: undefined,
+        nullable: true,
         enum: ["source", "destination"],
       },
     },

diff --git a/src/services/transfers.service.ts b/src/services/transfers.service.ts
@@ -2,7 +2,7 @@
 The Licensed Work is (c) 2023 Sygma
 SPDX-License-Identifier: LGPL-3.0-only
 */
-import { PrismaClient, Transfer, TransferStatus } from "@prisma/client"
+import { Prisma, PrismaClient, Transfer, TransferStatus } from "@prisma/client"
 import { NotFound, getTransferQueryParams } from "../utils/helpers"
 
 export type Pagination = {
@@ -34,7 +34,7 @@ class TransfersService {
     }
   }
 
-  public async findTransfers(where: Partial<Transfer>, paginationParams: Pagination): Promise<Transfer[]> {
+  public async findTransfers(where: Prisma.TransferWhereInput, paginationParams: Pagination): Promise<Transfer[]> {
     const { skip, take } = this.calculatePaginationParams(paginationParams)
     const transfers = await this.transfers.findMany({
       where,
@@ -54,7 +54,7 @@ class TransfersService {
   }
 
   public async findAllTransfers(status: TransferStatus | undefined, paginationParams: Pagination): Promise<Transfer[]> {
-    const where: Partial<Transfer> = {
+    const where: Prisma.TransferWhereInput = {
       status: status,
     }
 
@@ -76,7 +76,7 @@ class TransfersService {
   }
 
   public async findTransfersByTxHash(txHash: string, domainID: number): Promise<Transfer[]> {
-    let where: Partial<any>
+    let where: Prisma.DepositWhereInput
     domainID == undefined
       ? (where = { txHash: txHash })
       : (where = {
@@ -97,7 +97,7 @@ class TransfersService {
   }
 
   public async findTransfersByAccountAddress(sender: string, status: TransferStatus | undefined, paginationParams: Pagination): Promise<Transfer[]> {
-    const where: Partial<Transfer> = {
+    const where: Prisma.TransferWhereInput = {
       accountId: sender,
       status: status,
     }
@@ -108,7 +108,7 @@ class TransfersService {
   }
 
   public async findTransfersByResourceID(resourceID: string, status: TransferStatus | undefined, paginationParams: Pagination): Promise<Transfer[]> {
-    const where: Partial<Transfer> = {
+    const where: Prisma.TransferWhereInput = {
       resourceID: resourceID,
       status: status,
     }
@@ -123,7 +123,7 @@ class TransfersService {
     destinationDomainID: number,
     paginationParams: Pagination,
   ): Promise<Transfer[]> {
-    const where: Partial<Transfer> = {
+    const where: Prisma.TransferWhereInput = {
       fromDomainId: sourceDomainID,
       toDomainId: destinationDomainID,
     }
@@ -139,7 +139,7 @@ class TransfersService {
     destinationDomainID: number,
     paginationParams: Pagination,
   ): Promise<Transfer[]> {
-    const where: Partial<Transfer> = {
+    const where: Prisma.TransferWhereInput = {
       resourceID: resourceID,
       fromDomainId: sourceDomainID,
       toDomainId: destinationDomainID,
@@ -156,10 +156,20 @@ class TransfersService {
     status: TransferStatus | undefined,
     paginationParams: Pagination,
   ): Promise<Transfer[]> {
-    let where: Partial<Transfer>
-
-    domain == DomainType.Source ? (where = { fromDomainId: domainID, status: status }) : (where = { toDomainId: domainID, status: status })
-
+    let where: Prisma.TransferWhereInput
+
+    if (domain == DomainType.Source) {
+      where = { fromDomainId: domainID, status: status }
+    } else if (domain == DomainType.Destination) {
+      where = { toDomainId: domainID, status: status }
+    } else {
+      where = {
+        OR: [
+          { fromDomainId: domainID, status: status },
+          { toDomainId: domainID, status: status },
+        ],
+      }
+    }
     const transfers = this.findTransfers(where, paginationParams)
 
     return transfers