stimulus_reflex 3.5.0.pre9 → 3.5.0.rc1

data/lib/stimulus_reflex/reflex.rb

@@ -1,57 +1,66 @@
 # frozen_string_literal: true
 
-ClientAttributes = Struct.new(:reflex_id, :tab_id, :reflex_controller, :xpath_controller, :xpath_element, :permanent_attribute_name, :version, :suppress_logging, keyword_init: true)
+require "stimulus_reflex/cable_readiness"
+require "stimulus_reflex/version_checker"
+
+# TODO remove xpath_controller and xpath_element for v4
+ClientAttributes = Struct.new(
+  :id,
+  :tab_id,
+  :reflex_controller,
+  :xpath_controller,
+  :xpath_element,
+  :permanent_attribute_name,
+  :version,
+  :npm_version,
+  :suppress_logging,
+  keyword_init: true
+)
 
 class StimulusReflex::Reflex
-  class VersionMismatchError < StandardError; end
-
-  include ActiveSupport::Rescuable
+  prepend StimulusReflex::CableReadiness
+  include StimulusReflex::VersionChecker
   include StimulusReflex::Callbacks
+  include ActiveSupport::Rescuable
   include ActionView::Helpers::TagHelper
   include CableReady::Identifiable
 
   attr_accessor :payload, :headers
-  attr_reader :cable_ready, :channel, :url, :element, :selectors, :method_name, :broadcaster, :client_attributes, :logger
+  attr_reader :channel, :url, :element, :selectors, :method_name, :broadcaster, :client_attributes, :logger
 
   alias_method :action_name, :method_name # for compatibility with controller libraries like Pundit that expect an action name
 
   delegate :connection, :stream_name, to: :channel
   delegate :controller_class, :flash, :session, to: :request
-  delegate :broadcast, :halted, :error, to: :broadcaster
-  delegate :reflex_id, :tab_id, :reflex_controller, :xpath_controller, :xpath_element, :permanent_attribute_name, :version, :suppress_logging, to: :client_attributes
+  delegate :broadcast, :broadcast_halt, :broadcast_forbid, :broadcast_error, to: :broadcaster
+  # TODO remove xpath_controller and xpath_element for v4
+  delegate :id, :tab_id, :reflex_controller, :xpath_controller, :xpath_element, :permanent_attribute_name, :version, :npm_version, :suppress_logging, to: :client_attributes
 
   def initialize(channel, url: nil, element: nil, selectors: [], method_name: nil, params: {}, client_attributes: {})
-    if is_a? CableReady::Broadcaster
-      message = <<~MSG
-
-        #{self.class.name} includes CableReady::Broadcaster, and you need to remove it.
-        Reflexes have their own CableReady interface. You can just assume that it's present.
-        See https://docs.stimulusreflex.com/rtfm/cableready#using-cableready-inside-a-reflex-action for more details.
-
-      MSG
-      raise TypeError.new(message.strip)
-    end
-
     @channel = channel
     @url = url
     @element = element
     @selectors = selectors
     @method_name = method_name
     @params = params
-    @broadcaster = StimulusReflex::PageBroadcaster.new(self)
     @client_attributes = ClientAttributes.new(client_attributes)
+    @broadcaster = StimulusReflex::PageBroadcaster.new(self)
     @logger = suppress_logging ? nil : StimulusReflex::Logger.new(self)
-    @cable_ready = StimulusReflex::CableReadyChannels.new(stream_name, reflex_id)
     @payload = {}
     @headers = {}
 
-    if version != StimulusReflex::VERSION && StimulusReflex.config.on_failed_sanity_checks != :ignore
-      raise VersionMismatchError.new("stimulus_reflex gem / NPM package version mismatch")
-    end
+    check_version!
 
     self.params
   end
 
+  # TODO: remove this for v4
+  def reflex_id
+    warn "Deprecation warning: reflex_id will be removed in v4. Use id instead!" if Rails.env.development?
+    id
+  end
+  # END TODO: remove
+
   def request
     @request ||= begin
       uri = URI.parse(url)
@@ -94,10 +103,10 @@ class StimulusReflex::Reflex
     when :page
       raise StandardError.new("Cannot call :page morph after :#{broadcaster.to_sym} morph") unless broadcaster.page?
     when :nothing
-      raise StandardError.new("Cannot call :nothing morph after :selector morph") if broadcaster.selector?
+      raise StandardError.new("#{broadcaster.to_sym} morph type has already been set") if broadcaster.selector?
       @broadcaster = StimulusReflex::NothingBroadcaster.new(self) unless broadcaster.nothing?
     else
-      raise StandardError.new("Cannot call :selector morph after :nothing morph") if broadcaster.nothing?
+      raise StandardError.new("#{broadcaster.to_sym} morph type has already been set") if broadcaster.nothing?
       @broadcaster = StimulusReflex::SelectorBroadcaster.new(self) unless broadcaster.selector?
       broadcaster.append_morph(selectors, html)
     end
@@ -128,12 +137,7 @@ class StimulusReflex::Reflex
 
   # Invoke the reflex action specified by `name` and run all callbacks
   def process(name, *args)
-    reflex_invoked = false
-    result = run_callbacks(:process) {
-      public_send(name, *args).tap { reflex_invoked = true }
-    }
-    @halted ||= result == false && !reflex_invoked
-    result
+    run_callbacks(:process) { public_send(name, *args) }
   end
 
   # Indicates if the callback chain was halted via a throw(:abort) in a before_reflex callback.
@@ -143,6 +147,11 @@ class StimulusReflex::Reflex
     !!@halted
   end
 
+  # Indicates if the callback chain was halted via a throw(:forbidden) in a before_reflex callback.
+  def forbidden?
+    !!@forbidden
+  end
+
   def default_reflex
     # noop default reflex to force page reloads
   end

data/lib/stimulus_reflex/reflex_data.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 class StimulusReflex::ReflexData
   attr_reader :data
 
@@ -49,18 +51,27 @@ class StimulusReflex::ReflexData
     Rack::Utils.parse_nested_query(data["formData"])
   end
 
+  def params
+    form_params.deep_merge(url_params)
+  end
+
   def form_params
     form_data.deep_merge(data["params"] || {})
   end
 
-  def reflex_id
-    data["reflexId"]
+  def url_params
+    Rack::Utils.parse_nested_query(URI.parse(url).query)
+  end
+
+  def id
+    data["id"]
   end
 
   def tab_id
     data["tabId"]
   end
 
+  # TODO: remove this in v4
   def xpath_controller
     data["xpathController"]
   end
@@ -68,13 +79,18 @@ class StimulusReflex::ReflexData
   def xpath_element
     data["xpathElement"]
   end
+  # END TODO remove
 
   def reflex_controller
     data["reflexController"]
   end
 
+  def npm_version
+    data["version"].to_s
+  end
+
   def version
-    data["version"].gsub("-pre", ".pre")
+    data["version"].to_s.gsub("-pre", ".pre")
   end
 
   private

data/lib/stimulus_reflex/reflex_factory.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 class StimulusReflex::ReflexFactory
   class << self
     attr_reader :reflex_data
@@ -9,16 +11,17 @@ class StimulusReflex::ReflexFactory
         element: reflex_data.element,
         selectors: reflex_data.selectors,
         method_name: reflex_data.method_name,
-        params: reflex_data.form_params,
+        params: reflex_data.params,
         client_attributes: {
-          reflex_id: reflex_data.reflex_id,
+          id: reflex_data.id,
           tab_id: reflex_data.tab_id,
           xpath_controller: reflex_data.xpath_controller,
           xpath_element: reflex_data.xpath_element,
           reflex_controller: reflex_data.reflex_controller,
           permanent_attribute_name: reflex_data.permanent_attribute_name,
           suppress_logging: reflex_data.suppress_logging,
-          version: reflex_data.version
+          version: reflex_data.version,
+          npm_version: reflex_data.npm_version
         })
     end
 

data/lib/stimulus_reflex/request_parameters.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module StimulusReflex
   class RequestParameters
     def initialize(params:, req:, url:)

data/lib/stimulus_reflex/utils/logger.rb

@@ -46,6 +46,15 @@ module StimulusReflex
       reflex.class.to_s + "#" + reflex.method_name
     end
 
+    def id_full
+      reflex.id
+    end
+
+    def id
+      id_full[0..7]
+    end
+
+    # TODO: remove for v4
     def reflex_id_full
       reflex.reflex_id
     end
@@ -53,6 +62,7 @@ module StimulusReflex
     def reflex_id
       reflex_id_full[0..7]
     end
+    # END TODO remove
 
     def mode
       reflex.broadcaster.to_s

data/lib/stimulus_reflex/utils/sanity_checker.rb

@@ -1,8 +1,6 @@
 # frozen_string_literal: true
 
 class StimulusReflex::SanityChecker
-  LATEST_VERSION_FORMAT = /^(\d+\.\d+\.\d+)$/
-
   class << self
     def check!
       return if ENV["SKIP_SANITY_CHECK"]
@@ -14,7 +12,6 @@ class StimulusReflex::SanityChecker
       instance = new
       instance.check_caching_enabled
       # instance.check_default_url_config
-      instance.check_new_version_available
     end
 
     private
@@ -48,7 +45,7 @@ class StimulusReflex::SanityChecker
     if using_null_store?
       warn_and_exit <<~WARN
         👉 StimulusReflex requires caching to be enabled.
-        
+
         Caching allows the session to be modified during ActionCable requests. Your config.cache_store is set to :null_store, so it won't work.
       WARN
     end
@@ -56,7 +53,7 @@ class StimulusReflex::SanityChecker
 
   def check_default_url_config
     return if StimulusReflex.config.on_missing_default_urls == :ignore
-    if default_url_config_set? == false
+    if default_url_config_missing?
       puts <<~WARN
         👉 StimulusReflex strongly suggests that you set default_url_options in your environment files. Otherwise, ActionController #{"and ActionMailer " if defined?(ActionMailer)}will default to example.com when rendering route helpers.
 
@@ -70,29 +67,6 @@ class StimulusReflex::SanityChecker
     end
   end
 
-  def check_new_version_available
-    return if StimulusReflex.config.on_new_version_available == :ignore
-    return if Rails.env.development? == false
-    return if using_preview_release?
-    begin
-      latest_version = URI.open("https://raw.githubusercontent.com/stimulusreflex/stimulus_reflex/master/LATEST", open_timeout: 1, read_timeout: 1).read.strip
-      if latest_version != StimulusReflex::VERSION
-        puts <<~WARN
-
-          👉 There is a new version of StimulusReflex available!
-          Current: #{StimulusReflex::VERSION} Latest: #{latest_version}
-
-          If you upgrade, it is very important that you update BOTH Gemfile and package.json
-          Then, run `bundle install && yarn install` to update to #{latest_version}.
-
-        WARN
-        exit if StimulusReflex.config.on_new_version_available == :exit
-      end
-    rescue
-      puts "👉 StimulusReflex #{StimulusReflex::VERSION} update check skipped: connection timeout"
-    end
-  end
-
   def caching_not_enabled?
     Rails.application.config.action_controller.perform_caching == false
   end
@@ -101,24 +75,18 @@ class StimulusReflex::SanityChecker
     Rails.application.config.cache_store == :null_store
   end
 
+  def initializer_missing?
+    File.exist?(Rails.root.join("config", "initializers", "stimulus_reflex.rb")) == false
+  end
+
   def default_url_config_set?
     if defined?(ActionMailer)
-      Rails.application.config.action_controller.default_url_options.blank? && Rails.application.config.action_mailer.default_url_options.blank?
+      Rails.application.config.action_controller.default_url_options.blank? || Rails.application.config.action_mailer.default_url_options.blank?
     else
       Rails.application.config.action_controller.default_url_options.blank?
     end
   end
 
-  def using_preview_release?
-    preview = StimulusReflex::VERSION.match?(LATEST_VERSION_FORMAT) == false
-    puts "👉 StimulusReflex #{StimulusReflex::VERSION} update check skipped: pre-release build" if preview
-    preview
-  end
-
-  def initializer_missing?
-    File.exist?(Rails.root.join("config", "initializers", "stimulus_reflex.rb")) == false
-  end
-
   def warn_and_exit(text)
     puts
     puts "Heads up! 🔥"
@@ -138,15 +106,7 @@ class StimulusReflex::SanityChecker
           end
 
       INFO
-      if initializer_missing?
-        puts <<~INFO
-          You can create a StimulusReflex initializer with the command:
-
-            bundle exec rails generate stimulus_reflex:initializer
-
-        INFO
-      end
-      exit false if Rails.env.test? == false
+      exit false unless Rails.env.test?
     end
   end
 end

data/lib/stimulus_reflex/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module StimulusReflex
-  VERSION = "3.5.0.pre9"
+  VERSION = "3.5.0.rc1"
 end

data/lib/stimulus_reflex/version_checker.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+module StimulusReflex
+  class VersionMismatchError < StandardError
+  end
+
+  module VersionChecker
+    def check_version!
+      return if StimulusReflex.config.on_failed_sanity_checks == :ignore
+      return if version == StimulusReflex::VERSION
+
+      level = (StimulusReflex.config.on_failed_sanity_checks == :exit) ? "error" : "warn"
+      reason = (level == "error") ? "failed to execute your reflex action due to" : "noticed"
+
+      mismatch = "StimulusReflex #{reason} a version mismatch between your gem and JavaScript version. Package versions must match exactly.\n\nstimulus_reflex gem: #{StimulusReflex::VERSION}\nstimulus_reflex npm: #{npm_version}"
+
+      StimulusReflex.config.logger.error("\n\e[31m#{mismatch}\e[0m")
+
+      log = {
+        message: mismatch,
+        level: level,
+        reflexId: id
+      }
+
+      event = {
+        name: "stimulus-reflex:version-mismatch",
+        reflexId: id,
+        detail: {
+          message: mismatch,
+          gem: StimulusReflex::VERSION,
+          npm: npm_version,
+          level: level
+        }
+      }
+
+      toast = {
+        text: mismatch.to_s,
+        destination: "https://docs.stimulusreflex.com/hello-world/setup#upgrading-package-versions-and-sanity",
+        reflexId: id,
+        level: level
+      }
+
+      CableReady::Channels.instance[@channel.stream_name].tap { |channel|
+        channel.console_log(log)
+        channel.dispatch_event(event)
+        channel.stimulus_reflex_version_mismatch(toast) if Rails.env.development?
+      }.broadcast
+
+      return if level == "warn"
+
+      raise VersionMismatchError.new(mismatch)
+    end
+  end
+end

data/lib/stimulus_reflex.rb

@@ -15,6 +15,8 @@ require "stimulus_reflex/cable_ready_channels"
 require "stimulus_reflex/concern_enhancer"
 require "stimulus_reflex/configuration"
 require "stimulus_reflex/callbacks"
+require "stimulus_reflex/html/document"
+require "stimulus_reflex/html/document_fragment"
 require "stimulus_reflex/request_parameters"
 require "stimulus_reflex/reflex"
 require "stimulus_reflex/reflex_data"

data/lib/tasks/stimulus_reflex/stimulus_reflex.rake

@@ -0,0 +1,250 @@
+# frozen_string_literal: true
+
+SR_STEPS = {
+  "action_cable" => "Action Cable",
+  "webpacker" => "Install StimulusReflex using Webpacker",
+  "shakapacker" => "Install StimulusReflex using Shakapacker",
+  "npm_packages" => "StimulusReflex and CableReady and related npm packages",
+  "reflexes" => "Create app/reflexes and related files",
+  "importmap" => "Install StimulusReflex using importmaps",
+  "esbuild" => "Install StimulusReflex using esbuild",
+  "config" => "Client initialization",
+  "initializers" => "StimulusReflex and CableReady initializers",
+  "example" => "Create an Example Reflex",
+  "development" => "development environment configuration",
+  "spring" => "Disable spring gem. Spring has been removed from Rails 7",
+  "mrujs" => "Swap out Rails UJS for mrujs",
+  "broadcaster" => "Make CableReady available to channels, controllers, jobs and models",
+  "updatable" => "Include CableReady::Updatable in Active Record model classes",
+  "yarn" => "Resolve npm dependency changes",
+  "bundle" => "Resolve gem dependency changes and install configuration changes",
+  "vite" => "Install StimulusReflex using Vite",
+  "compression" => "Compress WebSocket traffic with gzip"
+}
+
+SR_BUNDLERS = {
+  "webpacker" => ["npm_packages", "webpacker", "config", "action_cable", "reflexes", "development", "initializers", "broadcaster", "updatable", "example", "spring", "yarn", "bundle"],
+  "esbuild" => ["npm_packages", "esbuild", "config", "action_cable", "reflexes", "development", "initializers", "broadcaster", "updatable", "example", "spring", "yarn", "bundle"],
+  "vite" => ["npm_packages", "vite", "config", "action_cable", "reflexes", "development", "initializers", "broadcaster", "updatable", "example", "spring", "yarn", "bundle"],
+  "shakapacker" => ["npm_packages", "shakapacker", "config", "action_cable", "reflexes", "development", "initializers", "broadcaster", "updatable", "example", "spring", "yarn", "bundle"],
+  "importmap" => ["config", "action_cable", "importmap", "reflexes", "development", "initializers", "broadcaster", "updatable", "example", "spring", "bundle"]
+}
+
+def run_install_template(template, force: false, trace: false)
+  puts "--- [#{template}] ----"
+
+  if Rails.root.join("tmp/stimulus_reflex_installer/halt").exist?
+    FileUtils.rm(Rails.root.join("tmp/stimulus_reflex_installer/halt"))
+    puts "StimulusReflex installation halted. Please fix the issues above and try again."
+    exit
+  end
+  if Rails.root.join("tmp/stimulus_reflex_installer/#{template}").exist? && !force
+    puts "👍 #{SR_STEPS[template]}"
+    return
+  end
+
+  system "#{RbConfig.ruby} ./bin/rails app:template LOCATION=#{File.expand_path("../../install/#{template}.rb", __dir__)} SKIP_SANITY_CHECK=true #{"--trace" if trace}"
+
+  puts
+end
+
+namespace :stimulus_reflex do
+  desc "✨ Install StimulusReflex and CableReady ✨"
+  task :install do
+    FileUtils.mkdir_p(Rails.root.join("tmp/stimulus_reflex_installer/templates"))
+    FileUtils.mkdir_p(Rails.root.join("tmp/stimulus_reflex_installer/working"))
+    install_complete = Rails.root.join("tmp/stimulus_reflex_installer/complete")
+
+    bundler = nil
+    options = {}
+
+    ARGV.each do |arg|
+      # make sure we have a valid build tool specified, or proceed to automatic detection
+      if ["webpacker", "esbuild", "vite", "shakapacker", "importmap"].include?(arg)
+        bundler = arg
+      else
+        kv = arg.split("=")
+        if kv.length == 2
+          kv[1] = if kv[1] == "true"
+            true
+          else
+            (kv[1] == "false") ? false : kv[1]
+          end
+          options[kv[0]] = kv[1]
+        end
+      end
+    end
+
+    options_path = Rails.root.join("tmp/stimulus_reflex_installer/options")
+    options_path.write(options.to_yaml)
+
+    if install_complete.exist?
+      puts "✨ \e[38;5;220mStimulusReflex\e[0m and \e[38;5;220mCableReady\e[0m are already installed ✨"
+      puts
+      puts "To restart the installation process, run: \e[38;5;231mrails stimulus_reflex:install:restart\e[0m"
+      puts
+      puts "To get started, check out \e[4;97mhttps://docs.stimulusreflex.com/hello-world/quickstart\e[0m"
+      puts "or get help on Discord: \e[4;97mhttps://discord.gg/stimulus-reflex\e[0m. \e[38;5;196mWe are here for you.\e[0m 💙"
+      puts
+      exit
+    end
+
+    # if there is an installation in progress, continue where we left off
+    cached_entrypoint = Rails.root.join("tmp/stimulus_reflex_installer/entrypoint")
+    if cached_entrypoint.exist?
+      entrypoint = File.read(cached_entrypoint)
+      puts "✨ Resuming \e[38;5;220mStimulusReflex\e[0m and \e[38;5;220mCableReady\e[0m installation ✨"
+      puts
+      puts "If you have any setup issues, please consult \e[4;97mhttps://docs.stimulusreflex.com/hello-world/setup\e[0m"
+      puts "or get help on Discord: \e[4;97mhttps://discord.gg/stimulus-reflex\e[0m. \e[38;5;196mWe are here for you.\e[0m 💙"
+      puts
+      puts "Resuming installation into \e[1m#{entrypoint}\e[22m"
+      puts "Run \e[1;94mrails stimulus_reflex:install:restart\e[0m to restart the installation process"
+      puts
+    else
+      puts "✨ Installing \e[38;5;220mStimulusReflex\e[0m and \e[38;5;220mCableReady\e[0m ✨"
+      puts
+      puts "If you have any setup issues, please consult \e[4;97mhttps://docs.stimulusreflex.com/hello-world/setup\e[0m"
+      puts "or get help on Discord: \e[4;97mhttps://discord.gg/stimulus-reflex\e[0m. \e[38;5;196mWe are here for you.\e[0m 💙"
+      if Rails.root.join(".git").exist?
+        puts
+        puts "We recommend running \e[1;94mgit commit\e[0m before proceeding. A diff will be generated at the end."
+      end
+
+      if options.key? "entrypoint"
+        entrypoint = options["entrypoint"]
+      else
+        entrypoint = [
+          "app/javascript",
+          "app/frontend"
+        ].find { |path| File.exist?(Rails.root.join(path)) } || "app/javascript"
+
+        puts
+        puts "Where do JavaScript files live in your app? Our best guess is: \e[1m#{entrypoint}\e[22m 🤔"
+        puts "Press enter to accept this, or type a different path."
+        print "> "
+        input = $stdin.gets.chomp
+        entrypoint = input unless input.blank?
+      end
+      File.write(cached_entrypoint, entrypoint)
+    end
+
+    # verify their bundler before starting, unless they explicitly specified on CLI
+    if !bundler
+      # auto-detect build tool based on existing packages and configuration
+      if Rails.root.join("config/importmap.rb").exist?
+        bundler = "importmap"
+      elsif Rails.root.join("package.json").exist?
+        package_json = File.read(Rails.root.join("package.json"))
+        bundler = "webpacker" if package_json.include?('"@rails/webpacker":')
+        bundler = "esbuild" if package_json.include?('"esbuild":')
+        bundler = "vite" if package_json.include?('"vite":')
+        bundler = "shakapacker" if package_json.include?('"shakapacker":')
+        if !bundler
+          puts "❌ You must be using a node-based bundler such as esbuild, webpacker, vite or shakapacker (package.json) or importmap (config/importmap.rb) to use StimulusReflex."
+          exit
+        end
+      else
+        puts "❌ You must be using a node-based bundler such as esbuild, webpacker, vite or shakapacker (package.json) or importmap (config/importmap.rb) to use StimulusReflex."
+        exit
+      end
+
+      puts
+      puts "It looks like you're using \e[1m#{bundler}\e[22m as your bundler. Is that correct? (Y/n)"
+      print "> "
+      input = $stdin.gets.chomp
+      if input.downcase == "n"
+        puts
+        puts "StimulusReflex installation supports: esbuild, webpacker, vite, shakapacker and importmap."
+        puts "Please run \e[1;94mrails stimulus_reflex:install [bundler]\e[0m to install StimulusReflex and CableReady."
+        exit
+      end
+    end
+
+    File.write("tmp/stimulus_reflex_installer/bundler", bundler)
+    FileUtils.touch("tmp/stimulus_reflex_installer/backups")
+    File.write("tmp/stimulus_reflex_installer/template_src", File.expand_path("../../generators/stimulus_reflex/templates/", __dir__))
+
+    `bin/spring stop` if defined?(Spring)
+
+    # do the things
+    SR_BUNDLERS[bundler].each do |template|
+      run_install_template(template, trace: !!options["trace"])
+    end
+
+    puts
+    puts "🎉 \e[1;92mStimulusReflex and CableReady have been successfully installed!\e[22m 🎉"
+    puts
+    puts "👉 \e[4;97mhttps://docs.stimulusreflex.com/hello-world/quickstart\e[0m"
+    puts
+    puts "Join over 2000 StimulusReflex developers on Discord: \e[4;97mhttps://discord.gg/stimulus-reflex\e[0m"
+    puts
+
+    backups = File.readlines("tmp/stimulus_reflex_installer/backups").map(&:chomp)
+
+    if backups.any?
+      puts "🙆 The following files were modified during installation:"
+      puts
+      backups.each { |backup| puts "  #{backup}" }
+      puts
+      puts "Each of these files has been backed up with a .bak extension. Please review the changes carefully."
+      puts "If you're happy with the changes, you can delete the .bak files."
+      puts
+    end
+
+    if Rails.root.join(".git").exist?
+      system "git diff > tmp/stimulus_reflex_installer.diff"
+      puts "🏮 A diff of all changes has been saved to \e[1mtmp/stimulus_reflex_installer.diff\e[22m"
+      puts
+    end
+
+    if Rails.root.join("app/reflexes/example_reflex.rb").exist?
+      launch = Rails.root.join("bin/dev").exist? ? "bin/dev" : "rails s"
+      puts "🚀 Launch \e[1;94m#{launch}\e[0m to access the example at ⚡ \e[4;97mhttp://localhost:3000/example\e[0m ⚡"
+      puts "Once you're finished with the example, you can remove it with \e[1;94mrails destroy stimulus_reflex example\e[0m"
+      puts
+    end
+
+    FileUtils.touch(install_complete)
+    `pkill -f spring` if Rails.root.join("tmp/stimulus_reflex_installer/kill_spring").exist?
+    exit
+  end
+
+  namespace :install do
+    desc "Restart StimulusReflex and CableReady installation"
+    task :restart do
+      FileUtils.rm_rf Rails.root.join("tmp/stimulus_reflex_installer")
+      system "rails stimulus_reflex:install #{ARGV.join(" ")}"
+      exit
+    end
+
+    desc <<~DESC
+      Run specific StimulusReflex install steps
+
+      #{SR_STEPS.sort.map { |step, description| "#{step.ljust(20)} #{description}" }.join("\n")}
+    DESC
+    task :step do
+      def warning(step = nil, force_exit = true)
+        return if step.to_s.include?("=")
+        if step
+          puts "⚠️  #{step} is not a valid step. Valid steps are: #{SR_STEPS.keys.join(", ")}"
+        else
+          puts "❌ You must specify a step to re-run. Valid steps are: #{SR_STEPS.keys.join(", ")}"
+          puts "Example: \e[1;94mrails stimulus_reflex:install:step initializers\e[0m"
+        end
+        exit if force_exit
+      end
+
+      warning if ARGV.empty?
+
+      ARGV.each do |step|
+        SR_STEPS.include?(step) ? run_install_template(step, force: true) : warning(step, false)
+      end
+
+      run_install_template(:bundle, force: true)
+      run_install_template(:yarn, force: true)
+
+      exit
+    end
+  end
+end