fix: format switch type statement

Author: ldez

shorten/format.go

@@ -145,10 +145,24 @@ func (s *Shortener) formatStmt(stmt dst.Stmt, force bool) {
 	case *dst.SwitchStmt:
 		s.formatStmt(st.Body, false)
 
+	case *dst.TypeSwitchStmt:
+		s.formatStmt(st.Body, false)
+
+	case *dst.BadStmt, *dst.EmptyStmt, *dst.LabeledStmt,
+		*dst.SendStmt, *dst.IncDecStmt, *dst.BranchStmt:
+		// These statements are explicitly defined to improve switch cases exhaustiveness.
+		// They may be handled in the future.
+		if shouldShorten {
+			s.logger.Debug(
+				"got a statement type that is not shortened",
+				slog.Any("stmt_type", stmtType),
+			)
+		}
+
 	default:
 		if shouldShorten {
 			s.logger.Debug(
-				"got a statement type that can't be shortened",
+				"got an unknown statement type",
 				slog.Any("stmt_type", stmtType),
 			)
 		}

shorten/testdata/switch_type/switch_type.go

@@ -0,0 +1,25 @@
+package testdata
+
+import (
+	"fmt"
+
+	"github.com/dave/dst"
+)
+
+func _(node dst.Node) error {
+	switch n := node.(type) {
+	case dst.Decl:
+		n.Decorations()
+
+		return fmt.Errorf("This is a really long line that can be broken up twice %s %s", fmt.Sprintf("This is a really long sub-line that should be broken up more because %s %s", "xxxx", "yyyy"), fmt.Sprintf("A short one %d", 3))
+
+	case dst.Expr:
+		return fmt.Errorf("This is a really long line that can be broken up twice %s %s", fmt.Sprintf("This is a really long sub-line that should be broken up more because %s %s", "xxxx", "yyyy"), fmt.Sprintf("A short one %d", 3))
+
+	case dst.Stmt:
+		return fmt.Errorf("This is a really long line that can be broken up twice %s %s", fmt.Sprintf("This is a really long sub-line that should be broken up more because %s %s", "xxxx", "yyyy"), fmt.Sprintf("A short one %d", 3))
+
+	default:
+		return fmt.Errorf("This is a really long line that can be broken up twice %s %s", fmt.Sprintf("This is a really long sub-line that should be broken up more because %s %s", "xxxx", "yyyy"), fmt.Sprintf("A short one %d", 3))
+	}
+}

shorten/testdata/switch_type/switch_type.go.golden

@@ -0,0 +1,57 @@
+package testdata
+
+import (
+	"fmt"
+
+	"github.com/dave/dst"
+)
+
+func _(node dst.Node) error {
+	switch n := node.(type) {
+	case dst.Decl:
+		n.Decorations()
+
+		return fmt.Errorf(
+			"This is a really long line that can be broken up twice %s %s",
+			fmt.Sprintf(
+				"This is a really long sub-line that should be broken up more because %s %s",
+				"xxxx",
+				"yyyy",
+			),
+			fmt.Sprintf("A short one %d", 3),
+		)
+
+	case dst.Expr:
+		return fmt.Errorf(
+			"This is a really long line that can be broken up twice %s %s",
+			fmt.Sprintf(
+				"This is a really long sub-line that should be broken up more because %s %s",
+				"xxxx",
+				"yyyy",
+			),
+			fmt.Sprintf("A short one %d", 3),
+		)
+
+	case dst.Stmt:
+		return fmt.Errorf(
+			"This is a really long line that can be broken up twice %s %s",
+			fmt.Sprintf(
+				"This is a really long sub-line that should be broken up more because %s %s",
+				"xxxx",
+				"yyyy",
+			),
+			fmt.Sprintf("A short one %d", 3),
+		)
+
+	default:
+		return fmt.Errorf(
+			"This is a really long line that can be broken up twice %s %s",
+			fmt.Sprintf(
+				"This is a really long sub-line that should be broken up more because %s %s",
+				"xxxx",
+				"yyyy",
+			),
+			fmt.Sprintf("A short one %d", 3),
+		)
+	}
+}