Optimize code

2024-01-12 23:45:33 +08:00
parent dc4e49343b
commit 8e880cc1ed
5 changed files with 17 additions and 34 deletions
--- a/src/components/Playground/files.ts
+++ b/src/components/Playground/files.ts
@@ -90,6 +90,13 @@ export const cssToJs = (file: IFile) => {
 `
 }

+export const addReactImport = (code: string) => {
+    if (!/import\s+React/g.test(code)) {
+        return `import React from 'react';\n${code}`
+    }
+    return code
+}
+
 export const initFiles: IFiles = getFilesFromUrl() || {
    [ENTRY_FILE_NAME]: {
        name: ENTRY_FILE_NAME,