* Add agent usage collectors that write display-ready data files
One omarchy-agent-usage-scan-<agent> collector per AI coding agent prints a
complete display-ready usage record — identity, tier, status, rate limits,
and today/week/all-time stats. omarchy-agent-usage-update runs every
collector it finds and writes the records atomically to
~/.local/state/omarchy/agents/usage/, so anything that displays usage only
ever reads JSON from there.
The Claude collector absorbs what the shell previously did in-process:
transcript scanning, the stats-cache/history fallback, credentials parsing,
and the OAuth limits probe, now with a probe throttle and last-good limits
kept across network failures. The Codex collector is the existing scanner
reshaped to the shared record contract.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
* Redo the model-usage plugin as omarchy.agents watching usage data files
The panel is now strictly a display. It discovers the JSON records that
omarchy-agent-usage-update maintains under
~/.local/state/omarchy/agents/usage/, watches them for changes, and draws
whatever appears — so adding an agent means shipping a collector, never
touching the panel. Marks resolve by convention (assets/<id>.svg with an
optional -light twin), the limits meters read a generic limits array, and
the per-provider QML adapters and in-plugin scanner scripts are gone.
Cross-device sync aggregation stays in the shell and keeps the snapshot
field names older versions wrote, so mixed-version fleets still merge in
both directions.
With the provider fan-out gone, the widget takes its real name: the plugin
id becomes omarchy.agents. A migration renames it wherever a user's config
mentions it — layout entries keep their settings and position, a disabled
widget stays disabled — then primes the data files once and drops the old
scanner cache. The migration test also drops a stale assertion that expected
migrations to restart the shell themselves, which c992cdff moved to
omarchy update.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
* Address Codex review: synced-only tabs, limits retry, history fallback
Three data-availability gaps from review. An agent whose records only exist
in synced snapshots — a collector installed on just one machine — now gets
its tab by unioning the synced aggregate into the provider list, with rate
limits blank since those never travel. A Claude limits probe that reaches no
server at all writes retryAdvised into its record, and the shell honors it
with one 30-second retry instead of waiting out the full refresh interval,
restoring the old boot-before-DHCP behavior. And a machine with only
history.jsonl — no transcripts, no stats-cache — still reports today's
prompt and session counts.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
* Address second Codex pass: history-only visibility, targeted retries
Today's prompt and session counts now count toward an agent's presence in
the bar, so a machine whose only Claude source is history.jsonl shows up
without waiting for limits. And the 30-second limits retry passes the
advising agent ids to the updater, so an outage at one provider no longer
puts every other collector on a retry treadmill.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
* Drop omarchy-cmd-present jq guards from the agents migrations
jq ships in the default package set, which makes it a runtime invariant per
AGENTS.md — call it directly. The migration tests lose their now-unused
omarchy-cmd-present stubs with it.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
* Drop the scan infix from the collector command names
Collectors are omarchy-agent-usage-<agent>; the updater skips its own name
when globbing them, and the update test proves it with a decoy.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
* Keep the credential store out of the printed usage record
The Claude collector now reads .credentials.json once into three scalars —
the access token, its expiry, and the plan label — instead of passing the
parsed store around. The token reaches nothing but the Authorization header
of the limits probe, and only the plan label may travel into the record,
which is what CodeQL's clear-text-logging alert on the record print was
unable to see when the whole dict flowed through.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
---------
Co-authored-by: Claude Fable 5 <noreply@anthropic.com>
183 lines
6.3 KiB
QML
183 lines
6.3 KiB
QML
import QtQuick
|
|
import Quickshell
|
|
import qs.Commons
|
|
|
|
ShellRoot {
|
|
id: root
|
|
|
|
readonly property string resultPath: Quickshell.env("OMARCHY_QML_TEST_RESULT")
|
|
readonly property string rootPath: Quickshell.env("OMARCHY_PATH")
|
|
property var failures: []
|
|
property var createdIds: []
|
|
property var createdObjects: []
|
|
|
|
function fail(message) {
|
|
failures.push(String(message))
|
|
}
|
|
|
|
function assertTrue(condition, message) {
|
|
if (!condition) fail(message)
|
|
}
|
|
|
|
function assertEqual(actual, expected, message) {
|
|
if (actual !== expected) fail(message + " expected=" + expected + " actual=" + actual)
|
|
}
|
|
|
|
function shellQuote(value) {
|
|
return "'" + String(value).replace(/'/g, "'\\''") + "'"
|
|
}
|
|
|
|
function writeResult() {
|
|
var payload = JSON.stringify({
|
|
ok: failures.length === 0,
|
|
failures: failures,
|
|
created: createdIds
|
|
})
|
|
|
|
if (resultPath) {
|
|
Quickshell.execDetached(["bash", "-lc", "printf '%s' " + shellQuote(payload) + " > " + shellQuote(resultPath)])
|
|
}
|
|
}
|
|
|
|
function widgets() {
|
|
try {
|
|
return JSON.parse(Qt.atob(Quickshell.env("OMARCHY_QML_BAR_WIDGETS") || "W10="))
|
|
} catch (error) {
|
|
fail("bar widget list failed to parse: " + error)
|
|
return []
|
|
}
|
|
}
|
|
|
|
function safeCall(item, method, entry) {
|
|
if (!item || typeof item[method] !== "function") return
|
|
try {
|
|
item[method]()
|
|
} catch (error) {
|
|
fail(entry.id + " " + method + "() threw: " + error)
|
|
}
|
|
}
|
|
|
|
function finiteDimension(value) {
|
|
var n = Number(value)
|
|
return isFinite(n) && n >= 0
|
|
}
|
|
|
|
function loadWidget(entry) {
|
|
var component = Qt.createComponent(entry.url, Component.PreferSynchronous)
|
|
if (component.status !== Component.Ready) {
|
|
fail(entry.id + " failed to load: " + component.errorString())
|
|
return
|
|
}
|
|
|
|
var item = component.createObject(host, {
|
|
moduleName: entry.id,
|
|
settings: {}
|
|
})
|
|
if (!item) {
|
|
fail(entry.id + " failed to instantiate without bar: " + component.errorString())
|
|
return
|
|
}
|
|
|
|
if ("bar" in item) {
|
|
root.assertTrue(item.bar === null || item.bar === undefined, entry.id + " starts without injected bar")
|
|
item.bar = fakeBar
|
|
root.assertTrue(item.bar === fakeBar, entry.id + " accepts delayed bar injection")
|
|
}
|
|
if ("moduleName" in item) {
|
|
item.moduleName = entry.id
|
|
root.assertEqual(item.moduleName, entry.id, entry.id + " accepts moduleName injection")
|
|
}
|
|
if ("settings" in item) {
|
|
item.settings = {}
|
|
root.assertTrue(item.settings !== null && item.settings !== undefined, entry.id + " accepts settings injection")
|
|
}
|
|
if (typeof item.setting === "function") {
|
|
root.assertEqual(item.setting("missing", "fallback"), "fallback", entry.id + " exposes setting fallback")
|
|
}
|
|
if (entry.id === "omarchy.agents") {
|
|
root.assertTrue(typeof item.iconCandidatesForProvider === "function", entry.id + " resolves provider marks by convention")
|
|
var darkIcons = item.iconCandidatesForProvider({ providerId: "codex" }, Qt.color("#1a1b26")).join(" ")
|
|
var lightIcons = item.iconCandidatesForProvider({ providerId: "codex" }, Qt.color("#ffffff")).join(" ")
|
|
root.assertTrue(darkIcons.indexOf("codex.svg") >= 0 && darkIcons.indexOf("codex-light.svg") < 0, entry.id + " uses the dark-theme Codex icon on dark surfaces")
|
|
root.assertTrue(lightIcons.indexOf("codex-light.svg") >= 0, entry.id + " prefers the light-theme Codex icon on light surfaces")
|
|
}
|
|
|
|
safeCall(item, "refresh", entry)
|
|
safeCall(item, "close", entry)
|
|
|
|
createdObjects.push(item)
|
|
createdIds.push(entry.id)
|
|
}
|
|
|
|
Item { id: host }
|
|
|
|
QtObject {
|
|
id: mockShell
|
|
property var bar: fakeBar
|
|
property var barConfig: ({ position: "top" })
|
|
property var shellConfig: ({ version: 1, idle: {}, plugins: [], bar: { layout: { left: [], center: [], right: [] } } })
|
|
function firstPartyServiceFor(id) { return null }
|
|
function serviceFor(id) { return null }
|
|
function summon(id, payloadJson) { return true }
|
|
function hide(id) { return true }
|
|
function toggle(id, payloadJson) { return true }
|
|
function updateEntryInline(moduleName, settings) { return true }
|
|
}
|
|
|
|
QtObject {
|
|
id: fakeBar
|
|
property bool vertical: false
|
|
property int barSize: 26
|
|
property string omarchyPath: root.rootPath
|
|
property string fontFamily: "monospace"
|
|
property color foreground: "white"
|
|
property color background: "black"
|
|
property color urgent: "red"
|
|
property var shell: mockShell
|
|
function run(command) {}
|
|
function showTooltip(target, text) {}
|
|
function hideTooltip(target) {}
|
|
function requestPopout(owner) {}
|
|
function releasePopout(owner) {}
|
|
function registerClickTarget(target) {}
|
|
function unregisterClickTarget(target) {}
|
|
}
|
|
|
|
Timer {
|
|
interval: 1
|
|
running: true
|
|
repeat: false
|
|
onTriggered: {
|
|
var entries = widgets()
|
|
root.assertTrue(entries.length > 0, "bar widget list is not empty")
|
|
for (var i = 0; i < entries.length; i++) root.loadWidget(entries[i])
|
|
|
|
Qt.callLater(function() {
|
|
for (var j = 0; j < root.createdObjects.length; j++) {
|
|
var item = root.createdObjects[j]
|
|
var id = root.createdIds[j]
|
|
root.assertTrue(root.finiteDimension(item.implicitWidth), id + " has a finite implicitWidth")
|
|
root.assertTrue(root.finiteDimension(item.implicitHeight), id + " has a finite implicitHeight")
|
|
}
|
|
|
|
fakeBar.vertical = true
|
|
fakeBar.barSize = Style.bar.sizeVertical
|
|
|
|
Qt.callLater(function() {
|
|
for (var k = 0; k < root.createdObjects.length; k++) {
|
|
var verticalItem = root.createdObjects[k]
|
|
var verticalId = root.createdIds[k]
|
|
if (verticalId === "omarchy.clock")
|
|
root.assertEqual(verticalItem.implicitHeight, Style.bar.iconSlot * 3, verticalId + " uses one slot per line")
|
|
else if (verticalId === "omarchy.weather" || verticalId === "omarchy.system-update")
|
|
root.assertEqual(verticalItem.implicitHeight, Style.bar.statusSlot, verticalId + " uses one compact status slot")
|
|
if (verticalItem && typeof verticalItem.destroy === "function") verticalItem.destroy()
|
|
}
|
|
root.assertTrue(root.createdIds.length === entries.length, "all bar widgets instantiate")
|
|
root.writeResult()
|
|
})
|
|
})
|
|
}
|
|
}
|
|
}
|