improve vite config creation

chicoxyzzy · chicoxyzzy · commit 010ef7a66701 · 2023-11-10T14:53:42.000+01:00
diff --git a/index.js b/index.js
@@ -97,16 +97,6 @@ viteProcess.on('close', (code) => {
   const modifiedData = indexHTML.replace('<head>', '<head>\n' + CSP + '\n')
   writeFileSync(indexHTMLPath, modifiedData, 'utf8')
 
-  process.stdout.write('\nPatching package.json\n')
-
-  const packageJSONPath = resolve(cwd, 'package.json')
-  const packageJSON = readFileSync(packageJSONPath, 'utf-8')
-  const modifiedPackageJSON = packageJSON.replace(
-    '"tsc && vite build"',
-    '"tsc && vite build --"'
-  )
-  writeFileSync(packageJSONPath, modifiedPackageJSON, 'utf8')
-
   // So TypeScript doesn't complain about about String#startsWith in the vite.config.ts
   if (existsSync(resolve(cwd, 'tsconfig.node.json'))) {
     process.stdout.write('\nPatching tsconfig.node.json\n')
@@ -123,21 +113,48 @@ viteProcess.on('close', (code) => {
     }
   }
 
-  // patching the vite.config.ts to make socket:* modules external
-  if (existsSync(resolve(cwd, 'vite.config.ts'))) {
-    process.stdout.write('\nPatching vite.config.ts\n')
+  let isViteConfigExists = false
+  ['js', 'ts'].forEach((ext) => {
+    // patching the vite.config.js to make socket:* modules external
+    if (existsSync(resolve(cwd, `vite.config.${ext}`))) {
+      process.stdout.write('\nPatching vite.config.' + ext + '\n')
+
+      const viteConfigPath = resolve(cwd, 'vite.config.js')
+      const viteConfig = readFileSync(viteConfigPath, 'utf-8')
+
+      // make socket:* modules external for build
+      if (!viteConfig.includes('external(id)')) {
+        const modifiedViteConfig = viteConfig.replace(
+          'defineConfig({',
+          'defineConfig({\n  build: {\n    rollupOptions: {\n      external(id) {\n        return id.startsWith(\'socket:\')\n      }\n    }\n  },'
+        )
+        writeFileSync(viteConfigPath, modifiedViteConfig, 'utf8')
+      }
+    }
+    isViteConfigExists = true
+  })
+
+  if (!isViteConfigExists) {
+    process.stdout.write('\nCreating vite.config.js\n')
 
-    const viteConfigTSPath = resolve(cwd, 'vite.config.ts')
-    const viteConfigTS = readFileSync(viteConfigTSPath, 'utf-8')
+    const viteConfigPath = resolve(cwd, 'vite.config.js')
+    const viteConfig = `
+import { defineConfig } from 'vite'
 
-    if (!viteConfigTS.includes('external(id)')) {
-      const modifiedViteConfigTS = viteConfigTS.replace(
-        'defineConfig({',
-        'defineConfig({\n  build: {\n    rollupOptions: {\n      external(id) {\n        return id.startsWith(\'socket:\')\n      }\n    }\n  },'
-      )
-      writeFileSync(viteConfigTSPath, modifiedViteConfigTS, 'utf8')
+// https://vitejs.dev/config/
+export default defineConfig({
+  build: { // HERE
+    rollupOptions: {
+      external(id) {
+        return id.startsWith('socket:')
+      }
     }
+  },
+})
+`
+    writeFileSync(viteConfigPath, viteConfig, 'utf8')
   }
 
+
   process.stdout.write(`\nDone!\n\nBuild and run your app with:\nssc build -r ${projectName}`)
 })
