From 14e8162d199cbf80dd1e4dfcce6006758e6c89ea Mon Sep 17 00:00:00 2001
From: Patrik Oldsberg <poldsberg@gmail.com>
Date: Fri, 12 Nov 2021 13:08:25 +0100
Subject: [PATCH] cli: added temp dir utility to create context

Signed-off-by: Patrik Oldsberg <poldsberg@gmail.com>
---
 packages/cli/src/commands/create/create.ts | 36 +++++++++++++++++-----
 packages/cli/src/lib/create/types.ts       |  3 ++
 2 files changed, 32 insertions(+), 7 deletions(-)

diff --git a/packages/cli/src/commands/create/create.ts b/packages/cli/src/commands/create/create.ts
index 07378579d2..d39fa2b9d0 100644
--- a/packages/cli/src/commands/create/create.ts
+++ b/packages/cli/src/commands/create/create.ts
@@ -14,7 +14,9 @@
  * limitations under the License.
  */
 
+import os from 'os';
 import fs from 'fs-extra';
+import { join as joinPath } from 'path';
 import { Command } from 'commander';
 import { FactoryRegistry } from '../../lib/create/FactoryRegistry';
 import { paths } from '../../lib/paths';
@@ -68,11 +70,31 @@ export default async (cmd: Command) => {
     }
   }
 
-  await factory.create(options, {
-    isMonoRepo,
-    defaultVersion,
-    scope: cmdOpts.scope.replace(/^@/, ''),
-    npmRegistry: cmdOpts.npmRegistry,
-    private: Boolean(cmdOpts.private),
-  });
+  const tempDirs = new Array<string>();
+  async function createTemporaryDirectory(name: string): Promise<string> {
+    const dir = await fs.mkdtemp(joinPath(os.tmpdir(), name));
+    tempDirs.push(dir);
+    return dir;
+  }
+
+  try {
+    await factory.create(options, {
+      isMonoRepo,
+      defaultVersion,
+      scope: cmdOpts.scope.replace(/^@/, ''),
+      npmRegistry: cmdOpts.npmRegistry,
+      private: Boolean(cmdOpts.private),
+      createTemporaryDirectory,
+    });
+  } finally {
+    for (const dir of tempDirs) {
+      try {
+        await fs.remove(dir);
+      } catch (error) {
+        console.error(
+          `Failed to remove temporary directory '${dir}', ${error}`,
+        );
+      }
+    }
+  }
 };
diff --git a/packages/cli/src/lib/create/types.ts b/packages/cli/src/lib/create/types.ts
index 82bdd4978a..629506b5f7 100644
--- a/packages/cli/src/lib/create/types.ts
+++ b/packages/cli/src/lib/create/types.ts
@@ -27,6 +27,9 @@ export interface CreateContext {
   isMonoRepo: boolean;
   /** The default version to use for new packages */
   defaultVersion: string;
+
+  /** Creates a temporary directory. This will always be deleted after creation is done. */
+  createTemporaryDirectory(name: string): Promise<string>;
 }
 
 export type AnyOptions = Record<string, string>;