dontbugme 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e3c88f3e8468ff48582e0dca26ddeea1ebb66b378c57a7e96b888bdb0ca991d7
-  data.tar.gz: ae3982763c6d291f3f1a4873fb168c9718dc296ace4053779b7c82d3f1b4b271
+  metadata.gz: 5b8559d5544650453a07a48f7a123270fa34537e4980167b96c5391cd21bd081
+  data.tar.gz: 8e87d2da3525e8def1b4fbd7d8f7c8e0c5941cc176c835a0cedd8f9635ead888
 SHA512:
-  metadata.gz: 5c3d6729e5e7fac64ca625d0e1e4a0ec54498f124acc3cfafa47afff56839e269aa64795f1060b5bddae665f58b94ee57247ec9d26de200e7f44976cd9d726ca
-  data.tar.gz: 86133c069405f2c75d9605fed551f2a5361ce5a0072fcfbcfde77d878da7f2934e4d0fe6ac0cb35c6301a0ddba1e9f1dcc7f44b879013338aaeaa5d0e8d8e7bb
+  metadata.gz: cc02e325d106483e274c53cd77258edc868bbd03241ec20aaa3e4962b5bfeea8f9345bb2e2d5b0d1fc04155e2f6caaa60a540211392f1c7a9a54c6c0ad4864f3
+  data.tar.gz: 45642d97da0b241ffd31a3b0585227195b17320e675ac9f2af671f07d38f7af11620ae9e11141a76aa8f4ae220505ddf2d1638b9f112e859ae0351e134a89038

data/README.md

@@ -114,18 +114,20 @@ puts trace.to_timeline
 
 ### Manual Spans and Snapshots
 
-Add custom spans and snapshots within a trace:
+Add custom spans and snapshots within a trace. Spans capture the **return value** by default so you can see outputs in the UI:
 
 ```ruby
 trace = Dontbugme.trace("checkout flow") do
   Dontbugme.span("Calculate tax") do
-    tax = order.calculate_tax
+    tax = order.calculate_tax  # output shown in UI
   end
   Dontbugme.snapshot(user: user.attributes.slice("id", "email"), total: order.total)
   Dontbugme.tag(customer_tier: "enterprise")
 end
 ```
 
+Use `capture_output: false` to skip capturing the return value for sensitive data.
+
 ### Span Categories
 
 Access spans by category for assertions or analysis:
@@ -185,6 +187,11 @@ Dontbugme.configure do |config|
   config.recording_mode = :always
   config.capture_sql_binds = true
   config.source_mode = :full
+
+  # Capture outputs for debugging (development only)
+  config.capture_span_output = true      # return values from Dontbugme.span
+  config.capture_http_body = true       # HTTP response bodies
+  config.capture_redis_return_values = true  # Redis command return values
 end
 ```
 

data/app/controllers/dontbugme/traces_controller.rb

@@ -24,7 +24,7 @@ module Dontbugme
       @trace_a = params[:a].present? ? store.find_trace(params[:a]) : nil
       @trace_b = params[:b].present? ? store.find_trace(params[:b]) : nil
       @diff_output = if @trace_a && @trace_b
-        Formatters::Diff.format(@trace_a, @trace_b)
+        Dontbugme::Formatters::Diff.format(@trace_a, @trace_b)
       else
         nil
       end

data/app/helpers/dontbugme/traces_helper.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+module Dontbugme
+  module TracesHelper
+    def span_category_badge_class(category)
+      case category.to_sym
+      when :sql then 'badge-sql'
+      when :http then 'badge-http'
+      when :redis then 'badge-redis'
+      when :cache then 'badge-cache'
+      when :mailer then 'badge-mailer'
+      when :enqueue then 'badge-enqueue'
+      when :custom then 'badge-custom'
+      when :snapshot then 'badge-snapshot'
+      else 'badge-default'
+      end
+    end
+
+    OUTPUT_KEYS = %w[output result response_body].freeze
+
+    def span_output(span)
+      return nil if span.payload.blank?
+
+      payload = span.payload.is_a?(Hash) ? span.payload : {}
+      key = OUTPUT_KEYS.find { |k| payload[k.to_sym] || payload[k] }
+      key ? (payload[key.to_sym] || payload[key]) : nil
+    end
+
+    def format_span_payload(span)
+      return [] if span.payload.blank?
+
+      payload = span.payload.is_a?(Hash) ? span.payload : {}
+      payload.map do |key, value|
+        next if value.nil?
+        next if OUTPUT_KEYS.include?(key.to_s)
+
+        display_value = case value
+        when Array then value.map { |v| v.is_a?(String) ? v : v.inspect }.join(', ')
+        when Hash then value.inspect
+        else value.to_s
+        end
+        display_value = "#{display_value[0, 500]}..." if display_value.length > 500
+        [key.to_s.tr('_', ' ').capitalize, display_value]
+      end.compact
+    end
+
+    def format_span_detail_for_display(span)
+      span.detail
+    end
+
+    def truncate_detail(str, max_len = 80)
+      return '' if str.blank?
+      str = str.to_s.strip
+      return str if str.length <= max_len
+      "#{str[0, max_len - 3]}..."
+    end
+  end
+end

data/app/views/dontbugme/traces/diff.html.erb

@@ -15,4 +15,4 @@
   <p class="empty">Enter two trace IDs above to see what changed between two executions.</p>
 <% end %>
 
-<p style="margin-top: 16px;"><%= link_to '← Back to traces', root_path %></p>
+<p style="margin-top: 16px;"><%= link_to '← Back to traces', root_path, class: 'btn btn-secondary', style: 'display: inline-block;' %></p>

data/app/views/dontbugme/traces/index.html.erb

@@ -1,5 +1,5 @@
 <%= form_with url: root_path, method: :get, local: true, class: 'filters' do |f| %>
-  <input type="text" name="q" value="<%= params[:q] %>" placeholder="Search identifier..." style="min-width: 200px;">
+  <input type="text" name="q" value="<%= params[:q] %>" placeholder="Search identifier..." style="min-width: 220px;">
   <select name="status">
     <option value="">All statuses</option>
     <option value="success" <%= params[:status] == 'success' ? 'selected' : '' %>>Success</option>
@@ -14,15 +14,15 @@
   <button type="submit">Filter</button>
 <% end %>
 
-<div class="card">
+<div class="card traces-list">
   <% if @traces.any? %>
     <% @traces.each do |trace| %>
-      <div class="trace-row">
-        <span class="trace-id"><%= link_to trace.id, trace_path(trace.id) %></span>
+      <%= link_to trace_path(trace.id), class: 'trace-row trace-row-link' do %>
+        <span class="trace-id"><%= trace.id %></span>
         <span class="trace-identifier"><%= trace.identifier %></span>
         <span class="badge <%= trace.status == :success ? 'badge-success' : 'badge-error' %>"><%= trace.status %></span>
         <span class="trace-duration"><%= trace.duration_ms ? "#{trace.duration_ms.round}ms" : '-' %></span>
-      </div>
+      <% end %>
     <% end %>
   <% else %>
     <p class="empty">No traces found. Run your app and execute some jobs or requests.</p>

data/app/views/dontbugme/traces/show.html.erb

@@ -1,15 +1,96 @@
-<div class="card">
-  <h2 style="margin: 0 0 8px; font-size: 16px;"><%= @trace.identifier %></h2>
-  <p style="margin: 0 0 16px; color: #6b7280; font-size: 12px;">
-    <%= @trace.id %> · <%= @trace.status %> · <%= @trace.duration_ms ? "#{@trace.duration_ms.round}ms" : 'N/A' %>
-    <% if @trace.correlation_id.present? %>
-      · <%= link_to "Follow chain", root_path(correlation_id: @trace.correlation_id) %>
+<div class="trace-header card">
+  <div class="trace-header-top">
+    <h2 class="trace-title"><%= @trace.identifier %></h2>
+    <div class="trace-meta">
+      <span class="badge <%= @trace.status == :success ? 'badge-success' : 'badge-error' %>"><%= @trace.status %></span>
+      <span class="trace-meta-item"><%= @trace.duration_ms ? "#{@trace.duration_ms.round}ms" : 'N/A' %></span>
+      <span class="trace-meta-item"><%= @trace.id %></span>
+      <% if @trace.correlation_id.present? %>
+        <%= link_to "Follow chain", root_path(correlation_id: @trace.correlation_id), class: 'trace-link' %>
+      <% end %>
+    </div>
+  </div>
+  <div class="trace-actions">
+    <%= link_to '← Back to traces', root_path, class: 'btn btn-secondary' %>
+    <%= link_to 'Compare with another', diff_path(a: @trace.id), class: 'btn btn-secondary' %>
+  </div>
+</div>
+
+<% if params[:only].present? %>
+  <p class="filter-hint">Filtering by: <strong><%= params[:only] %></strong> · <%= link_to 'Show all', trace_path(@trace.id) %></p>
+<% end %>
+
+<div class="timeline-viz">
+  <div class="timeline-bar" style="--total-ms: <%= [@trace.duration_ms || 1, 1].max %>;">
+    <% @trace.raw_spans.each do |span| %>
+      <% next if params[:only].present? && span.category.to_s != params[:only] %>
+      <% width = span.duration_ms.to_f > 0 ? [(span.duration_ms / (@trace.duration_ms || 1) * 100).round(1), 2].max : 2 %>
+      <span class="timeline-segment badge-<%= span.category %>" style="width: <%= width %>%;" title="<%= span.category %> <%= span.duration_ms.round(1) %>ms"></span>
     <% end %>
-  </p>
-  <div class="timeline"><%= Formatters::Timeline.format(@trace) %></div>
+  </div>
+</div>
+
+<div class="spans-list">
+  <% spans_to_show = params[:only].present? ? @trace.raw_spans.select { |s| s.category.to_s == params[:only] } : @trace.raw_spans %>
+  <% spans_to_show.each_with_index do |span, idx| %>
+    <details class="span-card card" data-category="<%= span.category %>">
+      <summary class="span-summary">
+        <span class="span-offset"><%= span.started_at.to_f.round(1) %>ms</span>
+        <span class="span-duration"><%= span.duration_ms ? "#{span.duration_ms.round(1)}ms" : '-' %></span>
+        <span class="badge <%= span_category_badge_class(span.category) %>"><%= span.category.to_s.upcase %></span>
+        <span class="span-operation"><%= span.operation %></span>
+        <span class="span-detail-preview"><%= truncate_detail(span.detail, 80) %></span>
+      </summary>
+      <div class="span-details">
+        <% if span_output(span).present? %>
+          <div class="span-section span-output-section">
+            <h4>Output <button type="button" class="copy-btn" data-copy-target="output-<%= idx %>" title="Copy">Copy</button></h4>
+            <pre class="code-block output-block" id="output-<%= idx %>"><code><%= h(span_output(span)) %></code></pre>
+          </div>
+        <% end %>
+        <div class="span-section">
+          <h4>Detail <button type="button" class="copy-btn" data-copy-target="detail-<%= idx %>" title="Copy">Copy</button></h4>
+          <pre class="code-block" id="detail-<%= idx %>"><code><%= h(format_span_detail_for_display(span)) %></code></pre>
+        </div>
+        <% if format_span_payload(span).any? %>
+          <div class="span-section">
+            <h4>Inputs</h4>
+            <dl class="payload-list">
+              <% format_span_payload(span).each do |key, value| %>
+                <div class="payload-row">
+                  <dt><%= key %></dt>
+                  <dd><pre class="payload-value"><%= h(value) %></pre></dd>
+                </div>
+              <% end %>
+            </dl>
+          </div>
+        <% end %>
+        <% if span.source.present? %>
+          <div class="span-section">
+            <h4>Source</h4>
+            <code class="source-location"><%= h(span.source) %></code>
+          </div>
+        <% end %>
+      </div>
+    </details>
+  <% end %>
 </div>
 
-<p style="margin-top: 16px;">
-  <%= link_to '← Back to traces', root_path %> ·
-  <%= link_to 'Compare with another trace', diff_path(a: @trace.id) %>
-</p>
+<% if @trace.truncated_spans_count.to_i.positive? %>
+  <p class="truncated-notice">+ <%= @trace.truncated_spans_count %> additional spans truncated</p>
+<% end %>
+
+<div class="trace-footer">
+  <p>Filter by category: <%= link_to 'All', trace_path(@trace.id) %> · <%= link_to 'SQL', trace_path(@trace.id, only: 'sql') %> · <%= link_to 'HTTP', trace_path(@trace.id, only: 'http') %> · <%= link_to 'Redis', trace_path(@trace.id, only: 'redis') %> · <%= link_to 'Cache', trace_path(@trace.id, only: 'cache') %> · <%= link_to 'Mailer', trace_path(@trace.id, only: 'mailer') %> · <%= link_to 'Enqueue', trace_path(@trace.id, only: 'enqueue') %></p>
+</div>
+
+<% if @trace.error.present? %>
+  <div class="card error-card">
+    <h3>Error</h3>
+    <pre class="code-block error-detail"><%= h(@trace.error[:message] || @trace.error['message']) %></pre>
+    <% bt = @trace.error[:backtrace] || @trace.error['backtrace'] %>
+    <% if bt.present? %>
+      <pre class="code-block backtrace"><%= Array(bt).map { |l| h(l) }.join("\n") %></pre>
+    <% end %>
+  </div>
+<% end %>

data/app/views/layouts/dontbugme/application.html.erb

@@ -3,41 +3,110 @@
 <head>
   <meta charset="UTF-8">
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  <%= csrf_meta_tags %>
   <title>Dontbugme</title>
   <style>
     * { box-sizing: border-box; }
-    body { font-family: ui-monospace, SFMono-Regular, "SF Mono", Menlo, monospace; font-size: 13px; line-height: 1.5; color: #1a1a1a; background: #f8f9fa; margin: 0; padding: 0; }
-    .container { max-width: 1000px; margin: 0 auto; padding: 24px; }
-    header { background: #fff; border-bottom: 1px solid #e5e7eb; padding: 16px 24px; margin-bottom: 24px; }
-    header h1 { margin: 0; font-size: 18px; font-weight: 600; }
+    body { font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, sans-serif; font-size: 14px; line-height: 1.5; color: #1a1a1a; background: #0f172a; margin: 0; padding: 0; }
+    .container { max-width: 1200px; margin: 0 auto; padding: 24px; }
+    header { background: #1e293b; border-bottom: 1px solid #334155; padding: 16px 24px; margin-bottom: 24px; }
+    header h1 { margin: 0; font-size: 18px; font-weight: 600; color: #f8fafc; }
     header nav { margin-top: 12px; }
-    header a { color: #4b5563; text-decoration: none; margin-right: 16px; }
-    header a:hover { color: #111; }
-    header a.active { color: #111; font-weight: 600; }
-    .card { background: #fff; border-radius: 6px; border: 1px solid #e5e7eb; padding: 16px; margin-bottom: 16px; }
-    .trace-row { display: flex; align-items: center; gap: 16px; padding: 0 0 12px; border-bottom: 1px solid #f3f4f6; }
-    .trace-row:last-child { border-bottom: none; padding-bottom: 0; }
-    .trace-row:hover { background: #f9fafb; margin: 0 -16px; padding: 0 16px 12px; }
-    .trace-id { font-family: monospace; font-size: 12px; color: #6b7280; min-width: 140px; }
-    .trace-id a { color: #2563eb; text-decoration: none; }
+    header a { color: #94a3b8; text-decoration: none; margin-right: 16px; }
+    header a:hover { color: #f8fafc; }
+    header a.active { color: #38bdf8; font-weight: 600; }
+    .card { background: #1e293b; border-radius: 8px; border: 1px solid #334155; padding: 20px; margin-bottom: 16px; }
+    .trace-row { display: flex; align-items: center; gap: 16px; padding: 12px 16px; border-radius: 6px; transition: background 0.15s; }
+    .trace-row:hover { background: #334155; }
+    .trace-row-link { text-decoration: none; color: inherit; display: flex; }
+    .trace-id { font-family: "JetBrains Mono", "Fira Code", monospace; font-size: 12px; color: #64748b; min-width: 140px; }
+    .trace-id a { color: #38bdf8; text-decoration: none; }
     .trace-id a:hover { text-decoration: underline; }
-    .trace-identifier { flex: 1; }
-    .badge { display: inline-block; padding: 2px 8px; border-radius: 4px; font-size: 11px; font-weight: 500; }
-    .badge-success { background: #d1fae5; color: #065f46; }
-    .badge-error { background: #fee2e2; color: #991b1b; }
-    .trace-duration { color: #6b7280; min-width: 50px; }
+    .trace-identifier { flex: 1; color: #e2e8f0; }
+    .badge { display: inline-block; padding: 3px 10px; border-radius: 6px; font-size: 11px; font-weight: 600; }
+    .badge-success { background: #064e3b; color: #34d399; }
+    .badge-error { background: #7f1d1d; color: #f87171; }
+    .badge-sql { background: #1e3a5f; color: #7dd3fc; }
+    .badge-http { background: #422006; color: #fbbf24; }
+    .badge-redis { background: #4c1d95; color: #c4b5fd; }
+    .badge-cache { background: #14532d; color: #86efac; }
+    .badge-mailer { background: #701a75; color: #f0abfc; }
+    .badge-enqueue { background: #1e293b; color: #94a3b8; }
+    .badge-custom { background: #0f172a; color: #cbd5e1; }
+    .badge-snapshot { background: #312e81; color: #a5b4fc; }
+    .badge-default { background: #334155; color: #94a3b8; }
+    .trace-duration { color: #64748b; min-width: 60px; font-family: monospace; }
     .filters { display: flex; gap: 12px; margin-bottom: 16px; flex-wrap: wrap; }
-    .filters input, .filters select { padding: 8px 12px; border: 1px solid #e5e7eb; border-radius: 4px; font-size: 13px; }
-    .filters button { padding: 8px 16px; background: #111; color: #fff; border: none; border-radius: 4px; cursor: pointer; font-size: 13px; }
-    .filters button:hover { background: #333; }
-    .timeline { font-size: 12px; white-space: pre-wrap; font-family: ui-monospace, monospace; }
-    .span-line { padding: 4px 0; }
-    .span-source { color: #6b7280; padding-left: 12px; font-size: 11px; }
+    .filters input, .filters select { padding: 10px 14px; border: 1px solid #334155; border-radius: 6px; font-size: 13px; background: #1e293b; color: #e2e8f0; }
+    .filters button { padding: 10px 20px; background: #38bdf8; color: #0f172a; border: none; border-radius: 6px; cursor: pointer; font-size: 13px; font-weight: 600; }
+    .filters button:hover { background: #7dd3fc; }
     .diff-form { display: flex; gap: 12px; align-items: center; margin-bottom: 24px; }
-    .diff-form input { flex: 1; padding: 8px 12px; border: 1px solid #e5e7eb; border-radius: 4px; }
-    .diff-output { white-space: pre-wrap; font-size: 12px; background: #1f2937; color: #e5e7eb; padding: 16px; border-radius: 6px; overflow-x: auto; }
-    .empty { color: #6b7280; padding: 24px; text-align: center; }
+    .diff-form input { flex: 1; padding: 10px 14px; border: 1px solid #334155; border-radius: 6px; background: #1e293b; color: #e2e8f0; }
+    .diff-output { white-space: pre-wrap; font-size: 12px; background: #0f172a; color: #e2e8f0; padding: 20px; border-radius: 8px; overflow-x: auto; font-family: monospace; }
+    .empty { color: #64748b; padding: 24px; text-align: center; }
+    .trace-header { display: flex; flex-wrap: wrap; justify-content: space-between; align-items: flex-start; gap: 16px; }
+    .trace-title { margin: 0 0 8px; font-size: 18px; font-weight: 600; color: #f8fafc; }
+    .trace-meta { display: flex; align-items: center; gap: 12px; flex-wrap: wrap; }
+    .trace-meta-item { color: #64748b; font-size: 13px; }
+    .trace-link { color: #38bdf8; text-decoration: none; }
+    .trace-link:hover { text-decoration: underline; }
+    .trace-actions { display: flex; gap: 8px; }
+    .btn { padding: 8px 16px; border-radius: 6px; font-size: 13px; text-decoration: none; font-weight: 500; }
+    .btn-secondary { background: #334155; color: #e2e8f0; }
+    .btn-secondary:hover { background: #475569; }
+    .timeline-viz { margin-bottom: 24px; }
+    .timeline-bar { display: flex; height: 24px; border-radius: 6px; overflow: hidden; background: #334155; }
+    .timeline-segment { min-width: 2px; transition: opacity 0.2s; }
+    .timeline-segment:hover { opacity: 0.8; }
+    .spans-list { display: flex; flex-direction: column; gap: 8px; }
+    .span-card { padding: 0; overflow: hidden; }
+    .span-card summary { padding: 14px 20px; cursor: pointer; display: flex; align-items: center; gap: 16px; flex-wrap: wrap; list-style: none; }
+    .span-card summary::-webkit-details-marker { display: none; }
+    .span-card summary::after { content: '▶'; color: #64748b; font-size: 10px; margin-left: auto; transition: transform 0.2s; }
+    .span-card[open] summary::after { transform: rotate(90deg); }
+    .span-card summary:hover { background: #334155; }
+    .span-offset { font-family: monospace; font-size: 12px; color: #64748b; min-width: 50px; }
+    .span-duration { font-family: monospace; font-size: 12px; color: #38bdf8; min-width: 55px; }
+    .span-operation { color: #94a3b8; font-size: 12px; }
+    .span-detail-preview { color: #cbd5e1; font-size: 13px; flex: 1; overflow: hidden; text-overflow: ellipsis; white-space: nowrap; }
+    .span-details { padding: 0 20px 20px; border-top: 1px solid #334155; }
+    .span-section { margin-top: 16px; }
+    .span-section h4 { margin: 0 0 8px; font-size: 11px; font-weight: 600; color: #64748b; text-transform: uppercase; letter-spacing: 0.05em; }
+    .code-block { background: #0f172a; color: #e2e8f0; padding: 16px; border-radius: 6px; overflow-x: auto; font-family: "JetBrains Mono", "Fira Code", monospace; font-size: 12px; line-height: 1.6; margin: 0; white-space: pre-wrap; word-break: break-all; }
+    .payload-list { margin: 0; }
+    .payload-row { display: grid; grid-template-columns: 140px 1fr; gap: 12px; margin-bottom: 12px; align-items: start; }
+    .payload-row dt { margin: 0; color: #64748b; font-size: 12px; font-weight: 500; }
+    .payload-value { margin: 0; padding: 8px 12px; background: #0f172a; border-radius: 4px; font-size: 12px; color: #e2e8f0; overflow-x: auto; white-space: pre-wrap; font-family: monospace; }
+    .source-location { display: block; padding: 8px 12px; background: #0f172a; border-radius: 4px; font-size: 11px; color: #94a3b8; }
+    .filter-hint { color: #94a3b8; margin-bottom: 16px; font-size: 13px; }
+    .filter-hint a { color: #38bdf8; }
+    .trace-footer { margin-top: 24px; }
+    .trace-footer p { color: #64748b; font-size: 13px; }
+    .trace-footer a { color: #38bdf8; }
+    .truncated-notice { color: #64748b; font-size: 13px; margin-top: 8px; }
+    .error-card { border-color: #7f1d1d; }
+    .error-card h3 { color: #f87171; margin: 0 0 12px; }
+    .error-detail { color: #f87171; }
+    .backtrace { font-size: 11px; color: #94a3b8; }
+    .span-section h4 { display: flex; align-items: center; gap: 8px; }
+    .span-output-section .output-block { border-left: 3px solid #38bdf8; }
+    .copy-btn { background: #334155; color: #94a3b8; border: none; padding: 4px 10px; border-radius: 4px; font-size: 11px; cursor: pointer; }
+    .copy-btn:hover { background: #475569; color: #e2e8f0; }
   </style>
+  <script>
+    document.querySelectorAll('.copy-btn').forEach(function(btn) {
+      btn.addEventListener('click', function() {
+        var target = document.getElementById(this.dataset.copyTarget);
+        if (!target) return;
+        var text = target.textContent;
+        navigator.clipboard.writeText(text).then(function() {
+          var orig = btn.textContent;
+          btn.textContent = 'Copied!';
+          setTimeout(function() { btn.textContent = orig; }, 1500);
+        });
+      });
+    });
+  </script>
 </head>
 <body>
   <header>
@@ -49,7 +118,7 @@
       </nav>
     </div>
   </header>
-  <main class="container">
+  <main class="container" style="color: #e2e8f0;">
     <%= yield %>
   </main>
 </body>

data/{lib/dontbugme/config → config}/routes.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-Rails.application.routes.draw do
-  root to: 'traces#index', as: :root
+Dontbugme::Engine.routes.draw do
+  root to: 'traces#index'
   get 'diff', to: 'traces#diff', as: :diff
   get ':id', to: 'traces#show', as: :trace, constraints: { id: /tr_[a-f0-9]+/ }
 end

data/lib/dontbugme/configuration.rb

@@ -16,6 +16,8 @@ module Dontbugme
                   :capture_http_headers,
                   :capture_http_body,
                   :capture_redis_values,
+                  :capture_redis_return_values,
+                  :capture_span_output,
                   :source_mode,
                   :source_filter,
                   :source_depth,
@@ -63,6 +65,8 @@ module Dontbugme
       self.capture_http_headers = []
       self.capture_http_body = false
       self.capture_redis_values = false
+      self.capture_redis_return_values = true
+      self.capture_span_output = true
       self.source_mode = :full
       self.source_filter = %w[app/ lib/]
       self.source_depth = 3
@@ -95,6 +99,8 @@ module Dontbugme
       self.recording_mode = :selective
       self.record_on_error = true
       self.capture_sql_binds = false
+      self.capture_redis_return_values = false
+      self.capture_span_output = false
       self.source_mode = :shallow
       self.source_depth = 1
       self.source_stack_limit = 30
@@ -134,13 +140,11 @@ module Dontbugme
     private
 
     def retention_seconds(value, unit)
-      return value if value.is_a?(Integer)
-      return value.to_i if defined?(ActiveSupport) && value.respond_to?(:to_i)
-
       case unit
-      when :hours then value * 3600
-      when :days then value * 86400
-      else value
+      when nil, :seconds then value.to_i
+      when :hours then (value.to_f * 3600).to_i
+      when :days then (value.to_f * 86400).to_i
+      else value.to_i
       end
     end
   end

data/lib/dontbugme/middleware/rack.rb

@@ -9,6 +9,7 @@ module Dontbugme
 
       def call(env)
         return @app.call(env) unless Dontbugme.config.recording?
+        return @app.call(env) unless Dontbugme.config.should_record_request?(env)
 
         request = ::Rack::Request.new(env)
         request_id = env['action_dispatch.request_id'] || request.get_header('HTTP_X_REQUEST_ID') || SecureRandom.uuid

data/lib/dontbugme/middleware/sidekiq.rb

@@ -7,6 +7,7 @@ module Dontbugme
         return yield unless Dontbugme.config.recording?
 
         job_class = job['class'] || job[:class] || 'Unknown'
+        return yield unless Dontbugme.config.should_record_job?(job_class)
         jid = job['jid'] || job[:jid] || SecureRandom.hex(8)
         correlation_id = job['correlation_id'] || job[:correlation_id] || Correlation.current
         Correlation.current = correlation_id

data/lib/dontbugme/span.rb

@@ -38,7 +38,7 @@ module Dontbugme
     end
 
     def self.from_h(hash)
-      new(
+      span = new(
         category: hash[:category] || hash['category'],
         operation: hash[:operation] || hash['operation'],
         detail: hash[:detail] || hash['detail'],
@@ -47,6 +47,8 @@ module Dontbugme
         duration_ms: hash[:duration_ms] || hash['duration_ms'],
         source: hash[:source] || hash['source']
       )
+      span.instance_variable_set(:@id, (hash[:id] || hash['id']).to_s) if hash[:id] || hash['id']
+      span
     end
 
     private

data/lib/dontbugme/store/async.rb

@@ -6,10 +6,12 @@ module Dontbugme
       def initialize(backend)
         @backend = backend
         @queue = Queue.new
+        @pid = Process.pid
         @thread = start_worker
       end
 
       def save_trace(trace)
+        restart_worker_if_forked
         @queue << [:save, trace]
       end
 
@@ -22,11 +24,19 @@ module Dontbugme
       end
 
       def cleanup(before:)
+        restart_worker_if_forked
         @queue << [:cleanup, before]
       end
 
       private
 
+      def restart_worker_if_forked
+        return if Process.pid == @pid
+
+        @pid = Process.pid
+        @thread = start_worker
+      end
+
       def start_worker
         Thread.new do
           loop do
@@ -37,6 +47,8 @@ module Dontbugme
             when :cleanup
               @backend.cleanup(before: arg)
             end
+          rescue StandardError => e
+            warn "[Dontbugme] Async store error: #{e.class} #{e.message}"
           end
         end
       end

data/lib/dontbugme/store/memory.rb

@@ -24,7 +24,7 @@ module Dontbugme
       def search(filters = {})
         traces = @mutex.synchronize { @traces.values.dup }
         traces = apply_filters(traces, filters)
-        traces.sort_by { |t| t[:started_at] || '' }.reverse.map { |h| Trace.from_h(h) }
+        traces.map { |h| Trace.from_h(h) }
       end
 
       def cleanup(before:)
@@ -37,8 +37,8 @@ module Dontbugme
       private
 
       def apply_filters(traces, filters)
-        traces = traces.select { |t| t[:status] == filters[:status].to_s } if filters[:status]
-        traces = traces.select { |t| t[:status] == filters['status'].to_s } if filters['status']
+        traces = traces.select { |t| t[:status].to_s == filters[:status].to_s } if filters[:status]
+        traces = traces.select { |t| t[:status].to_s == filters['status'].to_s } if filters['status']
         traces = traces.select { |t| t[:kind].to_s == filters[:kind].to_s } if filters[:kind]
         if filters[:identifier]
           pattern = /#{Regexp.escape(filters[:identifier].to_s)}/i
@@ -48,8 +48,8 @@ module Dontbugme
           cid = filters[:correlation_id].to_s
           traces = traces.select { |t| (t[:correlation_id] || t.dig(:metadata, :correlation_id)).to_s == cid }
         end
-        traces = traces.first(filters[:limit] || filters['limit'] || 100)
-        traces
+        limit = filters[:limit] || filters['limit'] || 100
+        traces.sort_by { |t| t[:started_at] || '' }.reverse.first(limit)
       end
 
       def parse_time(val)

data/lib/dontbugme/store/postgresql.rb

@@ -14,22 +14,7 @@ module Dontbugme
       def save_trace(trace)
         data = trace.to_h
         correlation_id = data[:correlation_id] || data[:metadata]&.dig(:correlation_id)
-        exec_params(
-          <<~SQL,
-            INSERT INTO dontbugme_traces
-            (id, kind, identifier, status, started_at, duration_ms, correlation_id, metadata_json, spans_json, error_json)
-            VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
-            ON CONFLICT (id) DO UPDATE SET
-              kind = EXCLUDED.kind,
-              identifier = EXCLUDED.identifier,
-              status = EXCLUDED.status,
-              started_at = EXCLUDED.started_at,
-              duration_ms = EXCLUDED.duration_ms,
-              correlation_id = EXCLUDED.correlation_id,
-              metadata_json = EXCLUDED.metadata_json,
-              spans_json = EXCLUDED.spans_json,
-              error_json = EXCLUDED.error_json
-          SQL
+        params = [
           data[:id],
           data[:kind].to_s,
           data[:identifier],
@@ -40,7 +25,22 @@ module Dontbugme
           data[:metadata].to_json,
           data[:spans].to_json,
           data[:error]&.to_json
-        )
+        ]
+        exec_params(<<~SQL, params)
+          INSERT INTO dontbugme_traces
+          (id, kind, identifier, status, started_at, duration_ms, correlation_id, metadata_json, spans_json, error_json)
+          VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
+          ON CONFLICT (id) DO UPDATE SET
+            kind = EXCLUDED.kind,
+            identifier = EXCLUDED.identifier,
+            status = EXCLUDED.status,
+            started_at = EXCLUDED.started_at,
+            duration_ms = EXCLUDED.duration_ms,
+            correlation_id = EXCLUDED.correlation_id,
+            metadata_json = EXCLUDED.metadata_json,
+            spans_json = EXCLUDED.spans_json,
+            error_json = EXCLUDED.error_json
+        SQL
       end
 
       def find_trace(trace_id)
@@ -92,6 +92,8 @@ module Dontbugme
         exec_params('DELETE FROM dontbugme_traces WHERE started_at < $1', [cutoff])
       end
 
+      private
+
       def query_result(sql, params)
         if conn.respond_to?(:exec_query)
           conn.exec_query(sql, 'Dontbugme', params)
@@ -106,8 +108,6 @@ module Dontbugme
         row
       end
 
-      private
-
       def conn
         @connection
       end
@@ -159,8 +159,8 @@ module Dontbugme
         conn.execute('CREATE INDEX IF NOT EXISTS idx_dontbugme_started_at ON dontbugme_traces(started_at)')
         conn.execute('CREATE INDEX IF NOT EXISTS idx_dontbugme_status ON dontbugme_traces(status)')
         conn.execute('CREATE INDEX IF NOT EXISTS idx_dontbugme_correlation_id ON dontbugme_traces(correlation_id)')
-      rescue StandardError
-        # Schema might already exist
+      rescue StandardError => e
+        raise e if e.message !~ /already exists/i
       end
 
       def row_to_trace(row)

data/lib/dontbugme/store/sqlite.rb

@@ -75,7 +75,10 @@ module Dontbugme
       def db
         @db ||= begin
           require 'sqlite3'
-          SQLite3::Database.new(@path)
+          db = SQLite3::Database.new(@path)
+          db.execute('PRAGMA journal_mode=WAL')
+          db.execute('PRAGMA busy_timeout=5000')
+          db
         end
       end
 

data/lib/dontbugme/subscribers/action_mailer.rb

@@ -2,7 +2,7 @@
 
 module Dontbugme
   module Subscribers
-    class ActionMailer
+    class ActionMailer < Base
       EVENT = 'deliver.action_mailer'
 
       def self.subscribe

data/lib/dontbugme/subscribers/active_job.rb

@@ -2,7 +2,7 @@
 
 module Dontbugme
   module Subscribers
-    class ActiveJob
+    class ActiveJob < Base
       EVENT = 'enqueue.active_job'
 
       def self.subscribe

data/lib/dontbugme/subscribers/net_http.rb

@@ -41,6 +41,10 @@ module Dontbugme
           }
           payload[:error] = error.message if error
 
+          if config.capture_http_body && response&.body
+            payload[:response_body] = truncate(response.body, config.max_http_body_size)
+          end
+
           if config.capture_http_headers&.any?
             payload[:request_headers] = capture_headers(req, config.capture_http_headers)
           end

data/lib/dontbugme/subscribers/redis.rb

@@ -20,7 +20,7 @@ module Dontbugme
           result = super
           duration_ms = (Process.clock_gettime(Process::CLOCK_MONOTONIC, :float_millisecond) - start_time).round(2)
 
-          record_span(command, start_wall, duration_ms)
+          record_span(command, start_wall, duration_ms, result: result)
           result
         rescue StandardError => e
           duration_ms = (Process.clock_gettime(Process::CLOCK_MONOTONIC, :float_millisecond) - start_time).round(2)
@@ -30,7 +30,7 @@ module Dontbugme
 
         private
 
-        def record_span(command, start_wall, duration_ms, error: nil)
+        def record_span(command, start_wall, duration_ms, result: nil, error: nil)
           cmd = Array(command).map(&:to_s)
           operation = cmd.first&.upcase || 'UNKNOWN'
           detail = cmd.join(' ')
@@ -40,6 +40,9 @@ module Dontbugme
           if config.capture_redis_values && cmd.size > 1
             payload[:args] = cmd[1..].map { |a| truncate(a, config.max_redis_value_size) }
           end
+          if config.capture_redis_return_values && result && !error
+            payload[:output] = truncate_value(result, config.max_redis_value_size)
+          end
           payload[:error] = error.message if error
 
           Dontbugme::Recorder.add_span(
@@ -57,6 +60,13 @@ module Dontbugme
 
           "#{str.to_s.byteslice(0, max)}[truncated]"
         end
+
+        def truncate_value(val, max)
+          str = val.to_s
+          return str if str.bytesize <= max
+
+          "#{str.byteslice(0, max)}[truncated]"
+        end
       end
     end
   end

data/lib/dontbugme/trace.rb

@@ -43,7 +43,7 @@ module Dontbugme
     end
 
     def raw_spans
-      @spans.freeze
+      @spans.dup
     end
 
     def finish!(error: nil)
@@ -74,12 +74,13 @@ module Dontbugme
     end
 
     def to_h
+      finished_at_time = @finished_at ? (@started_at_utc + (duration_ms || 0) / 1000.0) : nil
       {
         id: id,
         kind: kind,
         identifier: identifier,
         started_at: format_time(started_at_utc),
-        finished_at: @finished_at ? format_time(Time.at(@finished_at / 1000.0).utc) : nil,
+        finished_at: finished_at_time ? format_time(finished_at_time) : nil,
         duration_ms: duration_ms,
         status: status,
         error: error,

data/lib/dontbugme/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Dontbugme
-  VERSION = '0.1.3'
+  VERSION = '0.1.5'
 end

data/lib/dontbugme.rb

@@ -24,7 +24,7 @@ module Dontbugme
       Recorder.record(kind: :custom, identifier: identifier, metadata: metadata, return_trace: true, &block)
     end
 
-    def span(name, payload: {}, &block)
+    def span(name, payload: {}, capture_output: true, &block)
       return yield unless Context.active?
 
       start_mono = Process.clock_gettime(Process::CLOCK_MONOTONIC, :float_millisecond)
@@ -32,11 +32,16 @@ module Dontbugme
       result = yield
       duration_ms = (Process.clock_gettime(Process::CLOCK_MONOTONIC, :float_millisecond) - start_mono).round(2)
 
+      span_payload = payload.dup
+      if capture_output && config.capture_span_output
+        span_payload[:output] = format_output_value(result)
+      end
+
       Recorder.add_span(
         category: :custom,
         operation: 'span',
         detail: name.to_s,
-        payload: payload,
+        payload: span_payload,
         duration_ms: duration_ms,
         started_at: start_wall
       )
@@ -65,6 +70,52 @@ module Dontbugme
 
     private
 
+    def format_output_value(val)
+      return nil if val.nil?
+
+      max = config.max_span_detail_size
+      str = if val.is_a?(Array)
+              format_array(val)
+            elsif val.is_a?(Hash)
+              format_hash(val)
+            elsif defined?(ActiveRecord::Base) && val.is_a?(ActiveRecord::Base)
+              val.inspect
+            elsif defined?(ActiveRecord::Relation) && val.is_a?(ActiveRecord::Relation)
+              "#{val.to_sql} (relation)"
+            else
+              val.to_s
+            end
+      str.bytesize > max ? "#{str.byteslice(0, max)}...[truncated]" : str
+    rescue StandardError
+      val.class.name
+    end
+
+    def format_array(ary)
+      return '[]' if ary.empty?
+
+      preview = ary.first(5).map { |v| format_single_value(v) }.join(', ')
+      ary.size > 5 ? "[#{preview}, ... (#{ary.size} total)]" : "[#{preview}]"
+    end
+
+    def format_hash(hash)
+      return '{}' if hash.empty?
+
+      preview = hash.first(5).map { |k, v| "#{k}: #{format_single_value(v)}" }.join(', ')
+      hash.size > 5 ? "{#{preview}, ...}" : "{#{preview}}"
+    end
+
+    def format_single_value(v)
+      if defined?(ActiveRecord::Base) && v.is_a?(ActiveRecord::Base)
+        v.respond_to?(:id) ? "#<#{v.class.name} id=#{v.id}>" : "#<#{v.class.name}>"
+      elsif v.is_a?(Hash)
+        '{...}'
+      elsif v.is_a?(Array)
+        '[...]'
+      else
+        v.to_s
+      end
+    end
+
     def build_store
       store = case config.store
               when :sqlite

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: dontbugme
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.5
 platform: ruby
 authors:
 - Inspector Contributors
@@ -122,15 +122,16 @@ files:
 - LICENSE
 - README.md
 - app/controllers/dontbugme/traces_controller.rb
+- app/helpers/dontbugme/traces_helper.rb
 - app/views/dontbugme/traces/diff.html.erb
 - app/views/dontbugme/traces/index.html.erb
 - app/views/dontbugme/traces/show.html.erb
 - app/views/layouts/dontbugme/application.html.erb
 - bin/dontbugme
+- config/routes.rb
 - lib/dontbugme.rb
 - lib/dontbugme/cleanup_job.rb
 - lib/dontbugme/cli.rb
-- lib/dontbugme/config/routes.rb
 - lib/dontbugme/configuration.rb
 - lib/dontbugme/context.rb
 - lib/dontbugme/correlation.rb