Implement setup stage progress display in session logs

Author: Copilot

src/github/copilotRemoteAgent.ts

@@ -29,6 +29,7 @@ type RemoteAgentResult = RemoteAgentSuccessResult | RemoteAgentErrorResult;
 export interface IAPISessionLogs {
 	readonly info: SessionInfo;
 	readonly logs: string;
+	readonly setupLogs?: string[];
 }
 
 export interface ICopilotRemoteAgentCommandArgs {
@@ -629,7 +630,23 @@ export class CopilotRemoteAgentManager extends Disposable {
 		}
 
 		const logs = await capi.getLogsFromSession(session.id);
-		return { info: session, logs };
+
+		// If session is in progress, try to fetch setup logs from workflow
+		let setupLogs: string[] | undefined;
+		if (session.state === 'in_progress' || logs.trim().length === 0) {
+			try {
+				// Get the pull request model to access workflow logs
+				const pullRequest = await this.findPullRequestForSession(pullRequestId);
+				if (pullRequest) {
+					setupLogs = await this.getSessionLogsFromAction(pullRequest);
+				}
+			} catch (error) {
+				// If we can't fetch setup logs, don't fail the entire request
+				Logger.warn(`Failed to fetch setup logs for session ${session.id}: ${error}`, CopilotRemoteAgentManager.ID);
+			}
+		}
+
+		return { info: session, logs, setupLogs };
 	}
 
 	async getSessionUrlFromPullRequest(pullRequest: PullRequestModel): Promise<string | undefined> {
@@ -645,6 +662,28 @@ export class CopilotRemoteAgentManager extends Disposable {
 		return sessions.html_url;
 	}
 
+	private async findPullRequestForSession(pullRequestId: number): Promise<PullRequestModel | undefined> {
+		// Search through all repository managers to find the pull request
+		for (const manager of this.repositoriesManager.folderManagers) {
+			if (manager.gitHubRepositories.length === 0) {
+				continue;
+			}
+
+			for (const githubRepo of manager.gitHubRepositories) {
+				try {
+					const pullRequest = await githubRepo.getPullRequest(pullRequestId);
+					if (pullRequest) {
+						return pullRequest;
+					}
+				} catch (error) {
+					// Continue searching in other repositories
+					continue;
+				}
+			}
+		}
+		return undefined;
+	}
+
 	private getLatestRun<T extends { last_updated_at?: string; updated_at?: string }>(runs: T[]): T {
 		return runs
 			.slice()

src/view/sessionLogView.ts

@@ -409,7 +409,8 @@ class SessionLogView extends Disposable {
 			type: 'loaded',
 			pullInfo: this._source.type === 'pull' ? this._source.pullRequest : undefined,
 			info: apiResponse.info,
-			logs: apiResponse.logs
+			logs: apiResponse.logs,
+			setupLogs: apiResponse.setupLogs
 		} as messages.LoadedMessage);
 
 		if (apiResponse.info.state === 'in_progress') {
@@ -430,7 +431,8 @@ class SessionLogView extends Disposable {
 					type: 'update',
 					pullInfo: this._source.type === 'pull' ? this._source.pullRequest : undefined,
 					info: apiResult.info,
-					logs: apiResult.logs
+					logs: apiResult.logs,
+					setupLogs: apiResult.setupLogs
 				} as messages.UpdateMessage);
 
 				if (apiResult.info.state !== 'in_progress') {

webviews/sessionLogView/app.tsx

@@ -16,7 +16,7 @@ const themeName = 'vscode-theme';
 
 type SessionViewState =
 	| { state: 'loading' }
-	| { state: 'ready'; readonly info: SessionInfo; readonly logs: readonly SessionResponseLogChunk[]; readonly pullInfo: messages.PullInfo | undefined }
+	| { state: 'ready'; readonly info: SessionInfo; readonly logs: readonly SessionResponseLogChunk[]; readonly pullInfo: messages.PullInfo | undefined; readonly setupLogs?: readonly string[] }
 	| { state: 'error'; readonly url: string | undefined }
 	;
 
@@ -49,7 +49,8 @@ export function App() {
 						state: 'ready',
 						info: message.info,
 						logs: parseSessionLogs(message.logs),
-						pullInfo: message.pullInfo
+						pullInfo: message.pullInfo,
+						setupLogs: message.setupLogs
 					});
 					break;
 				}
@@ -89,7 +90,7 @@ export function App() {
 			</div>
 		);
 	} else {
-		return <SessionView info={state.info} logs={state.logs} pullInfo={state.pullInfo} />;
+		return <SessionView info={state.info} logs={state.logs} pullInfo={state.pullInfo} setupLogs={state.setupLogs} />;
 	}
 }
 

webviews/sessionLogView/index.css

@@ -185,4 +185,41 @@ button.codeview-expand:hover {
 	to {
 		transform: rotate(360deg);
 	}
+}
+
+.setup-stage-log {
+	margin: 1em 0;
+	border: 1px solid var(--vscode-widget-border);
+	border-radius: 4px;
+	overflow: hidden;
+}
+
+.setup-stage-title {
+	display: flex;
+	align-items: center;
+	padding: 8px 12px;
+	background: var(--vscode-sideBarSectionHeader-background);
+	border-bottom: 1px solid var(--vscode-sideBarSectionHeader-border);
+	color: var(--vscode-sideBarSectionHeader-foreground);
+	font-size: 13px;
+	font-weight: 600;
+	margin: 0;
+	gap: 8px;
+}
+
+.setup-log-content {
+	padding: 8px 12px;
+	background: var(--vscode-notebook-cellEditorBackground);
+	max-height: 200px;
+	overflow-y: auto;
+}
+
+.setup-log-line {
+	font-family: var(--vscode-editor-font-family);
+	font-size: var(--vscode-editor-font-size);
+	line-height: 1.4;
+	color: var(--vscode-foreground);
+	margin: 2px 0;
+	white-space: pre-wrap;
+	word-wrap: break-word;
 }

webviews/sessionLogView/messages.ts

@@ -20,13 +20,15 @@ export interface LoadedMessage {
 	pullInfo: PullInfo | undefined;
 	info: SessionInfo;
 	logs: string;
+	setupLogs?: string[];
 }
 
 export interface UpdateMessage {
 	type: 'update';
 	pullInfo: PullInfo | undefined;
 	info: SessionInfo;
 	logs: string;
+	setupLogs?: string[];
 }
 
 export interface ResetMessage {

webviews/sessionLogView/sessionView.tsx

@@ -19,17 +19,25 @@ interface SessionViewProps {
 	readonly pullInfo: PullInfo | undefined;
 	readonly info: SessionInfo;
 	readonly logs: readonly SessionResponseLogChunk[];
+	readonly setupLogs?: readonly string[];
 }
 
 export const SessionView: React.FC<SessionViewProps> = (props) => {
 	return (
 		<div className="session-container">
 			<SessionHeader info={props.info} pullInfo={props.pullInfo} />
+			{props.setupLogs && props.setupLogs.length > 0 && (
+				<SetupStageLog setupLogs={props.setupLogs} />
+			)}
 			<SessionLog logs={props.logs} />
 			{props.info.state === 'in_progress' && (
 				<div className="session-in-progress-indicator">
 					<span className="icon"><i className="codicon codicon-loading"></i></span>
-					Session is in progress...</div>
+					{props.logs.length === 0 && props.setupLogs && props.setupLogs.length > 0
+						? 'Configuring environment...'
+						: 'Session is in progress...'
+					}
+				</div>
 			)}
 		</div>
 	);
@@ -268,3 +276,34 @@ function toFileLabel(file: string): string {
 	const parts = file.split('/');
 	return parts.slice(6).join('/');
 }
+
+// Setup Stage Log component
+interface SetupStageLogProps {
+	readonly setupLogs: readonly string[];
+}
+
+const SetupStageLog: React.FC<SetupStageLogProps> = ({ setupLogs }) => {
+	if (!setupLogs || setupLogs.length === 0) {
+		return null;
+	}
+
+	const setupSteps = setupLogs
+		.filter(line => line.trim().length > 0)
+		.map((line, index) => (
+			<div key={index} className="setup-log-line">
+				{line}
+			</div>
+		));
+
+	return (
+		<div className="setup-stage-log">
+			<h3 className="setup-stage-title">
+				<span className="icon"><i className="codicon codicon-gear"></i></span>
+				Environment Setup
+			</h3>
+			<div className="setup-log-content">
+				{setupSteps}
+			</div>
+		</div>
+	);
+};