token-lens 0.1.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fc58978bdc3f4759309ded99d65da22be12a01287ea79acdb43ab873e7cb8771
-  data.tar.gz: 34c54363d13e72ed669e3f27bbf1fc8af2d3665e958d977baabfaac9ab80f849
+  metadata.gz: 5df4aa9d712d04ae90513777b2b34b541bb3b77038fadd637725f1bf1599db3f
+  data.tar.gz: 52f773f6c7bfd677a29bef81ba14bb23e2b01ec54071468ee6ba7be2e4632b9a
 SHA512:
-  metadata.gz: 3f3c35dedc0cd0efd6e12b8d136aab7f4328ba4619f912f07c94727e3bd6c4c2bb4184d07a9e3bbb117dd0eaeec28edd27c5f7100f140a79b12043773a066881
-  data.tar.gz: d0767d26c77b4ba6dfec6f3a40b70c7d1dde3b3a58094fa8707b4bf57bc860612ed1af91c161b31fc3f3faa78f3db25ed6f5ddc3ae2f9422ca351084e3d5ec41
+  metadata.gz: 0b271f96c07b9c74db5ca42e4ffff7a7259e5a63893e949a9ab0fc3dd2bc4761df0c783151769a42cd320dc6e5574ae408a5b2dbfe2133cc839bab60d12f638a
+  data.tar.gz: 8f296614fd7f957189e07961223fea18a2242fee5f090860a2b49d6f69daf90f92dadc87a289ce4b22cbaf739d0d25cfdfbd2c2cac062fe4ed7e23f952db156c

data/README.md

@@ -10,9 +10,27 @@ Record Data --> Interpret Data --> Render Data
 
 ## Data Sources
 
-1. Tap OTEL console output
-2. Tail session JSONL
-3. Local API proxy
+1. [unsupported] Tap OTEL console output
+2. [supported] Tail session JSONL
+3. [unsupported] Local API proxy
+
+## Quick Start
+
+
+**Locally**:
+```
+bin/token-lens record --duration-in-seconds=30 > capture.json
+bin/token-lens render --file-path=capture.json
+open flame.html
+```
+
+**Via gem**:
+```
+gem install token-lens
+token-lens record --duration-in-seconds=30 > capture.json
+token-lens render --file-path=capture.json
+open flame.html
+```
 
 ## Contributing
 

data/lib/token_lens/cli.rb

@@ -8,8 +8,12 @@ module TokenLens
   class CLI < Thor
     desc "record", "Tail the active session and capture events to stdout"
     option :duration_in_seconds, type: :numeric, default: 30, desc: "Seconds to record"
+    option :project_dir, type: :string, desc: "Working directory of the Claude Code session to record (default: auto-detect)"
     def record
-      Commands::Record.new(duration_in_seconds: options[:duration_in_seconds]).run
+      Commands::Record.new(
+        duration_in_seconds: options[:duration_in_seconds],
+        project_dir: options[:project_dir]
+      ).run
     end
 
     desc "render", "Render a captured session as a flame graph"

data/lib/token_lens/commands/record.rb

@@ -6,8 +6,9 @@ require "token_lens/sources/jsonl"
 module TokenLens
   module Commands
     class Record
-      def initialize(duration_in_seconds:)
+      def initialize(duration_in_seconds:, project_dir: nil)
         @duration_in_seconds = duration_in_seconds
+        @project_dir = project_dir
       end
 
       def run
@@ -16,7 +17,7 @@ module TokenLens
         queue = Queue.new
         events = []
 
-        thread = Thread.new { Sources::Jsonl.new(queue).start }
+        thread = Thread.new { Sources::Jsonl.new(queue, project_dir: @project_dir).start }
         drain_thread = Thread.new { loop { events << queue.pop } }
 
         trap("INT") { finish(thread, drain_thread, queue, events) }

data/lib/token_lens/session.rb

@@ -16,15 +16,28 @@ module TokenLens
 
     def self.active_jsonl(dir = Dir.pwd)
       project_dir = CLAUDE_DIR / encoded_cwd(dir)
-
-      # find all *.jsonl files in the project directory
       jsonl_files = project_dir.glob("*.jsonl")
       raise "No session files found in #{project_dir}" if jsonl_files.empty?
-
-      # get most recently modified file
       jsonl_files.max_by(&:mtime)
     end
 
+    # Returns the most recently modified session file across ALL projects.
+    def self.latest_jsonl
+      all = CLAUDE_DIR.glob("*/*.jsonl")
+      raise "No session files found in #{CLAUDE_DIR}" if all.empty?
+      all.max_by(&:mtime)
+    end
+
+    # Like active_jsonl but falls back to latest_jsonl with a warning when
+    # the current directory has no sessions (e.g. running from ~/Desktop).
+    def self.active_or_latest_jsonl(dir = Dir.pwd)
+      active_jsonl(dir)
+    rescue RuntimeError
+      path = latest_jsonl
+      warn "  [session] no sessions for #{dir}, using most recent: #{path}"
+      path
+    end
+
     def self.tail(path, &block)
       last_pos = File.size(path)
       loop do

data/lib/token_lens/sources/jsonl.rb

@@ -6,9 +6,9 @@ require "token_lens/session"
 module TokenLens
   module Sources
     class Jsonl
-      def initialize(queue)
+      def initialize(queue, project_dir: nil)
         @queue = queue
-        @path = Session.active_jsonl
+        @path = project_dir ? Session.active_jsonl(project_dir) : Session.active_or_latest_jsonl
       end
 
       def start

data/lib/token_lens/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module TokenLens
-  VERSION = "0.1.0"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: token-lens
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.3.0
 platform: ruby
 authors:
 - rob durst