fix: Fix $FlowFixMe comments for eslint 4 (#338)

gajus · Jun 1, 2018 · 647807f · 647807f
1 parent 7c57cea
commit 647807f
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/rules/noFlowFixMeComments.js b/src/rules/noFlowFixMeComments.js
@@ -31,16 +31,24 @@ const create = (context) => {
   };
 
   return {
-    BlockComment: handleComment,
-    GenericTypeAnnotation: (node) => {
+    GenericTypeAnnotation (node) {
       if (isIdentifier(node.id, /\$FlowFixMe/)) {
         context.report({
           message,
           node: node.id
         });
       }
     },
-    LineComment: handleComment
+
+    Program () {
+      context
+        .getSourceCode()
+        .getAllComments()
+        .filter((comment) => {
+          return comment.type === 'Block' || comment.type === 'Line';
+        })
+        .forEach(handleComment);
+    }
   };
 };