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

rollup · Dec 13, 2023 · 1c97b0a · 1c97b0a
1 parent 62b648e
commit 1c97b0a
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 1 deletion.
diff --git a/src/ast/nodes/LabeledStatement.ts b/src/ast/nodes/LabeledStatement.ts
@@ -28,13 +28,18 @@ export default class LabeledStatement extends StatementBase {
 
 	include(context: InclusionContext, includeChildrenRecursively: IncludeChildren): void {
 		this.included = true;
-		const brokenFlow = context.brokenFlow;
+		const { brokenFlow, includedLabels } = context;
+		context.includedLabels = new Set<string>();
 		this.body.include(context, includeChildrenRecursively);
+		// eslint-disable-next-line unicorn/consistent-destructuring
 		if (includeChildrenRecursively || context.includedLabels.has(this.label.name)) {
 			this.label.include();
+			// eslint-disable-next-line unicorn/consistent-destructuring
 			context.includedLabels.delete(this.label.name);
 			context.brokenFlow = brokenFlow;
 		}
+		// eslint-disable-next-line unicorn/consistent-destructuring
+		context.includedLabels = new Set([...includedLabels, ...context.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: {