Merge pull request #80 from erichalldev:master

Fix reference counter appearing in wrong spot in reading mode
TfTHacker · Feb 28, 2023 · ba88db3 · ba88db3
2 parents 83e65ac + 9bd52b8
commit ba88db3
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/view-extensions/references-preview.ts b/src/view-extensions/references-preview.ts
@@ -85,19 +85,23 @@ class snwChildComponent extends MarkdownRenderChild {
                 } catch (error) { /* nothing to do here */ }
 
                 for (const value of transformedCache.blocks) {
-                    if ( value.references[0]?.excludedFile!=true && value.references.length >= minRefCountThreshold && 
-                        (value.pos.start.line >= this.sectionInfo?.lineStart && value.pos.end.line <= this.sectionInfo?.lineEnd) &&
-                        !isThisAnEmbed ) {
+                    if (value.references[0]?.excludedFile != true
+                        && value.references.length >= minRefCountThreshold
+                        && (value.pos.start.line >= this.sectionInfo?.lineStart && value.pos.end.line <= this.sectionInfo?.lineEnd)
+                        && !isThisAnEmbed ) {
                         const referenceElement = htmlDecorationForReferencesElement(value.references.length, "block", value.key, value.references[0]?.resolvedFile?.path.replace(".md",""), "", value.pos.start.line);
-                        let blockElement: HTMLElement = this.containerEl.querySelector('p')
+                        let blockElement: HTMLElement = this.containerEl.querySelector('p');
+                        const valueLineInSection: number = value.pos.start.line - this.sectionInfo.lineStart;
                         if (!blockElement) {
-                            blockElement = this.containerEl.querySelector("li");
+                            blockElement = this.containerEl.querySelector(`li[data-line="${valueLineInSection}"]`);
+                        }
+                        if (!blockElement) {
+                            blockElement = this.containerEl.querySelector(`ol[data-line="${valueLineInSection}"]`)
                         }
                         try {
                             if (!blockElement.hasClass("snw-block-preview")) {
                                 referenceElement.addClass("snw-block-preview");
                                 blockElement.append(referenceElement);
-                                break;
                             } 
                         } catch (error) { /* nothing to do here */ }
                     }