Wrap update_files calls in conditionals (#147)

Co-authored-by: Grant Petersen-Speelman <grantspeelman@users.noreply.github.com>

CHANGELOG.md

@@ -3,6 +3,9 @@
 ### Changed
 * Removed the update generator and reduced options for install generator [PR 149](https://github.com/shakacode/cypress-on-rails/pull/149)
 
+### Fixed
+* fix update index.js in install generatpr [PR 147](https://github.com/shakacode/cypress-on-rails/pull/147) by [Judahmeek]
+
 ## [1.16.0]
 [Compare]: https://github.com/shakacode/cypress-on-rails/compare/v1.15.1...v1.16.0
 

lib/generators/cypress_on_rails/install_generator.rb

@@ -55,12 +55,16 @@ def add_initial_files
     end
 
     def update_files
-      append_to_file "#{options.cypress_folder}/support/index.js",
-                     "\nimport './on-rails'",
-                     after: 'import \'./commands\''
-      append_to_file "#{options.playwright_folder}/support/index.js",
-                     "\nimport './on-rails'",
-                     after: '// Import commands.js using ES2015 syntax:'
+      if options.framework == 'cypress'
+        append_to_file "#{options.install_folder}/cypress/support/index.js",
+                       "\nimport './on-rails'",
+                       after: 'import \'./commands\''
+      end
+      if options.framework == 'playwright'
+        append_to_file "#{options.install_folder}/playwright/support/index.js",
+                       "\nimport './on-rails'",
+                       after: '// Import commands.js using ES2015 syntax:'
+      end
     end
   end
 end