From 61a3685d6cf250394b4924fb4e204b1aeb16e568 Mon Sep 17 00:00:00 2001 From: MathieuG-P Date: Tue, 27 Dec 2022 00:50:15 +0100 Subject: [PATCH] sanitize models filename --- .../additional-content/local-models-manager.service.ts | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/main/services/additional-content/local-models-manager.service.ts b/src/main/services/additional-content/local-models-manager.service.ts index 268ec776..f266d93a 100644 --- a/src/main/services/additional-content/local-models-manager.service.ts +++ b/src/main/services/additional-content/local-models-manager.service.ts @@ -9,7 +9,8 @@ import { BSVersion } from "shared/bs-version.interface"; import { BSLocalVersionService } from "../bs-local-version.service"; import path from "path"; import { RequestService } from "../request.service"; -import { copyFileSync, constants } from "fs-extra"; +import { copyFileSync } from "fs-extra"; +import sanitize from "sanitize-filename"; export class LocalModelsManagerService { @@ -82,7 +83,7 @@ export class LocalModelsManagerService { public async downloadModel(model: MSModel, version: BSVersion): Promise{ const modelFolder = await this.getModelFolderPath(model.type, version); - const modelDest = path.join(modelFolder, path.basename(model.download)); + const modelDest = path.join(modelFolder, sanitize(path.basename(model.download))); return this.request.downloadFile(model.download, modelDest);