rails_pulse 0.1.4 → 0.2.2

data/app/views/rails_pulse/requests/index.html.erb

@@ -1,4 +1,4 @@
-<%= render 'rails_pulse/components/breadcrumbs' %>
+<%= render 'rails_pulse/components/page_header' %>
 
 <% unless turbo_frame_request? %>
   <div class="row">
@@ -9,69 +9,51 @@
   </div>
 <% end %>
 
-<div
-  data-controller="rails-pulse--index"
-  data-rails-pulse--index-chart-id-value="average_response_times_chart"
->
-  <%= render 'rails_pulse/components/panel', { title: 'Average Response Time', } do %>
-    <%= search_form_for @ransack_query, url: requests_path, class: "flex items-center justify-between gap mb-4" do |form| %>
-      <div class="flex items-center grow gap">
-        <%= form.select :period_start_range,
-          RailsPulse::RequestsController::TIME_RANGE_OPTIONS,
-          { selected: @selected_time_range },
-          { class: "input" }
-        %>
-        <%= form.select :avg_duration,
-          duration_options(:request),
-          { selected: @selected_response_range },
-          { class: "input" }
-        %>
-        <%= link_to "Reset", requests_path, class: "btn btn--borderless show@md" if params.has_key?(:q) %>
-        <%= form.submit "Search", class: "btn show@sm" %>
-      </div>
-    <% end %>
+<%= render 'rails_pulse/components/panel', { title: 'Requests', } do %>
+  <%= search_form_for @ransack_query, url: requests_path, class: "flex items-center justify-between gap mb-4", data: { controller: "rails-pulse--custom-range" } do |form| %>
+    <div class="flex items-center grow gap">
+      <%= time_range_selector(form,
+        time_range_options: RailsPulse::RequestsController::TIME_RANGE_OPTIONS,
+        selected_time_range: @selected_time_range,
+        mode: :recent_custom
+      ) %>
+
+      <%= form.search_field :route_path_cont,
+        placeholder: "Filter by route",
+        autocomplete: "off",
+        class: "input"
+      %>
 
-    <% if @has_data %>
-      <% if @chart_data && @chart_data.values.any? { |v| v > 0 } %>
-        <div
-          class="chart-container chart-container--slim"
-          data-rails-pulse--index-target="chart"
-        >
-          <%= bar_chart(
-            @chart_data,
-            code: false,
-            id: "average_response_times_chart",
-            height: "100%",
-            options: bar_chart_options(
-              units: "ms",
-              zoom: true,
-              chart_start: 0,
-              chart_end: @chart_data.length - 1,
-              xaxis_formatter: @xaxis_formatter,
-              tooltip_formatter: @tooltip_formatter,
-              zoom_start: @zoom_start,
-              zoom_end: @zoom_end,
-              chart_data: @chart_data
-            )
-          ) %>
-        </div>
-      <% else %>
-        <div class="p-4 text-center text-muted">
-          No response time data available for the selected filters.
-        </div>
-      <% end %>
+      <%= form.select :status_category_eq,
+        [
+          ["All statuses", ""],
+          ["2xx Success", "2"],
+          ["3xx Redirect", "3"],
+          ["4xx Client Error", "4"],
+          ["5xx Server Error", "5"]
+        ],
+        {},
+        { class: "input" }
+      %>
 
-      <div class="alert flex items-start mbs-8" role="alert">
-        <p>The chart above shows aggregated average response times grouped by time periods while the table below shows specific request details.</p>
-      </div>
+      <%= form.select :duration_gteq,
+        duration_options(:request),
+        { selected: @selected_response_range, prompt: "Min duration" },
+        { class: "input" }
+      %>
 
-      <%= turbo_frame_tag :index_table, data: { rails_pulse__index_target: "indexTable" } do %>
-        <%= render 'rails_pulse/requests/table' %>
-      <% end %>
-    <% else %>
-      <%= render 'rails_pulse/components/empty_state',
-          title: 'No request data found for the selected filters.',
-          description: 'Try adjusting your time range or filters to see results.' %>
+      <%= link_to "Reset", requests_path, class: "btn btn--borderless show@md" if params.has_key?(:q) %>
+      <%= form.submit "Search", class: "btn show@sm" %>
+    </div>
+  <% end %>
+
+  <% if @has_data %>
+    <%= turbo_frame_tag :index_table do %>
+      <%= render 'rails_pulse/requests/table' %>
     <% end %>
+  <% else %>
+    <%= render 'rails_pulse/components/empty_state',
+        title: 'No request data found for the selected filters.',
+        description: 'Try adjusting your time range or filters to see results.' %>
   <% end %>
-</div>
+<% end %>

data/app/views/rails_pulse/requests/show.html.erb

@@ -1,4 +1,4 @@
-<%= render 'rails_pulse/components/breadcrumbs' %>
+<%= render 'rails_pulse/components/page_header', taggable: @request %>
 
 <div class="row">
   <div class="grid-item">

data/app/views/rails_pulse/routes/_requests_table.html.erb

@@ -1,7 +1,8 @@
 <% columns = [
   { field: :occurred_at, label: 'Timestamp', class: 'w-auto' },
   { field: :duration, label: 'Response Time', class: 'w-36' },
-  { field: :status, label: 'Status', class: 'w-20' }
+  { field: :status, label: 'Status', class: 'w-20' },
+  { field: nil, label: 'Tags', class: 'w-32' }
 ] %>
 
 <table class="table mbs-4" data-controller="rails-pulse--table-sort">
@@ -31,6 +32,7 @@
             <span class="text-green-600"><%= request.status %></span>
           <% end %>
         </td>
+        <td class="whitespace-nowrap"><%= display_tag_badges(request) %></td>
       </tr>
     <% end %>
   </tbody>

data/app/views/rails_pulse/routes/_table.html.erb

@@ -2,7 +2,8 @@
   { field: :route_path, label: 'Route', class: 'w-auto' },
   { field: :avg_duration_sort, label: 'Average Response Time', class: 'w-48' },
   { field: :max_duration_sort, label: 'Max Response Time', class: 'w-44' },
-  { field: :count_sort, label: 'Requests', class: 'w-24' }
+  { field: :count_sort, label: 'Requests', class: 'w-24' },
+  { field: nil, label: 'Tags', class: 'w-32' }
 ] %>
 
 <table class="table mbs-4" data-controller="rails-pulse--table-sort">
@@ -15,6 +16,7 @@
         <td class="whitespace-nowrap"><%= summary.avg_duration.to_i %> ms</td>
         <td class="whitespace-nowrap"><%= summary.max_duration.to_i %> ms</td>
         <td class="whitespace-nowrap"><%= number_with_delimiter summary.count %></td>
+        <td class="whitespace-nowrap"><%= display_tag_badges(summary.tags) %></td>
       </tr>
     <% end %>
   </tbody>

data/app/views/rails_pulse/routes/index.html.erb

@@ -1,4 +1,4 @@
-<%= render 'rails_pulse/components/breadcrumbs' %>
+<%= render 'rails_pulse/components/page_header' %>
 
 <% unless turbo_frame_request? %>
   <div class="row">
@@ -14,14 +14,10 @@
   data-rails-pulse--index-chart-id-value="average_response_times_chart"
 >
   <%= render 'rails_pulse/components/panel', { title: 'Average Response Time', card_classes: 'b-full' } do %>
-    <%= search_form_for @ransack_query, url: routes_path, class: "flex items-center justify-between gap mb-4" do |form| %>
+    <%= search_form_for @ransack_query, url: routes_path, class: "flex items-center justify-between gap mb-4", data: { controller: "rails-pulse--custom-range" } do |form| %>
       <div class="flex items-center grow gap">
-        <%= form.search_field :route_path_cont, placeholder: "Filter by route", autocomplete: "off", class: "input", style: "max-inline-size: 250px" %>
-        <%= form.select :period_start_range,
-          RailsPulse::RoutesController::TIME_RANGE_OPTIONS,
-          { selected: @selected_time_range },
-          { class: "input" }
-        %>
+        <%= time_range_selector(form, time_range_options: RailsPulse::RoutesController::TIME_RANGE_OPTIONS, selected_time_range: @selected_time_range) %>
+        <%= form.search_field :route_path_cont, placeholder: "Filter by route", autocomplete: "off", class: "input" %>
         <%= form.select :avg_duration,
           duration_options(:route),
           { selected: @selected_response_range },

data/app/views/rails_pulse/routes/show.html.erb

@@ -1,4 +1,4 @@
-<%= render 'rails_pulse/components/breadcrumbs' %>
+<%= render 'rails_pulse/components/page_header', taggable: @route %>
 
 <% unless turbo_frame_request? %>
   <div class="row">
@@ -16,13 +16,9 @@
 >
   <div class="grid-item">
     <%= render 'rails_pulse/components/panel', { title: 'Route Reqeusts', } do %>
-      <%= search_form_for @ransack_query, url: route_path(@route), class: "flex items-center justify-between gap mb-4" do |form| %>
+      <%= search_form_for @ransack_query, url: route_path(@route), class: "flex items-center justify-between gap mb-4", data: { controller: "rails-pulse--custom-range" } do |form| %>
         <div class="flex items-center grow gap">
-          <%= form.select :period_start_range,
-            RailsPulse::RoutesController::TIME_RANGE_OPTIONS,
-            { selected: @selected_time_range },
-            { class: "input" }
-          %>
+          <%= time_range_selector(form, time_range_options: RailsPulse::RoutesController::TIME_RANGE_OPTIONS, selected_time_range: @selected_time_range) %>
           <%= form.select :avg_duration,
             duration_options(:route),
             { selected: @selected_response_range },

data/app/views/rails_pulse/tags/_tag_manager.html.erb

@@ -0,0 +1,73 @@
+<%
+  taggable_type = taggable.class.name.demodulize.underscore
+  available_tags = RailsPulse.configuration.tags
+  current_tags = taggable.tag_list
+  available_to_add = available_tags - current_tags
+%>
+
+<div
+  id="tag_manager_<%= taggable_type %>_<%= taggable.id %>"
+  class="tag-manager"
+>
+  <div class="tag-list">
+    <% current_tags.each do |tag| %>
+      <span class="tag">
+        <%= tag %>
+        <%= button_to remove_tag_path(taggable_type, taggable.id, tag: tag),
+              method: :delete,
+              class: "tag-remove",
+              data: {
+                turbo_frame: "_top"
+              } do %>
+          <span aria-hidden="true">&times;</span>
+        <% end %>
+      </span>
+    <% end %>
+
+    <% if available_to_add.any? %>
+      <div class="tag-add-container" data-controller="rails-pulse--popover" data-rails-pulse--popover-placement-value="bottom-start">
+        <button
+          type="button"
+          id="tag_menu_button_<%= taggable_type %>_<%= taggable.id %>"
+          class="btn btn--sm tag-add-button"
+          data-rails-pulse--popover-target="button"
+          data-action="rails-pulse--popover#toggle"
+          aria-haspopup="true"
+          aria-controls="tag_menu_<%= taggable_type %>_<%= taggable.id %>"
+        >
+          + tag
+        </button>
+
+        <div
+          popover
+          class="popover"
+          style="--popover-size: 12rem;"
+          data-rails-pulse--popover-target="menu"
+        >
+          <div
+            id="tag_menu_<%= taggable_type %>_<%= taggable.id %>"
+            class="menu"
+            data-controller="rails-pulse--menu"
+            data-action="keydown.up->rails-pulse--menu#prev keydown.down->rails-pulse--menu#next"
+            role="menu"
+            aria-labelledby="tag_menu_button_<%= taggable_type %>_<%= taggable.id %>"
+          >
+            <% available_to_add.each do |tag| %>
+              <%= button_to add_tag_path(taggable_type, taggable.id, tag: tag),
+                    method: :post,
+                    class: "btn menu__item i-full",
+                    data: {
+                      turbo_frame: "_top",
+                      rails_pulse__menu_target: "item"
+                    },
+                    style: "cursor: pointer;",
+                    role: "menuitem" do %>
+                <%= tag %>
+              <% end %>
+            <% end %>
+          </div>
+        </div>
+      </div>
+    <% end %>
+  </div>
+</div>

data/config/routes.rb

@@ -11,6 +11,11 @@ RailsPulse::Engine.routes.draw do
   resources :operations, only: %i[show]
   resources :caches, only: %i[show], as: :cache
   patch "pagination/limit", to: "application#set_pagination_limit"
+  patch "settings/global_filters", to: "application#set_global_filters"
+
+  # Tag management
+  post "tags/:taggable_type/:taggable_id/add", to: "tags#create", as: :add_tag
+  delete "tags/:taggable_type/:taggable_id/remove", to: "tags#destroy", as: :remove_tag
 
   # CSP compliance testing
   get "csp_test", to: "csp_test#show", as: :csp_test

data/db/rails_pulse_schema.rb

@@ -18,6 +18,7 @@ RailsPulse::Schema = lambda do |connection|
   connection.create_table :rails_pulse_routes do |t|
     t.string :method, null: false, comment: "HTTP method (e.g., GET, POST)"
     t.string :path, null: false, comment: "Request path (e.g., /posts/index)"
+    t.text :tags, comment: "JSON array of tags for filtering and categorization"
     t.timestamps
   end
 
@@ -34,6 +35,7 @@ RailsPulse::Schema = lambda do |connection|
     t.text :index_recommendations, comment: "JSON array of database index recommendations"
     t.text :n_plus_one_analysis, comment: "JSON object with enhanced N+1 query detection results"
     t.text :suggestions, comment: "JSON array of optimization recommendations"
+    t.text :tags, comment: "JSON array of tags for filtering and categorization"
     t.timestamps
   end
 
@@ -47,6 +49,7 @@ RailsPulse::Schema = lambda do |connection|
     t.string :request_uuid, null: false, comment: "Unique identifier for the request (e.g., UUID)"
     t.string :controller_action, comment: "Controller and action handling the request (e.g., PostsController#show)"
     t.timestamp :occurred_at, null: false, comment: "When the request started"
+    t.text :tags, comment: "JSON array of tags for filtering and categorization"
     t.timestamps
   end
 

data/lib/generators/rails_pulse/install_generator.rb

@@ -22,6 +22,25 @@ module RailsPulse
         create_file "db/rails_pulse_migrate/.keep"
       end
 
+      def copy_gem_migrations
+        gem_migrations_path = File.expand_path("../../../db/rails_pulse_migrate", __dir__)
+        destination_dir = separate_database? ? "db/rails_pulse_migrate" : "db/migrate"
+
+        if File.directory?(gem_migrations_path)
+          Dir.glob("#{gem_migrations_path}/*.rb").each do |migration_file|
+            migration_name = File.basename(migration_file)
+            destination_path = File.join(destination_dir, migration_name)
+
+            # Only copy if it doesn't already exist in the destination
+            # Use File.join with destination_root to check the actual location
+            full_destination_path = File.join(destination_root, destination_path)
+            unless File.exist?(full_destination_path)
+              copy_file migration_file, destination_path
+            end
+          end
+        end
+      end
+
       def copy_initializer
         copy_file "rails_pulse.rb", "config/initializers/rails_pulse.rb"
       end
@@ -84,7 +103,7 @@ module RailsPulse
           3. Restart your Rails server
 
           The schema file db/rails_pulse_schema.rb is your single source of truth.
-          Future schema changes will come as regular migrations in db/rails_pulse_migrate/
+          Future upgrades will automatically copy new migrations to db/rails_pulse_migrate/
 
         MESSAGE
       end
@@ -99,7 +118,7 @@ module RailsPulse
           2. Restart your Rails server
 
           The schema file db/rails_pulse_schema.rb is your single source of truth.
-          Future schema changes will come as regular migrations in db/migrate/
+          Future upgrades will automatically copy new migrations to db/migrate/
 
           Note: The installation migration loads from db/rails_pulse_schema.rb
           and includes all current Rails Pulse tables and columns.

data/lib/generators/rails_pulse/templates/db/rails_pulse_schema.rb

@@ -18,6 +18,7 @@ RailsPulse::Schema = lambda do |connection|
   connection.create_table :rails_pulse_routes do |t|
     t.string :method, null: false, comment: "HTTP method (e.g., GET, POST)"
     t.string :path, null: false, comment: "Request path (e.g., /posts/index)"
+    t.text :tags, comment: "JSON array of tags for filtering and categorization"
     t.timestamps
   end
 
@@ -34,6 +35,7 @@ RailsPulse::Schema = lambda do |connection|
     t.text :index_recommendations, comment: "JSON array of database index recommendations"
     t.text :n_plus_one_analysis, comment: "JSON object with enhanced N+1 query detection results"
     t.text :suggestions, comment: "JSON array of optimization recommendations"
+    t.text :tags, comment: "JSON array of tags for filtering and categorization"
     t.timestamps
   end
 
@@ -47,6 +49,7 @@ RailsPulse::Schema = lambda do |connection|
     t.string :request_uuid, null: false, comment: "Unique identifier for the request (e.g., UUID)"
     t.string :controller_action, comment: "Controller and action handling the request (e.g., PostsController#show)"
     t.timestamp :occurred_at, null: false, comment: "When the request started"
+    t.text :tags, comment: "JSON array of tags for filtering and categorization"
     t.timestamps
   end
 

data/lib/generators/rails_pulse/templates/rails_pulse.rb

@@ -71,6 +71,27 @@ RailsPulse.configure do |config|
   config.ignored_requests = []
   config.ignored_queries  = []
 
+  # ====================================================================================================
+  #                                                 TAGGING
+  # ====================================================================================================
+  # Define custom tags for categorizing routes, requests, and queries.
+  # You can add any custom tags you want for filtering and organization.
+  #
+  # Tag names should be in present tense and describe the current state or category.
+  # Examples of good tag names:
+  #   - "critical" (for high-priority endpoints)
+  #   - "experimental" (for routes under development)
+  #   - "deprecated" (for routes being phased out)
+  #   - "external" (for third-party API calls)
+  #   - "background" (for async job-related operations)
+  #   - "admin" (for administrative routes)
+  #   - "public" (for public-facing routes)
+  #
+  # Example configuration:
+  #   config.tags = ["ignored", "critical", "experimental", "deprecated", "external", "admin"]
+
+  config.tags = [ "ignored", "critical", "experimental" ]
+
   # ====================================================================================================
   #                                            DATABASE CONFIGURATION
   # ====================================================================================================

data/lib/generators/rails_pulse/upgrade_generator.rb

@@ -41,17 +41,36 @@ module RailsPulse
         # Check for existing Rails Pulse tables
         tables_exist = rails_pulse_tables_exist?
 
-        if !tables_exist && File.exist?("db/rails_pulse_schema.rb")
+        root_path = respond_to?(:destination_root) ? destination_root : Rails.root
+        schema_path = File.join(root_path, "db/rails_pulse_schema.rb")
+
+        if !tables_exist && File.exist?(schema_path)
           :schema_only
         elsif !tables_exist
           :not_installed
-        elsif File.exist?("db/rails_pulse_migrate")
+        elsif has_separate_database_config?
           :separate
         else
           :single
         end
       end
 
+      def has_separate_database_config?
+        root_path = respond_to?(:destination_root) ? destination_root : Rails.root
+        config_path = File.join(root_path, "config/database.yml")
+
+        return false unless File.exist?(config_path)
+
+        require "yaml"
+        db_config = YAML.load_file(config_path)
+
+        # Check if any environment has a rails_pulse database configuration
+        db_config.values.any? { |env| env.is_a?(Hash) && env.key?("rails_pulse") }
+      rescue => e
+        # If we can't read the file, assume single database
+        false
+      end
+
       def rails_pulse_tables_exist?
         return false unless defined?(ActiveRecord::Base)
 
@@ -65,7 +84,8 @@ module RailsPulse
 
       def get_rails_pulse_table_names
         # Load the schema file to get the table names dynamically
-        schema_file = File.join(Rails.root, "db/rails_pulse_schema.rb")
+        root_path = respond_to?(:destination_root) ? destination_root : Rails.root
+        schema_file = File.join(root_path, "db/rails_pulse_schema.rb")
 
         if File.exist?(schema_file)
           # Read the schema file and extract the required_tables array
@@ -84,44 +104,114 @@ module RailsPulse
       end
 
       def upgrade_single_database
-        missing_columns = detect_missing_columns
+        # Check for new migrations in gem
+        gem_migrations = get_gem_migrations
+        existing_migrations = get_user_migrations("db/migrate")
+        new_migrations = gem_migrations - existing_migrations
+
+        if new_migrations.any?
+          say "Found #{new_migrations.size} new migration(s) to copy:", :blue
+          new_migrations.each do |migration|
+            say "  - #{migration}", :blue
+            copy_gem_migration_to(migration, "db/migrate")
+          end
 
-        if missing_columns.empty?
-          say "Rails Pulse is up to date! No migration needed.", :green
-          return
-        end
+          say "\nMigrations copied successfully!", :green
+          say "\nNext steps:", :green
+          say "1. Run: rails db:migrate"
+          say "2. Restart your Rails server"
+        else
+          # Fall back to detecting missing columns
+          missing_columns = detect_missing_columns
 
-        # Format missing columns by table for the template
-        missing_by_table = format_missing_columns_by_table(missing_columns)
+          if missing_columns.empty?
+            say "Rails Pulse is up to date! No migration needed.", :green
+            return
+          end
 
-        say "Creating upgrade migration for missing columns: #{missing_columns.keys.join(', ')}", :blue
+          # Format missing columns by table for the template
+          missing_by_table = format_missing_columns_by_table(missing_columns)
 
-        # Set instance variables for template
-        @migration_version = ActiveRecord::Migration.current_version
-        @missing_columns = missing_by_table
+          say "Creating upgrade migration for missing columns: #{missing_columns.keys.join(', ')}", :blue
 
-        migration_template(
-          "migrations/upgrade_rails_pulse_tables.rb",
-          "db/migrate/upgrade_rails_pulse_tables.rb"
-        )
+          # Set instance variables for template
+          @migration_version = ActiveRecord::Migration.current_version
+          @missing_columns = missing_by_table
 
-        say <<~MESSAGE
+          migration_template(
+            "migrations/upgrade_rails_pulse_tables.rb",
+            "db/migrate/upgrade_rails_pulse_tables.rb"
+          )
 
-          Upgrade migration created successfully!
+          say <<~MESSAGE
 
-          Next steps:
-          1. Run: rails db:migrate
-          2. Restart your Rails server
+            Upgrade migration created successfully!
 
-          This migration will add: #{missing_columns.keys.join(', ')}
+            Next steps:
+            1. Run: rails db:migrate
+            2. Restart your Rails server
 
-        MESSAGE
+            This migration will add: #{missing_columns.keys.join(', ')}
+
+          MESSAGE
+        end
       end
 
       def upgrade_separate_database
-        # For separate database, we'd need to check the schema file and generate migrations
-        # in db/rails_pulse_migrate/ directory
-        say "Separate database upgrade not implemented yet. Please check db/rails_pulse_schema.rb for updates.", :yellow
+        # Check for new migrations in gem
+        gem_migrations = get_gem_migrations
+        existing_migrations = get_user_migrations("db/rails_pulse_migrate")
+        new_migrations = gem_migrations - existing_migrations
+
+        if new_migrations.any?
+          say "Found #{new_migrations.size} new migration(s) to copy:", :blue
+          new_migrations.each do |migration|
+            say "  - #{migration}", :blue
+            copy_gem_migration_to(migration, "db/rails_pulse_migrate")
+          end
+
+          say "\nMigrations copied successfully!", :green
+          say "\nNext steps:", :green
+          say "1. Run migrations for the rails_pulse database:"
+          say "   rails db:migrate (will run migrations for all databases)"
+          say "   OR manually run the migration files in db/rails_pulse_migrate/"
+          say "2. Restart your Rails server"
+        else
+          # Fall back to detecting missing columns
+          missing_columns = detect_missing_columns
+
+          if missing_columns.empty?
+            say "Rails Pulse is up to date! No migrations needed.", :green
+          else
+            # Format missing columns by table for the template
+            missing_by_table = format_missing_columns_by_table(missing_columns)
+
+            say "Creating upgrade migration for missing columns: #{missing_columns.keys.join(', ')}", :blue
+
+            # Set instance variables for template
+            @migration_version = ActiveRecord::Migration.current_version
+            @missing_columns = missing_by_table
+
+            migration_template(
+              "migrations/upgrade_rails_pulse_tables.rb",
+              "db/rails_pulse_migrate/upgrade_rails_pulse_tables.rb"
+            )
+
+            say <<~MESSAGE
+
+              Upgrade migration created successfully!
+
+              Next steps:
+              1. Run migrations for the rails_pulse database:
+                 rails db:migrate (will run migrations for all databases)
+                 OR manually run the migration files in db/rails_pulse_migrate/
+              2. Restart your Rails server
+
+              This migration will add: #{missing_columns.keys.join(', ')}
+
+            MESSAGE
+          end
+        end
       end
 
       def offer_conversion_to_migrations
@@ -165,7 +255,8 @@ module RailsPulse
       end
 
       def get_expected_schema_from_file
-        schema_file = File.join(Rails.root, "db/rails_pulse_schema.rb")
+        root_path = respond_to?(:destination_root) ? destination_root : Rails.root
+        schema_file = File.join(root_path, "db/rails_pulse_schema.rb")
         return {} unless File.exist?(schema_file)
 
         schema_content = File.read(schema_file)
@@ -221,6 +312,31 @@ module RailsPulse
 
         missing_by_table
       end
+
+      def get_gem_migrations
+        gem_migrations_path = File.expand_path("../../../db/rails_pulse_migrate", __dir__)
+        return [] unless File.directory?(gem_migrations_path)
+
+        Dir.glob("#{gem_migrations_path}/*.rb").map { |f| File.basename(f) }
+      end
+
+      def get_user_migrations(directory)
+        # Use destination_root in tests, Rails.root in production
+        root_path = respond_to?(:destination_root) ? destination_root : Rails.root
+        full_directory = File.join(root_path, directory)
+
+        return [] unless File.directory?(full_directory)
+
+        Dir.glob("#{full_directory}/*.rb").map { |f| File.basename(f) }
+      end
+
+      def copy_gem_migration_to(migration_name, destination)
+        gem_migrations_path = File.expand_path("../../../db/rails_pulse_migrate", __dir__)
+        source_file = File.join(gem_migrations_path, migration_name)
+        destination_file = File.join(destination, migration_name)
+
+        copy_file source_file, destination_file
+      end
     end
   end
 end