import type { Program } from './nodes';

import { walk, iterChildNodes } from './visit';

export function withParents(tree: Program): Program {
  for (let parent of walk(tree)) {
    for (let child of iterChildNodes(parent)) {
      // We only set it here, so ignore
      // @ts-ignore
      child.parent = parent;
    }
  }

  return tree;
}