From 6beec8d985ad17a62bf176d382679aefa5589136 Mon Sep 17 00:00:00 2001
From: Gaurav Bhagchandani <gbolosta@gmail.com>
Date: Tue, 8 Nov 2022 04:25:17 -0800
Subject: [PATCH] Added multiple element preview on hover of scss selector

---
 src/services/selectorPrinting.ts      | 48 +++++++++++++++++++--------
 src/test/css/selectorPrinting.test.ts |  2 +-
 2 files changed, 36 insertions(+), 14 deletions(-)

diff --git a/src/services/selectorPrinting.ts b/src/services/selectorPrinting.ts
index edc1654b..04d485f7 100644
--- a/src/services/selectorPrinting.ts
+++ b/src/services/selectorPrinting.ts
@@ -332,14 +332,23 @@ export class SelectorPrinting {
 	}
 
 	public selectorToMarkedString(node: nodes.Selector): MarkedString[] {
-		const root = selectorToElement(node);
-		if (root) {
-			const markedStrings = new MarkedStringPrinter('"').print(root);
-			markedStrings.push(this.selectorToSpecificityMarkedString(node));
-			return markedStrings;
-		} else {
-			return [];
+		let hasChild: boolean = true;
+		let markedStrings: Array<Array<MarkedString>> = Array();
+		let childNumber: number = 0;
+		let childCount: number = 0;
+		while(hasChild){
+			const elementObject = selectorToElement(node, childNumber);
+			let root = elementObject.root;
+			hasChild = elementObject.hasChild;
+			childNumber += 1;
+			if ((hasChild || !childCount) && root) {
+				childCount += 1;
+				const markedString = new MarkedStringPrinter('"').print(root);
+				markedString.push(this.selectorToSpecificityMarkedString(node));
+				markedStrings.push(markedString);
+			}
 		}
+		return markedStrings.flat(1);
 	}
 
 	public simpleSelectorToMarkedString(node: nodes.SimpleSelector): MarkedString[] {
@@ -452,7 +461,7 @@ export class SelectorPrinting {
 			return specificity;
 		};
 
-		const specificity = calculateScore(node);;
+		const specificity = calculateScore(node);
 		return localize('specificity', "[Selector Specificity](https://developer.mozilla.org/en-US/docs/Web/CSS/Specificity): ({0}, {1}, {2})", specificity.id, specificity.attr, specificity.tag);
 	}
 
@@ -525,10 +534,17 @@ function isNewSelectorContext(node: nodes.Node): boolean {
 	}
 	return false;
 }
+interface ElementObject {
+    root: Element | null,
+	hasChild: boolean  
+}
 
-export function selectorToElement(node: nodes.Selector): Element | null {
+export function selectorToElement(node: nodes.Selector, child: number = 0): ElementObject {
 	if (node.matches('@at-root')) {
-		return null;
+		return {
+			root: null,
+			hasChild: false
+		};
 	}
 	const root: Element = new RootElement();
 	const parentRuleSets: nodes.RuleSet[] = [];
@@ -548,14 +564,20 @@ export function selectorToElement(node: nodes.Selector): Element | null {
 	}
 
 	const builder = new SelectorElementBuilder(root);
-
+	let hasChild: boolean = false;
 	for (let i = parentRuleSets.length - 1; i >= 0; i--) {
-		const selector = <nodes.Selector>parentRuleSets[i].getSelectors().getChild(0);
+		const selector = <nodes.Selector>parentRuleSets[i].getSelectors().getChild(i === 0 ? child : 0);
 		if (selector) {
+			hasChild = true;
 			builder.processSelector(selector);
+		} else {
+			hasChild = false;
 		}
 	}
 
 	builder.processSelector(node);
-	return root;
+	return {
+		root,
+		hasChild
+	};
 }
diff --git a/src/test/css/selectorPrinting.test.ts b/src/test/css/selectorPrinting.test.ts
index b0c59523..e4d2d128 100644
--- a/src/test/css/selectorPrinting.test.ts
+++ b/src/test/css/selectorPrinting.test.ts
@@ -61,7 +61,7 @@ export function assertSelector(p: Parser, input: string, selectorName: string, e
 	let element = selectorPrinting.selectorToElement(selector!);
 	assert(element);
 
-	assert.equal(elementToString(element!), expected);
+	assert.equal(elementToString(element.root!), expected);
 }
 
 function assertElement(p: Parser, input: string, expected: { name: string; value?: string }[]): void {