Mangle exported symbols

build/lib/compilation.ts

@@ -17,7 +17,7 @@ import * as os from 'os';
 import ts = require('typescript');
 import * as File from 'vinyl';
 import * as task from './task';
-import { Mangler } from './mangleTypeScript';
+import { Mangler } from './mangle/index';
 import { RawSourceMap } from 'source-map';
 const watch = require('./watch');
 
@@ -126,19 +126,20 @@ export function compileTask(src: string, out: string, build: boolean, options: {
 		if (build && !options.disableMangle) {
 			let ts2tsMangler = new Mangler(compile.projectPath, (...data) => fancyLog(ansiColors.blue('[mangler]'), ...data));
 			const newContentsByFileName = ts2tsMangler.computeNewFileContents(new Set(['saveState']));
-			mangleStream = es.through(function write(data: File & { sourceMap?: RawSourceMap }) {
+			mangleStream = es.through(async function write(data: File & { sourceMap?: RawSourceMap }) {
 				type TypeScriptExt = typeof ts & { normalizePath(path: string): string };
 				const tsNormalPath = (<TypeScriptExt>ts).normalizePath(data.path);
-				const newContents = newContentsByFileName.get(tsNormalPath);
+				const newContents = (await newContentsByFileName).get(tsNormalPath);
 				if (newContents !== undefined) {
 					data.contents = Buffer.from(newContents.out);
 					data.sourceMap = newContents.sourceMap && JSON.parse(newContents.sourceMap);
 				}
 				this.push(data);
-			}, function end() {
-				this.push(null);
+			}, async function end() {
 				// free resources
-				newContentsByFileName.clear();
+				(await newContentsByFileName).clear();
+
+				this.push(null);
 				(<any>ts2tsMangler) = undefined;
 			});
 		}