Fix one innerHTML usage #106395

microsoft · Sep 10, 2020 · ad451c4 · ad451c4
1 parent 2e56d31
commit ad451c4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/vs/workbench/contrib/notebook/browser/view/renderers/cellRenderer.ts b/src/vs/workbench/contrib/notebook/browser/view/renderers/cellRenderer.ts
@@ -597,8 +597,9 @@ class CodeCellDragImageRenderer {
 	}
 
 	private getDragImageImpl(templateData: BaseCellRenderTemplate, editor: ICodeEditor, type: 'code' | 'markdown'): HTMLElement | null {
-		const dragImageContainer = DOM.$(`.cell-drag-image.monaco-list-row.focused.${type}-cell-row`);
-		dragImageContainer.innerHTML = templateData.container.innerHTML;
+		const dragImageContainer = templateData.container.cloneNode(true) as HTMLElement;
+		dragImageContainer.classList.forEach(c => dragImageContainer.classList.remove(c));
+		dragImageContainer.classList.add('cell-drag-image', 'monaco-list-row', 'focused', `${type}-cell-row`);
 
 		const editorContainer = dragImageContainer.querySelector('.cell-editor-container');
 		if (!editorContainer) {