From 295a1cdb9cbbbce8434e8d82b96eb63e57c46c1a Mon Sep 17 00:00:00 2001
From: Henry <75142172+ngnlTeto@users.noreply.github.com>
Date: Tue, 27 Feb 2024 10:16:55 +0100
Subject: [PATCH] fix: Windows malicious file analysis waiting problem (#398)

Co-authored-by: Steven <steven@ceriously.com>
---
 sources/corepackUtils.ts | 31 ++++++++++++++++++++++++++++++-
 1 file changed, 30 insertions(+), 1 deletion(-)

diff --git a/sources/corepackUtils.ts b/sources/corepackUtils.ts
index 9d17342..738a9f7 100644
--- a/sources/corepackUtils.ts
+++ b/sources/corepackUtils.ts
@@ -6,6 +6,7 @@ import type {Dir}                                              from 'fs';
 import Module                                                  from 'module';
 import path                                                    from 'path';
 import semver                                                  from 'semver';
+import {setTimeout as setTimeoutPromise}                       from 'timers/promises';
 
 import * as engine                                             from './Engine';
 import * as debugUtils                                         from './debugUtils';
@@ -213,7 +214,11 @@ export async function installVersion(installTarget: string, locator: Locator, {s
 
   await fs.promises.mkdir(path.dirname(installFolder), {recursive: true});
   try {
-    await fs.promises.rename(tmpFolder, installFolder);
+    if (process.platform === `win32`) {
+      await renameUnderWindows(tmpFolder, installFolder);
+    } else {
+      await fs.promises.rename(tmpFolder, installFolder);
+    }
   } catch (err) {
     if (
       (err as nodeUtils.NodeError).code === `ENOTEMPTY` ||
@@ -260,6 +265,30 @@ export async function installVersion(installTarget: string, locator: Locator, {s
   };
 }
 
+async function renameUnderWindows(oldPath: fs.PathLike, newPath: fs.PathLike) {
+  // Windows malicious file analysis blocks files currently under analysis, so we need to wait for file release
+  const retries = 5;
+  for (let i = 0; i < retries; i++) {
+    try {
+      await fs.promises.rename(oldPath, newPath);
+      break;
+    } catch (err) {
+      if (
+        (
+          (err as nodeUtils.NodeError).code === `ENOENT` ||
+          (err as nodeUtils.NodeError).code === `EPERM`
+        ) &&
+        i < (retries - 1)
+      ) {
+        await setTimeoutPromise(100 * 2 ** i);
+        continue;
+      } else {
+        throw err;
+      }
+    }
+  }
+}
+
 /**
  * Loads the binary, taking control of the current process.
  */