fix: moving a comment to the top node (#49)

* fix: moving a comment to the top node * fix: added moveSyntheticComments, before if statement
rbxts-flamework · Oct 18, 2024 · 4e7d33c · 4e7d33c
1 parent 7845311
commit 4e7d33c
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/transformations/transformFile.ts b/src/transformations/transformFile.ts
@@ -24,8 +24,14 @@ export function transformFile(state: TransformState, file: ts.SourceFile): ts.So
 
 		// steal comments from original first statement so that comment directives work properly
 		if (firstStatement && statements[0]) {
-			ts.copyComments(firstStatement, statements[0]);
-			ts.removeAllComments(firstStatement);
+			const original = ts.getParseTreeNode(firstStatement);
+
+			ts.moveSyntheticComments(statements[0], firstStatement);
+
+			if (original) {
+				ts.copyComments(original, statements[0]);
+				ts.removeAllComments(original);
+			}
 		}
 	}