fix(*): fix program update issue

Author: Stanislav Panferov

src/index.ts

@@ -90,13 +90,15 @@ async function compiler(webpack: IWebPack, text: string): Promise<void> {
             state.fileAnalyzer.dependencies.addCompiledModule(fileName, compiledModule.fileName);
             transformation = {
                 text: compiledModule.text,
-                map: JSON.parse(compiledModule.map)
+                map: compiledModule.map
+                    ? JSON.parse(compiledModule.map)
+                    : null
             }
         } else {
 
             function transform() {
                 let resultText;
-                let resultSourceMap;
+                let resultSourceMap = null;
                 let output = state.emit(fileName);
 
                 let result = helpers.findResultFor(output, fileName);
@@ -106,14 +108,20 @@ async function compiler(webpack: IWebPack, text: string): Promise<void> {
                 }
 
                 resultText = result.text;
-                resultSourceMap = JSON.parse(result.sourceMap);
-                resultSourceMap.sources = [ fileName ];
-                resultSourceMap.file = fileName;
-                resultSourceMap.sourcesContent = [ text ];
+
+                if (result.sourceMap) {
+                    resultSourceMap = JSON.parse(result.sourceMap);
+                    resultSourceMap.sources = [ loaderUtils.getRemainingRequest(webpack) ];
+                    resultSourceMap.file = fileName;
+                    resultSourceMap.sourcesContent = [ text ];
+
+                    resultText = resultText.replace(/^\/\/# sourceMappingURL=[^\r\n]*/gm, '');
+                }
 
                 if (instance.options.useBabel) {
                     let defaultOptions = {
                         inputSourceMap: resultSourceMap,
+                        sourceRoot: process.cwd(),
                         filename: fileName,
                         sourceMap: true
                     }

src/instance.ts

@@ -160,8 +160,9 @@ export function ensureInstance(webpack: IWebPack, options: ICompilerOptions, ins
 
     let babelImpl: any;
     if (options.useBabel) {
+        console.error('babel')
         try {
-            babelImpl = require(path.join(process.cwd(), 'node_modules', 'babel'));
+            babelImpl = require(path.join(process.cwd(), 'node_modules', 'babel-core'));
         } catch (e) {
             console.error(BABEL_ERROR);
             process.exit(1);
@@ -262,6 +263,12 @@ function setupAfterCompile(compiler, instanceName, forkChecker = false) {
                 payload
             });
         } else {
+            if (!state.program) {
+                // program may be undefined here, if all files
+                // will be loaded by tsconfig
+                state.updateProgram();
+            }
+
             let diagnostics = state.ts.getPreEmitDiagnostics(state.program);
             let emitError = (err) => {
                 if (compilation.bail) {