pg_reports 0.6.2 → 0.8.0

data/app/views/pg_reports/dashboard/_target_selector.html.erb

@@ -0,0 +1,27 @@
+<%# locals: (show_label: true) %>
+<% if @available_targets.present? && @available_targets.size > 1 %>
+  <form class="db-selector-form"
+        action="<%= switch_target_path %>"
+        method="post"
+        data-turbo="false">
+    <input type="hidden" name="authenticity_token" value="<%= form_authenticity_token %>">
+    <% if local_assigns.fetch(:show_label, true) %>
+      <label class="db-selector-label" for="pg-reports-target-select">
+        <%= t("pg_reports.ui.target_selector.label", default: "Target") %>
+      </label>
+    <% end %>
+    <select id="pg-reports-target-select"
+            name="target"
+            class="db-selector"
+            onchange="this.form.submit()">
+      <% default_target_name = PgReports.connection_registry.default_name %>
+      <% @available_targets.each do |target| %>
+        <option value="<%= target.name %>"
+                class="<%= "is-default" if target.name == default_target_name %>"
+                <%= "selected" if target.name == @selected_target %>>
+          <%= target.name %><%= " (#{target.default_database})" if target.default_database.present? %>
+        </option>
+      <% end %>
+    </select>
+  </form>
+<% end %>

data/app/views/pg_reports/dashboard/index.html.erb

@@ -1,5 +1,15 @@
 <%= csrf_meta_tags %>
 
+<% if @database_error %>
+  <div class="db-error-banner">
+    <strong><%= @database_error[:title] %></strong>
+    <span><%= @database_error[:detail] %></span>
+    <% if @database_error[:hint] %>
+      <code><%= @database_error[:hint] %></code>
+    <% end %>
+  </div>
+<% end %>
+
 <header class="header">
   <div class="logo">
     <div class="logo-icon">🐘</div>
@@ -138,9 +148,8 @@ pg_stat_statements.track = all</pre>
     <div class="live-monitoring-title">
       <span class="live-indicator"></span>
       <span><%= t("pg_reports.ui.monitoring.live_title") %></span>
-      <span class="database-badge">
-        <strong><%= @current_database %></strong>
-      </span>
+      <%= render "target_selector", show_label: false %>
+      <%= render "database_selector", show_label: false %>
     </div>
     <div class="live-monitoring-controls">
       <span class="live-monitoring-interval"><%= t("pg_reports.ui.monitoring.update_interval") %></span>
@@ -258,6 +267,9 @@ pg_stat_statements.track = all</pre>
     <div class="query-monitoring-title">
       <span class="monitoring-indicator" id="monitor-indicator"></span>
       <span><%= t("pg_reports.ui.monitoring.query_monitor_title") %></span>
+      <span class="query-monitoring-scope" title="<%= t("pg_reports.ui.monitoring.scope_note_long", default: "Query Monitor subscribes to ActiveSupport::Notifications in the host application's process. It captures every SQL the host app runs, on whichever database the host app is connected to. The dashboard's database selector does not change this scope.") %>">
+        <%= t("pg_reports.ui.monitoring.scope_note_short", default: "scope: host application") %>
+      </span>
       <span class="session-badge" id="session-badge" style="display: none;">
         <%= t("pg_reports.ui.monitoring.session_label") %> <strong id="session-id"></strong>
       </span>
@@ -297,11 +309,20 @@ pg_stat_statements.track = all</pre>
 
 <div class="categories-grid">
   <% @categories.each do |category_key, category| %>
-    <div class="category-card<%= ' disabled' if category_key == :queries && !@pg_stat_status[:ready] %>">
-      <% if category_key == :queries && !@pg_stat_status[:ready] %>
+    <%
+      requires_pg_stat = (category_key == :queries && !@pg_stat_status[:ready])
+      target_disabled_reason = category_disabled_reason(category_key)
+      is_disabled = requires_pg_stat || target_disabled_reason.present?
+    %>
+    <div class="category-card<%= ' disabled' if is_disabled %>">
+      <% if requires_pg_stat %>
         <div class="category-warning">
           <%= t("pg_reports.ui.categories.requires_pg_stat") %>
         </div>
+      <% elsif target_disabled_reason %>
+        <div class="category-warning">
+          <%= target_disabled_reason %>
+        </div>
       <% end %>
       <div class="category-header">
         <div class="category-icon" style="background: <%= category[:color] %>20; color: <%= category[:color] %>;">
@@ -313,7 +334,7 @@ pg_stat_statements.track = all</pre>
 
       <div class="reports-list">
         <% category[:reports].each do |report_key, report| %>
-          <% if category_key == :queries && !@pg_stat_status[:ready] %>
+          <% if is_disabled %>
             <div class="report-link disabled">
               <div class="report-link-info">
                 <span class="report-link-name">
@@ -613,22 +634,6 @@ pg_stat_statements.track = all</pre>
     font-size: 1rem;
   }
 
-  .database-badge {
-    margin-left: 0.5rem;
-    padding: 0.375rem 0.75rem;
-    background: var(--bg-tertiary);
-    border: 1px solid var(--border-color);
-    border-radius: 8px;
-    font-size: 0.75rem;
-    font-weight: 500;
-    color: var(--text-secondary);
-  }
-
-  .database-badge strong {
-    color: var(--text-primary);
-    font-weight: 600;
-  }
-
   .live-indicator {
     width: 8px;
     height: 8px;
@@ -990,13 +995,12 @@ pg_stat_statements.track = all</pre>
     if (!panel) return;
 
     panel.style.display = 'block';
-    panel.innerHTML = `
-      <div class="live-monitoring-header">
-        <div class="live-monitoring-title">
-          <span class="badge-dot error"></span>
-          <span>${PG_REPORTS_I18N.status.monitoring_unavailable}</span>
-        </div>
-      </div>
+
+    // Replace only the metrics grid; preserve the header (which holds the
+    // database selector) so the user can still switch databases when metrics
+    // are unavailable on the current one.
+    const grid = panel.querySelector('.live-metrics-grid');
+    const fallbackHTML = `
       <div style="padding: 2rem; text-align: center; color: var(--text-muted);">
         <p style="margin-bottom: 1rem;">${PG_REPORTS_I18N.errors.unable_fetch_metrics}</p>
         <p style="font-size: 0.875rem;">${PG_REPORTS_I18N.errors.possible_causes}</p>
@@ -1010,6 +1014,16 @@ pg_stat_statements.track = all</pre>
         </button>
       </div>
     `;
+
+    if (grid) {
+      grid.outerHTML = fallbackHTML;
+    } else {
+      panel.insertAdjacentHTML('beforeend', fallbackHTML);
+    }
+
+    // Surface the unavailable status next to the live indicator.
+    const indicator = panel.querySelector('.live-indicator');
+    if (indicator) indicator.classList.add('error');
   }
 
   function startPolling() {

data/app/views/pg_reports/dashboard/show.html.erb

@@ -7,6 +7,10 @@
     <span><%= @categories[@category][:name] %></span>
     <span>/</span>
     <span><%= @report_info[:name] %></span>
+
+    <span class="breadcrumb-spacer"></span>
+    <%= render "target_selector" %>
+    <%= render "database_selector" %>
   </nav>
 
   <div class="report-header">

data/config/locales/en.yml

@@ -592,6 +592,10 @@ en:
       navigation:
         dashboard: "Dashboard"
         back: "← Back"
+      database_selector:
+        label: "Database"
+      target_selector:
+        label: "Target"
       actions:
         cancel: "Cancel"
         retry: "Retry"
@@ -633,7 +637,7 @@ en:
         extension_installed: "Extension installed, not preloaded"
         preloaded: "Preloaded, extension not created"
         not_configured: "Not configured"
-        monitoring_unavailable: "Live Monitoring Unavailable"
+        monitoring_unavailable: "Status Unavailable"
       modals:
         enable_pg_stat_title: "Enable pg_stat_statements"
         enable_pg_stat_intro: "To enable pg_stat_statements, follow these steps:"
@@ -665,7 +669,7 @@ en:
         ide_cursor_wsl: "Cursor (WSL)"
         ide_cursor: "Cursor"
       monitoring:
-        live_title: "Live Monitoring"
+        live_title: "Status"
         update_interval: "Updates every 5s"
         toggle_title: "Toggle live monitoring"
         query_monitor_title: "SQL Query Monitor"
@@ -675,6 +679,8 @@ en:
         feed_no_queries: "No queries captured yet..."
         unknown_source: "Unknown source"
         expand_collapse_title: "Expand/Collapse"
+        scope_note_short: "scope: host application"
+        scope_note_long: "Query Monitor subscribes to ActiveSupport::Notifications in the host application's process. It captures every SQL the host app runs, on whichever database the host app is connected to. The dashboard's database selector does not change this scope."
       metrics:
         connections_label: "Connections"
         tps_label: "TPS"
@@ -693,6 +699,7 @@ en:
       categories:
         requires_pg_stat: "🔒 Requires pg_stat_statements"
         reports_count_suffix: "reports"
+        primary_only_reason: "🔒 This category only runs on the primary database — it inspects the host application's models. Switch back to the default database to use it."
       documentation:
         toggle_title: "📖 What does this report show?"
         what_section: "📋 What"

data/config/locales/ru.yml

@@ -557,6 +557,10 @@ ru:
       navigation:
         dashboard: "Дашборд"
         back: "← Назад"
+      database_selector:
+        label: "База данных"
+      target_selector:
+        label: "Сервер"
       actions:
         cancel: "Отмена"
         retry: "Повторить"
@@ -598,7 +602,7 @@ ru:
         extension_installed: "Расширение установлено, не предзагружено"
         preloaded: "Предзагружено, расширение не создано"
         not_configured: "Не настроено"
-        monitoring_unavailable: "Live-мониторинг недоступен"
+        monitoring_unavailable: "Статус недоступен"
       modals:
         enable_pg_stat_title: "Включение pg_stat_statements"
         enable_pg_stat_intro: "Чтобы включить pg_stat_statements, выполните следующие шаги:"
@@ -630,7 +634,7 @@ ru:
         ide_cursor_wsl: "Cursor (WSL)"
         ide_cursor: "Cursor"
       monitoring:
-        live_title: "Live-мониторинг"
+        live_title: "Статус"
         update_interval: "Обновление каждые 5с"
         toggle_title: "Переключить live-мониторинг"
         query_monitor_title: "Монитор SQL-запросов"
@@ -640,6 +644,8 @@ ru:
         feed_no_queries: "Запросы пока не захвачены..."
         unknown_source: "Источник неизвестен"
         expand_collapse_title: "Развернуть / свернуть"
+        scope_note_short: "область: хост-приложение"
+        scope_note_long: "Монитор SQL-запросов подписан на ActiveSupport::Notifications в процессе хост-приложения и захватывает каждый SQL, который выполняет приложение, в той базе, к которой приложение подключено. Селектор базы данных в дашборде на эту область не влияет."
       metrics:
         connections_label: "Соединения"
         tps_label: "TPS"
@@ -658,6 +664,7 @@ ru:
       categories:
         requires_pg_stat: "🔒 Требуется pg_stat_statements"
         reports_count_suffix: "отчётов"
+        primary_only_reason: "🔒 Эта категория работает только на первичной базе — она проверяет модели хост-приложения. Чтобы пользоваться, переключитесь обратно на базу по умолчанию."
       documentation:
         toggle_title: "📖 Что показывает этот отчёт?"
         what_section: "📋 Что"

data/config/locales/uk.yml

@@ -557,6 +557,10 @@ uk:
       navigation:
         dashboard: "Дашборд"
         back: "← Назад"
+      database_selector:
+        label: "База даних"
+      target_selector:
+        label: "Сервер"
       actions:
         cancel: "Скасувати"
         retry: "Повторити"
@@ -598,7 +602,7 @@ uk:
         extension_installed: "Розширення встановлено, не передзавантажено"
         preloaded: "Передзавантажено, розширення не створено"
         not_configured: "Не налаштовано"
-        monitoring_unavailable: "Live-моніторинг недоступний"
+        monitoring_unavailable: "Статус недоступний"
       modals:
         enable_pg_stat_title: "Увімкнення pg_stat_statements"
         enable_pg_stat_intro: "Щоб увімкнути pg_stat_statements, виконайте такі кроки:"
@@ -630,7 +634,7 @@ uk:
         ide_cursor_wsl: "Cursor (WSL)"
         ide_cursor: "Cursor"
       monitoring:
-        live_title: "Live-моніторинг"
+        live_title: "Статус"
         update_interval: "Оновлення кожні 5с"
         toggle_title: "Перемкнути live-моніторинг"
         query_monitor_title: "Монітор SQL-запитів"
@@ -640,6 +644,8 @@ uk:
         feed_no_queries: "Запитів поки не захоплено..."
         unknown_source: "Джерело невідоме"
         expand_collapse_title: "Розгорнути / згорнути"
+        scope_note_short: "область: хост-застосунок"
+        scope_note_long: "Монітор SQL-запитів підписаний на ActiveSupport::Notifications у процесі хост-застосунку та захоплює кожен SQL, який виконує застосунок, у тій базі, до якої він підключений. Селектор бази даних у дашборді на цю область не впливає."
       metrics:
         connections_label: "З'єднання"
         tps_label: "TPS"
@@ -658,6 +664,7 @@ uk:
       categories:
         requires_pg_stat: "🔒 Потрібен pg_stat_statements"
         reports_count_suffix: "звітів"
+        primary_only_reason: "🔒 Ця категорія працює лише на первинній базі — вона перевіряє моделі хост-застосунку. Щоб користуватися, перемкніться на базу за замовчуванням."
       documentation:
         toggle_title: "📖 Що показує цей звіт?"
         what_section: "📋 Що"

data/config/routes.rb

@@ -3,8 +3,13 @@
 PgReports::Engine.routes.draw do
   root to: "dashboard#index"
 
+  get "metrics", to: "metrics#show", as: :metrics
+
   get "live_metrics", to: "dashboard#live_metrics", as: :live_metrics
 
+  post "switch_database", to: "dashboard#switch_database", as: :switch_database
+  post "switch_target", to: "dashboard#switch_target", as: :switch_target
+
   post "enable_pg_stat_statements", to: "dashboard#enable_pg_stat_statements", as: :enable_pg_stat_statements
   post "reset_statistics", to: "dashboard#reset_statistics", as: :reset_statistics
   post "explain_analyze", to: "dashboard#explain_analyze", as: :explain_analyze

data/lib/pg_reports/configuration.rb

@@ -20,7 +20,7 @@ module PgReports
     attr_accessor :dead_rows_threshold          # Tables with more dead rows need vacuum
 
     # Connection settings
-    attr_accessor :connection_pool              # Custom connection pool (optional)
+    attr_accessor :connection_pool              # Custom connection pool (optional, legacy override)
 
     # Output settings
     attr_accessor :max_query_length             # Truncate query text to this length
@@ -41,6 +41,12 @@ module PgReports
 
     # Security settings
     attr_accessor :allow_raw_query_execution    # Allow execute_query and explain_analyze from dashboard
+    attr_accessor :allow_migration_creation     # Allow dashboard's "Generate Migration" button to write files into db/migrate/
+
+    # Grafana / Prometheus exporter settings
+    attr_accessor :grafana_favorites            # Reports exposed at /metrics (Array of keys or Hash with per-report opts)
+    attr_accessor :grafana_metrics_token        # Bearer token required to access /metrics (nil = no auth)
+    attr_accessor :grafana_cache_ttl            # Default cache TTL for collected reports, seconds
 
     def initialize
       # Telegram
@@ -87,10 +93,44 @@ module PgReports
       @allow_raw_query_execution = ActiveModel::Type::Boolean.new.cast(
         ENV.fetch("PG_REPORTS_ALLOW_RAW_QUERY_EXECUTION", false)
       )
+      # Migration creation defaults to Rails dev — preserves prior behavior —
+      # but is now explicitly togglable so it can be disabled even in dev.
+      env_override = ENV["PG_REPORTS_ALLOW_MIGRATION_CREATION"]
+      @allow_migration_creation = if env_override.nil?
+        defined?(Rails) && Rails.respond_to?(:env) && Rails.env.development?
+      else
+        ActiveModel::Type::Boolean.new.cast(env_override)
+      end
+
+      # Grafana / Prometheus exporter
+      @grafana_favorites = []
+      @grafana_metrics_token = ENV.fetch("PG_REPORTS_METRICS_TOKEN", nil)
+      @grafana_cache_ttl = 60
     end
 
     def connection
-      @connection_pool || ActiveRecord::Base.connection
+      return @connection_pool if @connection_pool
+
+      PgReports.connection_registry.current_connection
+    end
+
+    # Register an additional named target (besides the auto-detected :primary).
+    # Spec accepts the same keys as ActiveRecord::Base.establish_connection.
+    #
+    # Example:
+    #   config.add_target :analytics,
+    #     host: "...", user: "...", password: ENV["..."], database: "warehouse"
+    def add_target(name, spec)
+      PgReports.connection_registry.register(name, spec)
+    end
+
+    # Override the default target name (default is :primary).
+    def default_target=(name)
+      PgReports.connection_registry.default_name = name
+    end
+
+    def default_target
+      PgReports.connection_registry.default_name
     end
 
     def telegram_configured?

data/lib/pg_reports/connection/error_translator.rb

@@ -0,0 +1,109 @@
+# frozen_string_literal: true
+
+module PgReports
+  module Connection
+    # Translates raw PG / ActiveRecord exceptions into human-readable messages
+    # with concrete remediation hints (typically a GRANT statement).
+    #
+    # Usage:
+    #   PgReports::Connection::ErrorTranslator.translate(error)
+    #   # => { title: "...", detail: "...", hint: "GRANT ...", code: "42501" }
+    module ErrorTranslator
+      module_function
+
+      # Returns a Hash with :title, :detail, :hint, :code, :raw_message.
+      # The shape is suitable for rendering in the dashboard.
+      def translate(error)
+        sqlstate = sqlstate_for(error)
+        message = error.message.to_s
+
+        info = case sqlstate
+        when "42501" then permission_denied(message)
+        when "3D000" then database_does_not_exist(message)
+        when "28000", "28P01" then auth_failed(message)
+        when "08001", "08006", "08000", "08003", "08004" then connection_refused(message)
+        when "53300" then too_many_connections(message)
+        else generic(error)
+        end
+
+        info.merge(code: sqlstate, raw_message: message)
+      end
+
+      def sqlstate_for(error)
+        case error
+        when PG::Error
+          error.result&.error_field(PG::Result::PG_DIAG_SQLSTATE)
+        when ActiveRecord::StatementInvalid, ActiveRecord::ConnectionNotEstablished
+          sqlstate_for(error.cause) if error.cause && !error.cause.equal?(error)
+        end
+      rescue
+        nil
+      end
+
+      def permission_denied(message)
+        target = extract_object(message, /permission denied for (?:database|schema|table|relation|view) "?([\w.]+)"?/)
+        kind = extract_object(message, /permission denied for (database|schema|table|relation|view)/)
+
+        hint = if kind && target
+          case kind
+          when "database" then "GRANT CONNECT ON DATABASE #{target} TO <role>;"
+          when "schema" then "GRANT USAGE ON SCHEMA #{target} TO <role>;"
+          when "table", "relation", "view" then "GRANT SELECT ON #{target} TO <role>;"
+          end
+        end
+
+        {
+          title: "Permission denied",
+          detail: (kind && target) ? "The connecting role does not have the required privilege on #{kind} \"#{target}\"." : "The connecting role lacks the required privilege.",
+          hint: hint
+        }
+      end
+
+      def database_does_not_exist(message)
+        target = extract_object(message, /database "([^"]+)" does not exist/)
+        {
+          title: "Database not found",
+          detail: target ? "PostgreSQL has no database named \"#{target}\"." : "The requested database does not exist on this server.",
+          hint: nil
+        }
+      end
+
+      def auth_failed(_message)
+        {
+          title: "Authentication failed",
+          detail: "PostgreSQL rejected the credentials for this target.",
+          hint: "Verify the username/password in the target configuration; check pg_hba.conf for the connecting host."
+        }
+      end
+
+      def connection_refused(_message)
+        {
+          title: "Cannot reach PostgreSQL",
+          detail: "The server is unreachable or refused the connection.",
+          hint: "Check host/port, network reachability, and that PostgreSQL is accepting connections."
+        }
+      end
+
+      def too_many_connections(_message)
+        {
+          title: "Too many connections",
+          detail: "PostgreSQL refused the connection because max_connections is reached.",
+          hint: "Wait, increase max_connections, or use a connection pooler (PgBouncer)."
+        }
+      end
+
+      def generic(error)
+        {
+          title: error.class.name.split("::").last,
+          detail: error.message,
+          hint: nil
+        }
+      end
+
+      def extract_object(message, regex)
+        match = message.match(regex)
+        match && match[1]
+      end
+    end
+  end
+end

data/lib/pg_reports/connection/registry.rb

@@ -0,0 +1,150 @@
+# frozen_string_literal: true
+
+module PgReports
+  module Connection
+    # Registry of named PostgreSQL targets.
+    #
+    # In a Rails app the :primary target is auto-registered on first access from
+    # ActiveRecord::Base.connection_db_config — no configuration is required.
+    # Additional targets can be registered explicitly via Configuration#add_target
+    # for setups where the dashboard should reach databases the host app cannot.
+    #
+    # Current target/database is tracked in a Thread.current slot, switched via
+    # PgReports.with_target / PgReports.with_database for block-scoped usage.
+    class Registry
+      THREAD_KEY_TARGET = :pg_reports_current_target
+      THREAD_KEY_DATABASE = :pg_reports_current_database
+
+      class UnknownTarget < PgReports::Error; end
+
+      def initialize
+        @targets = {}
+        @default_name = :primary
+        @auto_registered = false
+        @mutex = Mutex.new
+      end
+
+      attr_reader :default_name
+
+      def default_name=(name)
+        @default_name = name.to_sym
+      end
+
+      # Register or overwrite a target.
+      def register(name, spec)
+        @targets[name.to_sym] = Target.new(name, spec)
+      end
+
+      # All known targets (auto-registers :primary if needed first).
+      def targets
+        ensure_default_registered!
+        @targets.values
+      end
+
+      def target_names
+        targets.map(&:name)
+      end
+
+      def target?(name)
+        targets.any? { |t| t.name == name.to_sym }
+      end
+
+      def fetch(name = nil)
+        ensure_default_registered!
+        key = (name || current_name || @default_name).to_sym
+        @targets.fetch(key) { raise UnknownTarget, "Unknown target #{key.inspect}. Known: #{@targets.keys.inspect}" }
+      end
+
+      # Resolve the AR connection to use right now (current target + database).
+      # Honors PgReports.with_target / with_database thread-local context.
+      def current_connection
+        target = fetch(current_name)
+        target.connection_for(current_database)
+      end
+
+      # Returns the target that current_connection would resolve to.
+      def current_target
+        fetch(current_name)
+      end
+
+      # The database name in effect for the current target.
+      def current_database_name
+        current_database || fetch(current_name).default_database
+      end
+
+      def current_name
+        Thread.current[THREAD_KEY_TARGET]
+      end
+
+      def current_database
+        Thread.current[THREAD_KEY_DATABASE]
+      end
+
+      # Switch target and/or database for the duration of the block. Semantics:
+      # - target given      → switches target AND clears the database override
+      #                       (the previous database belongs to the previous
+      #                       target's cluster and would not be valid on a new
+      #                       one). Pass `database:` explicitly to override on
+      #                       the new target.
+      # - target nil        → keeps the active target, switches only database.
+      # - database nil      → uses the (possibly new) target's default database.
+      def with_context(target: nil, database: nil)
+        prev_target = Thread.current[THREAD_KEY_TARGET]
+        prev_database = Thread.current[THREAD_KEY_DATABASE]
+
+        if target
+          Thread.current[THREAD_KEY_TARGET] = target.to_sym
+          Thread.current[THREAD_KEY_DATABASE] = database&.to_s
+        elsif database
+          Thread.current[THREAD_KEY_DATABASE] = database.to_s
+        end
+
+        yield
+      ensure
+        Thread.current[THREAD_KEY_TARGET] = prev_target
+        Thread.current[THREAD_KEY_DATABASE] = prev_database
+      end
+
+      # For tests / reload scenarios. Restores the registry to a fresh state —
+      # closes all derived pools, drops all registered targets, resets
+      # default_name back to :primary, and re-arms auto-registration.
+      def reset!
+        @mutex.synchronize do
+          @targets.each_value(&:disconnect!)
+          @targets.clear
+          @default_name = :primary
+          @auto_registered = false
+        end
+      end
+
+      # Auto-discover the :primary target from ActiveRecord on first need.
+      # Idempotent and thread-safe.
+      def ensure_default_registered!
+        return if @auto_registered
+        return unless defined?(ActiveRecord::Base)
+
+        @mutex.synchronize do
+          return if @auto_registered
+
+          unless @targets.key?(:primary)
+            spec = primary_spec_from_active_record
+            @targets[:primary] = Target.new(:primary, spec) if spec
+          end
+
+          @auto_registered = true
+        end
+      end
+
+      private
+
+      def primary_spec_from_active_record
+        config = ActiveRecord::Base.connection_db_config
+        return nil unless config
+
+        config.configuration_hash.transform_keys(&:to_sym)
+      rescue ActiveRecord::ConnectionNotEstablished
+        nil
+      end
+    end
+  end
+end