claude_usage 0.1.2

data/lib/claude_usage/usage_aggregator.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+module ClaudeUsage
+  class UsageAggregator
+    def initialize(project_name: nil)
+      @project_name = project_name || ClaudeUsage.configuration.resolved_project_name
+      @file_reader = FileReader.new(project_name: @project_name)
+      @cost_calculator = CostCalculator.new
+    end
+
+    def daily_usage
+      entries = @file_reader.read_usage_data
+      grouped = entries.group_by { |e| e[:timestamp].to_date }
+
+      grouped.map do |date, day_entries|
+        aggregate_entries(date, day_entries)
+      end.sort_by { |d| d[:date] }.reverse
+    end
+
+    def total_usage
+      daily = daily_usage
+
+      {
+        total_input_tokens: daily.sum { |d| d[:input_tokens] },
+        total_output_tokens: daily.sum { |d| d[:output_tokens] },
+        total_cache_creation_tokens: daily.sum { |d| d[:cache_creation_tokens] },
+        total_cache_read_tokens: daily.sum { |d| d[:cache_read_tokens] },
+        total_tokens: daily.sum { |d| d[:total_tokens] },
+        total_cost: daily.sum { |d| d[:cost] },
+        models_used: daily.flat_map { |d| d[:models_used] }.uniq,
+        first_usage: daily.last&.dig(:date),
+        last_usage: daily.first&.dig(:date),
+        days_active: daily.size
+      }
+    end
+
+    private
+
+    def aggregate_entries(date, entries)
+      {
+        date: date,
+        input_tokens: entries.sum { |e| e[:input_tokens] },
+        output_tokens: entries.sum { |e| e[:output_tokens] },
+        cache_creation_tokens: entries.sum { |e| e[:cache_creation_tokens] },
+        cache_read_tokens: entries.sum { |e| e[:cache_read_tokens] },
+        total_tokens: entries.sum do |e|
+          e[:input_tokens] + e[:output_tokens] + e[:cache_creation_tokens] + e[:cache_read_tokens]
+        end,
+        cost: entries.sum { |e| @cost_calculator.calculate_cost(e) },
+        models_used: entries.map { |e| e[:model] }.compact.uniq,
+        request_count: entries.size
+      }
+    end
+  end
+end

data/lib/claude_usage/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module ClaudeUsage
+  VERSION = '0.1.2'
+end

data/lib/claude_usage.rb

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+require_relative 'claude_usage/version'
+require_relative 'claude_usage/project_name_resolver'
+
+# Only require engine if Rails is defined
+require_relative 'claude_usage/engine' if defined?(Rails)
+
+module ClaudeUsage
+  class Error < StandardError; end
+
+  # Configuration
+  class Configuration
+    attr_accessor :project_name, :claude_paths, :cache_expiry, :offline_mode, :verify_ssl
+
+    def initialize
+      @project_name = nil # Will auto-detect from Rails.root if not set
+      @claude_paths = nil # Will use defaults if not set
+      @cache_expiry = 3600 # 1 hour in seconds
+      @offline_mode = false
+      @verify_ssl = false # Disabled by default for macOS compatibility
+    end
+
+    # Auto-detect project name from Rails.root
+    # Converts: /Users/name/projects/myapp → -Users-name-projects-myapp
+    def resolved_project_name
+      ProjectNameResolver.resolve(@project_name, fallback: nil)
+    end
+  end
+
+  class << self
+    attr_writer :configuration
+
+    def configuration
+      @configuration ||= Configuration.new
+    end
+
+    def configure
+      yield(configuration)
+    end
+
+    def reset_configuration!
+      @configuration = Configuration.new
+    end
+  end
+end
+
+# Load service objects
+require_relative 'claude_usage/file_reader'
+require_relative 'claude_usage/litellm_pricing_fetcher'
+require_relative 'claude_usage/cost_calculator'
+require_relative 'claude_usage/usage_aggregator'

data/lib/generators/claude_usage/install_generator.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+require 'rails/generators'
+
+module ClaudeUsage
+  module Generators
+    class InstallGenerator < Rails::Generators::Base
+      source_root File.expand_path('templates', __dir__)
+
+      desc 'Creates ClaudeUsage initializer and shows setup instructions'
+
+      def copy_initializer
+        @detected_project_name = detect_project_name
+        template 'initializer.rb', 'config/initializers/claude_usage.rb'
+      end
+
+      def show_readme
+        readme 'README' if behavior == :invoke
+      end
+
+      private
+
+      def detect_project_name
+        # Use shared resolver to avoid duplication
+        require_relative '../../claude_usage/project_name_resolver'
+        ClaudeUsage::ProjectNameResolver.resolve(nil, fallback: 'your-project-name')
+      end
+    end
+  end
+end

data/lib/generators/claude_usage/templates/initializer.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+ClaudeUsage.configure do |config|
+  # Project name - AUTO-DETECTED from Rails.root by default
+  #
+  # Claude Code stores projects using the full absolute path with dashes, like:
+  # /Users/name/projects/myapp → -Users-name-projects-myapp
+  #
+  # The gem automatically detects this, so you usually don't need to set it manually.
+  #
+  # Only uncomment if auto-detection fails or you need a custom name:
+  # config.project_name = "<%= @detected_project_name %>"
+  #
+  # To see what's auto-detected, run: rake claude_usage:show_project
+
+  # Custom Claude paths (optional)
+  # Default paths are already configured:
+  # - ~/.config/claude/projects
+  # - ~/.claude/projects
+  # config.claude_paths = [
+  #   File.expand_path("~/.config/claude/projects"),
+  #   File.expand_path("~/.claude/projects")
+  # ]
+
+  # Pricing cache expiry in seconds (default: 3600 = 1 hour)
+  # config.cache_expiry = 3600
+
+  # Offline mode (use cached pricing only, recommended for production)
+  # config.offline_mode = false
+
+  # SSL verification (disabled by default for macOS compatibility)
+  # Enable in production if your certificates are up-to-date
+  # config.verify_ssl = false
+end

data/lib/tasks/claude_usage_tasks.rake

@@ -0,0 +1,313 @@
+# frozen_string_literal: true
+
+require_relative '../claude_usage/formatters'
+
+namespace :claude_usage do
+  desc 'Show daily usage report'
+  task daily: :environment do
+    project_name = ClaudeUsage.configuration.resolved_project_name
+    aggregator = ClaudeUsage::UsageAggregator.new
+    daily_usage = aggregator.daily_usage
+
+    puts "\n🤖 Claude Code - Daily Usage Report\n"
+    puts '=' * 80
+    puts "📁 Project: #{project_name}"
+    puts '=' * 80
+
+    if daily_usage.empty?
+      puts "\n❌ No usage data found!"
+      puts "\nTroubleshooting:"
+      puts '  1. Run: rake claude_usage:show_project'
+      puts '  2. Check that Claude Code has been used with this project'
+      puts "  3. Verify project name matches Claude Code's folder name"
+      puts "\n"
+    else
+      daily_usage.each do |day|
+        puts "\n📅 #{day[:date].strftime('%Y-%m-%d')}"
+        puts "   Input Tokens:        #{day[:input_tokens].to_s.rjust(10)}"
+        puts "   Output Tokens:       #{day[:output_tokens].to_s.rjust(10)}"
+        puts "   Cache Creation:      #{day[:cache_creation_tokens].to_s.rjust(10)}"
+        puts "   Cache Read:          #{day[:cache_read_tokens].to_s.rjust(10)}"
+        puts "   Total Tokens:        #{day[:total_tokens].to_s.rjust(10)}"
+        puts "   Cost:                $#{format('%.4f', day[:cost]).rjust(9)}"
+        puts "   Requests:            #{day[:request_count].to_s.rjust(10)}"
+        puts "   Models:              #{day[:models_used].join(', ')}"
+      end
+
+      puts "\n#{'=' * 80}"
+    end
+  end
+
+  desc 'Show total usage'
+  task total: :environment do
+    project_name = ClaudeUsage.configuration.resolved_project_name
+    aggregator = ClaudeUsage::UsageAggregator.new
+    totals = aggregator.total_usage
+
+    puts "\n🤖 Claude Code - Total Usage Report\n"
+    puts '=' * 80
+    puts "📁 Project: #{project_name}"
+    puts '=' * 80
+
+    if totals[:days_active].zero?
+      puts "\n❌ No usage data found!"
+      puts "\nTroubleshooting:"
+      puts '  1. Run: rake claude_usage:show_project'
+      puts '  2. Check that Claude Code has been used with this project'
+      puts "  3. Verify project name matches Claude Code's folder name"
+      puts "\n"
+    else
+      puts "\n💰 Cost:                  $#{format('%.4f', totals[:total_cost])}"
+
+      # Show warning if cost is 0 but tokens exist
+      if totals[:total_cost].zero? && totals[:total_tokens].positive?
+        puts '   ⚠️  Cost is $0 - model pricing may not be available'
+        puts '   Run: rake claude_usage:debug_pricing'
+      end
+
+      puts "📊 Total Tokens:          #{ClaudeUsage::Formatters.format_number(totals[:total_tokens])}"
+      puts "   - Input:               #{ClaudeUsage::Formatters.format_number(totals[:total_input_tokens])}"
+      puts "   - Output:              #{ClaudeUsage::Formatters.format_number(totals[:total_output_tokens])}"
+      puts "   - Cache Creation:      #{ClaudeUsage::Formatters.format_number(totals[:total_cache_creation_tokens])}"
+      puts "   - Cache Read:          #{ClaudeUsage::Formatters.format_number(totals[:total_cache_read_tokens])}"
+      puts "📅 Days Active:           #{totals[:days_active]}"
+      puts "📅 First Usage:           #{totals[:first_usage]}"
+      puts "📅 Last Usage:            #{totals[:last_usage]}"
+      puts "🤖 Models Used:           #{totals[:models_used].join(', ')}"
+      puts "\n#{'=' * 80}\n"
+    end
+  end
+
+  desc 'Clear pricing cache'
+  task clear_cache: :environment do
+    Rails.cache.delete('claude_usage/litellm_pricing_data')
+    puts '✅ Pricing cache cleared!'
+  end
+
+  desc 'Show project name being tracked'
+  task show_project: :environment do
+    manual_config = ClaudeUsage.configuration.project_name
+    auto_detected = ClaudeUsage.configuration.resolved_project_name
+
+    puts "\n📁 Project Detection:"
+    puts "   Rails.root:      #{Rails.root}"
+
+    if manual_config
+      puts "   Manual config:   #{manual_config} (overriding auto-detection)"
+      project_name = manual_config
+    else
+      puts "   Auto-detected:   #{auto_detected}"
+      puts '   Status:          ✅ Automatic (no config needed)'
+      project_name = auto_detected
+    end
+
+    claude_paths = ClaudeUsage.configuration.claude_paths || ClaudeUsage::FileReader::CLAUDE_PATHS
+    puts "\n📂 Searching in directories:"
+    found_any = false
+
+    claude_paths.each do |path|
+      if Dir.exist?(path)
+        puts "   ✅ #{path} (exists)"
+
+        # List all projects in this directory
+        projects = Dir.glob(File.join(path, '*')).select { |f| File.directory?(f) }
+        if projects.any?
+          puts '      Available projects:'
+          projects.each do |p|
+            project_folder = File.basename(p)
+            jsonl_count = Dir.glob(File.join(p, '**', '*.jsonl')).size
+            marker = project_folder == project_name ? '👉' : '  '
+            puts "      #{marker} #{project_folder} (#{jsonl_count} JSONL files)"
+            found_any = true if project_folder == project_name && jsonl_count.positive?
+          end
+        else
+          puts '      (no projects found)'
+        end
+      else
+        puts "   ❌ #{path} (not found)"
+      end
+      puts ''
+    end
+
+    if found_any
+      puts "\n✅ Found data for project '#{project_name}'!"
+      puts '   Run: rake claude_usage:total'
+    else
+      puts "\n⚠️  No data found for project '#{project_name}'"
+      puts "\n💡 Solutions:"
+      puts '   1. If your project name differs, set it in config/initializers/claude_usage.rb:'
+      puts '      ClaudeUsage.configure do |config|'
+      puts "        config.project_name = 'actual-project-name'"
+      puts '      end'
+      puts "\n   2. Check available project names above (marked with 👉 when matching)"
+      puts "\n   3. Ensure you've used Claude Code with this project\n"
+    end
+  end
+
+  desc 'Debug pricing for current project models'
+  task debug_pricing: :environment do
+    puts "\n🔬 Debugging Model Pricing\n"
+    puts '=' * 80
+
+    ClaudeUsage.configuration.project_name || Rails.root.basename.to_s
+    aggregator = ClaudeUsage::UsageAggregator.new
+    totals = aggregator.total_usage
+
+    if totals[:models_used].empty?
+      puts '❌ No models found in usage data'
+      puts "\n"
+      next
+    end
+
+    puts "\n📋 Models in your usage data:"
+    totals[:models_used].each { |model| puts "   - #{model}" }
+    puts ''
+
+    fetcher = ClaudeUsage::LiteLLMPricingFetcher.new
+
+    # First, let's check if we can fetch pricing at all
+    puts '🔍 Testing pricing fetch...'
+    pricing_data = fetcher.fetch_pricing_data
+
+    if pricing_data.empty?
+      puts '❌ Failed to fetch pricing data from LiteLLM!'
+      puts '   This might be a network issue or the API is down.'
+      puts '   Check your internet connection and try again.'
+      puts "\n"
+      next
+    else
+      puts "✅ Successfully fetched pricing for #{pricing_data.keys.size} models"
+
+      # Check if our specific models are in there
+      test_models = ['claude-haiku-4-5-20251001', 'claude-sonnet-4-5-20250929']
+      test_models.each do |test_model|
+        if pricing_data.key?(test_model)
+          puts "   ✅ #{test_model} found in database"
+        else
+          puts "   ❌ #{test_model} NOT in database"
+        end
+      end
+    end
+    puts ''
+
+    totals[:models_used].each do |model|
+      puts "\n🤖 Model: #{model}"
+      puts "   #{'-' * 76}"
+
+      # Skip synthetic models
+      if model == '<synthetic>'
+        puts '   ℹ️  System/internal messages (not billable)'
+        next
+      end
+
+      pricing, debug_info = fetcher.get_model_pricing(model, debug: true)
+
+      if pricing
+        puts '   ✅ Pricing found!'
+        puts "   Input:          $#{pricing['input_cost_per_token'] || pricing[:input_cost_per_token]} per token"
+        puts "   Output:         $#{pricing['output_cost_per_token'] || pricing[:output_cost_per_token]} per token"
+        cache_creation = pricing['cache_creation_input_token_cost'] ||
+                         pricing[:cache_creation_input_token_cost] || 'N/A'
+        puts "   Cache Creation: $#{cache_creation} per token"
+        cache_read = pricing['cache_read_input_token_cost'] ||
+                     pricing[:cache_read_input_token_cost] || 'N/A'
+        puts "   Cache Read:     $#{cache_read} per token"
+
+        # Show how it was found
+        if debug_info[:fuzzy_match]
+          puts "   (Found via fuzzy match: #{debug_info[:fuzzy_match]})"
+        elsif debug_info[:fallbacks]&.any? { |f| f[:found] }
+          found = debug_info[:fallbacks].find { |f| f[:found] }
+          puts "   (Found via fallback: #{found[:candidate]})"
+        end
+      else
+        puts '   ❌ No pricing found!'
+        puts '   This model will contribute $0 to cost calculations'
+        puts ''
+        puts '   Debugging info:'
+        puts "   - Original model name: #{model}"
+
+        if debug_info[:tried].any?
+          puts '   - Provider prefixes tried:'
+          debug_info[:tried].each do |attempt|
+            puts "     * #{attempt[:candidate]}"
+          end
+        end
+
+        if debug_info[:fallbacks].any?
+          puts '   - Fallback models tried:'
+          debug_info[:fallbacks].each do |attempt|
+            puts "     * #{attempt[:candidate]}"
+          end
+        end
+
+        puts ''
+        puts '   💡 Possible solutions:'
+        puts '   1. Model may be too new for LiteLLM pricing database'
+        puts '   2. Model name format may differ (check LiteLLM docs)'
+        puts '   3. Clear cache and retry: rake claude_usage:clear_cache'
+        puts '   4. Check LiteLLM database:'
+        puts '      https://raw.githubusercontent.com/BerriAI/litellm/main/model_prices_and_context_window.json'
+      end
+    end
+
+    puts "\n#{'=' * 80}\n"
+  end
+
+  desc 'Test LiteLLM pricing fetch'
+  task test_fetch: :environment do
+    puts "\n🧪 Testing LiteLLM Pricing Fetch\n"
+    puts '=' * 80
+    puts ''
+
+    fetcher = ClaudeUsage::LiteLLMPricingFetcher.new
+
+    puts 'Fetching pricing data...'
+    pricing_data = fetcher.fetch_pricing_data
+
+    if pricing_data.empty?
+      puts '❌ FAILED - No pricing data received'
+      puts ''
+      puts 'Possible causes:'
+      puts '  1. Network connectivity issue'
+      puts '  2. LiteLLM API is down'
+      puts '  3. HTTP gem not working correctly'
+      puts ''
+      puts 'Try:'
+      puts "  curl #{ClaudeUsage::LiteLLMPricingFetcher::LITELLM_PRICING_URL}"
+      puts ''
+    else
+      puts "✅ SUCCESS - Fetched #{pricing_data.keys.size} models"
+      puts ''
+
+      # Test specific models
+      test_models = {
+        'claude-haiku-4-5-20251001' => 'Claude 4.5 Haiku',
+        'claude-sonnet-4-5-20250929' => 'Claude 4.5 Sonnet',
+        'claude-3-5-haiku-20241022' => 'Claude 3.5 Haiku',
+        'claude-3-5-sonnet-20241022' => 'Claude 3.5 Sonnet'
+      }
+
+      puts 'Testing specific models:'
+      test_models.each do |model_key, model_name|
+        if pricing_data.key?(model_key)
+          pricing = pricing_data[model_key]
+          puts "  ✅ #{model_name} (#{model_key})"
+          puts "     Input: $#{pricing[:input_cost_per_token]}"
+          puts "     Output: $#{pricing[:output_cost_per_token]}"
+        else
+          puts "  ❌ #{model_name} (#{model_key}) - NOT FOUND"
+        end
+      end
+      puts ''
+
+      # Show sample of available models
+      puts 'Sample of available models:'
+      pricing_data.keys.select { |k| k.include?('claude') }.first(10).each do |key|
+        puts "  - #{key}"
+      end
+    end
+
+    puts "\n#{'=' * 80}\n"
+  end
+end

data/sig/claude_usage.rbs

@@ -0,0 +1,4 @@
+module ClaudeUsage
+  VERSION: String
+  # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+end

metadata

@@ -0,0 +1,139 @@
+--- !ruby/object:Gem::Specification
+name: claude_usage
+version: !ruby/object:Gem::Version
+  version: 0.1.2
+platform: ruby
+authors:
+- Shoeb
+bindir: exe
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: httparty
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.21'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.21'
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '6.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '6.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: rspec-rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+- !ruby/object:Gem::Dependency
+  name: webmock
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+description: A Rails engine for monitoring Claude Code usage, calculating costs using
+  LiteLLM pricing, and displaying usage analytics in your Rails app
+email:
+- shoebtamboli98@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".rubocop.yml"
+- CHANGELOG.md
+- CODE_OF_CONDUCT.md
+- LICENSE.txt
+- README.md
+- Rakefile
+- app/controllers/claude_usage/application_controller.rb
+- app/controllers/claude_usage/usage_controller.rb
+- app/helpers/claude_usage/usage_helper.rb
+- app/views/claude_usage/usage/index.html.erb
+- config/routes.rb
+- lib/claude_usage.rb
+- lib/claude_usage/cost_calculator.rb
+- lib/claude_usage/engine.rb
+- lib/claude_usage/file_reader.rb
+- lib/claude_usage/formatters.rb
+- lib/claude_usage/litellm_pricing_fetcher.rb
+- lib/claude_usage/project_name_resolver.rb
+- lib/claude_usage/usage_aggregator.rb
+- lib/claude_usage/version.rb
+- lib/generators/claude_usage/install_generator.rb
+- lib/generators/claude_usage/templates/initializer.rb
+- lib/tasks/claude_usage_tasks.rake
+- sig/claude_usage.rbs
+homepage: https://github.com/Shoebtamboli/claude_usage
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/Shoebtamboli/claude_usage
+  source_code_uri: https://github.com/Shoebtamboli/claude_usage
+  changelog_uri: https://github.com/Shoebtamboli/claude_usage/blob/main/CHANGELOG.md
+  rubygems_mfa_required: 'true'
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.2.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.9
+specification_version: 4
+summary: Track Claude Code token usage and costs in Rails applications
+test_files: []