feat: backup ENB files before launching MO2/game

Wildlander-mod · Dec 16, 2021 · 3ff5525 · 3ff5525
1 parent 13e32df
commit 3ff5525
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 3 deletions.
diff --git a/src/main/ENB.ts b/src/main/ENB.ts
@@ -1,14 +1,17 @@
 import fs from "fs";
 import { logger } from "@/main/logger";
 import {
+  backupDirectory,
   skyrimDirectory,
   USER_PREFERENCE_KEYS,
   userPreferences,
 } from "@/main/config";
-import { copy } from "fs-extra";
+import { copy, existsSync } from "fs-extra";
 import { not as isNotJunk } from "junk";
 import { FriendlyDirectoryMap } from "@/modpack-metadata";
 
+const ENBNameMappingFile = "namesENB.json";
+
 const ENBDirectory = () =>
   `${userPreferences.get(USER_PREFERENCE_KEYS.MOD_DIRECTORY)}/ENB Presets`;
 
@@ -17,7 +20,7 @@ export const getENBPresets = async (): Promise<FriendlyDirectoryMap[]> => {
     await fs.promises.readFile(
       `${userPreferences.get(
         USER_PREFERENCE_KEYS.MOD_DIRECTORY
-      )}/ENB Presets/namesENB.json`,
+      )}/ENB Presets/${ENBNameMappingFile}`,
       "utf-8"
     )
   ) as FriendlyDirectoryMap[];
@@ -41,6 +44,21 @@ export const getENBPresets = async (): Promise<FriendlyDirectoryMap[]> => {
   return [...mappedENBs, ...unmappedENBs];
 };
 
+export const backupOriginalENBs = async () => {
+  const ENBBackupDirectory = `${backupDirectory()}/ENB Presets`;
+  const backupExists = existsSync(ENBBackupDirectory);
+  logger.debug(`Backup for ENBs exists: ${backupExists}`);
+
+  if (!backupExists) {
+    logger.info("No ENB backup exists. Backing up...");
+    await fs.promises.mkdir(backupDirectory(), {
+      recursive: true,
+    });
+
+    await copy(ENBDirectory(), ENBBackupDirectory);
+  }
+};
+
 /**
  * Get all ENB files from all presets.
  * Different presets can have different files,

diff --git a/src/main/config.ts b/src/main/config.ts
@@ -28,3 +28,5 @@ export const skyrimDirectory = () =>
 
 export const modDirectory = () =>
   userPreferences.get(USER_PREFERENCE_KEYS.MOD_DIRECTORY);
+
+export const backupDirectory = () => `${modDirectory()}/.launcher_backups`;
diff --git a/src/main/modOrganizer.ts b/src/main/modOrganizer.ts
@@ -6,7 +6,12 @@ import {
   userPreferences,
 } from "@/main/config";
 import { logger } from "@/main/logger";
-import { checkENBFilesExist, copyENBFiles, getENBPresets } from "@/main/ENB";
+import {
+  backupOriginalENBs,
+  checkENBFilesExist,
+  copyENBFiles,
+  getENBPresets,
+} from "@/main/ENB";
 import { handleError } from "@/main/errorHandler";
 import find from "find-process";
 import { dialog } from "electron";
@@ -151,6 +156,8 @@ const prepareForLaunch = async (): Promise<boolean> => {
     }
   }
 
+  await backupOriginalENBs();
+
   await copyENBFilesIfNotExist();
 
   await setResolution();