Deoptimize awaited expressions

src/ast/nodes/AwaitExpression.ts

@@ -1,6 +1,5 @@
-import { NormalizedTreeshakingOptions } from '../../rollup/types';
-import { NO_ARGS } from '../CallOptions';
-import { HasEffectsContext, InclusionContext } from '../ExecutionContext';
+import { InclusionContext } from '../ExecutionContext';
+import { UNKNOWN_PATH } from '../utils/PathTracker';
 import ArrowFunctionExpression from './ArrowFunctionExpression';
 import * as NodeType from './NodeType';
 import FunctionNode from './shared/FunctionNode';
@@ -9,25 +8,15 @@ import { ExpressionNode, IncludeChildren, Node, NodeBase } from './shared/Node';
 export default class AwaitExpression extends NodeBase {
 	argument!: ExpressionNode;
 	type!: NodeType.tAwaitExpression;
+	protected deoptimized = false;
 
-	hasEffects(context: HasEffectsContext): boolean {
-		const { propertyReadSideEffects } = this.context.options
-			.treeshake as NormalizedTreeshakingOptions;
-		return (
-			!context.ignore.returnAwaitYield ||
-			this.argument.hasEffects(context) ||
-			this.argument.hasEffectsWhenCalledAtPath(
-				['then'],
-				{ args: NO_ARGS, thisParam: null, withNew: false },
-				context
-			) ||
-			(propertyReadSideEffects &&
-				(propertyReadSideEffects === 'always' ||
-					this.argument.hasEffectsWhenAccessedAtPath(['then'], context)))
-		);
+	hasEffects(): boolean {
+		if (!this.deoptimized) this.applyDeoptimizations();
+		return true;
 	}
 
 	include(context: InclusionContext, includeChildrenRecursively: IncludeChildren): void {
+		if (!this.deoptimized) this.applyDeoptimizations();
 		if (!this.included) {
 			this.included = true;
 			checkTopLevelAwait: if (!this.context.usesTopLevelAwait) {
@@ -41,4 +30,9 @@ export default class AwaitExpression extends NodeBase {
 		}
 		this.argument.include(context, includeChildrenRecursively);
 	}
+
+	protected applyDeoptimizations(): void {
+		this.deoptimized = true;
+		this.argument.deoptimizePath(UNKNOWN_PATH);
+	}
 }

test/form/samples/async-function-effects/main.js

@@ -77,14 +77,6 @@
 	return { then() {} };
 })();
 
-// removed
-(async function () {
-	await {
-		then: function () {}
-	};
-	return { then() {} };
-})();
-
 (async function () {
 	await {
 		get then() {
@@ -104,16 +96,6 @@
 	return { then() {} };
 })();
 
-// removed
-(async function () {
-	await {
-		get then() {
-			return () => {};
-		}
-	};
-	return { then() {} };
-})();
-
 (async function () {
 	await await {
 		then(resolve) {

test/form/samples/ignore-property-access-side-effects/main.js

@@ -38,13 +38,3 @@ const foo7 = (async () => ({
 		return () => {};
 	}
 }))();
-
-const foo8 = (async function () {
-	await {
-		get then() {
-			console.log('effect');
-			return () => {};
-		}
-	};
-	return { then() {} };
-})();

test/function/samples/track-awaited-object/_config.js

@@ -0,0 +1,10 @@
+const assert = require('assert');
+
+module.exports = {
+	description: 'tracks object mutations through await',
+	async exports(exports) {
+		assert.strictEqual(exports.toggled, false);
+		await exports.test();
+		assert.strictEqual(exports.toggled, true);
+	}
+};

test/function/samples/track-awaited-object/main.js

@@ -0,0 +1,19 @@
+function fn() {
+	const obj = {
+		test() {
+			if (typeof obj.testfn === 'function') {
+				obj.testfn();
+			}
+		}
+	};
+
+	return obj;
+}
+
+export let toggled = false;
+
+export const test = async function () {
+	const obj = await fn();
+	obj.testfn = () => (toggled = true);
+	obj.test();
+};