packages/babel-traverse/src/cache.ts - main - githubmirror/babel

packages/babel-traverse/src/cache.ts

39 строк894 B

liuxingbaoyu

Enable `strictFunctionTypes` (#17946)

22 май 2026, 17:12

22 май 2026, 17:12c6d71f3

import type { Node } from "@babel/types";
import type NodePath from "./path/index.ts";
import type Scope from "./scope/index.ts";

let pathsCache = new WeakMap<Node, Map<Node, NodePath>>();
export { pathsCache as path };
export let scope = new WeakMap<Node, Scope>();

export function clear() {
  clearPath();
  clearScope();
}

export function clearPath() {
  pathsCache = new WeakMap();
}

export function clearScope() {
  scope = new WeakMap();
}

export function getCachedPaths(path: NodePath<Node | null>) {
  const { parent, parentPath } = path;
  return parentPath ? parentPath._store : pathsCache.get(parent);
}

export function getOrCreateCachedPaths(
  node: Node,
  parentPath?: NodePath | null,
) {
  if (parentPath) {
    return (parentPath._store ||= new Map());
  }

  let paths = pathsCache.get(node);
  if (!paths) pathsCache.set(node, (paths = new Map()));

  return paths;
}