From 4e47b97eabe547a04a333d9c556810ace786b2fc Mon Sep 17 00:00:00 2001 From: Ryan Schlesinger Date: Thu, 10 Apr 2025 09:27:08 -0300 Subject: [PATCH] Generalize EnsureRailsInit --- vite_rails/lib/vite_rails/cli.rb | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/vite_rails/lib/vite_rails/cli.rb b/vite_rails/lib/vite_rails/cli.rb index 31418f6a..c1a40d84 100644 --- a/vite_rails/lib/vite_rails/cli.rb +++ b/vite_rails/lib/vite_rails/cli.rb @@ -5,7 +5,7 @@ module ViteRails::CLI end -module ViteRails::CLI::Build +module ViteRails::CLI::EnsureRailsInit def call(**options) ensure_rails_init super @@ -15,7 +15,7 @@ def call(**options) def ensure_rails_init require File.expand_path("config/environment", Dir.pwd) rescue StandardError, LoadError => error - $stderr << "Unable to initialize Rails application before Vite build:\n\n\t#{error.message}\n\n" + $stderr << "Unable to initialize Rails application before Vite:\n\n\t#{error.message}\n\n" end end @@ -80,5 +80,6 @@ def install_sample_files end end -ViteRuby::CLI::Build.prepend(ViteRails::CLI::Build) +ViteRuby::CLI::Build.prepend(ViteRails::CLI::EnsureRailsInit) +ViteRuby::CLI::Dev.prepend(ViteRails::CLI::EnsureRailsInit) ViteRuby::CLI::Install.prepend(ViteRails::CLI::Install)