57 lines
1.7 KiB
TypeScript
57 lines
1.7 KiB
TypeScript
import { Gtk } from "ags/gtk3"
|
|
import { createPoll } from "ags/time"
|
|
import { shell } from "../lib/utils"
|
|
|
|
type ClaudeUsage = {
|
|
available: boolean
|
|
tokens: number
|
|
messages: number
|
|
}
|
|
|
|
const usage = createPoll<ClaudeUsage>(
|
|
{ available: false, tokens: 0, messages: 0 },
|
|
60000,
|
|
shell("bash $HOME/.config/ags/lib/claude-usage.sh"),
|
|
(stdout) => {
|
|
try {
|
|
return JSON.parse(stdout.trim())
|
|
} catch {
|
|
return { available: false, tokens: 0, messages: 0 }
|
|
}
|
|
},
|
|
)
|
|
|
|
function formatTokens(tokens: number) {
|
|
if (tokens >= 1000000) return `${(tokens / 1000000).toFixed(2)}M`
|
|
if (tokens >= 1000) return `${(tokens / 1000).toFixed(1)}K`
|
|
return `${tokens}`
|
|
}
|
|
|
|
export function ClaudeUsageCard() {
|
|
return (
|
|
<box class="panel widget-card" vertical spacing={6}>
|
|
<box spacing={8}>
|
|
<label class="widget-card-icon" label={""} />
|
|
<label class="widget-card-title" label="Claude Code" xalign={0} halign={Gtk.Align.START} hexpand />
|
|
</box>
|
|
<label
|
|
class="widget-card-empty"
|
|
label="No usage logs found"
|
|
xalign={0}
|
|
halign={Gtk.Align.START}
|
|
visible={usage.as((u) => !u.available)}
|
|
/>
|
|
<box vertical spacing={4} visible={usage.as((u) => u.available)}>
|
|
<box spacing={8}>
|
|
<label class="widget-card-label" label="Tokens today" xalign={0} halign={Gtk.Align.START} hexpand />
|
|
<label class="widget-card-value" label={usage.as((u) => formatTokens(u.tokens))} />
|
|
</box>
|
|
<box spacing={8}>
|
|
<label class="widget-card-label" label="Messages today" xalign={0} halign={Gtk.Align.START} hexpand />
|
|
<label class="widget-card-value" label={usage.as((u) => `${u.messages}`)} />
|
|
</box>
|
|
</box>
|
|
</box>
|
|
)
|
|
}
|