const babel = require("@babel/core");

async function plugin(bundle, { name, data, mode }) {
  return new Promise((resolve, reject) => {
    // only transform javascript files
    if (name.match(/\.js$|\.cjs$|\.mjs$/)) {
      babel.transform(data.toString(), (err, result) => {
        if (err) reject(err);
        // overwrite the old code with the transpiled code in the bundle
        bundle.write(name, result.code, { mode });
        resolve();
      });
    } else {
      resolve();
    }
  });
}

module.exports = plugin;