fix: prettier run on newText

.changeset/silver-books-yell.md

@@ -0,0 +1,6 @@
+---
+'@xstate/cli': patch
+'@xstate/tools-shared': patch
+---
+
+feat: import statement used to determine quote style

apps/cli/src/bin.ts

@@ -42,20 +42,26 @@ function getPrettierInstance(cwd: string): typeof import('prettier') {
   }
 }
 
+// Used to prettify text before writing
+const prettify = async (
+  uri: string,
+  text: string,
+  { cwd }: { cwd: string },
+) => {
+  const prettierInstance = getPrettierInstance(cwd);
+  return prettierInstance.format(text, {
+    ...(await prettierInstance.resolveConfig(uri)),
+    parser: 'typescript',
+  });
+};
+
 const writeToTypegenFile = async (
   typegenUri: string,
   types: TypegenData[],
   { cwd }: { cwd: string },
 ) => {
-  const prettierInstance = getPrettierInstance(cwd);
-  await fs.writeFile(
-    typegenUri,
-    // // Prettier v3 returns a promise
-    await prettierInstance.format(getTypegenOutput(types), {
-      ...(await prettierInstance.resolveConfig(typegenUri)),
-      parser: 'typescript',
-    }),
-  );
+  const output = await prettify(typegenUri, getTypegenOutput(types), { cwd });
+  await fs.writeFile(typegenUri, output);
 };
 
 // TODO: just use the native one when support for node 12 gets dropped
@@ -88,6 +94,20 @@ const writeToFiles = async (uriArray: string[], { cwd }: { cwd: string }) => {
           return;
         }
 
+        // Find the first import statement and check what quote style it uses
+        // We will consider that the quote style to use for our import statement
+        const quoteMatch = fileContents.match(/import\s(.|\n)*?(['"])/);
+        // Default quote style to "
+        let quoteStyle = '"';
+        if (quoteMatch !== null && quoteMatch.length > 2) {
+          // Get the second group from the match
+          quoteStyle = quoteMatch[2];
+          // Ensure that the quote is either a ' or "
+          if (quoteStyle !== '"' && quoteStyle !== "'") {
+            quoteStyle = '"';
+          }
+        }
+
         const typegenUri =
           uri.slice(0, -path.extname(uri).length) + '.typegen.ts';
 
@@ -108,7 +128,7 @@ const writeToFiles = async (uriArray: string[], { cwd }: { cwd: string }) => {
           await removeFile(typegenUri);
         }
 
-        const edits = getTsTypesEdits(types);
+        const edits = getTsTypesEdits(types, quoteStyle);
         if (edits.length > 0) {
           const newFile = processFileEdits(fileContents, edits);
           await fs.writeFile(uri, newFile);

packages/shared/src/getTsTypesEdits.ts

@@ -1,14 +1,14 @@
 import { TypegenData } from './getTypegenData';
 
-export const getTsTypesEdits = (types: TypegenData[]) =>
+export const getTsTypesEdits = (types: TypegenData[], quoteStyle = '"') =>
   types
     .filter(
-      (type) =>
+      type =>
         !type.typesNode.value ||
         type.typesNode.value.argument !== type.data.tsTypesValue.argument ||
         type.typesNode.value.qualifier !== type.data.tsTypesValue.qualifier,
     )
-    .map((type) => ({
+    .map(type => ({
       range: type.typesNode.range,
-      newText: `{} as import("${type.data.tsTypesValue.argument}").${type.data.tsTypesValue.qualifier}`,
+      newText: `{} as import(${quoteStyle}${type.data.tsTypesValue.argument}${quoteStyle}).${type.data.tsTypesValue.qualifier}`,
     }));