consider that literals cannot following switch case

TrickyPi · TrickyPi · commit da694cc69c4c · 2025-01-06T22:04:50.000+08:00
diff --git a/src/ast/nodes/SwitchCase.ts b/src/ast/nodes/SwitchCase.ts
@@ -39,17 +39,18 @@ export default class SwitchCase extends NodeBase {
 	}
 
 	render(code: MagicString, options: RenderOptions, nodeRenderOptions?: NodeRenderOptions): void {
-		if (this.consequent.length > 0) {
-			if (this.test) {
-				this.test.render(code, options);
+		if (this.test) {
+			this.test.render(code, options);
+			if (this.test.start === this.start + 4) {
+				code.prependLeft(this.test.start, ' ');
 			}
+		}
+		if (this.consequent.length > 0) {
 			const testEnd = this.test
 				? this.test.end
 				: findFirstOccurrenceOutsideComment(code.original, 'default', this.start) + 7;
 			const consequentStart = findFirstOccurrenceOutsideComment(code.original, ':', testEnd) + 1;
 			renderStatementList(this.consequent, code, consequentStart, nodeRenderOptions!.end!, options);
-		} else {
-			super.render(code, options);
 		}
 	}
 }
diff --git a/src/ast/nodes/UnaryExpression.ts b/src/ast/nodes/UnaryExpression.ts
@@ -90,7 +90,7 @@ export default class UnaryExpression extends NodeBase {
 		if (this.renderedLiteralValue !== UNASSIGNED) return this.renderedLiteralValue;
 		return (this.renderedLiteralValue = includeChildrenRecursively
 			? UnknownValue
-			: getSimplifiedLiterals(
+			: getRenderedLiteralValue(
 					this.getLiteralValueAtPath(EMPTY_PATH, SHARED_RECURSION_TRACKER, this)
 				));
 	}
@@ -120,7 +120,7 @@ export default class UnaryExpression extends NodeBase {
 	}
 }
 
-function getSimplifiedLiterals(value: unknown) {
+function getRenderedLiteralValue(value: unknown) {
 	if (value === undefined || typeof value === 'boolean') {
 		return String(value);
 	}
diff --git a/test/form/samples/treeshake-unary/_expected.js b/test/form/samples/treeshake-unary/_expected.js
@@ -27,3 +27,21 @@ function foo3() {
 }
 
 console.log(delete foo3.a);
+
+switch (bar) {
+	case false:
+		console.log('false');
+		break;
+	case true:
+		console.log('true');
+		break;
+	case 1:
+		console.log('1');
+		break;
+	case 1:
+		console.log('1');
+		break;
+	case 1:
+		console.log('1');
+		break;
+}
diff --git a/test/form/samples/treeshake-unary/main.js b/test/form/samples/treeshake-unary/main.js
@@ -32,3 +32,21 @@ function foo3() {
 }
 
 console.log(delete foo3.a);
+
+switch (bar) {
+	case!1:
+		console.log('false');
+		break;
+	case!0:
+		console.log('true');
+		break;
+	case+1:
+		console.log('1');
+		break;
+	case~-2:
+		console.log('1');
+		break;
+	case ~-2:
+		console.log('1');
+		break;
+}

Original file line number	Diff line number	Diff line change
`@@ -39,17 +39,18 @@ export default class SwitchCase extends NodeBase {`
`39`	`39`	`}`
`40`	`40`
`41`	`41`	`render(code: MagicString, options: RenderOptions, nodeRenderOptions?: NodeRenderOptions): void {`
`42`		`- if (this.consequent.length > 0) {`
`43`		`- if (this.test) {`
`44`		`- this.test.render(code, options);`
	`42`	`+ if (this.test) {`
	`43`	`+ this.test.render(code, options);`
	`44`	`+ if (this.test.start === this.start + 4) {`
	`45`	`+ code.prependLeft(this.test.start, ' ');`
`45`	`46`	`}`
	`47`	`+ }`
	`48`	`+ if (this.consequent.length > 0) {`
`46`	`49`	`const testEnd = this.test`
`47`	`50`	`? this.test.end`
`48`	`51`	`: findFirstOccurrenceOutsideComment(code.original, 'default', this.start) + 7;`
`49`	`52`	`const consequentStart = findFirstOccurrenceOutsideComment(code.original, ':', testEnd) + 1;`
`50`	`53`	`renderStatementList(this.consequent, code, consequentStart, nodeRenderOptions!.end!, options);`
`51`		`- } else {`
`52`		`- super.render(code, options);`
`53`	`54`	`}`
`54`	`55`	`}`
`55`	`56`	`}`
Original file line number	Diff line number	Diff line change
`@@ -90,7 +90,7 @@ export default class UnaryExpression extends NodeBase {`
`90`	`90`	`if (this.renderedLiteralValue !== UNASSIGNED) return this.renderedLiteralValue;`
`91`	`91`	`return (this.renderedLiteralValue = includeChildrenRecursively`
`92`	`92`	`? UnknownValue`
`93`		`- : getSimplifiedLiterals(`
	`93`	`+ : getRenderedLiteralValue(`
`94`	`94`	`this.getLiteralValueAtPath(EMPTY_PATH, SHARED_RECURSION_TRACKER, this)`
`95`	`95`	`));`
`96`	`96`	`}`
`@@ -120,7 +120,7 @@ export default class UnaryExpression extends NodeBase {`
`120`	`120`	`}`
`121`	`121`	`}`
`122`	`122`
`123`		`-function getSimplifiedLiterals(value: unknown) {`
	`123`	`+function getRenderedLiteralValue(value: unknown) {`
`124`	`124`	`if (value === undefined \|\| typeof value === 'boolean') {`
`125`	`125`	`return String(value);`
`126`	`126`	`}`