claude-matrix 1.0.3 → 1.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f3f9b8a20130e696e07a7cbeb49f63da290f3b28d48f81cac0733c71f141a6a1
-  data.tar.gz: c05db374a55b0ce90d261a72f9eb09d148fbde729601484c1be1884d03df7130
+  metadata.gz: ac8dd503ff936a79b5f07dae6c531a47c06124e845bd2799b563f6264216f02c
+  data.tar.gz: 3d427f9e25d31ddcf16a4325ab5553f88624bfa17185b9e3a5a5722a9fb4f5a8
 SHA512:
-  metadata.gz: d8ff3db3c0c1ec863067a7765913367c8880d173c2fb8fbc2957562acb323f30ba1124601fae4397271f32ed2264f05a70294f808b24afe9d4c71b6723017f6a
-  data.tar.gz: 97ba05746729e2d5dc1c423bdd5401a59b649cb2cd14d1997b46110c357d02c4d06688b0f63201e4100b269fa4f86681aa6e3b6e17e7f414ec0d3c6400fde815
+  metadata.gz: c7a759af7385dbcde61f05d5a91032804a499b703e8479e4a5c3d8352f5bbca106aafd7d6dfbd7405a6efd1c09067ea3977bd11a4b8bf1cc32dab4e0d63dbc38
+  data.tar.gz: '0319c7e331e47c51d31500156c378dfb0014e358e259b7445b3af3cea15d0f211b8e44d4582ab106d0a62b11eaa0951771fa0cf7ec4be0f6e350104c0d6f26fb'

data/lib/claude_matrix/analyzers/metrics.rb

@@ -15,6 +15,16 @@ module ClaudeMatrix
 
       DEFAULT_PRICE = { input: 3.0, output: 15.0, cache_read: 0.3, cache_write: 3.75 }.freeze
 
+      STOPWORDS = %w[
+        the a an and or but in on at to for of is it i my with this that you we
+        can be are was were have had do does did will would could should may might
+        must shall from by about as into through during before after above below
+        up down out off over under again further then once here there when where
+        why how all both each few more most other some such no nor not only own
+        same so than too very just because if while please let me make get how
+        what use new add your its im its just also want need its
+      ].freeze
+
       WORK_MODE_TOOLS = {
         exploration: %w[Read Glob Grep WebSearch WebFetch read_file list_files search_files grep],
         building:    %w[Write Edit NotebookEdit write_to_file edit_file new_file],
@@ -42,6 +52,10 @@ module ClaudeMatrix
         projects = sessions.group_by { |s| s[:project] }
         most_active_project = projects.max_by { |_, ss| ss.sum { |s| s[:message_count] } }&.first
 
+        all_prompt_words = sessions.flat_map { |s| s[:prompt_words] || [] }
+        total_prompts    = sessions.sum { |s| s[:prompt_count] || 0 }
+        avg_words        = total_prompts > 0 ? (all_prompt_words.size.to_f / total_prompts).round : 0
+
         {
           total_sessions:    sessions.size,
           total_messages:    sessions.sum { |s| s[:message_count] },
@@ -59,6 +73,7 @@ module ClaudeMatrix
           top_tools:         tool_counts.first(10).to_h,
           favorite_tool:     tool_counts.first&.first,
           daily_activity:    daily,
+          daily_cost:        build_daily_cost(sessions),
           streaks:           streaks,
           hour_counts:       hour_counts,
           hour_day_grid:     build_hour_day_grid(sessions),
@@ -68,6 +83,10 @@ module ClaudeMatrix
           most_productive_day: most_productive_day(daily),
           busiest_hour:      hour_counts.max_by { |_, c| c }&.first,
           models_used:       sessions.map { |s| s[:model] }.compact.uniq,
+          models_breakdown:  build_models_breakdown(sessions),
+          total_prompts:     total_prompts,
+          avg_prompt_words:  avg_words,
+          top_prompt_words:  top_prompt_words(all_prompt_words),
           per_project:       build_per_project(sessions)
         }
       end
@@ -165,6 +184,15 @@ module ClaudeMatrix
         { date: date, sessions: data[:sessions], messages: data[:messages] }
       end
 
+      def self.build_daily_cost(sessions)
+        daily = Hash.new(0.0)
+        sessions.each do |s|
+          date = s[:started_at].to_date.to_s
+          daily[date] += estimate_cost([s])
+        end
+        daily.sort.to_h
+      end
+
       def self.build_hour_day_grid(sessions)
         # 7 rows (wday: 0=Sun..6=Sat) × 24 cols (hours)
         grid = Array.new(7) { Array.new(24, 0) }
@@ -186,6 +214,25 @@ module ClaudeMatrix
         end.sort_by { |_, v| -v[:sessions] }.to_h
       end
 
+      def self.build_models_breakdown(sessions)
+        models = sessions.map { |s| s[:model] }.compact
+        total  = models.size.to_f
+        return {} if total.zero?
+
+        models.tally
+              .sort_by { |_, c| -c }
+              .to_h
+              .transform_values { |c| { count: c, pct: (c / total * 100).round } }
+      end
+
+      def self.top_prompt_words(words)
+        words.tally
+             .reject { |w, _| STOPWORDS.include?(w) || w.length < 3 || w.match?(/^\d+$/) }
+             .sort_by { |_, c| -c }
+             .first(6)
+             .to_h
+      end
+
       def self.empty_metrics
         {
           total_sessions: 0, total_messages: 0, total_tool_calls: 0,
@@ -194,11 +241,13 @@ module ClaudeMatrix
           estimated_cost: 0.0, first_session: nil, projects: 0,
           project_names: [], most_active_project: nil,
           tool_counts: {}, top_tools: {}, favorite_tool: nil,
-          daily_activity: {}, streaks: { current: 0, longest: 0, active_days: 0 },
+          daily_activity: {}, daily_cost: {}, streaks: { current: 0, longest: 0, active_days: 0 },
           hour_counts: {}, work_modes: { exploration: 0, building: 0, testing: 0 },
           longest_session: nil, most_messages_session: nil,
           most_productive_day: nil, busiest_hour: nil,
-          models_used: [], per_project: {}
+          models_used: [], models_breakdown: {},
+          total_prompts: 0, avg_prompt_words: 0, top_prompt_words: {},
+          per_project: {}
         }
       end
     end

data/lib/claude_matrix/readers/session_parser.rb

@@ -49,6 +49,9 @@ module ClaudeMatrix
         session_id    = nil
         cwd           = nil
 
+        prompt_words = []
+        prompt_count = 0
+
         lines.each do |line|
           data = JSON.parse(line)
 
@@ -75,6 +78,22 @@ module ClaudeMatrix
             content.each do |block|
               tools << block["name"] if block["type"] == "tool_use" && block["name"]
             end
+
+          elsif data["type"] == "user"
+            # Extract prompt text (skip tool_result blocks — only human text)
+            content = data.dig("message", "content")
+            texts = case content
+                    when Array  then content.select { |b| b["type"] == "text" }.map { |b| b["text"].to_s }
+                    when String then [content]
+                    else []
+                    end
+            text = texts.join(" ").strip
+            unless text.empty?
+              words = text.split(/\s+/).map { |w| w.downcase.gsub(/[^a-z0-9']/, "") }
+                         .reject { |w| w.length < 2 }
+              prompt_words.concat(words)
+              prompt_count += 1
+            end
           end
         rescue JSON::ParserError
           next
@@ -101,7 +120,9 @@ module ClaudeMatrix
           input_tokens:  input_tokens,
           output_tokens: output_tokens,
           cache_read:    cache_read,
-          cache_create:  cache_create
+          cache_create:  cache_create,
+          prompt_words:  prompt_words,
+          prompt_count:  prompt_count
         }
       rescue => e
         nil

data/lib/claude_matrix/version.rb

@@ -1,3 +1,3 @@
 module ClaudeMatrix
-  VERSION = "1.0.3"
+  VERSION = "1.0.4"
 end

data/lib/claude_matrix/visualizers/dashboard.rb

@@ -199,46 +199,101 @@ module ClaudeMatrix
 
       def build_right
         rows = []
-        rows += section("TOP TOOLS", @rw, tool_lines)
+        rows += tools_and_insights_section
         rows += section("TOKENS & COST", @rw, token_lines)
         rows += section("ACTIVITY  (last 30 days)", @rw, activity_lines)
-        rows += section("HEATMAP  day × hour", @rw, heatmap_lines) if @show_heatmap
+        rows += heatmap_and_cost_section if @show_heatmap
         rows
       end
 
-      # ── Section helper ────────────────────────────────────────────────────
+      # ── Tools + Insights (side by side) ──────────────────────────────────
 
-      def section(title, w, lines)
-        [divr,
-         cr(sec_title(title, w), w)] + lines.map { |l| cr(l, w) }
-      end
+      def tools_and_insights_section
+        @tools_w = [(@rw * 0.56).to_i, 26].max.clamp(26, 52)
+        @ins_w   = @rw - @tools_w - 1
 
-      def sec_title(title, w)
-        fill = [w - vis(title) - 2, 0].max
-        " #{bold(bright_cyan(title))} #{dim("─" * fill)}"
-      end
+        t_lines = raw_tool_lines(@tools_w)
+        i_lines = raw_insights_lines(@ins_w)
+
+        n = [t_lines.size, i_lines.size, @n_tools].max
+        t_lines.fill(pad("", @tools_w), t_lines.size...n)
+        i_lines.fill(pad("", @ins_w),   i_lines.size...n)
+
+        t_title = " #{bold(bright_cyan("TOP TOOLS"))} #{dim("─" * [@tools_w - 12, 0].max)}"
+        i_title = " #{bold(bright_cyan("MODELS & PROMPTS"))} #{dim("─" * [@ins_w - 19, 0].max)}"
+        title_row = cr(pad(t_title, @tools_w) + dim("│") + pad(i_title, @ins_w), @rw)
+
+        content = t_lines.zip(i_lines).map do |tl, il|
+          cr(pad(tl, @tools_w) + dim("│") + pad(il, @ins_w), @rw)
+        end
 
-      # ── Tool bars ─────────────────────────────────────────────────────────
+        [divr, title_row] + content
+      end
 
-      def tool_lines
+      def raw_tool_lines(w)
         tools = @m[:top_tools].to_a
         return [ind("No data yet")] if tools.empty?
 
         max_cnt = tools.first.last.to_f
         cnt_w   = tools.first.last.to_s.length
-        bar_w   = [@rw - 20 - cnt_w, 4].max
+        bar_w   = [w - 18 - cnt_w, 2].max
 
         tools.first(@n_tools).map do |tool, count|
           ratio   = count.to_f / max_cnt
           bw      = [(ratio * bar_w).round, 1].max
-          # Color gradient: high→bright_cyan, mid→cyan, low→blue
           bar_col = ratio > 0.7 ? method(:bright_cyan) : ratio > 0.35 ? method(:cyan) : method(:blue)
           bar     = bar_col.call("█" * bw) + dim("░" * [bar_w - bw, 0].max)
           cnt_s   = bright_white(count.to_s.rjust(cnt_w))
-          "  #{dim(tool[0, 14].ljust(14))} #{bar}  #{cnt_s}"
+          "  #{dim(tool[0, 12].ljust(12))} #{bar}  #{cnt_s}"
         end
       end
 
+      def raw_insights_lines(w)
+        lines  = []
+        mb     = @m[:models_breakdown] || {}
+        bar_w  = [w - 20, 2].max
+        colors = [method(:bright_cyan), method(:cyan), method(:blue), method(:green)]
+
+        if mb.any?
+          mb.each_with_index do |(model, info), idx|
+            short  = model.to_s.sub("claude-", "").sub(/-\d{8,}$/, "").sub(/-(\d+)-(\d+)$/, " \\1.\\2")
+            ratio  = info[:pct] / 100.0
+            bw     = [(ratio * bar_w).round, 1].max
+            col    = colors[idx % colors.size]
+            bar    = col.call("█" * bw) + dim("░" * [bar_w - bw, 0].max)
+            pct_s  = col.call("#{info[:pct]}%".rjust(4))
+            lines << "  #{dim(short[0, 12].ljust(12))} #{bar} #{pct_s}"
+          end
+        else
+          lines << "  #{dim("No model data")}"
+        end
+
+        lines << "  #{dim("─" * [w - 4, 2].max)}"
+
+        tp = @m[:total_prompts]    || 0
+        ap = @m[:avg_prompt_words] || 0
+        lines << "  #{dim("Prompts")} #{bright_cyan(tp.to_s)}  #{dim("avg")} #{bright_white("#{ap} words")}"
+
+        tw = @m[:top_prompt_words] || {}
+        if tw.any?
+          lines << "  #{dim("Top:")} #{bright_yellow(tw.keys.first(5).join(" · "))}"
+        end
+
+        lines
+      end
+
+      # ── Section helper ────────────────────────────────────────────────────
+
+      def section(title, w, lines)
+        [divr,
+         cr(sec_title(title, w), w)] + lines.map { |l| cr(l, w) }
+      end
+
+      def sec_title(title, w)
+        fill = [w - vis(title) - 2, 0].max
+        " #{bold(bright_cyan(title))} #{dim("─" * fill)}"
+      end
+
       # ── Token / Cost bars ─────────────────────────────────────────────────
 
       def token_lines
@@ -308,21 +363,83 @@ module ClaudeMatrix
         lines
       end
 
-      # ── Heatmap ───────────────────────────────────────────────────────────
+      # ── Heatmap + Daily Cost (side by side) ──────────────────────────────
+
+      def heatmap_and_cost_section
+        # Split the right column: heatmap left, cost right
+        @hmap_w = [(@rw * 0.54).to_i, 33].max.clamp(33, 50)
+        @cost_w = @rw - @hmap_w - 1  # -1 for inner separator
+
+        h_lines = raw_heatmap_lines
+        c_lines = raw_cost_lines
 
-      def heatmap_lines
+        # Align heights
+        n = [h_lines.size, c_lines.size].max
+        h_lines.fill(pad("", @hmap_w), h_lines.size...n)
+        c_lines.fill(pad("", @cost_w), c_lines.size...n)
+
+        # Build split title row
+        h_title = " #{bold(bright_cyan("HEATMAP"))} #{dim("─" * [@hmap_w - 10, 0].max)}"
+        c_title = " #{bold(bright_cyan("DAILY COST"))} #{dim("─" * [@cost_w - 13, 0].max)}"
+        title_row = cr(pad(h_title, @hmap_w) + dim("│") + pad(c_title, @cost_w), @rw)
+
+        content = h_lines.zip(c_lines).map do |hl, cl|
+          cr(pad(hl, @hmap_w) + dim("│") + pad(cl, @cost_w), @rw)
+        end
+
+        [divr, title_row] + content
+      end
+
+      def raw_heatmap_lines
         grid    = @m[:hour_day_grid]
         max_val = [grid.flatten.max, 1].max.to_f
-        step    = 3
-        hlabels = (0..21).step(step).map { |h| h.to_s.rjust(3) }.join
-        lines   = ["  #{dim("     #{hlabels}")}"]
+
+        # Choose step to fill @hmap_w: prefix "Mon  " = 5 + 1 indent, each cell = 1 + sep chars
+        inner = @hmap_w - 9  # available for cells after label + indent
+        step  = inner >= 46 ? 1 : inner >= 23 ? 2 : 3
+        hours = (0...24).step(step).to_a
+        sep   = " " * step
+
+        # cells string is exactly: n cells + (n-1) separators
+        cells_vis = hours.size + (hours.size - 1) * step
+
+        # Build header: place key-hour labels at their exact cell positions
+        hlabel_arr = Array.new(cells_vis, " ")
+        [0, 6, 12, 18].each do |h|
+          idx_in_hours = hours.index(h); next unless idx_in_hours
+          pos = idx_in_hours * (1 + step)
+          h.to_s.chars.each_with_index { |c, li| hlabel_arr[pos + li] = c if pos + li < cells_vis }
+        end
+        hlabels = hlabel_arr.join
+        lines   = [" #{dim("     #{hlabels}")}"]
+
         DAY_WDAY.each_with_index do |wday, i|
-          cells = (0..21).step(step).map { |h| heat_block(grid[wday][h], max_val) }.join("  ")
-          lines << "  #{cyan(DAY_NAMES[i])}  #{cells}"
+          cells = hours.map { |h| heat_block(grid[wday][h], max_val) }.join(sep)
+          lines << " #{cyan(DAY_NAMES[i])}  #{cells}"
         end
         lines
       end
 
+      def raw_cost_lines
+        daily_cost = @m[:daily_cost]
+        return [" #{dim("No cost data")}"] if daily_cost.empty?
+
+        max_cost = [daily_cost.values.max, 0.01].max.to_f
+        # Layout: " MMM DD  bar  $XXX.XX"
+        # overhead: 1(indent) + 6(label) + 2(gap) + 1(space) = 10, amount up to 7 → bar_w = @cost_w - 17
+        bar_w  = [@cost_w - 17, 2].max
+        n_days = [daily_cost.size, 14].min
+
+        daily_cost.to_a.last(n_days).reverse.map do |date, cost|
+          label  = Date.parse(date).strftime("%b %-d").rjust(6)
+          ratio  = cost / max_cost
+          bw     = [(ratio * bar_w).round, cost > 0 ? 1 : 0].max
+          bar    = cost_color(cost, "█" * bw) + dim("░" * [bar_w - bw, 0].max)
+          amount = cost_color(cost, "$#{"%.2f" % cost}")
+          " #{dim(label)}  #{bar} #{amount}"
+        end
+      end
+
       def heat_block(count, max)
         r = count.to_f / max
         if    r == 0   then dim("░")

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: claude-matrix
 version: !ruby/object:Gem::Version
-  version: 1.0.3
+  version: 1.0.4
 platform: ruby
 authors:
 - Kapil Bhosale