19 lines
510 B
TypeScript
19 lines
510 B
TypeScript
import * as t from "@babel/types";
|
|
import type { Node } from "@babel/types";
|
|
|
|
const PURE_ANNOTATION = "#__PURE__";
|
|
|
|
const isPureAnnotated = ({ leadingComments }: Node): boolean =>
|
|
!!leadingComments &&
|
|
leadingComments.some(comment => /[@#]__PURE__/.test(comment.value));
|
|
|
|
export default function annotateAsPure(
|
|
pathOrNode: Node | { node: Node },
|
|
): void {
|
|
const node = pathOrNode["node"] || pathOrNode;
|
|
if (isPureAnnotated(node)) {
|
|
return;
|
|
}
|
|
t.addComment(node, "leading", PURE_ANNOTATION);
|
|
}
|