Skip to content

feat(tab): agent usage SFTP fallback for hosts without wsh #39

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
Atreus-X wants to merge 5 commits into from
Closed

feat(tab): agent usage SFTP fallback for hosts without wsh #39

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
898023f
feat(tab): agent usage falls back to SFTP on hosts without wsh
Atreus-X Sep 29, 2026
398e806
fix(tab): leave time for the SFTP fallback when the wsh route hangs
Atreus-X Sep 29, 2026
3ea501e
fix(tab): agent usage "Off" sticks by using an explicit "off" value
Atreus-X Sep 29, 2026
f4d0ff7
feat(tab): show Claude's real plan percentages and resets from the st…
Atreus-X Sep 29, 2026
15de027
fix(tab): drop the daily limit and Today reset countdown from agent u…
Atreus-X Sep 29, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 2 additions & 3 deletions docs/docs/config.mdx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -108,11 +108,10 @@ wsh editconfig
| tab:preset <DeprecatedBadge /> | string | a "bg@" preset to automatically apply to new tabs. e.g. `bg@green`. should match the preset key. deprecated in favor of `tab:background` |
| tab:background <VersionBadge version="v0.14.4" /> | string | a "bg@" preset to automatically apply to new tabs. e.g. `bg@green`. should match the preset key |
| library:taborder | string | order of the Library block's tabs, `"snippets,notes"` (default) or `"notes,snippets"`; the first tab opens by default. Set it by dragging one tab onto the other, or right-click the tab bar → Swap Tab Order |
| agentusage:agent | string | AI agent whose usage the tab bar meter tracks, `"claude"` (default) or `"codex"` (set to an empty string to turn the meter off; it then shows only a gauge icon). Usage is read from the agent's local session logs (`~/.claude/projects`, `~/.codex/sessions`) |
| agentusage:conn | string | connection the agent runs on, e.g. `user@host`; empty means this machine. Remote hosts need wsh installed |
| agentusage:agent | string | AI agent whose usage the tab bar meter tracks, `"claude"` (default) or `"codex"` (set to `"off"` to turn the meter off; it then shows only a gauge icon). Usage is read from the agent's local session logs (`~/.claude/projects`, `~/.codex/sessions`). For Claude Code, exact plan percentages and reset times are used when `~/.claude/rate-limits.json` exists: have your statusline script save them with `echo "$input" \| jq '.rate_limits' > ~/.claude/rate-limits.json` (it updates while a Claude Code session is running) |
| agentusage:conn | string | connection the agent runs on, e.g. `user@host`; empty means this machine. Remote hosts are read through wsh when installed, otherwise over SFTP |
| agentusage:sessionhours | int | length in hours of the plan's rolling usage session (default 5, as on Claude Pro/Max); the session opens with the first message, floored to the hour |
| agentusage:sessionlimit | int | your own token budget for one session; 0 shows raw usage without a bar |
| agentusage:dailylimit | int | your own token budget for today (input + output + cache-write tokens); 0 shows raw usage without a bar. The plan's real limits are not in the logs |
| agentusage:weeklylimit | int | your own token budget for the last 7 days; 0 shows raw usage without a bar |
| tab:confirmclose | bool | if set to true, a confirmation dialog will be shown before closing a tab (defaults to false) |
| tab:confirmcloserunning | bool | when true (default), closing a tab whose terminals are running a command shows a warning listing them, with Cancel as the default button. Detected via shell integration, for `term:tmux` terminals by asking tmux what each pane runs, and for plain SSH shells by the processes tagged with the terminal's `LC_WAVETERM_BLOCK` (needs the host's sshd to accept `LC_*`, the common default, and a `/proc` filesystem) |
Expand Down
101 changes: 60 additions & 41 deletions frontend/app/tab/agentusage.tsx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -81,29 +81,51 @@ const UsageBar = memo(({ used, limit }: { used: number; limit: number }) => {
});
UsageBar.displayName = "UsageBar";

const WindowRow = memo(({ label, win, limit }: { label: string; win: AgentUsageWindow; limit: number }) => {
const pct = limit > 0 ? Math.round((win.total / limit) * 100) : null;
const remaining = formatRemaining(win.resetat, Date.now());
return (
<div className="flex flex-col gap-1">
<div className="flex justify-between text-xs">
<span className="font-medium">
{label}
{remaining != null && <span className="font-normal text-muted"> ({remaining} until reset)</span>}
</span>
<span className="text-secondary">
{formatTokens(win.total)}
{limit > 0 ? ` / ${formatTokens(limit)} (${pct}%)` : ""}
</span>
</div>
{limit > 0 && <UsageBar used={win.total} limit={limit} />}
<div className="text-[11px] text-muted">
in {formatTokens(win.input)} · out {formatTokens(win.output)} · cache write{" "}
{formatTokens(win.cachewrite)} · cache read {formatTokens(win.cacheread)} · {win.messages} msgs
const WindowRow = memo(
({
label,
win,
limit = 0,
showReset = true,
}: {
label: string;
win: AgentUsageWindow;
limit?: number;
showReset?: boolean;
}) => {
const pct = limit > 0 ? Math.round((win.total / limit) * 100) : null;
const remaining = showReset ? formatRemaining(win.resetat, Date.now()) : null;
return (
<div className="flex flex-col gap-1">
<div className="flex justify-between text-xs">
<span className="font-medium">
{label}
{remaining != null && (
<span className="font-normal text-muted"> ({remaining} until reset)</span>
)}
</span>
<span className="text-secondary">
{formatTokens(win.total)}
{win.planpct != null
? ` · ${Math.round(win.planpct)}% of plan`
: limit > 0
? ` / ${formatTokens(limit)} (${pct}%)`
: ""}
</span>
</div>
{win.planpct != null ? (
<UsageBar used={win.planpct} limit={100} />
) : (
limit > 0 && <UsageBar used={win.total} limit={limit} />
)}
<div className="text-[11px] text-muted">
in {formatTokens(win.input)} · out {formatTokens(win.output)} · cache write{" "}
{formatTokens(win.cachewrite)} · cache read {formatTokens(win.cacheread)} · {win.messages} msgs
</div>
</div>
</div>
);
});
);
}
);
WindowRow.displayName = "WindowRow";

const LimitInput = memo(
Expand Down Expand Up @@ -148,7 +170,6 @@ LimitInput.displayName = "LimitInput";
const AgentUsageWidgetComponent = () => {
const agent = useAtomValue(getSettingsKeyAtom("agentusage:agent")) ?? "claude";
const conn = useAtomValue(getSettingsKeyAtom("agentusage:conn")) ?? "";
const dailyLimit = useAtomValue(getSettingsKeyAtom("agentusage:dailylimit")) ?? 0;
const weeklyLimit = useAtomValue(getSettingsKeyAtom("agentusage:weeklylimit")) ?? 0;
const sessionLimit = useAtomValue(getSettingsKeyAtom("agentusage:sessionlimit")) ?? 0;
const sessionHours = useAtomValue(getSettingsKeyAtom("agentusage:sessionhours")) ?? 5;
Expand All @@ -168,7 +189,7 @@ const AgentUsageWidgetComponent = () => {
useEffect(() => {
setUsage(null);
setError(null);
if (agent === "") {
if (agent === "off") {
return;
}
let cancelled = false;
Expand All @@ -177,7 +198,7 @@ const AgentUsageWidgetComponent = () => {
const data = await RpcApi.AgentUsageCommand(
TabRpcClient,
{ agent, conn: conn === "" ? undefined : conn, sessionhours: sessionHours },
{ timeout: 30000 }
{ timeout: 90000 }
);
if (cancelled) {
return;
Expand Down Expand Up @@ -214,7 +235,7 @@ const AgentUsageWidgetComponent = () => {
const connOptions = conn !== "" && !connList.includes(conn) ? [conn, ...connList] : connList;

let meter: React.ReactNode;
if (agent === "") {
if (agent === "off") {
meter = <i className="fa fa-gauge-high" />;
} else if (error != null) {
meter = <i className="fa fa-triangle-exclamation text-yellow-500" />;
Expand All @@ -230,10 +251,12 @@ const AgentUsageWidgetComponent = () => {
<>
{segments.map((seg, i) => {
const remaining = formatRemaining(seg.win.resetat, now);
const value =
seg.limit > 0
? `${Math.round((seg.win.total / seg.limit) * 100)}%`
: formatTokens(seg.win.total);
let value = formatTokens(seg.win.total);
if (seg.win.planpct != null) {
value = `${Math.round(seg.win.planpct)}%`;
} else if (seg.limit > 0) {
value = `${Math.round((seg.win.total / seg.limit) * 100)}%`;
}
return (
<span key={seg.label} className="whitespace-nowrap">
{i > 0 && <span className="text-muted"> · </span>}
Expand All @@ -255,7 +278,7 @@ const AgentUsageWidgetComponent = () => {
{...getReferenceProps()}
className="flex items-center gap-1.5 px-2 mb-1 h-[22px] text-xs rounded-sm cursor-pointer hover:bg-hover transition-colors"
style={{ WebkitAppRegion: "no-drag" } as any}
title={agent === "" ? "AI agent usage" : `${AgentLabels[agent] ?? agent} usage`}
title={agent === "off" ? "AI agent usage" : `${AgentLabels[agent] ?? agent} usage`}
>
{meter}
</div>
Expand All @@ -274,7 +297,7 @@ const AgentUsageWidgetComponent = () => {
win={usage.session}
limit={sessionLimit}
/>
<WindowRow label="Today" win={usage.today} limit={dailyLimit} />
<WindowRow label="Today" win={usage.today} showReset={false} />
<WindowRow label="Last 7 days" win={usage.week} limit={weeklyLimit} />
</>
)}
Expand All @@ -285,8 +308,9 @@ const AgentUsageWidgetComponent = () => {
)}
{error != null && <div className="text-xs text-red-400 break-words">{error}</div>}
<div className="text-[11px] text-muted">
Counts input + output + cache-write tokens read from the agent's local logs; the plan's real
limits aren't in them, so set your own budgets below.
Counts input + output + cache-write tokens read from the agent's local logs. For Claude
Code, plan percentages and reset times come from ~/.claude/rate-limits.json when its
statusline script saves them (see the docs); otherwise the budgets below are used.
</div>
<div className="flex flex-col gap-2 pt-2 border-t border-border">
<label className="flex items-center justify-between gap-2 text-xs">
Expand All @@ -296,7 +320,7 @@ const AgentUsageWidgetComponent = () => {
value={agent}
onChange={(e) => setSetting({ "agentusage:agent": e.target.value })}
>
<option value="">Off</option>
<option value="off">Off</option>
<option value="claude">Claude Code</option>
<option value="codex">Codex</option>
</select>
Expand Down Expand Up @@ -338,19 +362,14 @@ const AgentUsageWidgetComponent = () => {
value={sessionLimit}
onCommit={(v) => setSetting({ "agentusage:sessionlimit": v })}
/>
<LimitInput
label="Daily limit (tokens)"
value={dailyLimit}
onCommit={(v) => setSetting({ "agentusage:dailylimit": v })}
/>
<LimitInput
label="Weekly limit (tokens)"
value={weeklyLimit}
onCommit={(v) => setSetting({ "agentusage:weeklylimit": v })}
/>
{conn !== "" && (
<div className="text-[11px] text-muted">
Remote hosts need wsh installed on the connection.
Remote hosts are read through wsh when it is installed, otherwise over SFTP.
</div>
)}
</div>
Expand Down
2 changes: 1 addition & 1 deletion frontend/types/gotypes.d.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -99,6 +99,7 @@ declare global {
messages: number;
firstts: number;
resetat: number;
planpct?: number;
};

// wshrpc.AiMessageData
Expand Down Expand Up @@ -1738,7 +1739,6 @@ declare global {
"agentusage:conn"?: string;
"agentusage:sessionhours"?: number;
"agentusage:sessionlimit"?: number;
"agentusage:dailylimit"?: number;
"agentusage:weeklylimit"?: number;
"tab:confirmcloserunning"?: boolean;
"widget:*"?: boolean;
Expand Down
Loading