import CodexBarCoreimport Commanderimport Foundation extension CodexBarCLI { private static let costSupportedProviders: Set<UsageProvider> = [.claude, .codex] static func runCost(_ values: ParsedValues) async { let selection = Self.decodeProvider(from: values) let providers = Self.costProviders(from: selection) let unsupported = selection.asList.filter { !Self.costSupportedProviders.contains($0) } if !unsupported.isEmpty { let names = unsupported .map { ProviderDescriptorRegistry.descriptor(for: $0).metadata.displayName } .sorted() .joined(separator: ", ") Self.writeStderr("Skipping providers without local cost usage: \(names)\n") } guard !providers.isEmpty else { Self.exit(code: .failure, message: "Error: cost is only supported for Claude and Codex.") } let format = Self.decodeFormat(from: values) let pretty = values.flags.contains("pretty") let forceRefresh = values.flags.contains("refresh") let useColor = Self.shouldUseColor(noColor: values.flags.contains("noColor"), format: format) let fetcher = CostUsageFetcher() var sections: [String] = [] var payload: [CostPayload] = [] var exitCode: ExitCode = .success for provider in providers { do { // Cost usage is local-only; it does not require web/CLI provider fetches. let snapshot = try await fetcher.loadTokenSnapshot( provider: provider, forceRefresh: forceRefresh) switch format { case .text: sections.append(Self.renderCostText(provider: provider, snapshot: snapshot, useColor: useColor)) case .json: payload.append(Self.makeCostPayload(provider: provider, snapshot: snapshot)) } } catch { exitCode = Self.mapError(error) Self.printError(error) } } switch format { case .text: if !sections.isEmpty { print(sections.joined(separator: "\n\n")) } case .json: if !payload.isEmpty { let encoder = JSONEncoder() encoder.dateEncodingStrategy = .iso8601 encoder.outputFormatting = pretty ? [.prettyPrinted, .sortedKeys] : [] if let data = try? encoder.encode(payload), let output = String(data: data, encoding: .utf8) { print(output) } } } Self.exit(code: exitCode) } static func costHelp(version: String) -> String { """ CodexBar \(version) Usage: codexbar cost [--format text|json] [--json] [--json-output] [--log-level <trace|verbose|debug|info|warning|error|critical>] [-v|--verbose] [--provider \(ProviderHelp.list)] [--no-color] [--pretty] [--refresh] Description: Print local token cost usage from Claude/Codex JSONL logs. This does not require web or CLI access. Uses cached scan results unless --refresh is provided. Examples: codexbar cost codexbar cost --provider claude --format json --pretty """ } static func renderCostText( provider: UsageProvider, snapshot: CostUsageTokenSnapshot, useColor: Bool) -> String { let name = ProviderDescriptorRegistry.descriptor(for: provider).metadata.displayName let header = Self.costHeaderLine("\(name) Cost (local)", useColor: useColor) let todayCost = snapshot.sessionCostUSD.map { UsageFormatter.usdString($0) } ?? "—" let todayTokens = snapshot.sessionTokens.map { UsageFormatter.tokenCountString($0) } let todayLine = todayTokens.map { "Today: \(todayCost) · \($0) tokens" } ?? "Today: \(todayCost)" let monthCost = snapshot.last30DaysCostUSD.map { UsageFormatter.usdString($0) } ?? "—" let monthTokens = snapshot.last30DaysTokens.map { UsageFormatter.tokenCountString($0) } let monthLine = monthTokens.map { "Last 30 days: \(monthCost) · \($0) tokens" } ?? "Last 30 days: \(monthCost)" return [header, todayLine, monthLine].joined(separator: "\n") } private static func costHeaderLine(_ header: String, useColor: Bool) -> String { guard useColor else { return header } return "\u{001B}[1;36m\(header)\u{001B}[0m" } private static func costProviders(from selection: ProviderSelection) -> [UsageProvider] { selection.asList.filter { Self.costSupportedProviders.contains($0) } } private static func makeCostPayload(provider: UsageProvider, snapshot: CostUsageTokenSnapshot) -> CostPayload { let daily = snapshot.daily.map { entry in CostDailyEntryPayload( date: entry.date, inputTokens: entry.inputTokens, outputTokens: entry.outputTokens, cacheReadTokens: entry.cacheReadTokens, cacheCreationTokens: entry.cacheCreationTokens, totalTokens: entry.totalTokens, costUSD: entry.costUSD, modelsUsed: entry.modelsUsed, modelBreakdowns: entry.modelBreakdowns?.map { breakdown in CostModelBreakdownPayload(modelName: breakdown.modelName, costUSD: breakdown.costUSD) }) } return CostPayload( provider: provider.rawValue, source: "local", updatedAt: snapshot.updatedAt, sessionTokens: snapshot.sessionTokens, sessionCostUSD: snapshot.sessionCostUSD, last30DaysTokens: snapshot.last30DaysTokens, last30DaysCostUSD: snapshot.last30DaysCostUSD, daily: daily, totals: Self.costTotals(from: snapshot)) } private static func costTotals(from snapshot: CostUsageTokenSnapshot) -> CostTotalsPayload? { let entries = snapshot.daily guard !entries.isEmpty else { guard snapshot.last30DaysTokens != nil || snapshot.last30DaysCostUSD != nil else { return nil } return CostTotalsPayload( totalInputTokens: nil, totalOutputTokens: nil, cacheReadTokens: nil, cacheCreationTokens: nil, totalTokens: snapshot.last30DaysTokens, totalCostUSD: snapshot.last30DaysCostUSD) } var totalInput = 0 var totalOutput = 0 var totalCacheRead = 0 var totalCacheCreation = 0 var totalTokens = 0 var totalCost = 0.0 var sawInput = false var sawOutput = false var sawCacheRead = false var sawCacheCreation = false var sawTokens = false var sawCost = false for entry in entries { if let input = entry.inputTokens { totalInput += input sawInput = true } if let output = entry.outputTokens { totalOutput += output sawOutput = true } if let cacheRead = entry.cacheReadTokens { totalCacheRead += cacheRead sawCacheRead = true } if let cacheCreation = entry.cacheCreationTokens { totalCacheCreation += cacheCreation sawCacheCreation = true } if let tokens = entry.totalTokens { totalTokens += tokens sawTokens = true } if let cost = entry.costUSD { totalCost += cost sawCost = true } } // Prefer totals derived from daily rows; fall back to snapshot aggregates when rows omit fields. return CostTotalsPayload( totalInputTokens: sawInput ? totalInput : nil, totalOutputTokens: sawOutput ? totalOutput : nil, cacheReadTokens: sawCacheRead ? totalCacheRead : nil, cacheCreationTokens: sawCacheCreation ? totalCacheCreation : nil, totalTokens: sawTokens ? totalTokens : snapshot.last30DaysTokens, totalCostUSD: sawCost ? totalCost : snapshot.last30DaysCostUSD) }} struct CostOptions: CommanderParsable { @Flag(names: [.short("v"), .long("verbose")], help: "Enable verbose logging") var verbose: Bool = false @Flag(name: .long("json-output"), help: "Emit machine-readable logs") var jsonOutput: Bool = false @Option(name: .long("log-level"), help: "Set log level (trace|verbose|debug|info|warning|error|critical)") var logLevel: String? @Option( name: .long("provider"), help: ProviderHelp.optionHelp) var provider: ProviderSelection? @Option(name: .long("format"), help: "Output format: text | json") var format: OutputFormat? @Flag(name: .long("json"), help: "") var jsonShortcut: Bool = false @Flag(name: .long("pretty"), help: "Pretty-print JSON output") var pretty: Bool = false @Flag(name: .long("no-color"), help: "Disable ANSI colors in text output") var noColor: Bool = false @Flag(name: .long("refresh"), help: "Force refresh by ignoring cached scans") var refresh: Bool = false} struct CostPayload: Encodable { let provider: String let source: String let updatedAt: Date let sessionTokens: Int? let sessionCostUSD: Double? let last30DaysTokens: Int? let last30DaysCostUSD: Double? let daily: [CostDailyEntryPayload] let totals: CostTotalsPayload?} struct CostDailyEntryPayload: Encodable { let date: String let inputTokens: Int? let outputTokens: Int? let cacheReadTokens: Int? let cacheCreationTokens: Int? let totalTokens: Int? let costUSD: Double? let modelsUsed: [String]? let modelBreakdowns: [CostModelBreakdownPayload]? private enum CodingKeys: String, CodingKey { case date case inputTokens case outputTokens case cacheReadTokens case cacheCreationTokens case totalTokens case costUSD = "totalCost" case modelsUsed case modelBreakdowns }} struct CostModelBreakdownPayload: Encodable { let modelName: String let costUSD: Double? private enum CodingKeys: String, CodingKey { case modelName case costUSD = "cost" }} struct CostTotalsPayload: Encodable { let totalInputTokens: Int? let totalOutputTokens: Int? let cacheReadTokens: Int? let cacheCreationTokens: Int? let totalTokens: Int? let totalCostUSD: Double? private enum CodingKeys: String, CodingKey { case totalInputTokens = "inputTokens" case totalOutputTokens = "outputTokens" case cacheReadTokens case cacheCreationTokens case totalTokens case totalCostUSD = "totalCost" }} #if DEBUGextension CodexBarCLI { static func _costSignatureForTesting() -> CommandSignature { CommandSignature.describe(CostOptions()) }}#endif
import CodexBarCore
import Commander
import Foundation
extension CodexBarCLI {
private static let costSupportedProviders: Set<UsageProvider> = [.claude, .codex]
static func runCost(_ values: ParsedValues) async {
let selection = Self.decodeProvider(from: values)
let providers = Self.costProviders(from: selection)
let unsupported = selection.asList.filter { !Self.costSupportedProviders.contains($0) }
if !unsupported.isEmpty {
let names = unsupported
.map { ProviderDescriptorRegistry.descriptor(for: $0).metadata.displayName }
.sorted()
.joined(separator: ", ")
Self.writeStderr("Skipping providers without local cost usage: \(names)\n")
}
guard !providers.isEmpty else {
Self.exit(code: .failure, message: "Error: cost is only supported for Claude and Codex.")
}
let format = Self.decodeFormat(from: values)
let pretty = values.flags.contains("pretty")
let forceRefresh = values.flags.contains("refresh")
let useColor = Self.shouldUseColor(noColor: values.flags.contains("noColor"), format: format)
let fetcher = CostUsageFetcher()
var sections: [String] = []
var payload: [CostPayload] = []
var exitCode: ExitCode = .success
for provider in providers {
do {
// Cost usage is local-only; it does not require web/CLI provider fetches.
let snapshot = try await fetcher.loadTokenSnapshot(
provider: provider,
forceRefresh: forceRefresh)
switch format {
case .text:
sections.append(Self.renderCostText(provider: provider, snapshot: snapshot, useColor: useColor))
case .json:
payload.append(Self.makeCostPayload(provider: provider, snapshot: snapshot))
}
} catch {
exitCode = Self.mapError(error)
Self.printError(error)
}
}
switch format {
case .text:
if !sections.isEmpty {
print(sections.joined(separator: "\n\n"))
}
case .json:
if !payload.isEmpty {
let encoder = JSONEncoder()
encoder.dateEncodingStrategy = .iso8601
encoder.outputFormatting = pretty ? [.prettyPrinted, .sortedKeys] : []
if let data = try? encoder.encode(payload),
let output = String(data: data, encoding: .utf8)
{
print(output)
}
}
}
Self.exit(code: exitCode)
}
static func costHelp(version: String) -> String {
"""
CodexBar \(version)
Usage:
codexbar cost [--format text|json]
[--json]
[--json-output] [--log-level <trace|verbose|debug|info|warning|error|critical>] [-v|--verbose]
[--provider \(ProviderHelp.list)]
[--no-color] [--pretty] [--refresh]
Description:
Print local token cost usage from Claude/Codex JSONL logs. This does not require web or CLI access.
Uses cached scan results unless --refresh is provided.
Examples:
codexbar cost
codexbar cost --provider claude --format json --pretty
"""
}
static func renderCostText(
provider: UsageProvider,
snapshot: CostUsageTokenSnapshot,
useColor: Bool) -> String
{
let name = ProviderDescriptorRegistry.descriptor(for: provider).metadata.displayName
let header = Self.costHeaderLine("\(name) Cost (local)", useColor: useColor)
let todayCost = snapshot.sessionCostUSD.map { UsageFormatter.usdString($0) } ?? "—"
let todayTokens = snapshot.sessionTokens.map { UsageFormatter.tokenCountString($0) }
let todayLine = todayTokens.map { "Today: \(todayCost) · \($0) tokens" } ?? "Today: \(todayCost)"
let monthCost = snapshot.last30DaysCostUSD.map { UsageFormatter.usdString($0) } ?? "—"
let monthTokens = snapshot.last30DaysTokens.map { UsageFormatter.tokenCountString($0) }
let monthLine = monthTokens.map { "Last 30 days: \(monthCost) · \($0) tokens" } ?? "Last 30 days: \(monthCost)"
return [header, todayLine, monthLine].joined(separator: "\n")
}
private static func costHeaderLine(_ header: String, useColor: Bool) -> String {
guard useColor else { return header }
return "\u{001B}[1;36m\(header)\u{001B}[0m"
}
private static func costProviders(from selection: ProviderSelection) -> [UsageProvider] {
selection.asList.filter { Self.costSupportedProviders.contains($0) }
}
private static func makeCostPayload(provider: UsageProvider, snapshot: CostUsageTokenSnapshot) -> CostPayload {
let daily = snapshot.daily.map { entry in
CostDailyEntryPayload(
date: entry.date,
inputTokens: entry.inputTokens,
outputTokens: entry.outputTokens,
cacheReadTokens: entry.cacheReadTokens,
cacheCreationTokens: entry.cacheCreationTokens,
totalTokens: entry.totalTokens,
costUSD: entry.costUSD,
modelsUsed: entry.modelsUsed,
modelBreakdowns: entry.modelBreakdowns?.map { breakdown in
CostModelBreakdownPayload(modelName: breakdown.modelName, costUSD: breakdown.costUSD)
})
}
return CostPayload(
provider: provider.rawValue,
source: "local",
updatedAt: snapshot.updatedAt,
sessionTokens: snapshot.sessionTokens,
sessionCostUSD: snapshot.sessionCostUSD,
last30DaysTokens: snapshot.last30DaysTokens,
last30DaysCostUSD: snapshot.last30DaysCostUSD,
daily: daily,
totals: Self.costTotals(from: snapshot))
}
private static func costTotals(from snapshot: CostUsageTokenSnapshot) -> CostTotalsPayload? {
let entries = snapshot.daily
guard !entries.isEmpty else {
guard snapshot.last30DaysTokens != nil || snapshot.last30DaysCostUSD != nil else { return nil }
return CostTotalsPayload(
totalInputTokens: nil,
totalOutputTokens: nil,
cacheReadTokens: nil,
cacheCreationTokens: nil,
totalTokens: snapshot.last30DaysTokens,
totalCostUSD: snapshot.last30DaysCostUSD)
}
var totalInput = 0
var totalOutput = 0
var totalCacheRead = 0
var totalCacheCreation = 0
var totalTokens = 0
var totalCost = 0.0
var sawInput = false
var sawOutput = false
var sawCacheRead = false
var sawCacheCreation = false
var sawTokens = false
var sawCost = false
for entry in entries {
if let input = entry.inputTokens {
totalInput += input
sawInput = true
}
if let output = entry.outputTokens {
totalOutput += output
sawOutput = true
}
if let cacheRead = entry.cacheReadTokens {
totalCacheRead += cacheRead
sawCacheRead = true
}
if let cacheCreation = entry.cacheCreationTokens {
totalCacheCreation += cacheCreation
sawCacheCreation = true
}
if let tokens = entry.totalTokens {
totalTokens += tokens
sawTokens = true
}
if let cost = entry.costUSD {
totalCost += cost
sawCost = true
}
}
// Prefer totals derived from daily rows; fall back to snapshot aggregates when rows omit fields.
return CostTotalsPayload(
totalInputTokens: sawInput ? totalInput : nil,
totalOutputTokens: sawOutput ? totalOutput : nil,
cacheReadTokens: sawCacheRead ? totalCacheRead : nil,
cacheCreationTokens: sawCacheCreation ? totalCacheCreation : nil,
totalTokens: sawTokens ? totalTokens : snapshot.last30DaysTokens,
totalCostUSD: sawCost ? totalCost : snapshot.last30DaysCostUSD)
}
}
struct CostOptions: CommanderParsable {
@Flag(names: [.short("v"), .long("verbose")], help: "Enable verbose logging")
var verbose: Bool = false
@Flag(name: .long("json-output"), help: "Emit machine-readable logs")
var jsonOutput: Bool = false
@Option(name: .long("log-level"), help: "Set log level (trace|verbose|debug|info|warning|error|critical)")
var logLevel: String?
@Option(
name: .long("provider"),
help: ProviderHelp.optionHelp)
var provider: ProviderSelection?
@Option(name: .long("format"), help: "Output format: text | json")
var format: OutputFormat?
@Flag(name: .long("json"), help: "")
var jsonShortcut: Bool = false
@Flag(name: .long("pretty"), help: "Pretty-print JSON output")
var pretty: Bool = false
@Flag(name: .long("no-color"), help: "Disable ANSI colors in text output")
var noColor: Bool = false
@Flag(name: .long("refresh"), help: "Force refresh by ignoring cached scans")
var refresh: Bool = false
}
struct CostPayload: Encodable {
let provider: String
let source: String
let updatedAt: Date
let sessionTokens: Int?
let sessionCostUSD: Double?
let last30DaysTokens: Int?
let last30DaysCostUSD: Double?
let daily: [CostDailyEntryPayload]
let totals: CostTotalsPayload?
}
struct CostDailyEntryPayload: Encodable {
let date: String
let inputTokens: Int?
let outputTokens: Int?
let cacheReadTokens: Int?
let cacheCreationTokens: Int?
let totalTokens: Int?
let costUSD: Double?
let modelsUsed: [String]?
let modelBreakdowns: [CostModelBreakdownPayload]?
private enum CodingKeys: String, CodingKey {
case date
case inputTokens
case outputTokens
case cacheReadTokens
case cacheCreationTokens
case totalTokens
case costUSD = "totalCost"
case modelsUsed
case modelBreakdowns
}
}
struct CostModelBreakdownPayload: Encodable {
let modelName: String
let costUSD: Double?
private enum CodingKeys: String, CodingKey {
case modelName
case costUSD = "cost"
}
}
struct CostTotalsPayload: Encodable {
let totalInputTokens: Int?
let totalOutputTokens: Int?
let cacheReadTokens: Int?
let cacheCreationTokens: Int?
let totalTokens: Int?
let totalCostUSD: Double?
private enum CodingKeys: String, CodingKey {
case totalInputTokens = "inputTokens"
case totalOutputTokens = "outputTokens"
case cacheReadTokens
case cacheCreationTokens
case totalTokens
case totalCostUSD = "totalCost"
}
}
#if DEBUG
extension CodexBarCLI {
static func _costSignatureForTesting() -> CommandSignature {
CommandSignature.describe(CostOptions())
}
}
#endif