import { Compiler, Plugin } from 'webpack'
import { extname } from 'path'

// Prevents outputting client pages when they are not needed
export class DropClientPage implements Plugin {
  ampPages = new Set();

  apply(compiler: Compiler) {
    compiler.hooks.emit.tap('DropClientPage', compilation => {
      Object.keys(compilation.assets).forEach(assetKey => {
        const asset = compilation.assets[assetKey]

        if (asset && asset._value && asset._value.includes('__NEXT_DROP_CLIENT_FILE__')) {
          const page = '/' + assetKey.split('pages/')[1]
          const pageNoExt = page.split(extname(page))[0]

          this.ampPages.add(pageNoExt.replace(/\/index$/, '') || '/')
          delete compilation.assets[assetKey]
        }
      })
    })
  }
}