Flow and insertPragma

fisker · Nov 2, 2020 · f52eb3d · f52eb3d
1 parent dba72f4
commit f52eb3d
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/src/language-js/clean.js b/src/language-js/clean.js
@@ -21,10 +21,20 @@ function clean(ast, newObj, parent) {
   }
 
   if (ast.type === "Line" || ast.type === "CommentLine") {
-    newObj.value = newObj.value.trimEnd();
-  }
-  if ( ast.type === "Block" || ast.type === "CommentBlock") {
-    newObj.value = newObj.value.split("\n").map(line => line.trim()).join("\n");
+    newObj.value = ast.value.trimEnd();
+  }
+  if (ast.type === "Block" || ast.type === "CommentBlock") {
+    const comment = ast.value.trim();
+    // Flow comment
+    if (comment.startsWith("::") || comment === "* @format") {
+      newObj.value = "";
+    } else {
+      const lines = comment
+        .split("\n")
+        .map((line) => line.trim())
+        .filter((line) => line === "* @format");
+      newObj.value = lines.join("\n");
+    }
   }
 
   if (