Merge pull request #10 from labscommunity/kranthi/upload-get-handlers

feat: add get handlers for upload module

Author: kranthicodes

src/common/utils/paginator.ts

@@ -0,0 +1,44 @@
+export interface PaginatedResult<T> {
+    data: T[]
+    meta: {
+        total: number
+        lastPage: number
+        currentPage: number
+        perPage: number
+        prev: number | null
+        next: number | null
+    }
+}
+
+export type PaginateOptions = { page?: number | string, perPage?: number | string }
+export type PaginateFunction = <T, K>(model: any, args?: K, options?: PaginateOptions) => Promise<PaginatedResult<T>>
+
+export const paginator = (defaultOptions: PaginateOptions): PaginateFunction => {
+    return async (model, args: any = { where: undefined }, options) => {
+        const page = Number(options?.page || defaultOptions?.page) || 1;
+        const perPage = Number(options?.perPage || defaultOptions?.perPage) || 10;
+
+        const skip = page > 0 ? perPage * (page - 1) : 0;
+        const [total, data] = await Promise.all([
+            model.count({ where: args.where }),
+            model.findMany({
+                ...args,
+                take: perPage,
+                skip,
+            }),
+        ]);
+        const lastPage = Math.ceil(total / perPage);
+
+        return {
+            data,
+            meta: {
+                total,
+                lastPage,
+                currentPage: page,
+                perPage,
+                prev: page > 1 ? page - 1 : null,
+                next: page < lastPage ? page + 1 : null,
+            },
+        };
+    };
+};

src/modules/upload/upload.controller.ts

@@ -1,7 +1,7 @@
 import { existsSync, mkdirSync } from 'node:fs';
 import { extname, parse } from 'node:path';
 
-import { BadRequestException, Body, Controller, Post, Req, UploadedFiles, UseInterceptors } from '@nestjs/common';
+import { BadRequestException, Body, Controller, Get, Param, Post, Query, Req, UploadedFiles, UseInterceptors } from '@nestjs/common';
 import { FilesInterceptor } from '@nestjs/platform-express';
 import { UploadStatus, UploadType, User } from '@prisma/client';
 import * as crypto from 'crypto';
@@ -19,43 +19,11 @@ import { UploadService } from './upload.service';
 export class UploadController {
   constructor(private readonly uploadService: UploadService) { }
 
-  @Post('cost')
-  getEstimate(@Body() body: EstimatesDto) {
-    return this.uploadService.getCostEstimate(body);
+  @Get()
+  async getUploadRequests(@Query('page') page: number, @Query('limit') limit: number) {
+    return this.uploadService.getUploadRequests({ page, perPage: limit });
   }
 
-  @Post('create')
-  createUploadRequest(@Body() body: CreateUploadRequestDto, @Req() req: Request) {
-    const { totalChunks, uploadType, fileName, size } = body
-    const user = (req as any).user as User
-    if (!user) {
-      throw new BadRequestException('User not found');
-    }
-
-    if (totalChunks !== 1) {
-      throw new BadRequestException('Total chunks must be 1');
-    }
-
-    if (uploadType === UploadType.MULTIPART_FILE) {
-      throw new BadRequestException('Invalid upload type. Only single file upload is supported.');
-    }
-
-    // Validate filename has proper extension
-    if (!fileName.includes('.')) {
-      throw new BadRequestException('Filename must include a file extension');
-    }
-
-    const extension = fileName.split('.').pop();
-    if (!extension || extension.length === 0) {
-      throw new BadRequestException('Invalid file. No extension found');
-    }
-
-    if (size <= 0) {
-      throw new BadRequestException('File size must be greater than 0');
-    }
-
-    return this.uploadService.createUploadRequest(body, user);
-  }
 
   @Post()
   @UseInterceptors(FilesInterceptor('file', 1, {
@@ -127,6 +95,45 @@ export class UploadController {
     return receipt
   }
 
+
+  @Post('cost')
+  getEstimate(@Body() body: EstimatesDto) {
+    return this.uploadService.getCostEstimate(body);
+  }
+
+  @Post('create')
+  createUploadRequest(@Body() body: CreateUploadRequestDto, @Req() req: Request) {
+    const { totalChunks, uploadType, fileName, size } = body
+    const user = (req as any).user as User
+    if (!user) {
+      throw new BadRequestException('User not found');
+    }
+
+    if (totalChunks !== 1) {
+      throw new BadRequestException('Total chunks must be 1');
+    }
+
+    if (uploadType === UploadType.MULTIPART_FILE) {
+      throw new BadRequestException('Invalid upload type. Only single file upload is supported.');
+    }
+
+    // Validate filename has proper extension
+    if (!fileName.includes('.')) {
+      throw new BadRequestException('Filename must include a file extension');
+    }
+
+    const extension = fileName.split('.').pop();
+    if (!extension || extension.length === 0) {
+      throw new BadRequestException('Invalid file. No extension found');
+    }
+
+    if (size <= 0) {
+      throw new BadRequestException('File size must be greater than 0');
+    }
+
+    return this.uploadService.createUploadRequest(body, user);
+  }
+
   @Post('chunk')
   async uploadChunk(@Req() req: Request) {
     const currentChunk = req.headers['x-current-chunk'];
@@ -199,4 +206,19 @@ export class UploadController {
 
     return rest;
   }
+
+  @Get('receipt')
+  async getReceipts(@Query('page') page: number, @Query('limit') limit: number) {
+    return this.uploadService.getReceipts({ page, perPage: limit });
+  }
+
+  @Get('receipt/:id')
+  async getReceipt(@Param('id') id: string) {
+    return this.uploadService.getReceiptById(id);
+  }
+
+  @Get(':id')
+  async getUploadRequest(@Param('id') id: string) {
+    return this.uploadService.getUploadRequest(id);
+  }
 }

src/modules/upload/upload.service.ts

@@ -6,6 +6,7 @@ import BigNumber from 'bignumber.js';
 import { Contract, getAddress, Interface, Provider } from 'ethers';
 import * as fs from 'fs';
 import path from 'path';
+import { paginator } from 'src/common/utils/paginator';
 import { DatabaseService } from 'src/database/database.service';
 import { pipeline } from 'stream';
 import { promisify } from 'util';
@@ -24,6 +25,8 @@ const ERC20_ABI = [
 ];
 const pump = promisify(pipeline);
 
+const paginate = paginator({ perPage: 10 });
+
 @Injectable()
 export class UploadService {
   constructor(
@@ -67,6 +70,22 @@ export class UploadService {
     };
   }
 
+  async getUploadRequests({ page = 1, perPage = 10 }: { page?: number, perPage?: number }) {
+    return await paginate(this.databaseService.upload, {
+      orderBy: {
+        createdAt: 'desc'
+      }
+    }, { page, perPage });
+  }
+
+  async getReceipts({ page = 1, perPage = 10 }: { page?: number, perPage?: number }) {
+    return await paginate(this.databaseService.receipt, {
+      orderBy: {
+        createdAt: 'desc'
+      }
+    }, { page, perPage });
+  }
+
   async createUploadRequest(createUploadRequestDto: CreateUploadRequestDto, user: User) {
     const { totalChunks, uploadType, fileName, size, tokenTicker, mimeType, network, chainId, tags } = createUploadRequestDto
     const { chainType } = user
@@ -404,6 +423,14 @@ export class UploadService {
     return feeTransaction;
   }
 
+  async getReceiptById(receiptId: string) {
+    return await this.databaseService.receipt.findFirst({
+      where: {
+        id: receiptId,
+      },
+    });
+  }
+
   async getReceiptByUploadId(uploadId: string) {
     return await this.databaseService.receipt.findFirst({
       where: {