fix: close three stale-completion holes from the round-16 suppressed block (#2144)

Copilot's round 16 reported "no new comments", but its suppressed block carried
three real findings.

- `AuthTab` released `clearInFlightRef` before checking whether the completion
  was still current, so server A settling after the user moved to B and started
  a clear there dropped B's lock and let a second B clear run concurrently
  against the same store entry. `current()` is now checked before either shared
  ref is touched: a stale clear owns nothing.
- Switching servers retired the clear but left `ownClearRef` set. The retired
  clear returns before its `oauthRevision` bump, so the marker had no bump to
  skip and would swallow the first unrelated revision on the newly selected
  server, stranding that server's banner.
- The web session check compared only the server id, which a
  disconnect/reconnect to the SAME server passes — that flow builds a
  replacement `InspectorClient`, so the old clear would have run its
  session-wide cleanup against the new session. The client identity is part of
  the check now.

Both new tests were verified to fail with their fix reverted.

Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_0177mPHAdECD18nLLCTwR5rh
Signed-off-by: cliffhall <cliff@futurescale.com>
This commit is contained in:
cliffhall 2026-08-28 06:47:31 -04:00
commit b9012da988
4 changed files with 102 additions and 3 deletions

View file

@ -187,6 +187,46 @@ describe("AuthTab", () => {
expect(onClearOAuth).toHaveBeenCalledTimes(2);
});
// A stale completion owns nothing: server A settling after the user moved to
// B and started a clear there must not drop B's lock, or a second B clear
// could run concurrently against the same store entry.
it("a stale completion does not release the current server's lock", async () => {
const settlers: Array<() => void> = [];
const onClearOAuth = vi.fn(
() =>
new Promise<void>((resolve) => {
settlers.push(resolve);
}),
);
const props = (serverName: string) => ({
...baseProps,
serverName,
onClearOAuth,
inspectorClient: null,
oauthStatus: "idle" as const,
oauthMessage: null,
focused: true,
});
const { stdin, rerender } = render(<AuthTab {...props("a")} />);
await tick();
stdin.write("s"); // A's clear starts
await tick();
rerender(<AuthTab {...props("b")} />);
await tick();
stdin.write("s"); // B's clear starts
await tick();
expect(onClearOAuth).toHaveBeenCalledTimes(2);
// A settles late. B's clear is still running, so its lock must hold.
settlers[0]!();
await tick();
stdin.write("s");
await tick();
expect(onClearOAuth).toHaveBeenCalledTimes(2);
});
// A rejection is NOT a revocation failure — those come back as outcomes and
// are reported through the message line. This is the local clear or the
// disconnect itself failing, so reporting success would be a plain lie.

View file

@ -123,6 +123,10 @@ export function AuthTab({
useEffect(() => {
clearAttemptRef.current++;
clearInFlightRef.current = false;
// The retired clear returns before its `oauthRevision` bump, so the marker
// has no bump to skip. Left set, it would swallow the first unrelated
// revision on the server just selected and strand that server's banner.
ownClearRef.current = false;
setClearState("idle");
setClearFailure(null);
setLastClearDisconnected(false);
@ -231,10 +235,16 @@ export function AuthTab({
serverNameRef.current === attemptServer;
void Promise.resolve(onClearOAuth()).then(
() => {
// `current()` FIRST, before touching either shared ref. A stale
// completion — server A settling after the user moved to B and
// started a clear there — would otherwise drop B's lock and let a
// second B clear run concurrently. A stale clear owns nothing.
if (!current()) return;
clearInFlightRef.current = false;
if (current()) setClearState("cleared");
setClearState("cleared");
},
(err: unknown) => {
if (!current()) return;
clearInFlightRef.current = false;
// A rejection is NOT a revocation failure — those come back as
// outcomes and are reported through the message line. This is the
@ -246,7 +256,6 @@ export function AuthTab({
// Left set, it would swallow the next *unrelated* revision change
// and strand this banner after the OAuth state moved on.
ownClearRef.current = false;
if (!current()) return;
setClearFailure(err instanceof Error ? err.message : String(err));
setClearState("failed");
},

View file

@ -1409,6 +1409,50 @@ describe("useOAuthRecovery", () => {
expect(client.disconnect).not.toHaveBeenCalled();
});
// A disconnect/reconnect to the SAME server builds a replacement client, so
// an id-only check passes again and the old clear would run its
// session-wide cleanup — including the disconnect — against the new session.
it("does not disconnect a replacement client for the same server", async () => {
let settle: (r: { cleared: boolean }) => void = () => {};
clearServerOAuthStateMock.mockImplementation(
() =>
new Promise((resolve) => {
settle = resolve as typeof settle;
}),
);
const original = fakeClient();
const h = harness({
servers: [entry("a")],
activeServerId: "a",
client: original,
});
let done: Promise<void>;
await act(async () => {
done = h.api().clearServerOAuthAndDisconnect(entry("a"));
await Promise.resolve();
});
// Same server, new client — a reconnect while the clear is pending.
const replacement = fakeClient();
h.rerender({
servers: [entry("a")],
activeServerId: "a",
client: replacement,
});
await act(async () => {
settle({ cleared: true });
await done;
});
// Neither client is torn down: the replacement is the live session and
// must not be touched, and the original is already gone — disconnecting
// it would only drag `finalizeExplicitDisconnect` across the new one.
expect(original.disconnect).not.toHaveBeenCalled();
expect(replacement.disconnect).not.toHaveBeenCalled();
});
it("clears the resume snapshot on an explicit disconnect", () => {
writeOAuthResumeSnapshot({
version: 1,

View file

@ -1265,8 +1265,14 @@ export function useOAuthRecovery({
// The session ref is the live answer. Only the parts that touch the
// *session* are gated on it; the store write already happened and the
// toast still belongs to the server the user asked about.
// The client identity is part of the check, not just the server id: a
// disconnect/reconnect to the SAME server builds a replacement
// `InspectorClient`, so an id-only check passes again and the old clear
// would run its session-wide cleanup against the new session.
const stillTargetsActiveSession = (): boolean =>
isActive && sessionRef.current.activeServerId === server.id;
isActive &&
sessionRef.current.activeServerId === server.id &&
sessionRef.current.inspectorClient === client;
const { cleared, revocation } = await clearServerOAuthState({
config: server.config,