simplify fix

Author: eleanorjboyd

src/vs/workbench/browser/labels.ts

@@ -492,40 +492,34 @@ class ResourceLabelWidget extends IconLabel {
 		if (!options.forceLabel && !isSideBySideEditor && resource?.scheme === Schemas.vscodeNotebookCellOutput) {
 			const notebookDocument = this.notebookDocumentService.getNotebook(resource);
 			const outputUriData = extractCellOutputDetails(resource);
-
-			let cellIndex: number | undefined;
-			if (!outputUriData?.notebook) {
-				return;
-			}
-			if (outputUriData.cellHandle) {
-				// if we have a cell handle, we can use that to get the cell index
+			if (outputUriData?.cellFragment) {
+				if (!outputUriData.notebook) {
+					return;
+				}
 				const cellUri = outputUriData.notebook.with({
 					scheme: Schemas.vscodeNotebookCell,
 					fragment: outputUriData.cellFragment
 				});
-				cellIndex = notebookDocument?.getCellIndex(cellUri);
-			}
-			const outputIndex = outputUriData.outputIndex;
-
-			if (cellIndex !== undefined && outputIndex !== undefined && typeof label.name === 'string') {
-				label.name = localize(
-					'notebookCellOutputLabel',
-					"{0} • Cell {1} • Output {2}",
-					label.name,
-					`${cellIndex + 1}`,
-					`${outputIndex + 1}`
-				);
-			} else if (cellIndex !== undefined && typeof label.name === 'string') {
-				label.name = localize(
-					'notebookCellOutputLabelSimple',
-					"{0} • Cell {1} • Output",
-					label.name,
-					`${cellIndex + 1}`
-				);
-			} else if (typeof label.name === 'string') {
-				label.name = localize('notebookCellOutputLabelJustFile', "{0}", label.name);
+				const cellIndex = notebookDocument?.getCellIndex(cellUri);
+				const outputIndex = outputUriData.outputIndex;
+
+				if (cellIndex !== undefined && outputIndex !== undefined && typeof label.name === 'string') {
+					label.name = localize(
+						'notebookCellOutputLabel',
+						"{0} • Cell {1} • Output {2}",
+						label.name,
+						`${cellIndex + 1}`,
+						`${outputIndex + 1}`
+					);
+				} else if (cellIndex !== undefined && typeof label.name === 'string') {
+					label.name = localize(
+						'notebookCellOutputLabelSimple',
+						"{0} • Cell {1} • Output",
+						label.name,
+						`${cellIndex + 1}`
+					);
+				}
 			}
-
 		}
 
 		const hasResourceChanged = this.hasResourceChanged(label);

src/vs/workbench/services/notebook/common/notebookDocumentService.ts

@@ -78,34 +78,23 @@ export function extractCellOutputDetails(uri: URI): { notebook: URI; openIn: str
 	}
 	const outputId = params.get('outputId') ?? undefined;
 	const notebookScheme = params.get('notebookScheme') ?? undefined;
-	const outputIndex = params.get('outputIndex') ? parseInt(params.get('outputIndex') || '', 10) : undefined;
-
-	if (openIn === 'editor') {
-		const notebookUri = uri.with({
-			scheme: notebookScheme || Schemas.file,
-			fragment: null,
-			query: null,
-		});
-		return {
-			notebook: notebookUri,
-			openIn: openIn,
-			outputId: outputId,
-			outputIndex: outputIndex,
-		};
-	}
-	// If openIn is not editor, it needs to be opened in the notebook editor
 	const parsedCell = parse(uri.with({ scheme: Schemas.vscodeNotebookCell, query: null }));
-
-	if (parsedCell?.notebook === undefined || parsedCell?.handle === undefined) {
-		throw new Error('Invalid notebook output cell URI');
+	const outputIndex = params.get('outputIndex') ? parseInt(params.get('outputIndex') || '', 10) : undefined;
+	const notebookUri = parsedCell ? parsedCell.notebook : uri.with({
+		scheme: notebookScheme || Schemas.file,
+		fragment: null,
+		query: null,
+	});
+	if (notebookUri === undefined) {
+		throw new Error('Invalid cell URI');
 	}
 
 	return {
-		notebook: parsedCell.notebook,
+		notebook: notebookUri,
 		openIn: openIn,
 		outputId: outputId,
 		outputIndex: outputIndex,
-		cellHandle: parsedCell.handle,
+		cellHandle: parsedCell?.handle,
 		cellFragment: uri.fragment,
 	};
 }