fix(plugins/table): line map for table, table cell close tokens

add line mapping to yfm_td_close and yfm_table_close tokens
diplodoc-platform · Jul 12, 2023 · 9600b7f · 9600b7f
1 parent 69d9ebe
commit 9600b7f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/transform/plugins/table/index.ts b/src/transform/plugins/table/index.ts
@@ -260,6 +260,7 @@ const yfmTable: MarkdownItPluginCb = (md) => {
                     state.eMarks[end.line] = oldEMark;
 
                     token = state.push('yfm_td_close', 'td', -1);
+                    state.tokens[state.tokens.length - 1].map = [end.line, end.line + 1];
                 }
 
                 if (rowLength < maxRowLength) {
@@ -276,6 +277,7 @@ const yfmTable: MarkdownItPluginCb = (md) => {
             token = state.push('yfm_tbody_close', 'tbody', -1);
 
             token = state.push('yfm_table_close', 'table', -1);
+            state.tokens[state.tokens.length - 1].map = [endOfTable, endOfTable + 1];
 
             state.lineMax = oldParentLineMax;
             state.line = endOfTable;