Find anchor more robustly

clear-code · Apr 30, 2024 · e1bcb03 · e1bcb03
1 parent 84b98ee
commit e1bcb03
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/background.js b/background.js
@@ -29,7 +29,9 @@ function generate() {
     }
     var toc = [];
     for (let heading of headings) {
-      let anchor = heading.querySelector('a[href]');
+      let anchor = heading.querySelector('a[href]') || heading.parentNode.querySelector('a.anchor[href]');
+      if (!anchor)
+        throw new Error(`Could not find out anchor for a heading: ${heading.textContent}`);
       let href = anchor.getAttribute('href');
       let indent = '';
       let headingLevel = parseInt(heading.localName.charAt(1));