fix(plugins/notes): preserve line mapping

diplodoc-platform · May 22, 2023 · 29e4c5c · 29e4c5c
1 parent 8d7b28f
commit 29e4c5c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/transform/plugins/notes.ts b/src/transform/plugins/notes.ts
@@ -79,7 +79,10 @@ const notes: MarkdownItPluginCb = (md, {lang, path: optPath, log}) => {
                 const newOpenToken = new state.Token('yfm_note_open', 'div', 1);
                 newOpenToken.attrSet('class', `yfm-note yfm-accent-${type}`);
                 newOpenToken.attrSet('note-type', type);
+                newOpenToken.map = tokens[i].map;
+
                 const newCloseToken = new state.Token('yfm_note_close', 'div', -1);
+                newCloseToken.map = tokens[closeTokenIdx].map;
 
                 // Add extra paragraph
                 const titleOpen = new state.Token('yfm_note_title_open', 'p', 1);