import type { Node } from 'ohm-js';

import * as ast from '../ast/nodes';

export function withLocations<T extends ast.ASTRootBase>(astNode: T, intermediateNode: Node): T {
  // It exists, but is not listed in TS definitions.
  // @ts-ignore
  const lineAndColumn = intermediateNode.source.getLineAndColumn();

  // This is the only place where we set it, ignore the type error
  // @ts-ignore
  astNode.locations = {
    line: lineAndColumn.lineNum,
    column: lineAndColumn.colNum,
  };
  return astNode;
}