fix: use a new includedLabels in the body of the LabeledStatement

rollup · Dec 12, 2023 · d55788a · d55788a
1 parent 62b648e
commit d55788a
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 4 deletions.
diff --git a/src/ast/nodes/LabeledStatement.ts b/src/ast/nodes/LabeledStatement.ts
@@ -28,13 +28,17 @@ export default class LabeledStatement extends StatementBase {
 
 	include(context: InclusionContext, includeChildrenRecursively: IncludeChildren): void {
 		this.included = true;
-		const brokenFlow = context.brokenFlow;
-		this.body.include(context, includeChildrenRecursively);
-		if (includeChildrenRecursively || context.includedLabels.has(this.label.name)) {
+		const { brokenFlow, includedLabels } = context;
+		const nestContext = { ...context, includedLabels: new Set<string>() };
+		this.body.include(nestContext, includeChildrenRecursively);
+		if (includeChildrenRecursively || nestContext.includedLabels.has(this.label.name)) {
 			this.label.include();
-			context.includedLabels.delete(this.label.name);
+			nestContext.includedLabels.delete(this.label.name);
 			context.brokenFlow = brokenFlow;
+		} else {
+			context.brokenFlow = nestContext.brokenFlow;
 		}
+		context.includedLabels = new Set([...includedLabels, ...nestContext.includedLabels]);
 	}
 
 	render(code: MagicString, options: RenderOptions): void {

diff --git a/test/form/samples/break-control-flow/break-statement-labels/_expected.js b/test/form/samples/break-control-flow/break-statement-labels/_expected.js
@@ -31,6 +31,17 @@ outer: {
 	console.log('retained');
 }
 
+outer: {
+	switch (globalThis.unknown) {
+		case 1:
+			(() => {
+				console.log('retained');
+			})();
+		case 2:
+			break outer;
+	}
+}
+
 function withConsequentReturn() {
 	{
 		inner: {

diff --git a/test/form/samples/break-control-flow/break-statement-labels/main.js b/test/form/samples/break-control-flow/break-statement-labels/main.js
@@ -48,6 +48,17 @@ outer: {
 	console.log('retained');
 }
 
+outer: {
+	switch (globalThis.unknown) {
+		case 1:
+			(() => {
+				outer: console.log('retained');
+			})();
+		case 2:
+			break outer;
+	}
+}
+
 function withConsequentReturn() {
 	outer: {
 		inner: {