Function getAstByKey

packages/compiler/src/utils/ast-key.ts:22–57

(ast: t.File, key: string)

}

export function getAstByKey(ast: t.File, key: string) {
  const programPath = _getProgramNodePath(ast);
  if (!programPath) {
    return null;
  }

  const keyParts = key.split("/").reverse();

  let result: NodePath = programPath;

  while (true) {
    let currentKeyPart = keyParts.pop();
    if (!currentKeyPart) {
      break;
    }
    const isIntegerPart = Number.isInteger(Number(currentKeyPart));
    if (isIntegerPart) {
      const maybeBodyItemsArray = result.get("body");
      const bodyItemsArray = Array.isArray(maybeBodyItemsArray)
        ? maybeBodyItemsArray
        : [maybeBodyItemsArray];
      const index = Number(currentKeyPart);
      const subResult = bodyItemsArray[index];
      result = subResult as NodePath;
    } else {
      const maybeSubResultArray = result.get(currentKeyPart);
      const subResultArray = Array.isArray(maybeSubResultArray)
        ? maybeSubResultArray
        : [maybeSubResultArray];
      const subResult = subResultArray[0];
      result = subResult;
    }
  }

  return result;
}

function _getProgramNodePath(ast: t.File): NodePath<t.Program> | null {
  let result: NodePath<t.Program> | null = null;