Refactor function getAptContainer

david-tejada · Jan 9, 2024 · 4237df5 · 4237df5
1 parent 3bde0e0
commit 4237df5
Showing 1 changed file with 16 additions and 21 deletions.
diff --git a/src/content/hints/getContextForHint.ts b/src/content/hints/getContextForHint.ts
@@ -171,28 +171,23 @@ export function getAptContainer(origin: Element) {
 
 		if (!(current instanceof HTMLElement) || current.shadowRoot) {
 			current = current.parentNode;
-			continue;
-		}
-
-		const { display } = getCachedStyle(current);
-
-		if (
-			current.matches("thead, tbody, tfoot, caption, colgroup, col, tr, th, td")
-		) {
-			current = current.closest("table") ?? current.parentElement;
-			continue;
-		}
-
-		if (current.tagName === "TABLE" || display.startsWith("table")) {
-			current = current.parentElement;
-			continue;
-		}
-
-		if (current.tagName !== "DETAILS" && display !== "contents") {
-			return current;
+		} else {
+			const { display } = getCachedStyle(current);
+
+			if (
+				current.matches(
+					"thead, tbody, tfoot, caption, colgroup, col, tr, th, td"
+				)
+			) {
+				current = current.closest("table") ?? current.parentElement;
+			} else if (current.tagName === "TABLE" || display.startsWith("table")) {
+				current = current.parentElement;
+			} else if (current.tagName !== "DETAILS" && display !== "contents") {
+				return current;
+			} else {
+				current = current.parentElement;
+			}
 		}
-
-		current = current.parentElement;
 	}
 
 	return document.body;