code refactoring

index.js

@@ -1,7 +1,7 @@
 const fs = require('fs');
 const path = require('path');
 const CDPConnection = require('./src/cdp.js');
-const {injectRendererName, injectScript} = require('./src/utils/index.js');
+const {injectRendererName, injectScript} = require('./src/utils/');
 
 module.exports = function viteNightwatchPlugin(options = {}) {
   const enableCdpConnection = options.enableCdpConnection || false;

src/utils/injectScript.js

@@ -50,11 +50,17 @@ const svelteScript = (path) =>
 	window['@component_class'] = Component;
 `;
 
+
 module.exports = function (content, componentType, path) {
-  return content.replace(
-    '<!-- script -->',
-    `<script type="module">
-			${componentType === 'react' ? reactScript(path) : (componentType === 'svelte' ? svelteScript(path) : vueScript(path))}
-		</script>`
-  );
+  let scriptType;
+
+  if (componentType === 'react') {
+    scriptType = reactScript(path);
+  } else if (componentType === 'svelte') {
+    scriptType = svelteScript(path);
+  } else {
+    scriptType = vueScript(path);
+  }
+
+  return content.replace('<!-- script -->', `<script type="module">${scriptType}</script>`);
 };

test/lib/vite.config-svelte.js

@@ -1,12 +1,11 @@
 import { defineConfig } from 'vite';
 import nightwatchPlugin from '../../index.js'; 
 
-let asyncSvelte = async () => {
+const asyncSvelte = async () => {
   const SveltPlugin = await import('@sveltejs/vite-plugin-svelte');
   return SveltPlugin.svelte();
 };
 
-
 export default defineConfig({
   plugins: [
     asyncSvelte(),