fix: WrapContent plugin

Author: chengyu.chengyulia

src/utils/webpack-plugin-wrap-content.ts

@@ -1,5 +1,6 @@
 import * as webpack from 'webpack';
 import * as ConcatSource from 'webpack-sources';
+import * as path from 'path';
 import * as _ from 'lodash';
 import { pri } from '../node';
 
@@ -14,16 +15,20 @@ export class WrapContent {
   }
 
   public apply(compiler: webpack.Compiler) {
-    compiler.plugin('compilation', compilation => {
-      compilation.plugin('optimize-chunk-assets', (chunks: any[], done: any) => {
+    compiler.hooks.compilation.tap('WrapContent', compilation => {
+      compilation.hooks.optimizeChunkAssets.tapAsync('WrapContent', (chunks, done) => {
         chunks.forEach(chunk => {
-          chunk.files.forEach((fileName: string) => {
+          chunk.files.forEach(fileName => {
             // Ignore workers
             if (fileName.indexOf('worker.js') > -1) {
               return;
             }
 
-            if (chunk.name === 'main' || _.has(pri.sourceConfig.entries, chunk.name)) {
+            if (
+              chunk.name === 'main' ||
+              chunk.name === path.basename(pri.sourceConfig.outFileName, '.js') ||
+              _.has(pri.sourceConfig.entries, chunk.name)
+            ) {
               compilation.assets[fileName] = new ConcatSource.ConcatSource(
                 this.header,
                 compilation.assets[fileName],