import { NodePath } from '@babel/traverse';
import { Program } from '@babel/types';
import { Helper } from '../helper';
import { NodePathState } from '../types';

export default {
  enter(path: NodePath<Program>) {
    path.state = {} as NodePathState;
    path.state.helper = new Helper(path);
  },
  exit(path: NodePath<Program>) {
    const state = path.state as NodePathState;
    if (state.helper.hasSpecifier) {
      path.node.body.unshift(state.helper.helperImportDeclaration);
    }
  }
}