From edd006ebbb0addbb8e22ac96ccea99b01938759d Mon Sep 17 00:00:00 2001
From: John Pan <john.pan@mail.mcgill.ca>
Date: Wed, 6 Nov 2024 16:39:33 -0800
Subject: [PATCH] Add highlight of quickinput during cmd+i step

---
 .../onboardingShadow/shadowOverlayService.ts  | 109 ++++++++++++++++++
 src/vs/workbench/browser/workbench.ts         |   6 +
 2 files changed, 115 insertions(+)
 create mode 100644 src/vs/workbench/browser/parts/overlay/onboardingShadow/shadowOverlayService.ts

diff --git a/src/vs/workbench/browser/parts/overlay/onboardingShadow/shadowOverlayService.ts b/src/vs/workbench/browser/parts/overlay/onboardingShadow/shadowOverlayService.ts
new file mode 100644
index 0000000000000..ff48587da1157
--- /dev/null
+++ b/src/vs/workbench/browser/parts/overlay/onboardingShadow/shadowOverlayService.ts
@@ -0,0 +1,109 @@
+import { createDecorator } from 'vs/platform/instantiation/common/instantiation';
+import { registerSingleton, InstantiationType } from 'vs/platform/instantiation/common/extensions';
+import { Disposable, IDisposable } from 'vs/base/common/lifecycle';
+import { CommandsRegistry } from 'vs/platform/commands/common/commands';
+
+export const IShadowOverlayService = createDecorator<IShadowOverlayService>('shadowOverlayService');
+
+export interface IShadowOverlayService extends IDisposable {
+    readonly _serviceBrand: undefined;
+
+    highlight(elements: string[]): void;
+	restoreStyles(elements: string[]): void;
+}
+
+
+export class ShadowOverlayService extends Disposable implements IShadowOverlayService {
+    declare readonly _serviceBrand: undefined;
+	private highlightedElements: Map<string, {
+        position: string;
+        zIndex: string;
+        isolation: string;
+        pointerEvents: string;
+        backgroundColor: string;
+        boxShadow: string;
+		transition: string,
+    }> = new Map();
+
+    constructor(
+    ) {
+		super();
+		this.registerCommands();
+	}
+
+	private registerCommands(): void {
+		CommandsRegistry.registerCommand('pearai.highlightElements', (accessor, ...args) => {
+
+			const selectors = args[0] as string[];  // array of CSS selectors
+			this.highlight(selectors);
+		});
+
+		CommandsRegistry.registerCommand('pearai.removeHighlight', (accessor, ...args) => {
+			const selectors = args[0] as string[];  // array of CSS selectors
+			 // Convert selectors to elements
+			 console.dir("in app UNHIGHLIGHT COMMAND, GOT SELECTORS:");
+			 console.dir(selectors);
+			this.restoreStyles(selectors);
+		});
+	}
+
+	restoreStyles(selectors: string[]): void {
+		selectors.forEach(selector => {
+			console.dir("in app RESTORE STYLES, GOT ELEMENT:");
+			console.dir(selector);
+			const originalStyles = this.highlightedElements.get(selector);
+			console.dir(originalStyles);
+
+			const element = document.querySelector(selector) as HTMLElement
+
+			if (originalStyles) {
+				element.style.position = originalStyles.position;
+				element.style.zIndex = originalStyles.zIndex;
+				element.style.isolation = originalStyles.isolation;
+				element.style.pointerEvents = originalStyles.pointerEvents;
+				element.style.backgroundColor = originalStyles.backgroundColor;
+				element.style.boxShadow = originalStyles.boxShadow;
+
+				// Remove this element from the tracked elements
+				this.highlightedElements.delete(selector);
+			}
+		});
+	}
+
+	highlight(selectors: string[]): void {
+        selectors.forEach(selector => {
+            if (selector) {
+				const element = document.querySelector(selector) as HTMLElement
+
+				// save original styles
+				if (!this.highlightedElements.has(selector)) {
+                    this.highlightedElements.set(selector, {
+                        position: element.style.position,
+                        zIndex: element.style.zIndex,
+                        isolation: element.style.isolation,
+                        pointerEvents: element.style.pointerEvents,
+                        backgroundColor: element.style.backgroundColor,
+                        boxShadow: element.style.boxShadow,
+                        transition: element.style.transition
+                    });
+
+                    console.dir("SAVED ORIGINAL STYLES:");
+                    console.dir(this.highlightedElements.get(selector));
+                }
+
+				console.dir("in app EXCLUDE FROM DIMMING, GOT ELEMENT:");
+				console.dir(element);
+				element.style.transition = 'box-shadow 0.3s ease-in-out';
+
+                element.style.position = 'absolute';
+                element.style.zIndex = '3000';
+				element.style.isolation = 'isolate';
+				element.style.pointerEvents = 'auto';
+				element.style.backgroundColor = 'transparent';
+            	element.style.boxShadow = '0 0 0 9999px rgba(0, 0, 0, 0.6)';
+            }
+        });
+    }
+}
+
+registerSingleton(IShadowOverlayService, ShadowOverlayService, InstantiationType.Eager,);
diff --git a/src/vs/workbench/browser/workbench.ts b/src/vs/workbench/browser/workbench.ts
index 0885c15b572f1..2a316d073f1d0 100644
--- a/src/vs/workbench/browser/workbench.ts
+++ b/src/vs/workbench/browser/workbench.ts
@@ -51,6 +51,7 @@ import { setProgressAcccessibilitySignalScheduler } from 'vs/base/browser/ui/pro
 import { AccessibleViewRegistry } from 'vs/platform/accessibility/browser/accessibleViewRegistry';
 import { NotificationAccessibleView } from 'vs/workbench/browser/parts/notifications/notificationAccessibleView';
 import { IPearOverlayService } from 'vs/workbench/browser/parts/overlay/pearOverlayService';
+import { IShadowOverlayService } from 'vs/workbench/browser/parts/overlay/onboardingShadow/shadowOverlayService';
 
 export interface IWorkbenchOptions {
 
@@ -380,6 +381,11 @@ export class Workbench extends Layout {
 				});
 			}
 
+			// instantiate highlighting
+			instantiationService.invokeFunction(accessor => {
+				accessor.get(IShadowOverlayService);
+			});
+
 			mark(`code/willCreatePart/${id}`);
 			this.getPart(id).create(partContainer, options);
 			mark(`code/didCreatePart/${id}`);