import { NodePath, Visitor } from 'babel-traverse';
import * as t from 'babel-types';
import { letExpression } from '@stopify/util';

const functionDecl: Visitor = {
  FunctionDeclaration: function (path: NodePath<t.FunctionDeclaration>) {
    const { id, params, body } = path.node;
    path.replaceWith(letExpression(id,
      t.functionExpression(undefined, params, body), 'var'));
  }
};

module.exports = function() {
  return { visitor: functionDecl };
};