Function plugin

tools/challenge-parser/parser/plugins/add-hooks.js:3–21

()

const { getSection } = require('./utils/get-section');

function plugin() {
  return transformer;

  function transformer(tree, file) {
    const beforeAll = getHook(tree, '--before-all--');
    const beforeEach = getHook(tree, '--before-each--');
    const afterEach = getHook(tree, '--after-each--');
    const afterAll = getHook(tree, '--after-all--');

    if (!beforeAll && !beforeEach && !afterEach && !afterAll) return;

    file.data.hooks = file.data.hooks = {
      ...(beforeAll && { beforeAll }),
      ...(beforeEach && { beforeEach }),
      ...(afterEach && { afterEach }),
      ...(afterAll && { afterAll })
    };
  }
}

function getHook(tree, sectionName) {
  const section = getSection(tree, sectionName);