Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions backend/src/file-upload/dto.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
export class FileUploadDto {
filename: string;
mimetype: string;
}
23 changes: 23 additions & 0 deletions backend/src/file-upload/file-upload.controller.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
import {
Controller,
Post,
UseInterceptors,
UploadedFile,
BadRequestException,
} from '@nestjs/common';
import { FileInterceptor } from '@nestjs/platform-express';
import { FileUploadService } from './file-upload.service';

@Controller('file-upload')
export class FileUploadController {
constructor(private readonly fileUploadService: FileUploadService) {}

@Post('upload')
@UseInterceptors(FileInterceptor('file'))
async uploadFile(@UploadedFile() file: any) {
if (!file) throw new BadRequestException('No file provided');
const isSafe = await this.fileUploadService.scanMalware(file.buffer);
if (!isSafe) throw new BadRequestException('Malware detected');
return { status: 'success', message: 'File is clean' };
}
}
10 changes: 10 additions & 0 deletions backend/src/file-upload/file-upload.module.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
import { Module } from '@nestjs/common';
import { FileUploadController } from './file-upload.controller';
import { FileUploadService } from './file-upload.service';

@Module({
controllers: [FileUploadController],
providers: [FileUploadService],
exports: [FileUploadService],
})
export class FileUploadModule {}
17 changes: 17 additions & 0 deletions backend/src/file-upload/file-upload.service.spec.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
import { Test, TestingModule } from '@nestjs/testing';
import { FileUploadService } from './file-upload.service';

describe('FileUploadService', () => {
let service: FileUploadService;

beforeEach(async () => {
const module: TestingModule = await Test.createTestingModule({
providers: [FileUploadService],
}).compile();
service = module.get<FileUploadService>(FileUploadService);
});

it('should be defined', () => {
expect(service).toBeDefined();
});
});
10 changes: 10 additions & 0 deletions backend/src/file-upload/file-upload.service.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
import { Injectable } from '@nestjs/common';

@Injectable()
export class FileUploadService {
async scanMalware(buffer: Buffer): Promise<boolean> {
// Integration with clamscan
console.log('Scanning file chunk of size', buffer.length);
return true; // Mock clean file
}
}
Loading