Release context claims on preflight abort

Author: esokullu

src/chrome/src/ui/sidepanel.js

@@ -7140,6 +7140,7 @@ async function sendMessage(extraChatParams = {}) {
 
   await prepareChatHistoryForTurn(tabId, modeForSend);
   if (isTabAbortRequested(tabId)) {
+    await releaseOwnedContextMenuClaim();
     setTabProcessing(tabId, false);
     setTabAbortRequested(tabId, false);
     syncSendButtonState();

src/firefox/src/ui/sidepanel.js

@@ -6871,6 +6871,7 @@ async function sendMessage(extraChatParams = {}) {
 
   await prepareChatHistoryForTurn(tabId, modeForSend);
   if (isTabAbortRequested(tabId)) {
+    await releaseOwnedContextMenuClaim();
     setTabProcessing(tabId, false);
     setTabAbortRequested(tabId, false);
     syncSendButtonState();

test/run.js

@@ -21939,14 +21939,19 @@ test('context-menu ownership and stale-panel persistence guards are wired in bot
     const sendMessageBody = panel.slice(sendMessageStart, sendMessageEnd);
     assert.equal(
       (sendMessageBody.match(/await releaseOwnedContextMenuClaim\(\);/g) || []).length,
-      4,
-      `${label}: every visibility exit should release the claim and schedule retry`,
+      5,
+      `${label}: every visibility or preflight-abort exit should release the claim and schedule retry`,
     );
     assert.match(
       sendMessageBody,
       /if \(!retryOptions\) text = await parseSlashCommands\(text, tabId, \{ permissionSkipContext \}\);[\s\S]*?if \(!renderToCurrentTab\) \{\s*await releaseOwnedContextMenuClaim\(\);/,
       `${label}: the first visibility failure after refresh and command parsing should relinquish initial ownership`,
     );
+    assert.match(
+      sendMessageBody,
+      /await prepareChatHistoryForTurn\(tabId, modeForSend\);\s*if \(isTabAbortRequested\(tabId\)\) \{\s*await releaseOwnedContextMenuClaim\(\);/,
+      `${label}: aborting preflight should relinquish initial ownership before resetting run state`,
+    );
     assert.match(
       sendMessageBody,
       /await prepareChatHistoryForTurn\(tabId, modeForSend\);[\s\S]*?if \(!renderToCurrentTab\) \{\s*await releaseOwnedContextMenuClaim\(\);[\s\S]*?contextMenuClaimOwned = renewedClaim\?\.claimed === true;[\s\S]*?const claimStillVisible[\s\S]*?if \(contextMenuClaimOwned && !claimStillVisible\) \{[\s\S]*?await releaseOwnedContextMenuClaim\(\);[\s\S]*?renderToCurrentTab = document\.visibilityState !== 'hidden'[\s\S]*?if \(!renderToCurrentTab\) \{\s*await releaseOwnedContextMenuClaim\(\);/,