refactor: move @babel/helper-annotate-as-pure to ts (babel#12415)

* refactor: move @babel/helper-annotate-as-pure to ts * chore: add flow types
nicolo-ribaudo · Dec 2, 2020 · 130c103 · 130c103
1 parent c13853b
commit 130c103
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/lib/babel-packages.js.flow b/lib/babel-packages.js.flow
@@ -49,3 +49,13 @@ declare module "@babel/helper-get-function-arity" {
     node: BabelNodeFunction
   ): number;
 }
+
+declare module "@babel/helper-annotate-as-pure" {
+  declare export default function annotateAsPure(
+    pathOrNode:
+      | BabelNode
+      | {
+          node: BabelNode,
+        }
+  ): void;
+}
diff --git a/...abel-helper-annotate-as-pure/src/index.js → ...abel-helper-annotate-as-pure/src/index.ts b/...abel-helper-annotate-as-pure/src/index.js → ...abel-helper-annotate-as-pure/src/index.ts
@@ -1,13 +1,16 @@
 import * as t from "@babel/types";
+import type { Node } from "@babel/types";
 
 const PURE_ANNOTATION = "#__PURE__";
 
-const isPureAnnotated = ({ leadingComments }) =>
+const isPureAnnotated = ({ leadingComments }: Node): boolean =>
   !!leadingComments &&
   leadingComments.some(comment => /[@#]__PURE__/.test(comment.value));
 
-export default function annotateAsPure(pathOrNode) {
-  const node = pathOrNode.node || pathOrNode;
+export default function annotateAsPure(
+  pathOrNode: Node | { node: Node },
+): void {
+  const node = pathOrNode["node"] || pathOrNode;
   if (isPureAnnotated(node)) {
     return;
   }