refactor: move preserveStrings & restoreStrings

Author: omermecitoglu

src/core/injectSchemas.ts

@@ -1,26 +1,4 @@
-function generateRandomString(length: number) {
-  return [...Array(length)].map(() => Math.random().toString(36)[2]).join("");
-}
-
-function preserveStrings(code: string) {
-  let replacements = {} as Record<string, string>;
-
-  const output = code.replace(/(['"`])([^'`"]+)\1/g, replacedString => {
-    const replacementId = generateRandomString(32);
-    replacements = {
-      ...replacements,
-      [replacementId]: replacedString,
-    };
-    return `<@~${replacementId}~@>`;
-  });
-  return { output, replacements };
-}
-
-function restoreStrings(code: string, replacements: Record<string, string>) {
-  return code.replace(/<@~(.*?)~@>/g, (_, replacementId) => {
-    return replacements[replacementId];
-  });
-}
+import { preserveStrings, restoreStrings } from "~/utils/string-preservation";
 
 export default function injectSchemas(code: string, refName: string) {
   const { output: preservedCode, replacements } = preserveStrings(code);

src/utils/string-preservation.ts

@@ -0,0 +1,23 @@
+export function preserveStrings(code: string) {
+  let replacements = {} as Record<string, string>;
+
+  const output = code.replace(/(['"`])([^'`"]+)\1/g, replacedString => {
+    const replacementId = generateRandomString(32);
+    replacements = {
+      ...replacements,
+      [replacementId]: replacedString,
+    };
+    return `<@~${replacementId}~@>`;
+  });
+  return { output, replacements };
+}
+
+export function restoreStrings(code: string, replacements: Record<string, string>) {
+  return code.replace(/<@~(.*?)~@>/g, (_, replacementId) => {
+    return replacements[replacementId];
+  });
+}
+
+function generateRandomString(length: number) {
+  return [...Array(length)].map(() => Math.random().toString(36)[2]).join("");
+}