Rollup plugin to enable banner/footer when using vite

rollup-banner-plugin.js

/* Include the following code inside your viteconfig then build.rollupOptions.output[banner|footer] will work as intended */
const RollupBannerPlugin = {
  name: 'banner',
  enforce: 'post',
  generateBundle(options, bundle) {
    const banner = options.banner() || ''
    const footer = options.footer() || ''
    for (const module of Object.values(bundle)) {
      if (module.type === 'chunk') {
        const shebang = module.code.match(/^#![^\n]*\n/)
        if (shebang) {
          module.code = shebang[0] + banner + module.code.slice(shebang[0].length) + footer
        } else {
          module.code = banner + module.code + footer
        }
        // should change source map if needed
      }
    }
  }
}