Fix entrypoint issue

CHANGELOG.md

@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning].
 
 ## [Unreleased]
 
+### Fixed
+
+- Fix entrypoint issue in the install generator. ([@skryukov])
+
 ## [0.3.0] - 2024-05-31
 
 ### Added

lib/generators/turbo_mount/install_generator.rb

@@ -54,7 +54,11 @@ def install_nodejs
 
         say "Creating Turbo Mount initializer"
         template "turbo-mount.js", File.join("app/javascript/turbo-mount.js")
-        append_to_file "app/javascript/entrypoints/application.js", %(import "./turbo-mount"\n)
+        begin
+          append_to_file js_entrypoint, %(import "./turbo-mount"\n)
+        rescue
+          say 'Could not find the application entrypoint, please add `import "./turbo-mount"` manually.', :yellow
+        end
         warn_about_vite_plugin if vite?
       end
 
@@ -72,6 +76,14 @@ def install_importmap
         run "bin/importmap pin #{FRAMEWORKS[framework][:pins]}"
       end
 
+      def js_entrypoint
+        if vite?
+          "app/javascript/entrypoints/application.js"
+        else
+          "app/javascript/application.js"
+        end
+      end
+
       def vite?
         Dir.glob(Rails.root.join("vite.config.*")).any?
       end