rails-worktrees 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fc7ddb67fbc5fde18c04ab164229207c6f54284dc4bb5c1fb955f40544706cbb
-  data.tar.gz: a4005286b6c1dbad990d94ca3999b1490f2532c981fa8e5a7fb9b5768e4f30d6
+  metadata.gz: 5cf1debd660652a5429a2d5485e944006685268b423866a96e15af71f9cb3471
+  data.tar.gz: 14f6a3f5efaa6b829026b3d4d085c9b55d09236cb7b610bafa8f8834a534fb5b
 SHA512:
-  metadata.gz: 8eef6c43ad39e5fc38c3dbdc10fc2618fc07f0ccd65c5c25bef94ae3991aeff143687ee45a37983846cc7ec8fc4c031d39b2b4cd6abcdd37d31243bb705a24a0
-  data.tar.gz: dfdeeb6c73a81e8c3c8147883b5ca752b79ff4a63e318db409792c6723c4fd84ae3f9effc93edc1d456f74fa22f326e954ac7239d044a27709436d1e462810b6
+  metadata.gz: e16d066bfb55c97b959e92b4c25d1b2ac08404a728cadf077852a937a1a4c00ea17ca0c08dcfecc571654f836ba0aeccc6f81b08b6a96f7a4443093681d3b25d
+  data.tar.gz: f7fdc3e954463fed68f62934d153fbe4b086fb86f4be97c8f8e172b54c0bcc4d402587b99c2b171cc024f7ef53be14e950da311399e49ef0a491d696ac15dc64

data/.release-please-manifest.json

@@ -1 +1 @@
-{".":"0.1.0"}
+{".":"0.2.0"}

data/CHANGELOG.md

@@ -1,5 +1,23 @@
 # Changelog
 
+## [0.2.0](https://github.com/asjer/rails-worktrees/compare/v0.1.1...v0.2.0) (2026-03-30)
+
+
+### Features
+
+* add `--yolo` mode to install common follow-ups without manual edits ([1ec82ec](https://github.com/asjer/rails-worktrees/commit/1ec82ec83726b5ca9cbaee39697c607fdb825f26))
+
+## [0.1.1](https://github.com/asjer/rails-worktrees/compare/v0.1.0...v0.1.1) (2026-03-30)
+
+
+### Bug Fixes
+
+* **bin/wt:** remove unnecessary frozen_string_literal comment ([e3752c8](https://github.com/asjer/rails-worktrees/commit/e3752c86e4003b80e401253e3c07bd0107ba1514))
+* **ci:** update gem installation steps to prevent `Gemfile.lock` freeze ([e3c75e8](https://github.com/asjer/rails-worktrees/commit/e3c75e85528e575185adb417b6bd9efa995087f1))
+* **frozen-string-literal:** remove unnecessary frozen_string_literal comments ([a8583d2](https://github.com/asjer/rails-worktrees/commit/a8583d2bab1097a6c6a1906a3d6f6092d7b8d867))
+* **generator:** update generator command to use shorter namespace ([cef728e](https://github.com/asjer/rails-worktrees/commit/cef728e3642c36da054f0d53ab5d8a932b2511c4))
+* **installation:** show setup instructions on boot when generator hasn't run ([af31a3f](https://github.com/asjer/rails-worktrees/commit/af31a3f3849408cf7098e2a4e90a9246c89b0bbe))
+
 ## 0.1.0 (2026-03-30)
 
 

data/README.md

@@ -12,7 +12,9 @@
 
 ```bash
 bundle add rails-worktrees
-bin/rails generate rails:worktrees:install
+bin/rails generate worktrees:install
+# or, to apply the common Procfile.dev + mise follow-ups automatically:
+bin/rails generate worktrees:install --yolo
 ```
 
 The installer adds:
@@ -22,6 +24,11 @@ The installer adds:
 - `Procfile.dev.worktree.example` — a copy-paste helper for `${DEV_PORT:-3000}` in `Procfile.dev`
 - a safe update to `config/database.yml` for common development/test database names
 
+With `--yolo`, the installer also:
+
+- replaces the existing `web:` entry in `Procfile.dev` with the DEV_PORT-aware command when `Procfile.dev` already exists
+- updates `mise.toml` or `.mise.toml` to load `.env` from `[env]` when either file already exists
+
 ## Usage
 
 ```bash
@@ -115,12 +122,17 @@ When `bin/wt` creates a worktree it writes a worktree-local `.env` with:
 
 Existing `.env` values are never overwritten.
 
-The gem does **not** edit your `Procfile.dev` or add dotenv. The installer generates `Procfile.dev.worktree.example` with a ready-to-copy line:
+By default, the installer does **not** edit your `Procfile.dev` or `mise` config. It generates `Procfile.dev.worktree.example` with a ready-to-copy line:
 
 ```text
 web: env RUBY_DEBUG_OPEN=true bin/rails server -b 0.0.0.0 -p ${DEV_PORT:-3000}
 ```
 
+If you run `bin/rails generate worktrees:install --yolo`, the installer applies the two common follow-ups for you when the files already exist:
+
+- replace the existing `web:` entry in `Procfile.dev`
+- add `_.file = ".env"` to the `[env]` section of `mise.toml` or `.mise.toml`
+
 Use a project-local env loader like `mise` with `_.file = ".env"` to keep values scoped per-worktree.
 
 ## Development
@@ -141,8 +153,8 @@ This smoke test:
 
 - creates a temporary Rails app from a compatible Rails version
 - installs `rails-worktrees` from the current checkout path
-- runs `bin/rails generate rails:worktrees:install`
-- verifies `bin/wt`, the generated initializer, the Procfile example, `config/database.yml` patching, and worktree `.env` bootstrapping
+- runs `bin/rails generate worktrees:install --yolo`
+- verifies `bin/wt`, the generated initializer, the Procfile example, yolo updates to `Procfile.dev` and `mise.toml`, `config/database.yml` patching, and worktree `.env` bootstrapping
 - creates a temporary bare `origin` and confirms `bin/wt smoke-branch` creates a real worktree
 
 By default, the script cleans up all temp directories after the run. Set `KEEP_SMOKE_TEST_ARTIFACTS=1` to keep them around for debugging, or set `RAILS_WORKTREES_SMOKE_RAILS_VERSION` to try a different compatible Rails version.

data/Rakefile

@@ -1,5 +1,3 @@
-# frozen_string_literal: true
-
 require 'bundler/gem_tasks'
 require 'rspec/core/rake_task'
 

data/exe/wt

@@ -1,6 +1,4 @@
 #!/usr/bin/env ruby
-# frozen_string_literal: true
-
 lib = File.expand_path('../lib', __dir__)
 $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 

data/lib/generators/rails/worktrees/mise_follow_up.rb

@@ -1,5 +1,3 @@
-# frozen_string_literal: true
-
 module Rails
   module Worktrees
     module Generators
@@ -13,7 +11,8 @@ module Rails
           [
             '',
             '      Tip:',
-            '        Detected mise.toml. To auto-load the worktree-local .env when you enter a worktree,',
+            "        Detected #{File.basename(mise_toml_path)}. To auto-load the worktree-local .env when",
+            '        you enter a worktree,',
             '        consider adding:',
             '          [env]',
             '          _.file = ".env"'
@@ -21,15 +20,24 @@ module Rails
         end
 
         def suggest_mise_env_file?
-          File.file?(mise_toml_path) && !mise_env_file_configured?
+          mise_toml_path && !mise_env_file_configured?
         end
 
         def mise_env_file_configured?
-          File.read(mise_toml_path).match?(/^\s*_.file\s*=\s*["']\.env["']\s*$/)
+          return false unless mise_toml_path
+
+          File.read(mise_toml_path).match?(/^\s*_.file\s*=\s*["']\.env["']\s*(?:#.*)?\s*$/)
         end
 
         def mise_toml_path
-          File.join(destination_root, 'mise.toml')
+          @mise_toml_path ||= mise_toml_paths.find { |path| File.file?(path) }
+        end
+
+        def mise_toml_paths
+          [
+            File.join(destination_root, 'mise.toml'),
+            File.join(destination_root, '.mise.toml')
+          ]
         end
       end
     end

data/lib/generators/rails/worktrees/templates/bin/wt

@@ -1,6 +1,4 @@
 #!/usr/bin/env ruby
-# frozen_string_literal: true
-
 ENV['BUNDLE_GEMFILE'] ||= File.expand_path('../Gemfile', __dir__)
 
 require 'bundler/setup'

data/lib/generators/rails/worktrees/templates/rails_worktrees.rb.tt

@@ -1,5 +1,3 @@
-# frozen_string_literal: true
-
 Rails::Worktrees.configure do |config|
 <% if options['conductor'] -%>
   config.workspace_root = <%= conductor_workspace_root %>

data/lib/generators/worktrees/install/install_generator.rb

@@ -0,0 +1,203 @@
+require 'open3'
+require 'rails/generators'
+
+require_relative '../../rails/worktrees/mise_follow_up'
+require_relative '../../../rails/worktrees/database_config_updater'
+require_relative '../../../rails/worktrees/procfile_updater'
+require_relative '../../../rails/worktrees/mise_toml_updater'
+
+module Worktrees
+  module Generators
+    # Installs the wt wrapper, configuration, and safe database.yml updates.
+    # rubocop:disable Metrics/ClassLength
+    class InstallGenerator < ::Rails::Generators::Base
+      include ::Rails::Worktrees::Generators::MiseFollowUp
+
+      namespace 'worktrees:install'
+      desc 'Installs bin/wt, a Rails::Worktrees initializer, and updates config/database.yml when safe.'
+      source_root File.expand_path('../../rails/worktrees/templates', __dir__)
+      class_option :conductor, type: :boolean, default: false,
+                               desc: 'Configure the installer for ~/Sites/conductor/workspaces'
+      class_option :yolo, type: :boolean, default: false,
+                          desc: 'Apply common Procfile.dev and mise .env follow-up edits when safe'
+
+      FOLLOW_UP_TEMPLATE = <<~TEXT.freeze
+          ============================================
+            rails-worktrees installed successfully! 🚂
+
+            Installed:
+        %<installed>s
+
+            Get started:
+              $ bin/wt
+              $ bin/wt my-feature
+
+            Configure:
+              config/initializers/rails_worktrees.rb
+        %<notes>s
+          ============================================
+      TEXT
+
+      def create_bin_wrapper
+        template('bin/wt', 'bin/wt')
+        chmod('bin/wt', 0o755)
+      end
+
+      def create_initializer
+        template('rails_worktrees.rb.tt', 'config/initializers/rails_worktrees.rb')
+      end
+
+      def create_procfile_worktree_example
+        template('Procfile.dev.worktree.example.tt', 'Procfile.dev.worktree.example')
+      end
+
+      def apply_yolo_follow_ups
+        return unless options[:yolo]
+
+        update_procfile
+        update_mise_toml
+      end
+
+      def update_database_configuration
+        unless File.exist?(database_config_path)
+          say_status(:skip, 'config/database.yml not found', :yellow)
+          @database_outcome = :not_found
+          return
+        end
+
+        result = database_update_result
+        @database_outcome = result.changed? ? :updated : :identical
+        announce_database_update(result)
+      end
+
+      def verify_installation
+        if git_repo?
+          say_status(:ok, 'git repository detected', :green)
+        else
+          say_status(:warning, 'run inside a git repository for bin/wt to work', :yellow)
+        end
+      end
+
+      def show_follow_up
+        say(follow_up_message)
+      end
+
+      private
+
+      def database_config_path
+        File.join(destination_root, 'config/database.yml')
+      end
+
+      def procfile_path
+        File.join(destination_root, 'Procfile.dev')
+      end
+
+      def mise_toml_paths
+        [
+          File.join(destination_root, 'mise.toml'),
+          File.join(destination_root, '.mise.toml')
+        ]
+      end
+
+      def database_update_result
+        result = ::Rails::Worktrees::DatabaseConfigUpdater.new(
+          content: File.read(database_config_path)
+        ).call
+
+        File.write(database_config_path, result.content) if result.changed?
+
+        result
+      end
+
+      def follow_up_message
+        "\n#{format(FOLLOW_UP_TEMPLATE, installed: installed_items_text, notes: follow_up_notes_text)}"
+      end
+
+      def installed_items_text
+        items = [
+          '      • bin/wt',
+          '      • config/initializers/rails_worktrees.rb',
+          '      • Procfile.dev.worktree.example'
+        ]
+        items << database_follow_up_line if database_follow_up_line
+        items.join("\n")
+      end
+
+      def database_follow_up_line
+        case @database_outcome
+        when :updated
+          '      • config/database.yml (updated with WORKTREE_DATABASE_SUFFIX)'
+        when :not_found
+          '      • config/database.yml was not found — see README for manual setup'
+        end
+      end
+
+      def announce_database_update(result)
+        status = result.changed? ? :update : :identical
+        color = result.changed? ? :green : :blue
+
+        say_status(status, 'config/database.yml', color)
+        result.messages.each { |message| say(message) }
+      end
+
+      def update_procfile
+        unless File.exist?(procfile_path)
+          say_status(:skip, 'Procfile.dev not found', :yellow)
+          say('Skipped Procfile.dev yolo update because the file does not exist yet.')
+          return
+        end
+
+        result = ::Rails::Worktrees::ProcfileUpdater.new(content: File.read(procfile_path)).call
+        File.write(procfile_path, result.content) if result.changed?
+        announce_updater_result('Procfile.dev', result)
+      end
+
+      def update_mise_toml
+        path = first_mise_toml_path
+        return announce_missing_mise_toml unless path
+
+        result = ::Rails::Worktrees::MiseTomlUpdater.new(
+          content: File.read(path),
+          file_name: File.basename(path)
+        ).call
+
+        File.write(path, result.content) if result.changed?
+        announce_updater_result(File.basename(path), result)
+      end
+
+      def announce_updater_result(path, result)
+        status, color = case result.status
+                        when :updated then %i[update green]
+                        when :identical then %i[identical blue]
+                        else %i[skip yellow]
+                        end
+
+        say_status(status, path, color)
+        result.messages.each { |message| say(message) }
+      end
+
+      def first_mise_toml_path
+        mise_toml_paths.find { |candidate| File.file?(candidate) }
+      end
+
+      def announce_missing_mise_toml
+        say_status(:skip, 'mise.toml/.mise.toml not found', :yellow)
+        say('Skipped mise yolo update because no supported mise config file was found.')
+      end
+
+      def git_repo?
+        _stdout_str, _stderr_str, status = Open3.capture3(
+          'git', 'rev-parse', '--is-inside-work-tree', chdir: destination_root
+        )
+        status.success?
+      rescue Errno::ENOENT
+        false
+      end
+
+      def conductor_workspace_root
+        "File.expand_path('~/Sites/conductor/workspaces')"
+      end
+    end
+    # rubocop:enable Metrics/ClassLength
+  end
+end

data/lib/rails/worktrees/cli.rb

@@ -1,5 +1,3 @@
-# frozen_string_literal: true
-
 module Rails
   module Worktrees
     # Shell entrypoint for the wt executable.

data/lib/rails/worktrees/command/environment_support.rb

@@ -1,5 +1,3 @@
-# frozen_string_literal: true
-
 module Rails
   module Worktrees
     class Command

data/lib/rails/worktrees/command/git_operations.rb

@@ -1,5 +1,3 @@
-# frozen_string_literal: true
-
 require 'open3'
 
 module Rails

data/lib/rails/worktrees/command/name_picking.rb

@@ -1,5 +1,3 @@
-# frozen_string_literal: true
-
 module Rails
   module Worktrees
     class Command

data/lib/rails/worktrees/command/output.rb

@@ -1,5 +1,3 @@
-# frozen_string_literal: true
-
 module Rails
   module Worktrees
     class Command

data/lib/rails/worktrees/command/workspace_paths.rb

@@ -1,5 +1,3 @@
-# frozen_string_literal: true
-
 module Rails
   module Worktrees
     class Command

data/lib/rails/worktrees/command.rb

@@ -1,5 +1,3 @@
-# frozen_string_literal: true
-
 require 'fileutils'
 require_relative 'command/environment_support'
 require_relative 'command/git_operations'

data/lib/rails/worktrees/configuration.rb

@@ -1,11 +1,9 @@
-# frozen_string_literal: true
-
 module Rails
   module Worktrees
     # Stores application-level settings for the wt command.
     class Configuration
       DEFAULT_BOOTSTRAP_ENV = true
-      DEFAULT_BRANCH_PREFIX = '🚂'
+      DEFAULT_BRANCH_PREFIX = '🚂'.freeze
       DEFAULT_DEV_PORT_RANGE = (3000..3999)
       DEFAULT_USED_NAMES_DIRECTORY = File.join(
         ENV.fetch('XDG_STATE_HOME', File.join(Dir.home, '.local/state')),

data/lib/rails/worktrees/database_config_updater.rb

@@ -1,5 +1,3 @@
-# frozen_string_literal: true
-
 module Rails
   module Worktrees
     # Safely patches common database.yml layouts for worktree suffixes.
@@ -10,7 +8,7 @@ module Rails
         end
       end
 
-      SUFFIX_TEMPLATE = "<%= ENV.fetch('WORKTREE_DATABASE_SUFFIX', '') %>"
+      SUFFIX_TEMPLATE = "<%= ENV.fetch('WORKTREE_DATABASE_SUFFIX', '') %>".freeze
       SUPPORTED_ENVIRONMENTS = %w[development test].freeze
       DATABASE_LINE_PATTERN = /\A(\s*database:\s*)(.+?)(\s*(?:#.*)?\n?)\z/
       SECTION_PATTERN = /\A([A-Za-z0-9_]+):(?:\s|$)/

data/lib/rails/worktrees/env_bootstrapper.rb

@@ -1,5 +1,3 @@
-# frozen_string_literal: true
-
 require 'pathname'
 require 'zlib'
 
@@ -14,7 +12,7 @@ module Rails
         attr_accessor :values
       end
 
-      ENV_FILE_NAME = '.env'
+      ENV_FILE_NAME = '.env'.freeze
 
       def initialize(target_dir:, worktree_name:, configuration:)
         @target_dir = target_dir

data/lib/rails/worktrees/mise_toml_updater.rb

@@ -0,0 +1,82 @@
+module Rails
+  module Worktrees
+    # Safely updates mise config to load the worktree-local .env.
+    class MiseTomlUpdater
+      Result = Struct.new(:content, :changed, :status, :messages) do
+        def changed?
+          changed
+        end
+      end
+
+      ENV_SECTION_PATTERN = /\A\s*\[env\]\s*(?:#.*)?\z/
+      ENV_FILE_PATTERN = /\A\s*_.file\s*=\s*["']\.env["']\s*(?:#.*)?\z/
+      SECTION_PATTERN = /\A\s*\[[^\]]+\]\s*(?:#.*)?\z/
+      ENV_FILE_ENTRY = '_.file = ".env"'.freeze
+
+      def initialize(content:, file_name:)
+        @content = content
+        @file_name = file_name
+      end
+
+      def call
+        lines = @content.lines(chomp: true)
+        return update_existing_env_section(lines) if env_section_present?(lines)
+
+        append_env_section(lines)
+      end
+
+      private
+
+      def update_existing_env_section(lines)
+        env_section_start, env_section_end = env_section_bounds(lines)
+        return identical_result if env_file_configured?(lines, env_section_start, env_section_end)
+
+        updated_lines = lines.dup
+        updated_lines.insert(env_section_end, ENV_FILE_ENTRY)
+        updated_result(rebuild_content(updated_lines, trailing_newline: @content.end_with?("\n")))
+      end
+
+      def append_env_section(lines)
+        updated_lines = lines.dup
+        updated_lines << '' if updated_lines.any? && !updated_lines.last.empty?
+        updated_lines << '[env]'
+        updated_lines << ENV_FILE_ENTRY
+
+        updated_result(rebuild_content(updated_lines, trailing_newline: true))
+      end
+
+      def updated_result(content)
+        Result.new(content, true, :updated, ["Configured #{@file_name} to load .env from [env]."])
+      end
+
+      def identical_result
+        Result.new(@content, false, :identical, ["#{@file_name} already loads .env from [env]."])
+      end
+
+      def env_section_bounds(lines)
+        start_index = lines.find_index { |line| line.match?(ENV_SECTION_PATTERN) }
+        return [nil, nil] unless start_index
+
+        end_index = ((start_index + 1)...lines.length).find do |index|
+          lines[index].match?(SECTION_PATTERN)
+        end || lines.length
+        [start_index, end_index]
+      end
+
+      def env_section_present?(lines)
+        env_section_bounds(lines).first
+      end
+
+      def env_file_configured?(lines, start_index, end_index)
+        lines[(start_index + 1)...end_index].any? { |line| line.match?(ENV_FILE_PATTERN) }
+      end
+
+      def rebuild_content(lines, trailing_newline:)
+        content = lines.join("\n")
+        return content if content.empty? || !trailing_newline
+
+        "#{content}\n"
+      end
+    end
+  end
+end

data/lib/rails/worktrees/procfile_updater.rb

@@ -0,0 +1,69 @@
+module Rails
+  module Worktrees
+    # Safely updates Procfile.dev to use the worktree-local DEV_PORT.
+    class ProcfileUpdater
+      Result = Struct.new(:content, :changed, :status, :messages) do
+        def changed?
+          changed
+        end
+      end
+
+      STANDARD_WEB_ENTRY = 'web: env RUBY_DEBUG_OPEN=true bin/rails server -b 0.0.0.0 -p ${DEV_PORT:-3000}'.freeze
+      WEB_ENTRY_PATTERN = /\Aweb:\s*.*\z/
+
+      def initialize(content:)
+        @content = content
+      end
+
+      def call
+        lines = @content.lines(chomp: true)
+        web_entry_indexes = web_entry_indexes(lines)
+        return skip_result if web_entry_indexes.empty?
+
+        updated_content = rebuild_content(
+          replace_web_entries(lines, web_entry_indexes),
+          trailing_newline: @content.end_with?("\n")
+        )
+
+        build_result(updated_content)
+      end
+
+      private
+
+      def updated_result(content)
+        Result.new(content, true, :updated, ['Updated Procfile.dev web entry to use DEV_PORT.'])
+      end
+
+      def identical_result(content)
+        Result.new(content, false, :identical, ['Procfile.dev already uses the DEV_PORT-aware web entry.'])
+      end
+
+      def skip_result
+        Result.new(@content, false, :skip, ['No web: entry found in Procfile.dev; update it manually if needed.'])
+      end
+
+      def web_entry_indexes(lines)
+        lines.each_index.select { |index| lines[index].match?(WEB_ENTRY_PATTERN) }
+      end
+
+      def replace_web_entries(lines, web_entry_indexes)
+        lines.dup.tap do |updated_lines|
+          web_entry_indexes.each { |index| updated_lines[index] = STANDARD_WEB_ENTRY }
+        end
+      end
+
+      def build_result(content)
+        return identical_result(content) if content == @content
+
+        updated_result(content)
+      end
+
+      def rebuild_content(lines, trailing_newline:)
+        content = lines.join("\n")
+        return content if content.empty? || !trailing_newline
+
+        "#{content}\n"
+      end
+    end
+  end
+end

data/lib/rails/worktrees/railtie.rb

@@ -1,8 +1,10 @@
-# frozen_string_literal: true
-
 module Rails
   module Worktrees
+    # Hooks install guidance into Rails boot when the generator has not run yet.
     class Railtie < ::Rails::Railtie
+      initializer 'rails_worktrees.installation_hint' do
+        Rails::Worktrees.warn_about_missing_installation
+      end
     end
   end
 end

data/lib/rails/worktrees/version.rb

@@ -1,7 +1,5 @@
-# frozen_string_literal: true
-
 module Rails
   module Worktrees
-    VERSION = '0.1.0'
+    VERSION = '0.2.0'.freeze
   end
 end

data/lib/rails/worktrees.rb

@@ -1,4 +1,4 @@
-# frozen_string_literal: true
+require 'pathname'
 
 require_relative 'worktrees/version'
 require_relative 'worktrees/configuration'
@@ -6,12 +6,21 @@ require_relative 'worktrees/env_bootstrapper'
 require_relative 'worktrees/command'
 require_relative 'worktrees/cli'
 require_relative 'worktrees/database_config_updater'
+require_relative 'worktrees/procfile_updater'
+require_relative 'worktrees/mise_toml_updater'
 
 module Rails
   # Rails-specific git worktree helpers and installer support.
   module Worktrees
     class Error < StandardError; end
 
+    INSTALL_GENERATOR_COMMAND = 'bin/rails generate worktrees:install'.freeze
+    INSTALL_GENERATOR_NAMES = %w[worktrees:install].freeze
+    REQUIRED_INSTALLATION_PATHS = [
+      'bin/wt',
+      'config/initializers/rails_worktrees.rb'
+    ].freeze
+
     class << self
       def configuration
         @configuration ||= Configuration.new
@@ -25,6 +34,95 @@ module Rails
       def reset_configuration!
         @configuration = Configuration.new
       end
+
+      def installation_complete?(root = resolve_root)
+        return false unless root
+
+        required_installation_paths(root).all?(&:exist?)
+      end
+
+      def missing_installation_message(root: resolve_root)
+        return generic_missing_installation_message unless root
+
+        detailed_missing_installation_message(normalize_root(root))
+      end
+
+      def warn_about_missing_installation(root: resolve_root, stderr: $stderr, argv: ARGV)
+        return unless missing_installation_warning_needed?(root: root, argv: argv)
+
+        stderr.puts(missing_installation_message(root: root))
+      end
+
+      private
+
+      def missing_installation_warning_needed?(root:, argv:)
+        return false unless root
+        return false if installation_complete?(root)
+        return false if install_generator_invocation?(argv)
+
+        true
+      end
+
+      def install_generator_invocation?(argv)
+        normalized_args = Array(argv).map(&:to_s)
+        generator_commands = %w[generate g]
+
+        INSTALL_GENERATOR_NAMES.any? do |generator_name|
+          normalized_args.include?(generator_name) ||
+            normalized_args.each_cons(2).any? do |left, right|
+              generator_commands.include?(left) && right == generator_name
+            end
+        end
+      end
+
+      def required_installation_paths(root)
+        root_path = normalize_root(root)
+
+        REQUIRED_INSTALLATION_PATHS.map { |relative_path| root_path.join(relative_path) }
+      end
+
+      def generic_missing_installation_message
+        <<~MSG
+
+          rails-worktrees is in your bundle, but the app installer has not run yet.
+
+          Run:
+            $ #{INSTALL_GENERATOR_COMMAND}
+
+          Docs: https://github.com/asjer/rails-worktrees
+        MSG
+      end
+
+      def detailed_missing_installation_message(root_path)
+        <<~MSG
+
+          rails-worktrees is in your bundle, but the app installer has not run yet.
+
+          Run:
+            $ #{INSTALL_GENERATOR_COMMAND}
+
+          Missing expected files under #{root_path}:
+          #{missing_installation_items_text(root_path)}
+
+          Docs: https://github.com/asjer/rails-worktrees
+        MSG
+      end
+
+      def missing_installation_items_text(root)
+        required_installation_paths(root).reject(&:exist?).map do |path|
+          "  • #{path.relative_path_from(root)}"
+        end.join("\n")
+      end
+
+      def resolve_root
+        return unless defined?(Rails) && Rails.respond_to?(:root)
+
+        Rails.root
+      end
+
+      def normalize_root(root)
+        root.is_a?(Pathname) ? root : Pathname(root)
+      end
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rails-worktrees
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Asjer Querido
@@ -45,11 +45,11 @@ files:
 - Rakefile
 - exe/wt
 - lefthook.yml
-- lib/generators/rails/worktrees/install_generator.rb
 - lib/generators/rails/worktrees/mise_follow_up.rb
 - lib/generators/rails/worktrees/templates/Procfile.dev.worktree.example.tt
 - lib/generators/rails/worktrees/templates/bin/wt
 - lib/generators/rails/worktrees/templates/rails_worktrees.rb.tt
+- lib/generators/worktrees/install/install_generator.rb
 - lib/rails/worktrees.rb
 - lib/rails/worktrees/cli.rb
 - lib/rails/worktrees/command.rb
@@ -61,7 +61,9 @@ files:
 - lib/rails/worktrees/configuration.rb
 - lib/rails/worktrees/database_config_updater.rb
 - lib/rails/worktrees/env_bootstrapper.rb
+- lib/rails/worktrees/mise_toml_updater.rb
 - lib/rails/worktrees/names/cities.txt
+- lib/rails/worktrees/procfile_updater.rb
 - lib/rails/worktrees/railtie.rb
 - lib/rails/worktrees/version.rb
 - mise.toml
@@ -77,7 +79,7 @@ metadata:
   rubygems_mfa_required: 'true'
 post_install_message: "\n============================================\n  Thank you
   for installing rails-worktrees! \U0001F389\n\n  Run the installer:\n    $ bin/rails
-  generate rails:worktrees:install\n\n  Docs: https://github.com/asjer/rails-worktrees\n============================================\n\n"
+  generate worktrees:install\n\n  Docs: https://github.com/asjer/rails-worktrees\n============================================\n\n"
 rdoc_options: []
 require_paths:
 - lib

data/lib/generators/rails/worktrees/install_generator.rb

@@ -1,137 +0,0 @@
-# frozen_string_literal: true
-
-require 'open3'
-require 'rails/generators'
-require_relative 'mise_follow_up'
-require_relative '../../../rails/worktrees/database_config_updater'
-
-module Rails
-  module Worktrees
-    module Generators
-      # Installs the wt wrapper, configuration, and safe database.yml updates.
-      class InstallGenerator < ::Rails::Generators::Base
-        include MiseFollowUp
-
-        namespace 'rails:worktrees:install'
-        desc 'Installs bin/wt, a Rails::Worktrees initializer, and updates config/database.yml when safe.'
-        source_root File.expand_path('templates', __dir__)
-        class_option :conductor, type: :boolean, default: false,
-                                 desc: 'Configure the installer for ~/Sites/conductor/workspaces'
-
-        FOLLOW_UP_TEMPLATE = <<~TEXT
-            ============================================
-              rails-worktrees installed successfully! 🚂
-
-              Installed:
-          %<installed>s
-
-              Get started:
-                $ bin/wt
-                $ bin/wt my-feature
-
-              Configure:
-                config/initializers/rails_worktrees.rb
-          %<notes>s
-            ============================================
-        TEXT
-
-        def create_bin_wrapper
-          template('bin/wt', 'bin/wt')
-          chmod('bin/wt', 0o755)
-        end
-
-        def create_initializer
-          template('rails_worktrees.rb.tt', 'config/initializers/rails_worktrees.rb')
-        end
-
-        def create_procfile_worktree_example
-          template('Procfile.dev.worktree.example.tt', 'Procfile.dev.worktree.example')
-        end
-
-        def update_database_configuration
-          unless File.exist?(database_config_path)
-            say_status(:skip, 'config/database.yml not found', :yellow)
-            @database_outcome = :not_found
-            return
-          end
-
-          result = database_update_result
-          @database_outcome = result.changed? ? :updated : :identical
-          announce_database_update(result)
-        end
-
-        def verify_installation
-          if git_repo?
-            say_status(:ok, 'git repository detected', :green)
-          else
-            say_status(:warning, 'run inside a git repository for bin/wt to work', :yellow)
-          end
-        end
-
-        def show_follow_up
-          say(follow_up_message)
-        end
-
-        private
-
-        def database_config_path
-          File.join(destination_root, 'config/database.yml')
-        end
-
-        def database_update_result
-          result = ::Rails::Worktrees::DatabaseConfigUpdater.new(
-            content: File.read(database_config_path)
-          ).call
-
-          File.write(database_config_path, result.content) if result.changed?
-
-          result
-        end
-
-        def follow_up_message
-          "\n#{format(FOLLOW_UP_TEMPLATE, installed: installed_items_text, notes: follow_up_notes_text)}"
-        end
-
-        def installed_items_text
-          items = [
-            '      • bin/wt',
-            '      • config/initializers/rails_worktrees.rb',
-            '      • Procfile.dev.worktree.example'
-          ]
-          items << database_follow_up_line if database_follow_up_line
-          items.join("\n")
-        end
-
-        def database_follow_up_line
-          case @database_outcome
-          when :updated
-            '      • config/database.yml (updated with WORKTREE_DATABASE_SUFFIX)'
-          when :not_found
-            '      • config/database.yml was not found — see README for manual setup'
-          end
-        end
-
-        def announce_database_update(result)
-          status = result.changed? ? :update : :identical
-          color = result.changed? ? :green : :blue
-
-          say_status(status, 'config/database.yml', color)
-          result.messages.each { |message| say(message) }
-        end
-
-        def git_repo?
-          _stdout_str, _stderr_str, status = Open3.capture3(
-            'git', 'rev-parse', '--is-inside-work-tree', chdir: destination_root
-          )
-          status.success?
-        rescue Errno::ENOENT
-          false
-        end
-
-        def conductor_workspace_root
-          "File.expand_path('~/Sites/conductor/workspaces')"
-        end
-      end
-    end
-  end
-end