refactor: get rid of deprecated ast.Object

Author: tdakkota

asciicheck.go

@@ -3,6 +3,8 @@ package asciicheck
 import (
 	"fmt"
 	"go/ast"
+	"go/token"
+
 	"golang.org/x/tools/go/analysis"
 )
 
@@ -14,36 +16,78 @@ func NewAnalyzer() *analysis.Analyzer {
 	}
 }
 
-func run(pass *analysis.Pass) (interface{}, error) {
+func run(pass *analysis.Pass) (any, error) {
 	for _, file := range pass.Files {
-		alreadyViewed := map[*ast.Object]struct{}{}
 		ast.Inspect(
 			file, func(node ast.Node) bool {
-				cb(pass, node, alreadyViewed)
+				cb(pass, node)
 				return true
 			},
 		)
 	}
-
 	return nil, nil
 }
 
-func cb(pass *analysis.Pass, n ast.Node, m map[*ast.Object]struct{}) {
-	if v, ok := n.(*ast.Ident); ok {
-		if _, ok := m[v.Obj]; ok {
-			return
-		} else {
-			m[v.Obj] = struct{}{}
+func cb(pass *analysis.Pass, n ast.Node) {
+	switch n := n.(type) {
+	case *ast.File:
+		checkIdent(pass, n.Name)
+	case *ast.ImportSpec:
+		checkIdent(pass, n.Name)
+	case *ast.TypeSpec:
+		checkIdent(pass, n.Name)
+		checkFieldList(pass, n.TypeParams)
+	case *ast.ValueSpec:
+		for _, name := range n.Names {
+			checkIdent(pass, name)
 		}
+	case *ast.FuncDecl:
+		checkIdent(pass, n.Name)
+		checkFieldList(pass, n.Recv)
+	case *ast.StructType:
+		checkFieldList(pass, n.Fields)
+	case *ast.FuncType:
+		checkFieldList(pass, n.TypeParams)
+		checkFieldList(pass, n.Params)
+		checkFieldList(pass, n.Results)
+	case *ast.InterfaceType:
+		checkFieldList(pass, n.Methods)
+	case *ast.LabeledStmt:
+		checkIdent(pass, n.Label)
+	case *ast.AssignStmt:
+		if n.Tok == token.DEFINE {
+			for _, expr := range n.Lhs {
+				if ident, ok := expr.(*ast.Ident); ok {
+					checkIdent(pass, ident)
+				}
+			}
+		}
+	}
+}
+
+func checkIdent(pass *analysis.Pass, v *ast.Ident) {
+	if v == nil {
+		return
+	}
 
-		ch, ascii := isASCII(v.Name)
-		if !ascii {
-			pass.Report(
-				analysis.Diagnostic{
-					Pos:     v.Pos(),
-					Message: fmt.Sprintf("identifier \"%s\" contain non-ASCII character: %#U", v.Name, ch),
-				},
-			)
+	ch, ascii := isASCII(v.Name)
+	if !ascii {
+		pass.Report(
+			analysis.Diagnostic{
+				Pos:     v.Pos(),
+				Message: fmt.Sprintf("identifier \"%s\" contain non-ASCII character: %#U", v.Name, ch),
+			},
+		)
+	}
+}
+
+func checkFieldList(pass *analysis.Pass, f *ast.FieldList) {
+	if f == nil {
+		return
+	}
+	for _, f := range f.List {
+		for _, name := range f.Names {
+			checkIdent(pass, name)
 		}
 	}
 }

testdata/field_name.go

@@ -1,7 +1,6 @@
 package testdata
 
-type Field struct{}
-
-type JustStruct struct {
-	Tеst Field // want `identifier "Tеst" contain non-ASCII character: U\+0435 'е'`
+type _ struct {
+	Tést     int // want `identifier "Tést" contain non-ASCII character: U\+00E9 'é'`
+	_, Tést2 int // want `identifier "Tést2" contain non-ASCII character: U\+00E9 'é'`
 }

testdata/func_name.go

@@ -1,5 +1,7 @@
 package testdata
 
-func TеstFunc() { // want `identifier "TеstFunc" contain non-ASCII character: U\+0435 'е'`
+func TéstFunc() {} // want `identifier "TéstFunc" contain non-ASCII character: U\+00E9 'é'`
 
+type _ interface {
+	TéstFunc() // want `identifier "TéstFunc" contain non-ASCII character: U\+00E9 'é'`
 }

testdata/generic_name.go

@@ -0,0 +1,5 @@
+package testdata
+
+import téstImport "fmt" // want `identifier "téstImport" contain non-ASCII character: U\+00E9 'é'`
+
+var _ téstImport.Stringer

testdata/import_name.go

@@ -0,0 +1,7 @@
+package testdata
+
+func _() {
+téstLabel: // want `identifier "téstLabel" contain non-ASCII character: U\+00E9 'é'`
+	return
+	goto téstLabel
+}

testdata/label_name.go

@@ -0,0 +1 @@
+package téstdata // want `identifier "téstdata" contain non-ASCII character: U\+00E9 'é'`

testdata/pkg/package_name.go

@@ -0,0 +1,12 @@
+package testdata
+
+type (
+	_          struct{}
+	TéstStruct struct{}   // want `identifier "TéstStruct" contain non-ASCII character: U\+00E9 'é'`
+	TéstAlias  = struct{} // want `identifier "TéstAlias" contain non-ASCII character: U\+00E9 'é'`
+
+	_[_ any]           struct{}
+	_[TéstGeneric any] struct{} // want `identifier "TéstGeneric" contain non-ASCII character: U\+00E9 'é'`
+)
+
+func _[TéstGeneric any]() {} // want `identifier "TéstGeneric" contain non-ASCII character: U\+00E9 'é'`

testdata/struct_name.go

@@ -2,8 +2,30 @@ package testdata
 
 import "fmt"
 
-func f() {
-	var tеstVar int // want `identifier "tеstVar" contain non-ASCII character: U\+0435 'е'`
-	tеstVar = 0
-	fmt.Println(tеstVar)
+const téstGlobalConst = 0 // want `identifier "téstGlobalConst" contain non-ASCII character: U\+00E9 'é'`
+
+var téstGlobalVar int // want `identifier "téstGlobalVar" contain non-ASCII character: U\+00E9 'é'`
+
+func _() {
+	const téstConst = 0 // want `identifier "téstConst" contain non-ASCII character: U\+00E9 'é'`
+
+	var téstVar int // want `identifier "téstVar" contain non-ASCII character: U\+00E9 'é'`
+	téstVar = 0
+	fmt.Println(téstVar)
+
+	var ch chan int
+	téstVar2, _ := <-ch // want `identifier "téstVar2" contain non-ASCII character: U\+00E9 'é'`
+	fmt.Println(téstVar2)
 }
+
+func _(téstParam int)     {}         // want `identifier "téstParam" contain non-ASCII character: U\+00E9 'é'`
+func _() (téstResult int) { return } // want `identifier "téstResult" contain non-ASCII character: U\+00E9 'é'`
+
+type _ interface {
+	m1(téstParam int)     // want `identifier "téstParam" contain non-ASCII character: U\+00E9 'é'`
+	m2() (téstResult int) // want `identifier "téstResult" contain non-ASCII character: U\+00E9 'é'`
+}
+
+type Recv struct{}
+
+func (téstRecv Recv) _() {} // want `identifier "téstRecv" contain non-ASCII character: U\+00E9 'é'`