module.exports = function (source: string) {
  let registerNotifyFalse = false
  const options = this.getOptions()

  ;[
    ...source.matchAll(
      /roots\.register\.(plugin|block|format|filter|variation|style)s\(['"`](.*)['"`]\)/g,
    ),
  ].map(([match, type, query]) => {
    source = source.replace(match, makeCodeString(type, query))
  })
  ;[...source.matchAll(/roots\.register\.notify\(false\)/g)]?.map(
    ([match]) => {
      if (!match) return
      registerNotifyFalse = true
      source = source.replace(match, makeNoNotifyString())
    },
  )

  if (!registerNotifyFalse && options?.notify === false) {
    source = [source, makeNoNotifyString()].join(`\n`)
  }

  return source
}

const makeCodeString = (type: string, query: string, enabled = true) => `
import * as __roots_${type}s from '@roots/wordpress-hmr/${type}s';

__roots_${type}s.register(
  () => import.meta.webpackContext(
    '${query}',
    {recursive: true, regExp: /.*\\.${type}\\..*$/}
  ),
  (id, context) => {
    if (import.meta.webpackHot) return import.meta.webpackHot.accept(id, context)
  },
);`

const makeNoNotifyString = () => `
import * as __roots_editor from '@roots/wordpress-hmr/editor';
__roots_editor.setNotify(false);
`