Update lineMap after altering source.text (#2837)

Fixes #2680.
aws · May 27, 2021 · 137bb19 · 137bb19
1 parent c83a39a
commit 137bb19
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/packages/jsii/lib/transforms/ts-comment-replacer.ts b/packages/jsii/lib/transforms/ts-comment-replacer.ts
@@ -126,5 +126,8 @@ function whiteoutLeadingComments(node: ts.Node, source: ts.SourceFile) {
       text = text.slice(0, pos).padEnd(end, ' ') + text.slice(end);
     }
   });
-  source.text = text;
+  if (source.text !== text) {
+    source.text = text;
+    (source as any).lineMap = (ts as any).computeLineStarts(text);
+  }
 }