feat: format switch tag

Author: ldez

shorten/format.go

@@ -75,6 +75,8 @@ func (s *Shortener) formatDecl(decl dst.Decl) {
 // formatStmt formats an AST statement node.
 // Among other examples, these include assignments, case clauses,
 // for statements, if statements, and select statements.
+//
+//nolint:funlen // the number of statements is expected.
 func (s *Shortener) formatStmt(stmt dst.Stmt, force bool) {
 	stmtType := reflect.TypeOf(stmt)
 
@@ -145,6 +147,12 @@ func (s *Shortener) formatStmt(stmt dst.Stmt, force bool) {
 	case *dst.SwitchStmt:
 		s.formatStmt(st.Body, false)
 
+		// Ignored: st.Init
+
+		if st.Tag != nil {
+			s.formatExpr(st.Tag, shouldShorten, false)
+		}
+
 	case *dst.TypeSwitchStmt:
 		s.formatStmt(st.Body, false)
 

shorten/testdata/switch_blocks/switch_blocks.go

@@ -2,7 +2,7 @@ package fixtures
 
 import "fmt"
 
-func testFunc2() {
+func _(a int) error {
 	c := make(chan int)
 
 	for {
@@ -17,4 +17,30 @@ func testFunc2() {
 
 		break
 	}
+
+	return nil
+}
+
+func _() {
+	var pneumonoultramicroscopicsilicovolcanoconiosis string
+	var floccinaucinihilipilification string
+
+	switch myfunction(pneumonoultramicroscopicsilicovolcanoconiosis, floccinaucinihilipilification) {
+	case "a":
+		fmt.Println("a")
+	}
+}
+
+func _() {
+	var pneumonoultramicroscopicsilicovolcanoconiosis string
+	var floccinaucinihilipilification string
+
+	switch a := "taumatawhakatangihangakoauauotamateaturipukakapikimaungahoronukupokaiwhenuakitanatahu"; myfunction(pneumonoultramicroscopicsilicovolcanoconiosis, floccinaucinihilipilification) {
+	case "a":
+		fmt.Println(a)
+	}
+}
+
+func myfunction(a, b string) string {
+	return ""
 }

shorten/testdata/switch_blocks/switch_blocks.go.golden

@@ -2,7 +2,7 @@ package fixtures
 
 import "fmt"
 
-func testFunc2() {
+func _(a int) error {
 	c := make(chan int)
 
 	for {
@@ -25,4 +25,36 @@ func testFunc2() {
 
 		break
 	}
+
+	return nil
+}
+
+func _() {
+	var pneumonoultramicroscopicsilicovolcanoconiosis string
+	var floccinaucinihilipilification string
+
+	switch myfunction(
+		pneumonoultramicroscopicsilicovolcanoconiosis,
+		floccinaucinihilipilification,
+	) {
+	case "a":
+		fmt.Println("a")
+	}
+}
+
+func _() {
+	var pneumonoultramicroscopicsilicovolcanoconiosis string
+	var floccinaucinihilipilification string
+
+	switch a := "taumatawhakatangihangakoauauotamateaturipukakapikimaungahoronukupokaiwhenuakitanatahu"; myfunction(
+		pneumonoultramicroscopicsilicovolcanoconiosis,
+		floccinaucinihilipilification,
+	) {
+	case "a":
+		fmt.Println(a)
+	}
+}
+
+func myfunction(a, b string) string {
+	return ""
 }