RubyGems - claude_usage - Versions diffs - 0.1.2 - Mend

claude_usage 0.1.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (26) hide show

checksums.yaml +7 -0
data/.rubocop.yml +74 -0
data/CHANGELOG.md +53 -0
data/CODE_OF_CONDUCT.md +132 -0
data/LICENSE.txt +21 -0
data/README.md +212 -0
data/Rakefile +8 -0
data/app/controllers/claude_usage/application_controller.rb +11 -0
data/app/controllers/claude_usage/usage_controller.rb +55 -0
data/app/helpers/claude_usage/usage_helper.rb +20 -0
data/app/views/claude_usage/usage/index.html.erb +298 -0
data/config/routes.rb +6 -0
data/lib/claude_usage/cost_calculator.rb +57 -0
data/lib/claude_usage/engine.rb +23 -0
data/lib/claude_usage/file_reader.rb +112 -0
data/lib/claude_usage/formatters.rb +32 -0
data/lib/claude_usage/litellm_pricing_fetcher.rb +148 -0
data/lib/claude_usage/project_name_resolver.rb +23 -0
data/lib/claude_usage/usage_aggregator.rb +55 -0
data/lib/claude_usage/version.rb +5 -0
data/lib/claude_usage.rb +52 -0
data/lib/generators/claude_usage/install_generator.rb +30 -0
data/lib/generators/claude_usage/templates/initializer.rb +34 -0
data/lib/tasks/claude_usage_tasks.rake +313 -0
data/sig/claude_usage.rbs +4 -0
metadata +139 -0

data/app/views/claude_usage/usage/index.html.erb ADDED Viewed

@@ -0,0 +1,298 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <title>Claude Code Usage Report</title>
+  <style>
+    * {
+      margin: 0;
+      padding: 0;
+      box-sizing: border-box;
+    }
+    body {
+      font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, Oxygen, Ubuntu, Cantarell, sans-serif;
+      background: #f9fafb;
+      min-height: 100vh;
+      padding: 2rem;
+    }
+    h1 {
+      color: #111827;
+      text-align: center;
+      margin-bottom: 2rem;
+      font-size: 2.5rem;
+    }
+    .container {
+      max-width: 1200px;
+      margin: 0 auto;
+    }
+    .error {
+      background: #fee;
+      border: 2px solid #f66;
+      color: #c33;
+      padding: 1.5rem;
+      border-radius: 8px;
+      margin-bottom: 2rem;
+    }
+    .totals-grid {
+      display: grid;
+      grid-template-columns: repeat(auto-fit, minmax(250px, 1fr));
+      gap: 1.5rem;
+      margin-bottom: 2rem;
+    }
+    .stat-card {
+      background: white;
+      padding: 1.5rem;
+      border-radius: 12px;
+      box-shadow: 0 4px 6px rgba(0,0,0,0.1);
+      transition: transform 0.2s;
+    }
+    .stat-card:hover {
+      transform: translateY(-4px);
+      box-shadow: 0 8px 12px rgba(0,0,0,0.15);
+    }
+    .stat-label {
+      font-size: 0.875rem;
+      color: #666;
+      text-transform: uppercase;
+      letter-spacing: 0.5px;
+      margin-bottom: 0.5rem;
+    }
+    .stat-value {
+      font-size: 2rem;
+      font-weight: bold;
+      color: #667eea;
+    }
+    .stat-value.cost {
+      color: #10b981;
+    }
+    table {
+      width: 100%;
+      background: white;
+      border-radius: 12px;
+      overflow: hidden;
+      box-shadow: 0 4px 6px rgba(0,0,0,0.1);
+      border-collapse: collapse;
+    }
+    thead {
+      background: #374151;
+      color: white;
+    }
+    th {
+      padding: 1rem;
+      text-align: left;
+      font-weight: 600;
+      text-transform: uppercase;
+      font-size: 0.875rem;
+      letter-spacing: 0.5px;
+    }
+    td {
+      padding: 1rem;
+      border-bottom: 1px solid #e5e7eb;
+    }
+    tbody tr:last-child td {
+      border-bottom: none;
+    }
+    tbody tr:hover {
+      background: #f9fafb;
+    }
+    .date-cell {
+      font-weight: 600;
+      color: #374151;
+    }
+    .number-cell {
+      text-align: right;
+      font-variant-numeric: tabular-nums;
+    }
+    .cost-cell {
+      text-align: right;
+      font-weight: bold;
+      color: #10b981;
+      font-variant-numeric: tabular-nums;
+    }
+    .model-badge {
+      display: inline-block;
+      background: #e0e7ff;
+      color: #4338ca;
+      padding: 0.25rem 0.75rem;
+      border-radius: 9999px;
+      font-size: 0.75rem;
+      font-weight: 500;
+      margin: 0.125rem;
+    }
+    .footer {
+      text-align: center;
+      color: #6b7280;
+      margin-top: 2rem;
+      font-size: 0.875rem;
+    }
+    .footer a {
+      color: #4f46e5;
+      text-decoration: underline;
+    }
+    .api-link {
+      display: inline-block;
+      background: #4f46e5;
+      color: white;
+      padding: 0.75rem 1.5rem;
+      border-radius: 8px;
+      text-decoration: none;
+      font-weight: 600;
+      margin: 1rem auto;
+      display: block;
+      text-align: center;
+      max-width: 200px;
+      transition: all 0.2s;
+    }
+    .api-link:hover {
+      background: #4338ca;
+      transform: translateY(-2px);
+      box-shadow: 0 4px 6px rgba(0,0,0,0.1);
+    }
+    @media (max-width: 768px) {
+      body {
+        padding: 1rem;
+      }
+      h1 {
+        font-size: 1.75rem;
+      }
+      .totals-grid {
+        grid-template-columns: 1fr;
+      }
+      table {
+        font-size: 0.875rem;
+      }
+      th, td {
+        padding: 0.75rem 0.5rem;
+      }
+    }
+  </style>
+</head>
+<body>
+  <div class="container">
+    <h1>Claude Code Usage Report</h1>
+    <% if @error %>
+      <div class="error">
+        <strong>Error:</strong> <%= @error %>
+      </div>
+    <% else %>
+      <div class="totals-grid">
+        <div class="stat-card">
+          <div class="stat-label">Total Cost</div>
+          <div class="stat-value cost">$<%= sprintf('%.4f', @totals[:total_cost] || 0) %></div>
+        </div>
+        <div class="stat-card">
+          <div class="stat-label">Total Tokens</div>
+          <div class="stat-value"><%= format_number(@totals[:total_tokens] || 0) %></div>
+        </div>
+        <div class="stat-card">
+          <div class="stat-label">Input Tokens</div>
+          <div class="stat-value"><%= format_number(@totals[:total_input_tokens] || 0) %></div>
+        </div>
+        <div class="stat-card">
+          <div class="stat-label">Output Tokens</div>
+          <div class="stat-value"><%= format_number(@totals[:total_output_tokens] || 0) %></div>
+        </div>
+        <div class="stat-card">
+          <div class="stat-label">Cache Creation</div>
+          <div class="stat-value"><%= format_number(@totals[:total_cache_creation_tokens] || 0) %></div>
+        </div>
+        <div class="stat-card">
+          <div class="stat-label">Cache Read</div>
+          <div class="stat-value"><%= format_number(@totals[:total_cache_read_tokens] || 0) %></div>
+        </div>
+        <div class="stat-card">
+          <div class="stat-label">Days Active</div>
+          <div class="stat-value"><%= @totals[:days_active] || 0 %></div>
+        </div>
+        <div class="stat-card">
+          <div class="stat-label">Models Used</div>
+          <div class="stat-value"><%= (@totals[:models_used] || []).size %></div>
+        </div>
+      </div>
+      <% if @daily_usage.any? %>
+        <table>
+          <thead>
+            <tr>
+              <th>Date</th>
+              <th style="text-align: right;">Input</th>
+              <th style="text-align: right;">Output</th>
+              <th style="text-align: right;">Cache Create</th>
+              <th style="text-align: right;">Cache Read</th>
+              <th style="text-align: right;">Total Tokens</th>
+              <th style="text-align: right;">Cost</th>
+              <th style="text-align: right;">Requests</th>
+              <th>Models</th>
+            </tr>
+          </thead>
+          <tbody>
+            <% @daily_usage.each do |day| %>
+              <tr>
+                <td class="date-cell"><%= day[:date].strftime('%Y-%m-%d') %></td>
+                <td class="number-cell"><%= format_number(day[:input_tokens]) %></td>
+                <td class="number-cell"><%= format_number(day[:output_tokens]) %></td>
+                <td class="number-cell"><%= format_number(day[:cache_creation_tokens]) %></td>
+                <td class="number-cell"><%= format_number(day[:cache_read_tokens]) %></td>
+                <td class="number-cell"><%= format_number(day[:total_tokens]) %></td>
+                <td class="cost-cell">$<%= sprintf('%.4f', day[:cost]) %></td>
+                <td class="number-cell"><%= day[:request_count] %></td>
+                <td>
+                  <% day[:models_used].each do |model| %>
+                    <span class="model-badge"><%= model %></span>
+                  <% end %>
+                </td>
+              </tr>
+            <% end %>
+          </tbody>
+        </table>
+      <% else %>
+        <div class="error">
+          <strong>No usage data found.</strong> Make sure you have Claude Code usage logs at ~/.config/claude/projects/
+        </div>
+      <% end %>
+      <a href="<%= claude_usage_engine.json_path %>" class="api-link">📊 View JSON API</a>
+      <div class="footer">
+        <p>Powered by <strong>claude_usage</strong> gem | Pricing from <a href="https://github.com/BerriAI/litellm" target="_blank">LiteLLM</a></p>
+      </div>
+    <% end %>
+  </div>
+</body>
+</html>

data/config/routes.rb ADDED Viewed

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+ClaudeUsage::Engine.routes.draw do
+  root to: 'usage#index'
+  get 'json', to: 'usage#show_json', as: :json
+end

data/lib/claude_usage/cost_calculator.rb ADDED Viewed

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+module ClaudeUsage
+  class CostCalculator
+    TIERED_THRESHOLD = 200_000
+    def initialize
+      @pricing_fetcher = LiteLLMPricingFetcher.new
+    end
+    def calculate_cost(entry)
+      return entry[:pre_calculated_cost] if entry[:pre_calculated_cost]
+      # Skip if no model information
+      return 0 if entry[:model].nil? || entry[:model].empty?
+      model_pricing = @pricing_fetcher.get_model_pricing(entry[:model])
+      return 0 unless model_pricing
+      calculate_tiered_cost(
+        entry[:input_tokens],
+        model_pricing[:input_cost_per_token],
+        model_pricing[:input_cost_per_token_above_200k]
+      ) +
+        calculate_tiered_cost(
+          entry[:output_tokens],
+          model_pricing[:output_cost_per_token],
+          model_pricing[:output_cost_per_token_above_200k]
+        ) +
+        calculate_tiered_cost(
+          entry[:cache_creation_tokens],
+          model_pricing[:cache_creation_input_token_cost],
+          model_pricing[:cache_creation_cost_above_200k]
+        ) +
+        calculate_tiered_cost(
+          entry[:cache_read_tokens],
+          model_pricing[:cache_read_input_token_cost],
+          model_pricing[:cache_read_cost_above_200k]
+        )
+    end
+    private
+    def calculate_tiered_cost(tokens, base_price, tiered_price)
+      return 0 if tokens.nil? || tokens <= 0 || base_price.nil?
+      if tokens > TIERED_THRESHOLD && tiered_price
+        tokens_below = [tokens, TIERED_THRESHOLD].min
+        tokens_above = [0, tokens - TIERED_THRESHOLD].max
+        (tokens_below * base_price) + (tokens_above * tiered_price)
+      else
+        tokens * base_price
+      end
+    end
+  end
+end

data/lib/claude_usage/engine.rb ADDED Viewed

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+module ClaudeUsage
+  class Engine < ::Rails::Engine
+    isolate_namespace ClaudeUsage
+    config.generators do |g|
+      g.test_framework :rspec
+    end
+    # Auto-mount routes in host app
+    initializer 'claude_usage.routes' do |app|
+      app.routes.append do
+        mount ClaudeUsage::Engine => '/claude-usage', as: :claude_usage_engine
+      end
+    end
+    # Add view paths
+    initializer 'claude_usage.assets' do |app|
+      app.config.assets.paths << root.join('app/assets')
+    end
+  end
+end

data/lib/claude_usage/file_reader.rb ADDED Viewed

@@ -0,0 +1,112 @@
+# frozen_string_literal: true
+require 'json'
+module ClaudeUsage
+  class FileReader
+    class InvalidProjectNameError < StandardError; end
+    CLAUDE_PATHS = [
+      File.expand_path('~/.config/claude/projects'),
+      File.expand_path('~/.claude/projects')
+    ].freeze
+    def initialize(project_name: nil)
+      # Use provided name or auto-detect from Rails.root
+      @project_name = project_name || ClaudeUsage.configuration.resolved_project_name
+      validate_project_name!
+    end
+    def read_usage_data
+      entries = []
+      processed_hashes = Set.new
+      jsonl_files.each do |file|
+        File.foreach(file) do |line|
+          next if line.strip.empty?
+          begin
+            data = JSON.parse(line)
+            # Deduplication
+            message_id = data.dig('message', 'id')
+            request_id = data['requestId']
+            if message_id && request_id
+              unique_hash = "#{message_id}:#{request_id}"
+              next if processed_hashes.include?(unique_hash)
+              processed_hashes.add(unique_hash)
+            end
+            entries << parse_entry(data)
+          rescue JSON::ParserError => e
+            logger&.debug "Skipping invalid JSON line: #{e.message}"
+          end
+        end
+      end
+      entries
+    end
+    private
+    def validate_project_name!
+      return if @project_name.nil? # Allow nil for cases where no project is configured
+      # Prevent path traversal attacks by validating project name format
+      # Allow alphanumeric, dashes, underscores, and dots (for typical project names)
+      unless @project_name.match?(/\A[\w.-]+\z/)
+        raise InvalidProjectNameError,
+              "Invalid project name '#{@project_name}'. Only alphanumeric characters, " \
+              'dashes, underscores, and dots are allowed.'
+      end
+      # Prevent path traversal patterns
+      return unless @project_name.include?('..') || @project_name.start_with?('/')
+      raise InvalidProjectNameError,
+            "Invalid project name '#{@project_name}'. Path traversal patterns are not allowed."
+    end
+    def claude_paths
+      ClaudeUsage.configuration.claude_paths || CLAUDE_PATHS
+    end
+    def jsonl_files
+      files = []
+      claude_paths.each do |base_path|
+        next unless Dir.exist?(base_path)
+        project_dir = File.join(base_path, @project_name)
+        next unless Dir.exist?(project_dir)
+        files.concat(Dir.glob(File.join(project_dir, '**', '*.jsonl')))
+      end
+      files.sort_by { |f| File.mtime(f) }
+    end
+    def parse_entry(data)
+      # Handle missing or invalid timestamp
+      timestamp = begin
+        data['timestamp'] ? Time.parse(data['timestamp']) : Time.now
+      rescue ArgumentError
+        Time.now
+      end
+      {
+        timestamp: timestamp,
+        model: data.dig('message', 'model'),
+        input_tokens: data.dig('message', 'usage', 'input_tokens') || 0,
+        output_tokens: data.dig('message', 'usage', 'output_tokens') || 0,
+        cache_creation_tokens: data.dig('message', 'usage', 'cache_creation_input_tokens') || 0,
+        cache_read_tokens: data.dig('message', 'usage', 'cache_read_input_tokens') || 0,
+        pre_calculated_cost: data['costUSD']
+      }
+    end
+    def logger
+      defined?(Rails) ? Rails.logger : nil
+    end
+  end
+end

data/lib/claude_usage/formatters.rb ADDED Viewed

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+module ClaudeUsage
+  # Shared formatting utilities for displaying usage data
+  module Formatters
+    module_function
+    # Format a number with comma separators
+    # Example: 1234567 => "1,234,567"
+    def format_number(number)
+      return '0' if number.nil? || number.zero?
+      number.to_s.reverse.gsub(/(\d{3})(?=\d)/, '\\1,').reverse
+    end
+    # Format a dollar amount with 2 decimal places
+    # Example: 1.234 => "$1.23"
+    def format_currency(amount)
+      return '$0.00' if amount.nil? || amount.zero?
+      format('$%.2f', amount)
+    end
+    # Format a percentage with 1 decimal place
+    # Example: 0.1234 => "12.3%"
+    def format_percentage(decimal)
+      return '0.0%' if decimal.nil? || decimal.zero?
+      format('%.1f%%', decimal * 100)
+    end
+  end
+end

data/lib/claude_usage/litellm_pricing_fetcher.rb ADDED Viewed

@@ -0,0 +1,148 @@
+# frozen_string_literal: true
+require 'httparty'
+require 'json'
+module ClaudeUsage
+  class LiteLLMPricingFetcher
+    include HTTParty
+    LITELLM_PRICING_URL = 'https://raw.githubusercontent.com/BerriAI/litellm/main/model_prices_and_context_window.json'
+    def initialize(offline: nil)
+      @offline = offline.nil? ? ClaudeUsage.configuration.offline_mode : offline
+    end
+    def fetch_pricing_data
+      cache_key = 'claude_usage/litellm_pricing_data'
+      cache_expiry = ClaudeUsage.configuration.cache_expiry
+      return Rails.cache.read(cache_key) || {} if @offline
+      Rails.cache.fetch(cache_key, expires_in: cache_expiry) do
+        fetch_from_litellm
+      end
+    end
+    def get_model_pricing(model_name, debug: false)
+      return nil if model_name.nil? || model_name.empty?
+      pricing_data = fetch_pricing_data
+      debug_info = { tried: [], fallbacks: [] } if debug
+      # Direct match (this should work for claude-haiku-4-5-20251001, claude-sonnet-4-5-20250929)
+      if pricing_data.key?(model_name)
+        debug_info[:tried] << { candidate: model_name, found: true } if debug
+        return debug ? [pricing_data[model_name], debug_info] : pricing_data[model_name]
+      end
+      debug_info[:tried] << { candidate: model_name, found: false } if debug
+      # Try with anthropic/ prefix
+      anthropic_model = "anthropic/#{model_name}"
+      debug_info[:tried] << { candidate: anthropic_model, found: false } if debug
+      if pricing_data.key?(anthropic_model)
+        debug_info[:tried].last[:found] = true if debug
+        return debug ? [pricing_data[anthropic_model], debug_info] : pricing_data[anthropic_model]
+      end
+      # For fallbacks, only try if it's a known Claude 4.5 model that might not be in LiteLLM yet
+      if model_name.include?('haiku-4-5')
+        fallback_models = [
+          'claude-3-5-haiku-20241022',
+          'anthropic/claude-3-5-haiku-20241022',
+          'claude-3-haiku-20240307',
+          'anthropic/claude-3-haiku-20240307'
+        ]
+        fallback_models.each do |fallback|
+          debug_info[:fallbacks] << { candidate: fallback, found: false } if debug
+          if pricing_data.key?(fallback)
+            debug_info[:fallbacks].last[:found] = true if debug
+            return debug ? [pricing_data[fallback], debug_info] : pricing_data[fallback]
+          end
+        end
+      elsif model_name.include?('sonnet-4-5')
+        fallback_models = [
+          'claude-sonnet-4-5-20250929',
+          'claude-3-5-sonnet-20241022',
+          'anthropic/claude-3-5-sonnet-20241022'
+        ]
+        fallback_models.each do |fallback|
+          debug_info[:fallbacks] << { candidate: fallback, found: false } if debug
+          if pricing_data.key?(fallback)
+            debug_info[:fallbacks].last[:found] = true if debug
+            return debug ? [pricing_data[fallback], debug_info] : pricing_data[fallback]
+          end
+        end
+      end
+      debug ? [nil, debug_info] : nil
+    end
+    private
+    def fetch_from_litellm
+      logger.info 'Fetching pricing from LiteLLM...'
+      # HTTParty configuration with configurable SSL verification
+      # Note: verify_ssl defaults to false for macOS compatibility
+      # Users can enable it via: config.verify_ssl = true
+      options = {
+        timeout: 30,
+        verify: ClaudeUsage.configuration.verify_ssl
+      }
+      response = self.class.get(LITELLM_PRICING_URL, options)
+      if response.success?
+        # Manually parse JSON to ensure we get a hash
+        raw_data = JSON.parse(response.body)
+        parsed_data = parse_pricing_data(raw_data)
+        logger.info "Successfully fetched pricing for #{parsed_data.keys.size} models"
+        # Log if our specific models are found
+        ['claude-haiku-4-5-20251001', 'claude-sonnet-4-5-20250929'].each do |model|
+          if parsed_data.key?(model)
+            logger.info "✅ Found pricing for #{model}"
+          else
+            logger.warn "❌ Missing pricing for #{model}"
+          end
+        end
+        parsed_data
+      else
+        logger.error "Failed to fetch LiteLLM pricing: HTTP #{response.code}"
+        {}
+      end
+    rescue StandardError => e
+      logger.error "Error fetching LiteLLM pricing: #{e.class} - #{e.message}"
+      logger.error e.backtrace.first(5).join("\n")
+      {}
+    end
+    def parse_pricing_data(raw_data)
+      pricing = {}
+      raw_data.each do |model_name, model_data|
+        next unless model_data.is_a?(Hash)
+        pricing[model_name] = {
+          input_cost_per_token: model_data['input_cost_per_token'],
+          output_cost_per_token: model_data['output_cost_per_token'],
+          cache_creation_input_token_cost: model_data['cache_creation_input_token_cost'],
+          cache_read_input_token_cost: model_data['cache_read_input_token_cost'],
+          input_cost_per_token_above_200k: model_data['input_cost_per_token_above_200k_tokens'],
+          output_cost_per_token_above_200k: model_data['output_cost_per_token_above_200k_tokens'],
+          cache_creation_cost_above_200k: model_data['cache_creation_input_token_cost_above_200k_tokens'],
+          cache_read_cost_above_200k: model_data['cache_read_input_token_cost_above_200k_tokens'],
+          max_input_tokens: model_data['max_input_tokens']
+        }.compact
+      end
+      pricing
+    end
+    def logger
+      defined?(Rails) ? Rails.logger : Logger.new($stdout)
+    end
+  end
+end

data/lib/claude_usage/project_name_resolver.rb ADDED Viewed

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+module ClaudeUsage
+  # Shared utility for detecting and resolving project names
+  module ProjectNameResolver
+    module_function
+    # Auto-detect project name from Rails.root
+    # Converts: /Users/name/projects/myapp → -Users-name-projects-myapp
+    def detect_from_rails_root
+      return nil unless defined?(Rails) && Rails.root
+      # Claude Code uses the absolute path with dashes as the folder name
+      # Remove leading slash, then replace remaining slashes with dashes, then prepend dash
+      Rails.root.to_s.sub(%r{^/}, '').gsub('/', '-').prepend('-')
+    end
+    # Get the project name with fallback
+    def resolve(configured_name = nil, fallback: 'your-project-name')
+      configured_name || detect_from_rails_root || fallback
+    end
+  end
+end