From 6e4b0360f5fa87617bd34c4ce731cdd40840ea8f Mon Sep 17 00:00:00 2001
From: Caleb Owens <hello@calebowens.com>
Date: Sat, 27 Jan 2024 22:40:29 +0000
Subject: [PATCH] Invoke app:template via Rake::Task[].invoke to avoid
 reloading rakefile

---
 lib/tasks/importmap_tasks.rake |  5 ++++-
 test/installer_test.rb         | 14 +++++++++++++-
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/lib/tasks/importmap_tasks.rake b/lib/tasks/importmap_tasks.rake
index b49a522..ed747ef 100644
--- a/lib/tasks/importmap_tasks.rake
+++ b/lib/tasks/importmap_tasks.rake
@@ -1,6 +1,9 @@
 namespace :importmap do
   desc "Setup Importmap for the app"
   task :install do
-    system RbConfig.ruby, "./bin/rails", "app:template", "LOCATION=#{File.expand_path("../install/install.rb", __dir__)}"
+    previous_location = ENV["LOCATION"]
+    ENV["LOCATION"] = File.expand_path("../install/install.rb", __dir__)
+    Rake::Task["app:template"].invoke
+    ENV["LOCATION"] = previous_location
   end
 end
diff --git a/test/installer_test.rb b/test/installer_test.rb
index 9a907e8..9b6c526 100644
--- a/test/installer_test.rb
+++ b/test/installer_test.rb
@@ -31,6 +31,18 @@ class InstallerTest < ActiveSupport::TestCase
     end
   end
 
+  test "doesn't load rakefile twice" do
+    with_new_rails_app do |app_dir|
+      rakefile = File.read("#{app_dir}/Rakefile")
+      rakefile = "puts \"I've been logged twice!\" \n" + rakefile
+      File.write("#{app_dir}/Rakefile", rakefile)
+
+      out, err = run_command("bin/rails", "importmap:install")
+
+      assert_equal 1, out.scan(/I've been logged twice!/).size
+    end
+  end
+
   private
     def with_new_rails_app
       # Unset testing dummy app so app generator doesn't get confused in Rails 6.1 and 7.0.
@@ -54,7 +66,7 @@ def with_new_rails_app
 
           run_command("bundle", "install")
 
-          yield
+          yield(app_dir)
         end
       end
     end