fix: write to tmp file and rename to prevent cache corruption

Vitesco-Technologies · Jan 15, 2024 · bbec078 · bbec078
1 parent c064aff
commit bbec078
Showing 1 changed file with 17 additions and 5 deletions.
diff --git a/lib/config.mjs b/lib/config.mjs
@@ -1,5 +1,12 @@
 import os from "os";
-import { readFile, writeFile, access, mkdir, unlink } from "fs/promises";
+import {
+  readFile,
+  writeFile,
+  access,
+  mkdir,
+  unlink,
+  rename,
+} from "fs/promises";
 import { join } from "path";
 import ini from "ini";
 import crypto from "crypto";
@@ -36,7 +43,7 @@ async function prepareConfig() {
 }
 
 export async function getCliConfig() {
- return import("../config.json")
+  return import("../config.json");
 }
 
 export async function writeDefaultSessions() {
@@ -50,7 +57,7 @@ export async function writeDefaultSessions() {
     if (!awsConfig[key]) {
       awsConfig[key] = sessions[key];
     } else {
-      awsConfig[key] = {...sessions[key], ...awsConfig[key]};
+      awsConfig[key] = { ...sessions[key], ...awsConfig[key] };
     }
   });
 
@@ -100,7 +107,10 @@ export async function writeCache(sessionName, config) {
   };
 
   const cacheFile = join(cacheDir, getHash(sessionName) + ".json");
-  await writeFile(cacheFile, JSON.stringify(cache, null, 2));
+  const tmpFile = cacheFile + ".tmp";
+
+  await writeFile(tmpFile, JSON.stringify(cache, null, 2));
+  await rename(tmpFile, cacheFile);
 }
 
 export async function deleteCache(sessionName) {
@@ -132,6 +142,8 @@ export async function readAccountCache(sessionName) {
 
 export async function writeAccountCache(sessionName, accounts) {
   const cacheFile = join(cacheDir, sessionName + "-accounts.json");
+  const tmpFile = cacheFile + ".tmp";
 
-  return await writeFile(cacheFile, JSON.stringify(accounts));
+  await writeFile(tmpFile, JSON.stringify(accounts, null, 2));
+  await rename(tmpFile, cacheFile);
 }