iio/pingvin-share
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: use fs promises in file service

Browse Source
This commit is contained in:
Elias Schneider
2024-11-17 16:09:23 +01:00
parent 6cf5c66fe2
commit c52ec71920
1 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
backend/src/file/file.service.ts
View File

@@ -8,7 +8,8 @@ import {
} from "@nestjs/common"; } from "@nestjs/common";
import { JwtService } from "@nestjs/jwt"; import { JwtService } from "@nestjs/jwt";
import * as crypto from "crypto"; import * as crypto from "crypto";
import * as fs from "fs"; import { createReadStream } from "fs";
import * as fs from "fs/promises";
import * as mime from "mime-types"; import * as mime from "mime-types";
import { ConfigService } from "src/config/config.service"; import { ConfigService } from "src/config/config.service";
import { PrismaService } from "src/prisma/prisma.service"; import { PrismaService } from "src/prisma/prisma.service";
@@ -19,7 +20,6 @@ import { SHARE_DIRECTORY } from "../constants";
export class FileService { export class FileService {
constructor( constructor(
private prisma: PrismaService, private prisma: PrismaService,
private jwtService: JwtService,
private config: ConfigService, private config: ConfigService,
) {} ) {}
@@ -45,8 +45,8 @@ export class FileService {
let diskFileSize: number; let diskFileSize: number;
try { try {
diskFileSize = fs.statSync( diskFileSize = (
`${SHARE_DIRECTORY}/${shareId}/${file.id}.tmp-chunk`, await fs.stat(`${SHARE_DIRECTORY}/${shareId}/${file.id}.tmp-chunk`)
).size; ).size;
} catch { } catch {
diskFileSize = 0; diskFileSize = 0;
@@ -66,7 +66,7 @@ export class FileService {
const buffer = Buffer.from(data, "base64"); const buffer = Buffer.from(data, "base64");
// Check if there is enough space on the server // Check if there is enough space on the server
const space = await fs.promises.statfs(SHARE_DIRECTORY); const space = await fs.statfs(SHARE_DIRECTORY);
const availableSpace = space.bavail * space.bsize; const availableSpace = space.bavail * space.bsize;
if (availableSpace < buffer.byteLength) { if (availableSpace < buffer.byteLength) {
throw new InternalServerErrorException("Not enough space on the server"); throw new InternalServerErrorException("Not enough space on the server");
@@ -91,19 +91,19 @@ export class FileService {
); );
} }
fs.appendFileSync( await fs.appendFile(
`${SHARE_DIRECTORY}/${shareId}/${file.id}.tmp-chunk`, `${SHARE_DIRECTORY}/${shareId}/${file.id}.tmp-chunk`,
buffer, buffer,
); );
const isLastChunk = chunk.index == chunk.total - 1; const isLastChunk = chunk.index == chunk.total - 1;
if (isLastChunk) { if (isLastChunk) {
fs.renameSync( await fs.rename(
`${SHARE_DIRECTORY}/${shareId}/${file.id}.tmp-chunk`, `${SHARE_DIRECTORY}/${shareId}/${file.id}.tmp-chunk`,
`${SHARE_DIRECTORY}/${shareId}/${file.id}`, `${SHARE_DIRECTORY}/${shareId}/${file.id}`,
); );
const fileSize = fs.statSync( const fileSize = (
`${SHARE_DIRECTORY}/${shareId}/${file.id}`, await fs.stat(`${SHARE_DIRECTORY}/${shareId}/${file.id}`)
).size; ).size;
await this.prisma.file.create({ await this.prisma.file.create({
data: { data: {
@@ -125,7 +125,7 @@ export class FileService {
if (!fileMetaData) throw new NotFoundException("File not found"); if (!fileMetaData) throw new NotFoundException("File not found");
const file = fs.createReadStream(`${SHARE_DIRECTORY}/${shareId}/${fileId}`); const file = createReadStream(`${SHARE_DIRECTORY}/${shareId}/${fileId}`);
return { return {
metaData: { metaData: {
@@ -144,19 +144,19 @@ export class FileService {
if (!fileMetaData) throw new NotFoundException("File not found"); if (!fileMetaData) throw new NotFoundException("File not found");
fs.unlinkSync(`${SHARE_DIRECTORY}/${shareId}/${fileId}`); await fs.unlink(`${SHARE_DIRECTORY}/${shareId}/${fileId}`);
await this.prisma.file.delete({ where: { id: fileId } }); await this.prisma.file.delete({ where: { id: fileId } });
} }
async deleteAllFiles(shareId: string) { async deleteAllFiles(shareId: string) {
await fs.promises.rm(`${SHARE_DIRECTORY}/${shareId}`, { await fs.rm(`${SHARE_DIRECTORY}/${shareId}`, {
recursive: true, recursive: true,
force: true, force: true,
}); });
} }
getZip(shareId: string) { getZip(shareId: string) {
return fs.createReadStream(`${SHARE_DIRECTORY}/${shareId}/archive.zip`); return createReadStream(`${SHARE_DIRECTORY}/${shareId}/archive.zip`);
} }
} }