From 8f19297fa1e266dd1d70818babc6528a911cb0dd Mon Sep 17 00:00:00 2001
From: Mikkel Laursen <mlaursen03@gmail.com>
Date: Mon, 18 Oct 2021 19:25:50 -0600
Subject: [PATCH] chore(docs): Fix alert sandboxes

---
 packages/dev-utils/src/sandbox/createSandbox.ts | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/packages/dev-utils/src/sandbox/createSandbox.ts b/packages/dev-utils/src/sandbox/createSandbox.ts
index c006edb944..4e12609c62 100644
--- a/packages/dev-utils/src/sandbox/createSandbox.ts
+++ b/packages/dev-utils/src/sandbox/createSandbox.ts
@@ -34,7 +34,7 @@ function getExtension(filePath: string): string {
 function transformFileContents(
   contents: string,
   filePath: string,
-  demoName?: string
+  isDemoFile = false
 ): string {
   const parentFolders = filePath.replace("src/", "").split("/").length - 1;
 
@@ -53,10 +53,11 @@ function transformFileContents(
     .replace(/\.(\.\/TextFieldThemeConfig)/, "$1")
     .replace(aliasRegExp, `"${aliasReplacement}`);
 
-  if (demoName) {
-    transformed = transformed
-      .replace(`default function ${demoName}`, "default function Demo")
-      .replace(new RegExp(`<${demoName}(\\s+)`, "g"), "<Demo$1");
+  if (isDemoFile) {
+    transformed = transformed.replace(
+      /export default function [A-Z][A-z0-9]+\(/,
+      "export default function Demo("
+    );
   }
 
   return format(transformed);
@@ -169,7 +170,7 @@ export function createSandbox(
       content: transformFileContents(
         sourceFile.getFullText(),
         "src/Demo.tsx",
-        demoName
+        true
       ),
     },
   };