cmd/errtrace: Fix blank import handling

cmd/errtrace/main.go

@@ -398,7 +398,7 @@ type parsedFile struct {
 	file *ast.File
 
 	errtracePkg     string
-	importsErrtrace bool
+	importsErrtrace bool // includes blank imports
 	inserts         []insert
 	unusedOptouts   []int // list of line numbers
 }
@@ -410,21 +410,27 @@ func (cmd *mainCmd) parseFile(filename string, src []byte) (parsedFile, error) {
 		return parsedFile{}, errtrace.Wrap(err)
 	}
 
-	errtracePkg := "errtrace" // name to use for errtrace package
-	var importsErrtrace bool  // whether the file imports errtrace already
+	errtracePkg := "errtrace"  // name to use for errtrace package
+	var importsErrtrace bool   // whether there's any errtrace import, including blank imports
+	needErrtraceImport := true // whether to add a new import.
 	for _, imp := range f.Imports {
 		if imp.Path.Value == `"braces.dev/errtrace"` {
 			importsErrtrace = true
 			if imp.Name != nil {
+				if imp.Name.Name == "_" {
+					// Can't use a blank import, keep processing imports.
+					continue
+				}
 				// If the file already imports errtrace,
 				// we'll want to use the name it's imported under.
 				errtracePkg = imp.Name.Name
 			}
+			needErrtraceImport = false
 			break
 		}
 	}
 
-	if !importsErrtrace {
+	if needErrtraceImport {
 		// If the file doesn't import errtrace already,
 		// do a quick check to find an unused identifier name.
 		idents := make(map[string]struct{})
@@ -475,7 +481,7 @@ func (cmd *mainCmd) parseFile(filename string, src []byte) (parsedFile, error) {
 	// If errtrace isn't imported, but at least one insert was made,
 	// we'll need to import errtrace.
 	// Add an import declaration to the file.
-	if !importsErrtrace && len(inserts) > 0 {
+	if needErrtraceImport && len(inserts) > 0 {
 		// We want to insert the import after the last existing import.
 		// If the last import is part of a group, we'll make it part of the group.
 		//

cmd/errtrace/testdata/golden/imported_blank.go

@@ -0,0 +1,17 @@
+//go:build ignore
+
+package foo
+
+import (
+	"strconv"
+
+	_ "braces.dev/errtrace"
+)
+
+func Unwrapped(s string) (int, error) {
+	i, err := strconv.Atoi(s)
+	if err != nil {
+		return 0, err
+	}
+	return i + 42, nil
+}

cmd/errtrace/testdata/golden/imported_blank.go.golden

@@ -0,0 +1,17 @@
+//go:build ignore
+
+package foo
+
+import (
+	"strconv"
+
+	_ "braces.dev/errtrace"; "braces.dev/errtrace"
+)
+
+func Unwrapped(s string) (int, error) {
+	i, err := strconv.Atoi(s)
+	if err != nil {
+		return 0, errtrace.Wrap(err)
+	}
+	return i + 42, nil
+}

cmd/errtrace/testdata/toolexec-test/main.go

@@ -3,6 +3,7 @@ package main
 import (
 	"fmt"
 
+	_ "braces.dev/errtrace" // Opt-in to errtrace wrapping with toolexec.
 	"braces.dev/errtrace/cmd/errtrace/testdata/toolexec-test/p1"
 )
 

cmd/errtrace/toolexec.go

@@ -53,7 +53,7 @@ func (cmd *mainCmd) toolExecVersion(args []string) int {
 	}
 
 	// TODO: This version number should change whenever the rewriting changes.
-	fmt.Fprintf(cmd.Stdout, "%s-errtrace0\n", strings.TrimSpace(stdout.String()))
+	fmt.Fprintf(cmd.Stdout, "%s-errtrace1\n", strings.TrimSpace(stdout.String()))
 	return 0
 }