Restore focus to comment textarea when description page regains focus

Co-authored-by: alexr00 <38270282+alexr00@users.noreply.github.com>

Author: Copilot

src/github/pullRequestOverview.ts

@@ -186,6 +186,11 @@ export class PullRequestOverviewPanel extends IssueOverviewPanel<PullRequestMode
 		if (this._panel.visible && this._item) {
 			PullRequestOverviewPanel._onVisible.fire(this._item);
 		}
+
+		// If the panel becomes active, send a message to restore focus to the comment textarea if needed
+		if (this._panel.active) {
+			this._postMessage({ command: 'pr.restoreFocus' });
+		}
 	}
 
 	private setVisibilityContext() {

webviews/common/context.tsx

@@ -410,6 +410,17 @@ export class PRContext {
 					pendingReview.focus();
 				}
 				return;
+			case 'pr.restoreFocus':
+				// Restore focus to the comment textarea if it had focus before the webview lost focus
+				const commentTextarea = document.getElementById('comment-textarea');
+				if (commentTextarea instanceof HTMLTextAreaElement) {
+					// Check if there's a stored focus state
+					const shouldRestoreFocus = sessionStorage.getItem('comment-textarea-had-focus') === 'true';
+					if (shouldRestoreFocus) {
+						commentTextarea.focus();
+					}
+				}
+				return;
 			case 'pr.submitting-review':
 				return this.updatePR({ busy: true, lastReviewType: message.lastReviewType });
 			case 'pr.append-review':

webviews/editorWebview/app.tsx

@@ -21,6 +21,34 @@ export function Root({ children }) {
 		ctx.onchange = setPR;
 		setPR(ctx.pr);
 	}, []);
+
+	// Track focus on the comment textarea to restore it when the webview regains focus
+	useEffect(() => {
+		const handleFocusIn = (e: FocusEvent) => {
+			if (e.target instanceof HTMLTextAreaElement && e.target.id === 'comment-textarea') {
+				sessionStorage.setItem('comment-textarea-had-focus', 'true');
+			}
+		};
+
+		const handleFocusOut = (e: FocusEvent) => {
+			if (e.target instanceof HTMLTextAreaElement && e.target.id === 'comment-textarea') {
+				// Only clear the flag if we're switching to a non-textarea element
+				// This prevents clearing when the webview loses focus
+				if (e.relatedTarget instanceof HTMLElement) {
+					sessionStorage.setItem('comment-textarea-had-focus', 'false');
+				}
+			}
+		};
+
+		document.addEventListener('focusin', handleFocusIn);
+		document.addEventListener('focusout', handleFocusOut);
+
+		return () => {
+			document.removeEventListener('focusin', handleFocusIn);
+			document.removeEventListener('focusout', handleFocusOut);
+		};
+	}, []);
+
 	window.onscroll = debounce(() => {
 		ctx.postMessage({
 			command: 'scroll',