2018-12-28 23:15:05 +01:00
// Package simple contains a linter for Go source code.
package simple // import "honnef.co/go/tools/simple"
import (
2019-04-22 12:59:42 +02:00
"fmt"
2018-12-28 23:15:05 +01:00
"go/ast"
"go/constant"
"go/token"
"go/types"
2020-05-29 22:15:21 +02:00
"path/filepath"
2018-12-28 23:15:05 +01:00
"reflect"
2020-05-09 15:44:17 +02:00
"sort"
2018-12-28 23:15:05 +01:00
"strings"
2020-05-09 15:44:17 +02:00
"golang.org/x/tools/go/analysis"
"golang.org/x/tools/go/types/typeutil"
2019-02-18 20:32:41 +01:00
. "honnef.co/go/tools/arg"
2020-05-29 22:15:21 +02:00
"honnef.co/go/tools/code"
"honnef.co/go/tools/edit"
"honnef.co/go/tools/internal/passes/buildir"
2018-12-28 23:15:05 +01:00
"honnef.co/go/tools/internal/sharedcheck"
. "honnef.co/go/tools/lint/lintdsl"
2020-05-29 22:15:21 +02:00
"honnef.co/go/tools/pattern"
"honnef.co/go/tools/report"
2018-12-28 23:15:05 +01:00
)
2020-05-29 22:15:21 +02:00
var (
checkSingleCaseSelectQ1 = pattern . MustParse ( `
( ForStmt
nil nil nil
select @ ( SelectStmt
( CommClause
( Or
( UnaryExpr "<-" _ )
( AssignStmt _ _ ( UnaryExpr "<-" _ ) ) )
_ ) ) ) ` )
checkSingleCaseSelectQ2 = pattern . MustParse ( ` (SelectStmt (CommClause _ _)) ` )
)
2018-12-28 23:15:05 +01:00
2020-05-29 22:15:21 +02:00
func CheckSingleCaseSelect ( pass * analysis . Pass ) ( interface { } , error ) {
2018-12-28 23:15:05 +01:00
seen := map [ ast . Node ] struct { } { }
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
2020-05-29 22:15:21 +02:00
if m , ok := Match ( pass , checkSingleCaseSelectQ1 , node ) ; ok {
seen [ m . State [ "select" ] . ( ast . Node ) ] = struct { } { }
report . Report ( pass , node , "should use for range instead of for { select {} }" , report . FilterGenerated ( ) )
} else if _ , ok := Match ( pass , checkSingleCaseSelectQ2 , node ) ; ok {
if _ , ok := seen [ node ] ; ! ok {
report . Report ( pass , node , "should use a simple channel send/receive instead of select with a single case" ,
report . ShortRange ( ) ,
report . FilterGenerated ( ) )
2018-12-28 23:15:05 +01:00
}
}
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . ForStmt ) ( nil ) , ( * ast . SelectStmt ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
var (
checkLoopCopyQ = pattern . MustParse ( `
( Or
( RangeStmt
key value ":=" src @ ( Ident _ )
[ ( AssignStmt
( IndexExpr dst @ ( Ident _ ) key )
"="
value ) ] )
( RangeStmt
key nil ":=" src @ ( Ident _ )
[ ( AssignStmt
( IndexExpr dst @ ( Ident _ ) key )
"="
( IndexExpr src key ) ) ] ) ) ` )
checkLoopCopyR = pattern . MustParse ( ` (CallExpr (Ident "copy") [dst src]) ` )
)
2019-02-18 20:32:41 +01:00
2020-05-29 22:15:21 +02:00
func CheckLoopCopy ( pass * analysis . Pass ) ( interface { } , error ) {
fn := func ( node ast . Node ) {
m , edits , ok := MatchAndEdit ( pass , checkLoopCopyQ , checkLoopCopyR , node )
2018-12-28 23:15:05 +01:00
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
t1 := pass . TypesInfo . TypeOf ( m . State [ "src" ] . ( * ast . Ident ) )
t2 := pass . TypesInfo . TypeOf ( m . State [ "dst" ] . ( * ast . Ident ) )
if _ , ok := t1 . Underlying ( ) . ( * types . Slice ) ; ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
if ! types . Identical ( t1 , t2 ) {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
tv , err := types . Eval ( pass . Fset , pass . Pkg , node . Pos ( ) , "copy" )
if err == nil && tv . IsBuiltin ( ) {
report . Report ( pass , node ,
"should use copy() instead of a loop" ,
report . ShortRange ( ) ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "replace loop with call to copy()" , edits ... ) ) )
2018-12-28 23:15:05 +01:00
} else {
2020-05-29 22:15:21 +02:00
report . Report ( pass , node , "should use copy() instead of a loop" , report . FilterGenerated ( ) )
2018-12-28 23:15:05 +01:00
}
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . RangeStmt ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
func CheckIfBoolCmp ( pass * analysis . Pass ) ( interface { } , error ) {
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
2020-05-29 22:15:21 +02:00
if code . IsInTest ( pass , node ) {
return
}
2019-04-22 12:59:42 +02:00
expr := node . ( * ast . BinaryExpr )
if expr . Op != token . EQL && expr . Op != token . NEQ {
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
x := code . IsBoolConst ( pass , expr . X )
y := code . IsBoolConst ( pass , expr . Y )
2018-12-28 23:15:05 +01:00
if ! x && ! y {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
var other ast . Expr
var val bool
if x {
2020-05-29 22:15:21 +02:00
val = code . BoolConst ( pass , expr . X )
2018-12-28 23:15:05 +01:00
other = expr . Y
} else {
2020-05-29 22:15:21 +02:00
val = code . BoolConst ( pass , expr . Y )
2018-12-28 23:15:05 +01:00
other = expr . X
}
2020-05-09 15:44:17 +02:00
basic , ok := pass . TypesInfo . TypeOf ( other ) . Underlying ( ) . ( * types . Basic )
2018-12-28 23:15:05 +01:00
if ! ok || basic . Kind ( ) != types . Bool {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
op := ""
if ( expr . Op == token . EQL && ! val ) || ( expr . Op == token . NEQ && val ) {
op = "!"
}
2020-05-29 22:15:21 +02:00
r := op + report . Render ( pass , other )
2018-12-28 23:15:05 +01:00
l1 := len ( r )
r = strings . TrimLeft ( r , "!" )
if ( l1 - len ( r ) ) % 2 == 1 {
r = "!" + r
}
2020-05-29 22:15:21 +02:00
report . Report ( pass , expr , fmt . Sprintf ( "should omit comparison to bool constant, can be simplified to %s" , r ) ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "simplify bool comparison" , edit . ReplaceWithString ( pass . Fset , expr , r ) ) ) )
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . BinaryExpr ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
var (
checkBytesBufferConversionsQ = pattern . MustParse ( ` (CallExpr _ [(CallExpr sel@(SelectorExpr recv _) [])]) ` )
checkBytesBufferConversionsRs = pattern . MustParse ( ` (CallExpr (SelectorExpr recv (Ident "String")) []) ` )
checkBytesBufferConversionsRb = pattern . MustParse ( ` (CallExpr (SelectorExpr recv (Ident "Bytes")) []) ` )
)
2018-12-28 23:15:05 +01:00
2020-05-29 22:15:21 +02:00
func CheckBytesBufferConversions ( pass * analysis . Pass ) ( interface { } , error ) {
if pass . Pkg . Path ( ) == "bytes" || pass . Pkg . Path ( ) == "bytes_test" {
// The bytes package can use itself however it wants
return nil , nil
}
fn := func ( node ast . Node ) {
m , ok := Match ( pass , checkBytesBufferConversionsQ , node )
2018-12-28 23:15:05 +01:00
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
call := node . ( * ast . CallExpr )
sel := m . State [ "sel" ] . ( * ast . SelectorExpr )
2018-12-28 23:15:05 +01:00
2020-05-09 15:44:17 +02:00
typ := pass . TypesInfo . TypeOf ( call . Fun )
2020-05-29 22:15:21 +02:00
if typ == types . Universe . Lookup ( "string" ) . Type ( ) && code . IsCallToAST ( pass , call . Args [ 0 ] , "(*bytes.Buffer).Bytes" ) {
report . Report ( pass , call , fmt . Sprintf ( "should use %v.String() instead of %v" , report . Render ( pass , sel . X ) , report . Render ( pass , call ) ) ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "simplify conversion" , edit . ReplaceWithPattern ( pass , checkBytesBufferConversionsRs , m . State , node ) ) ) )
} else if typ , ok := typ . ( * types . Slice ) ; ok && typ . Elem ( ) == types . Universe . Lookup ( "byte" ) . Type ( ) && code . IsCallToAST ( pass , call . Args [ 0 ] , "(*bytes.Buffer).String" ) {
report . Report ( pass , call , fmt . Sprintf ( "should use %v.Bytes() instead of %v" , report . Render ( pass , sel . X ) , report . Render ( pass , call ) ) ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "simplify conversion" , edit . ReplaceWithPattern ( pass , checkBytesBufferConversionsRb , m . State , node ) ) ) )
2018-12-28 23:15:05 +01:00
}
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . CallExpr ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
func CheckStringsContains ( pass * analysis . Pass ) ( interface { } , error ) {
2018-12-28 23:15:05 +01:00
// map of value to token to bool value
allowed := map [ int64 ] map [ token . Token ] bool {
- 1 : { token . GTR : true , token . NEQ : true , token . EQL : false } ,
0 : { token . GEQ : true , token . LSS : false } ,
}
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
expr := node . ( * ast . BinaryExpr )
2018-12-28 23:15:05 +01:00
switch expr . Op {
case token . GEQ , token . GTR , token . NEQ , token . LSS , token . EQL :
default :
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
value , ok := code . ExprToInt ( pass , expr . Y )
2018-12-28 23:15:05 +01:00
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
allowedOps , ok := allowed [ value ]
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
b , ok := allowedOps [ expr . Op ]
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
call , ok := expr . X . ( * ast . CallExpr )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
sel , ok := call . Fun . ( * ast . SelectorExpr )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
pkgIdent , ok := sel . X . ( * ast . Ident )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
funIdent := sel . Sel
if pkgIdent . Name != "strings" && pkgIdent . Name != "bytes" {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
var r ast . Expr
2018-12-28 23:15:05 +01:00
switch funIdent . Name {
case "IndexRune" :
2020-05-29 22:15:21 +02:00
r = & ast . SelectorExpr {
X : pkgIdent ,
Sel : & ast . Ident { Name : "ContainsRune" } ,
}
2018-12-28 23:15:05 +01:00
case "IndexAny" :
2020-05-29 22:15:21 +02:00
r = & ast . SelectorExpr {
X : pkgIdent ,
Sel : & ast . Ident { Name : "ContainsAny" } ,
}
2018-12-28 23:15:05 +01:00
case "Index" :
2020-05-29 22:15:21 +02:00
r = & ast . SelectorExpr {
X : pkgIdent ,
Sel : & ast . Ident { Name : "Contains" } ,
}
2018-12-28 23:15:05 +01:00
default :
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
r = & ast . CallExpr {
Fun : r ,
Args : call . Args ,
}
2018-12-28 23:15:05 +01:00
if ! b {
2020-05-29 22:15:21 +02:00
r = & ast . UnaryExpr {
Op : token . NOT ,
X : r ,
}
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
report . Report ( pass , node , fmt . Sprintf ( "should use %s instead" , report . Render ( pass , r ) ) ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( fmt . Sprintf ( "simplify use of %s" , report . Render ( pass , call . Fun ) ) , edit . ReplaceWithNode ( pass . Fset , node , r ) ) ) )
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . BinaryExpr ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
var (
checkBytesCompareQ = pattern . MustParse ( ` (BinaryExpr (CallExpr (Function "bytes.Compare") args) op@(Or "==" "!=") (BasicLit "INT" "0")) ` )
checkBytesCompareRn = pattern . MustParse ( ` (CallExpr (SelectorExpr (Ident "bytes") (Ident "Equal")) args) ` )
checkBytesCompareRe = pattern . MustParse ( ` (UnaryExpr "!" (CallExpr (SelectorExpr (Ident "bytes") (Ident "Equal")) args)) ` )
)
func CheckBytesCompare ( pass * analysis . Pass ) ( interface { } , error ) {
if pass . Pkg . Path ( ) == "bytes" || pass . Pkg . Path ( ) == "bytes_test" {
// the bytes package is free to use bytes.Compare as it sees fit
return nil , nil
}
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
2020-05-29 22:15:21 +02:00
m , ok := Match ( pass , checkBytesCompareQ , node )
2018-12-28 23:15:05 +01:00
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
args := report . RenderArgs ( pass , m . State [ "args" ] . ( [ ] ast . Expr ) )
2018-12-28 23:15:05 +01:00
prefix := ""
2020-05-29 22:15:21 +02:00
if m . State [ "op" ] . ( token . Token ) == token . NEQ {
2018-12-28 23:15:05 +01:00
prefix = "!"
}
2020-05-29 22:15:21 +02:00
var fix analysis . SuggestedFix
switch tok := m . State [ "op" ] . ( token . Token ) ; tok {
case token . EQL :
fix = edit . Fix ( "simplify use of bytes.Compare" , edit . ReplaceWithPattern ( pass , checkBytesCompareRe , m . State , node ) )
case token . NEQ :
fix = edit . Fix ( "simplify use of bytes.Compare" , edit . ReplaceWithPattern ( pass , checkBytesCompareRn , m . State , node ) )
default :
panic ( fmt . Sprintf ( "unexpected token %v" , tok ) )
}
report . Report ( pass , node , fmt . Sprintf ( "should use %sbytes.Equal(%s) instead" , prefix , args ) , report . FilterGenerated ( ) , report . Fixes ( fix ) )
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . BinaryExpr ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
func CheckForTrue ( pass * analysis . Pass ) ( interface { } , error ) {
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
loop := node . ( * ast . ForStmt )
2018-12-28 23:15:05 +01:00
if loop . Init != nil || loop . Post != nil {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
if ! code . IsBoolConst ( pass , loop . Cond ) || ! code . BoolConst ( pass , loop . Cond ) {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
report . Report ( pass , loop , "should use for {} instead of for true {}" ,
report . ShortRange ( ) ,
report . FilterGenerated ( ) )
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . ForStmt ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
func CheckRegexpRaw ( pass * analysis . Pass ) ( interface { } , error ) {
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
call := node . ( * ast . CallExpr )
2020-05-29 22:15:21 +02:00
if ! code . IsCallToAnyAST ( pass , call , "regexp.MustCompile" , "regexp.Compile" ) {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
sel , ok := call . Fun . ( * ast . SelectorExpr )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2019-02-18 20:32:41 +01:00
lit , ok := call . Args [ Arg ( "regexp.Compile.expr" ) ] . ( * ast . BasicLit )
2018-12-28 23:15:05 +01:00
if ! ok {
// TODO(dominikh): support string concat, maybe support constants
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if lit . Kind != token . STRING {
// invalid function call
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if lit . Value [ 0 ] != '"' {
// already a raw string
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
val := lit . Value
if ! strings . Contains ( val , ` \\ ` ) {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2019-02-18 20:32:41 +01:00
if strings . Contains ( val , "`" ) {
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
}
2018-12-28 23:15:05 +01:00
bs := false
for _ , c := range val {
if ! bs && c == '\\' {
bs = true
continue
}
if bs && c == '\\' {
bs = false
continue
}
if bs {
// backslash followed by non-backslash -> escape sequence
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
}
2020-05-29 22:15:21 +02:00
report . Report ( pass , call , fmt . Sprintf ( "should use raw string (`...`) with regexp.%s to avoid having to escape twice" , sel . Sel . Name ) , report . FilterGenerated ( ) )
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . CallExpr ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
var (
checkIfReturnQIf = pattern . MustParse ( ` (IfStmt nil cond [(ReturnStmt [ret@(Ident _)])] nil) ` )
checkIfReturnQRet = pattern . MustParse ( ` (ReturnStmt [ret@(Ident _)]) ` )
)
func CheckIfReturn ( pass * analysis . Pass ) ( interface { } , error ) {
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
block := node . ( * ast . BlockStmt )
2018-12-28 23:15:05 +01:00
l := len ( block . List )
if l < 2 {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
n1 , n2 := block . List [ l - 2 ] , block . List [ l - 1 ]
if len ( block . List ) >= 3 {
if _ , ok := block . List [ l - 3 ] . ( * ast . IfStmt ) ; ok {
// Do not flag a series of if statements
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
}
2020-05-29 22:15:21 +02:00
m1 , ok := Match ( pass , checkIfReturnQIf , n1 )
2018-12-28 23:15:05 +01:00
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
m2 , ok := Match ( pass , checkIfReturnQRet , n2 )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
if op , ok := m1 . State [ "cond" ] . ( * ast . BinaryExpr ) ; ok {
2018-12-28 23:15:05 +01:00
switch op . Op {
case token . EQL , token . LSS , token . GTR , token . NEQ , token . LEQ , token . GEQ :
default :
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
}
2020-05-29 22:15:21 +02:00
ret1 := m1 . State [ "ret" ] . ( * ast . Ident )
if ! code . IsBoolConst ( pass , ret1 ) {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
ret2 := m2 . State [ "ret" ] . ( * ast . Ident )
if ! code . IsBoolConst ( pass , ret2 ) {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-09 15:44:17 +02:00
2020-05-29 22:15:21 +02:00
if ret1 . Name == ret2 . Name {
2020-05-09 15:44:17 +02:00
// we want the function to return true and false, not the
// same value both times.
return
}
2020-05-29 22:15:21 +02:00
cond := m1 . State [ "cond" ] . ( ast . Expr )
origCond := cond
if ret1 . Name == "false" {
cond = negate ( cond )
}
report . Report ( pass , n1 ,
fmt . Sprintf ( "should use 'return %s' instead of 'if %s { return %s }; return %s'" ,
report . Render ( pass , cond ) ,
report . Render ( pass , origCond ) , report . Render ( pass , ret1 ) , report . Render ( pass , ret2 ) ) ,
report . FilterGenerated ( ) )
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . BlockStmt ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
func negate ( expr ast . Expr ) ast . Expr {
switch expr := expr . ( type ) {
case * ast . BinaryExpr :
out := * expr
switch expr . Op {
case token . EQL :
out . Op = token . NEQ
case token . LSS :
out . Op = token . GEQ
case token . GTR :
out . Op = token . LEQ
case token . NEQ :
out . Op = token . EQL
case token . LEQ :
out . Op = token . GTR
case token . GEQ :
out . Op = token . LEQ
}
return & out
case * ast . Ident , * ast . CallExpr , * ast . IndexExpr :
return & ast . UnaryExpr {
Op : token . NOT ,
X : expr ,
}
default :
return & ast . UnaryExpr {
Op : token . NOT ,
X : & ast . ParenExpr {
X : expr ,
} ,
}
}
}
// CheckRedundantNilCheckWithLen checks for the following redundant nil-checks:
2018-12-28 23:15:05 +01:00
//
// if x == nil || len(x) == 0 {}
// if x != nil && len(x) != 0 {}
// if x != nil && len(x) == N {} (where N != 0)
// if x != nil && len(x) > N {}
// if x != nil && len(x) >= N {} (where N != 0)
//
2020-05-29 22:15:21 +02:00
func CheckRedundantNilCheckWithLen ( pass * analysis . Pass ) ( interface { } , error ) {
2018-12-28 23:15:05 +01:00
isConstZero := func ( expr ast . Expr ) ( isConst bool , isZero bool ) {
_ , ok := expr . ( * ast . BasicLit )
if ok {
2020-05-29 22:15:21 +02:00
return true , code . IsIntLiteral ( expr , "0" )
2018-12-28 23:15:05 +01:00
}
id , ok := expr . ( * ast . Ident )
if ! ok {
return false , false
}
2020-05-09 15:44:17 +02:00
c , ok := pass . TypesInfo . ObjectOf ( id ) . ( * types . Const )
2018-12-28 23:15:05 +01:00
if ! ok {
return false , false
}
return true , c . Val ( ) . Kind ( ) == constant . Int && c . Val ( ) . String ( ) == "0"
}
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
2018-12-28 23:15:05 +01:00
// check that expr is "x || y" or "x && y"
2019-04-22 12:59:42 +02:00
expr := node . ( * ast . BinaryExpr )
2018-12-28 23:15:05 +01:00
if expr . Op != token . LOR && expr . Op != token . LAND {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
eqNil := expr . Op == token . LOR
// check that x is "xx == nil" or "xx != nil"
x , ok := expr . X . ( * ast . BinaryExpr )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if eqNil && x . Op != token . EQL {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if ! eqNil && x . Op != token . NEQ {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
xx , ok := x . X . ( * ast . Ident )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
if ! code . IsNil ( pass , x . Y ) {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
// check that y is "len(xx) == 0" or "len(xx) ... "
y , ok := expr . Y . ( * ast . BinaryExpr )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if eqNil && y . Op != token . EQL { // must be len(xx) *==* 0
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
yx , ok := y . X . ( * ast . CallExpr )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
yxFun , ok := yx . Fun . ( * ast . Ident )
if ! ok || yxFun . Name != "len" || len ( yx . Args ) != 1 {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2019-02-18 20:32:41 +01:00
yxArg , ok := yx . Args [ Arg ( "len.v" ) ] . ( * ast . Ident )
2018-12-28 23:15:05 +01:00
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if yxArg . Name != xx . Name {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
if eqNil && ! code . IsIntLiteral ( y . Y , "0" ) { // must be len(x) == *0*
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if ! eqNil {
isConst , isZero := isConstZero ( y . Y )
if ! isConst {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
switch y . Op {
case token . EQL :
// avoid false positive for "xx != nil && len(xx) == 0"
if isZero {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
case token . GEQ :
// avoid false positive for "xx != nil && len(xx) >= 0"
if isZero {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
case token . NEQ :
// avoid false positive for "xx != nil && len(xx) != <non-zero>"
if ! isZero {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
case token . GTR :
// ok
default :
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
}
// finally check that xx type is one of array, slice, map or chan
// this is to prevent false positive in case if xx is a pointer to an array
var nilType string
2020-05-09 15:44:17 +02:00
switch pass . TypesInfo . TypeOf ( xx ) . ( type ) {
2018-12-28 23:15:05 +01:00
case * types . Slice :
nilType = "nil slices"
case * types . Map :
nilType = "nil maps"
case * types . Chan :
nilType = "nil channels"
default :
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
report . Report ( pass , expr , fmt . Sprintf ( "should omit nil check; len() for %s is defined as zero" , nilType ) , report . FilterGenerated ( ) )
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . BinaryExpr ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
var checkSlicingQ = pattern . MustParse ( ` (SliceExpr x@(Object _) low (CallExpr (Builtin "len") [x]) nil) ` )
func CheckSlicing ( pass * analysis . Pass ) ( interface { } , error ) {
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
2020-05-29 22:15:21 +02:00
if _ , ok := Match ( pass , checkSlicingQ , node ) ; ok {
expr := node . ( * ast . SliceExpr )
report . Report ( pass , expr . High ,
"should omit second index in slice, s[a:len(s)] is identical to s[a:]" ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "simplify slice expression" , edit . Delete ( expr . High ) ) ) )
2018-12-28 23:15:05 +01:00
}
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . SliceExpr ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
func refersTo ( pass * analysis . Pass , expr ast . Expr , ident types . Object ) bool {
2018-12-28 23:15:05 +01:00
found := false
fn := func ( node ast . Node ) bool {
ident2 , ok := node . ( * ast . Ident )
if ! ok {
return true
}
2020-05-29 22:15:21 +02:00
if ident == pass . TypesInfo . ObjectOf ( ident2 ) {
2018-12-28 23:15:05 +01:00
found = true
return false
}
return true
}
ast . Inspect ( expr , fn )
return found
}
2020-05-29 22:15:21 +02:00
var checkLoopAppendQ = pattern . MustParse ( `
( RangeStmt
( Ident "_" )
val @ ( Object _ )
_
x
[ ( AssignStmt [ lhs ] "=" [ ( CallExpr ( Builtin "append" ) [ lhs val ] ) ] ) ] ) ` )
func CheckLoopAppend ( pass * analysis . Pass ) ( interface { } , error ) {
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
2020-05-29 22:15:21 +02:00
m , ok := Match ( pass , checkLoopAppendQ , node )
2018-12-28 23:15:05 +01:00
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
val := m . State [ "val" ] . ( types . Object )
if refersTo ( pass , m . State [ "lhs" ] . ( ast . Expr ) , val ) {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
src := pass . TypesInfo . TypeOf ( m . State [ "x" ] . ( ast . Expr ) )
dst := pass . TypesInfo . TypeOf ( m . State [ "lhs" ] . ( ast . Expr ) )
2018-12-28 23:15:05 +01:00
if ! types . Identical ( src , dst ) {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
r := & ast . AssignStmt {
Lhs : [ ] ast . Expr { m . State [ "lhs" ] . ( ast . Expr ) } ,
Tok : token . ASSIGN ,
Rhs : [ ] ast . Expr {
& ast . CallExpr {
Fun : & ast . Ident { Name : "append" } ,
Args : [ ] ast . Expr {
m . State [ "lhs" ] . ( ast . Expr ) ,
m . State [ "x" ] . ( ast . Expr ) ,
} ,
Ellipsis : 1 ,
} ,
} ,
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
report . Report ( pass , node , fmt . Sprintf ( "should replace loop with %s" , report . Render ( pass , r ) ) ,
report . ShortRange ( ) ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "replace loop with call to append" , edit . ReplaceWithNode ( pass . Fset , node , r ) ) ) )
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . RangeStmt ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
var (
checkTimeSinceQ = pattern . MustParse ( ` (CallExpr (SelectorExpr (CallExpr (Function "time.Now") []) (Function "(time.Time).Sub")) [arg]) ` )
checkTimeSinceR = pattern . MustParse ( ` (CallExpr (SelectorExpr (Ident "time") (Ident "Since")) [arg]) ` )
)
func CheckTimeSince ( pass * analysis . Pass ) ( interface { } , error ) {
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
2020-05-29 22:15:21 +02:00
if _ , edits , ok := MatchAndEdit ( pass , checkTimeSinceQ , checkTimeSinceR , node ) ; ok {
report . Report ( pass , node , "should use time.Since instead of time.Now().Sub" ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "replace with call to time.Since" , edits ... ) ) )
2018-12-28 23:15:05 +01:00
}
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . CallExpr ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
var (
checkTimeUntilQ = pattern . MustParse ( ` (CallExpr (Function "(time.Time).Sub") [(CallExpr (Function "time.Now") [])]) ` )
checkTimeUntilR = pattern . MustParse ( ` (CallExpr (SelectorExpr (Ident "time") (Ident "Until")) [arg]) ` )
)
func CheckTimeUntil ( pass * analysis . Pass ) ( interface { } , error ) {
if ! code . IsGoVersion ( pass , 8 ) {
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
2020-05-29 22:15:21 +02:00
if _ , ok := Match ( pass , checkTimeUntilQ , node ) ; ok {
if sel , ok := node . ( * ast . CallExpr ) . Fun . ( * ast . SelectorExpr ) ; ok {
r := pattern . NodeToAST ( checkTimeUntilR . Root , map [ string ] interface { } { "arg" : sel . X } ) . ( ast . Node )
report . Report ( pass , node , "should use time.Until instead of t.Sub(time.Now())" ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "replace with call to time.Until" , edit . ReplaceWithNode ( pass . Fset , node , r ) ) ) )
} else {
report . Report ( pass , node , "should use time.Until instead of t.Sub(time.Now())" , report . FilterGenerated ( ) )
}
2018-12-28 23:15:05 +01:00
}
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . CallExpr ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
var (
checkUnnecessaryBlankQ1 = pattern . MustParse ( `
( AssignStmt
[ _ ( Ident "_" ) ]
_
( Or
( IndexExpr _ _ )
( UnaryExpr "<-" _ ) ) ) ` )
checkUnnecessaryBlankQ2 = pattern . MustParse ( `
( AssignStmt
( Ident "_" ) _ recv @ ( UnaryExpr "<-" _ ) ) ` )
)
2018-12-28 23:15:05 +01:00
2020-05-29 22:15:21 +02:00
func CheckUnnecessaryBlank ( pass * analysis . Pass ) ( interface { } , error ) {
fn1 := func ( node ast . Node ) {
if _ , ok := Match ( pass , checkUnnecessaryBlankQ1 , node ) ; ok {
r := * node . ( * ast . AssignStmt )
r . Lhs = r . Lhs [ 0 : 1 ]
report . Report ( pass , node , "unnecessary assignment to the blank identifier" ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "remove assignment to blank identifier" , edit . ReplaceWithNode ( pass . Fset , node , & r ) ) ) )
} else if m , ok := Match ( pass , checkUnnecessaryBlankQ2 , node ) ; ok {
report . Report ( pass , node , "unnecessary assignment to the blank identifier" ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "simplify channel receive operation" , edit . ReplaceWithNode ( pass . Fset , node , m . State [ "recv" ] . ( ast . Node ) ) ) ) )
2018-12-28 23:15:05 +01:00
}
}
fn3 := func ( node ast . Node ) {
2019-04-22 12:59:42 +02:00
rs := node . ( * ast . RangeStmt )
2018-12-28 23:15:05 +01:00
2020-05-29 22:15:21 +02:00
// for _
if rs . Value == nil && code . IsBlank ( rs . Key ) {
report . Report ( pass , rs . Key , "unnecessary assignment to the blank identifier" ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "remove assignment to blank identifier" , edit . Delete ( edit . Range { rs . Key . Pos ( ) , rs . TokPos + 1 } ) ) ) )
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
// for _, _
if code . IsBlank ( rs . Key ) && code . IsBlank ( rs . Value ) {
// FIXME we should mark both key and value
report . Report ( pass , rs . Key , "unnecessary assignment to the blank identifier" ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "remove assignment to blank identifier" , edit . Delete ( edit . Range { rs . Key . Pos ( ) , rs . TokPos + 1 } ) ) ) )
}
// for x, _
if ! code . IsBlank ( rs . Key ) && code . IsBlank ( rs . Value ) {
report . Report ( pass , rs . Value , "unnecessary assignment to the blank identifier" ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "remove assignment to blank identifier" , edit . Delete ( edit . Range { rs . Key . End ( ) , rs . Value . End ( ) } ) ) ) )
2018-12-28 23:15:05 +01:00
}
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn1 , ( * ast . AssignStmt ) ( nil ) )
if code . IsGoVersion ( pass , 4 ) {
code . Preorder ( pass , fn3 , ( * ast . RangeStmt ) ( nil ) )
2018-12-28 23:15:05 +01:00
}
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
func CheckSimplerStructConversion ( pass * analysis . Pass ) ( interface { } , error ) {
2018-12-28 23:15:05 +01:00
var skip ast . Node
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
2018-12-28 23:15:05 +01:00
// Do not suggest type conversion between pointers
if unary , ok := node . ( * ast . UnaryExpr ) ; ok && unary . Op == token . AND {
if lit , ok := unary . X . ( * ast . CompositeLit ) ; ok {
skip = lit
}
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if node == skip {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
lit , ok := node . ( * ast . CompositeLit )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-09 15:44:17 +02:00
typ1 , _ := pass . TypesInfo . TypeOf ( lit . Type ) . ( * types . Named )
2018-12-28 23:15:05 +01:00
if typ1 == nil {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
s1 , ok := typ1 . Underlying ( ) . ( * types . Struct )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
var typ2 * types . Named
var ident * ast . Ident
getSelType := func ( expr ast . Expr ) ( types . Type , * ast . Ident , bool ) {
sel , ok := expr . ( * ast . SelectorExpr )
if ! ok {
return nil , nil , false
}
ident , ok := sel . X . ( * ast . Ident )
if ! ok {
return nil , nil , false
}
2020-05-09 15:44:17 +02:00
typ := pass . TypesInfo . TypeOf ( sel . X )
2018-12-28 23:15:05 +01:00
return typ , ident , typ != nil
}
if len ( lit . Elts ) == 0 {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if s1 . NumFields ( ) != len ( lit . Elts ) {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
for i , elt := range lit . Elts {
var t types . Type
var id * ast . Ident
var ok bool
switch elt := elt . ( type ) {
case * ast . SelectorExpr :
t , id , ok = getSelType ( elt )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if i >= s1 . NumFields ( ) || s1 . Field ( i ) . Name ( ) != elt . Sel . Name {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
case * ast . KeyValueExpr :
var sel * ast . SelectorExpr
sel , ok = elt . Value . ( * ast . SelectorExpr )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if elt . Key . ( * ast . Ident ) . Name != sel . Sel . Name {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
t , id , ok = getSelType ( elt . Value )
}
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
// All fields must be initialized from the same object
if ident != nil && ident . Obj != id . Obj {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
typ2 , _ = t . ( * types . Named )
if typ2 == nil {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
ident = id
}
if typ2 == nil {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if typ1 . Obj ( ) . Pkg ( ) != typ2 . Obj ( ) . Pkg ( ) {
// Do not suggest type conversions between different
// packages. Types in different packages might only match
// by coincidence. Furthermore, if the dependency ever
// adds more fields to its type, it could break the code
// that relies on the type conversion to work.
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
s2 , ok := typ2 . Underlying ( ) . ( * types . Struct )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if typ1 == typ2 {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
if code . IsGoVersion ( pass , 8 ) {
2019-02-18 20:32:41 +01:00
if ! types . IdenticalIgnoreTags ( s1 , s2 ) {
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
}
} else {
if ! types . Identical ( s1 , s2 ) {
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
}
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
r := & ast . CallExpr {
Fun : lit . Type ,
Args : [ ] ast . Expr { ident } ,
}
report . Report ( pass , node ,
fmt . Sprintf ( "should convert %s (type %s) to %s instead of using struct literal" , ident . Name , typ2 . Obj ( ) . Name ( ) , typ1 . Obj ( ) . Name ( ) ) ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "use type conversion" , edit . ReplaceWithNode ( pass . Fset , node , r ) ) ) )
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . UnaryExpr ) ( nil ) , ( * ast . CompositeLit ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
func CheckTrim ( pass * analysis . Pass ) ( interface { } , error ) {
2018-12-28 23:15:05 +01:00
sameNonDynamic := func ( node1 , node2 ast . Node ) bool {
if reflect . TypeOf ( node1 ) != reflect . TypeOf ( node2 ) {
return false
}
switch node1 := node1 . ( type ) {
case * ast . Ident :
return node1 . Obj == node2 . ( * ast . Ident ) . Obj
case * ast . SelectorExpr :
2020-05-29 22:15:21 +02:00
return report . Render ( pass , node1 ) == report . Render ( pass , node2 )
2018-12-28 23:15:05 +01:00
case * ast . IndexExpr :
2020-05-29 22:15:21 +02:00
return report . Render ( pass , node1 ) == report . Render ( pass , node2 )
2018-12-28 23:15:05 +01:00
}
return false
}
isLenOnIdent := func ( fn ast . Expr , ident ast . Expr ) bool {
call , ok := fn . ( * ast . CallExpr )
if ! ok {
return false
}
if fn , ok := call . Fun . ( * ast . Ident ) ; ! ok || fn . Name != "len" {
return false
}
if len ( call . Args ) != 1 {
return false
}
2019-02-18 20:32:41 +01:00
return sameNonDynamic ( call . Args [ Arg ( "len.v" ) ] , ident )
2018-12-28 23:15:05 +01:00
}
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
2018-12-28 23:15:05 +01:00
var pkg string
var fun string
2019-04-22 12:59:42 +02:00
ifstmt := node . ( * ast . IfStmt )
2018-12-28 23:15:05 +01:00
if ifstmt . Init != nil {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if ifstmt . Else != nil {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if len ( ifstmt . Body . List ) != 1 {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
condCall , ok := ifstmt . Cond . ( * ast . CallExpr )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
condCallName := code . CallNameAST ( pass , condCall )
switch condCallName {
case "strings.HasPrefix" :
2019-02-18 20:32:41 +01:00
pkg = "strings"
fun = "HasPrefix"
2020-05-29 22:15:21 +02:00
case "strings.HasSuffix" :
2018-12-28 23:15:05 +01:00
pkg = "strings"
2019-02-18 20:32:41 +01:00
fun = "HasSuffix"
2020-05-29 22:15:21 +02:00
case "strings.Contains" :
2019-02-18 20:32:41 +01:00
pkg = "strings"
fun = "Contains"
2020-05-29 22:15:21 +02:00
case "bytes.HasPrefix" :
2018-12-28 23:15:05 +01:00
pkg = "bytes"
fun = "HasPrefix"
2020-05-29 22:15:21 +02:00
case "bytes.HasSuffix" :
2019-02-18 20:32:41 +01:00
pkg = "bytes"
2018-12-28 23:15:05 +01:00
fun = "HasSuffix"
2020-05-29 22:15:21 +02:00
case "bytes.Contains" :
2019-02-18 20:32:41 +01:00
pkg = "bytes"
fun = "Contains"
default :
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
assign , ok := ifstmt . Body . List [ 0 ] . ( * ast . AssignStmt )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if assign . Tok != token . ASSIGN {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if len ( assign . Lhs ) != 1 || len ( assign . Rhs ) != 1 {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if ! sameNonDynamic ( condCall . Args [ 0 ] , assign . Lhs [ 0 ] ) {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2019-02-18 20:32:41 +01:00
switch rhs := assign . Rhs [ 0 ] . ( type ) {
case * ast . CallExpr :
if len ( rhs . Args ) < 2 || ! sameNonDynamic ( condCall . Args [ 0 ] , rhs . Args [ 0 ] ) || ! sameNonDynamic ( condCall . Args [ 1 ] , rhs . Args [ 1 ] ) {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
rhsName := code . CallNameAST ( pass , rhs )
if condCallName == "strings.HasPrefix" && rhsName == "strings.TrimPrefix" ||
condCallName == "strings.HasSuffix" && rhsName == "strings.TrimSuffix" ||
condCallName == "strings.Contains" && rhsName == "strings.Replace" ||
condCallName == "bytes.HasPrefix" && rhsName == "bytes.TrimPrefix" ||
condCallName == "bytes.HasSuffix" && rhsName == "bytes.TrimSuffix" ||
condCallName == "bytes.Contains" && rhsName == "bytes.Replace" {
report . Report ( pass , ifstmt , fmt . Sprintf ( "should replace this if statement with an unconditional %s" , rhsName ) , report . FilterGenerated ( ) )
2018-12-28 23:15:05 +01:00
}
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
case * ast . SliceExpr :
slice := rhs
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2019-02-18 20:32:41 +01:00
if slice . Slice3 {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2019-02-18 20:32:41 +01:00
if ! sameNonDynamic ( slice . X , condCall . Args [ 0 ] ) {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2019-02-18 20:32:41 +01:00
var index ast . Expr
switch fun {
case "HasPrefix" :
// TODO(dh) We could detect a High that is len(s), but another
// rule will already flag that, anyway.
if slice . High != nil {
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
}
index = slice . Low
case "HasSuffix" :
if slice . Low != nil {
2020-05-29 22:15:21 +02:00
n , ok := code . ExprToInt ( pass , slice . Low )
2019-02-18 20:32:41 +01:00
if ! ok || n != 0 {
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
}
}
index = slice . High
}
switch index := index . ( type ) {
case * ast . CallExpr :
if fun != "HasPrefix" {
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
}
if fn , ok := index . Fun . ( * ast . Ident ) ; ! ok || fn . Name != "len" {
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
}
if len ( index . Args ) != 1 {
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
}
id3 := index . Args [ Arg ( "len.v" ) ]
switch oid3 := condCall . Args [ 1 ] . ( type ) {
case * ast . BasicLit :
if pkg != "strings" {
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
}
lit , ok := id3 . ( * ast . BasicLit )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
}
2020-05-29 22:15:21 +02:00
s1 , ok1 := code . ExprToString ( pass , lit )
s2 , ok2 := code . ExprToString ( pass , condCall . Args [ 1 ] )
2019-02-18 20:32:41 +01:00
if ! ok1 || ! ok2 || s1 != s2 {
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
}
default :
if ! sameNonDynamic ( id3 , oid3 ) {
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
}
}
case * ast . BasicLit , * ast . Ident :
if fun != "HasPrefix" {
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
}
2018-12-28 23:15:05 +01:00
if pkg != "strings" {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
string , ok1 := code . ExprToString ( pass , condCall . Args [ 1 ] )
int , ok2 := code . ExprToInt ( pass , slice . Low )
2019-02-18 20:32:41 +01:00
if ! ok1 || ! ok2 || int != int64 ( len ( string ) ) {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2019-02-18 20:32:41 +01:00
case * ast . BinaryExpr :
if fun != "HasSuffix" {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2019-02-18 20:32:41 +01:00
if index . Op != token . SUB {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2019-02-18 20:32:41 +01:00
if ! isLenOnIdent ( index . X , condCall . Args [ 0 ] ) ||
! isLenOnIdent ( index . Y , condCall . Args [ 1 ] ) {
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
}
default :
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2019-02-18 20:32:41 +01:00
var replacement string
switch fun {
case "HasPrefix" :
replacement = "TrimPrefix"
case "HasSuffix" :
replacement = "TrimSuffix"
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
report . Report ( pass , ifstmt , fmt . Sprintf ( "should replace this if statement with an unconditional %s.%s" , pkg , replacement ) ,
report . ShortRange ( ) ,
report . FilterGenerated ( ) )
2018-12-28 23:15:05 +01:00
}
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . IfStmt ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
var (
checkLoopSlideQ = pattern . MustParse ( `
( ForStmt
( AssignStmt initvar @ ( Ident _ ) _ ( BasicLit "INT" "0" ) )
( BinaryExpr initvar "<" limit @ ( Ident _ ) )
( IncDecStmt initvar "++" )
[ ( AssignStmt
( IndexExpr slice @ ( Ident _ ) initvar )
"="
( IndexExpr slice ( BinaryExpr offset @ ( Ident _ ) "+" initvar ) ) ) ] ) ` )
checkLoopSlideR = pattern . MustParse ( `
( CallExpr
( Ident "copy" )
[ ( SliceExpr slice nil limit nil )
( SliceExpr slice offset nil nil ) ] ) ` )
)
func CheckLoopSlide ( pass * analysis . Pass ) ( interface { } , error ) {
2018-12-28 23:15:05 +01:00
// TODO(dh): detect bs[i+offset] in addition to bs[offset+i]
// TODO(dh): consider merging this function with LintLoopCopy
// TODO(dh): detect length that is an expression, not a variable name
// TODO(dh): support sliding to a different offset than the beginning of the slice
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
loop := node . ( * ast . ForStmt )
2020-05-29 22:15:21 +02:00
m , edits , ok := MatchAndEdit ( pass , checkLoopSlideQ , checkLoopSlideR , loop )
2018-12-28 23:15:05 +01:00
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
if _ , ok := pass . TypesInfo . TypeOf ( m . State [ "slice" ] . ( * ast . Ident ) ) . Underlying ( ) . ( * types . Slice ) ; ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
report . Report ( pass , loop , "should use copy() instead of loop for sliding slice elements" ,
report . ShortRange ( ) ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "use copy() instead of loop" , edits ... ) ) )
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . ForStmt ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
var (
checkMakeLenCapQ1 = pattern . MustParse ( ` (CallExpr (Builtin "make") [typ size@(BasicLit "INT" "0")]) ` )
checkMakeLenCapQ2 = pattern . MustParse ( ` (CallExpr (Builtin "make") [typ size size]) ` )
)
func CheckMakeLenCap ( pass * analysis . Pass ) ( interface { } , error ) {
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
2020-05-29 22:15:21 +02:00
if pass . Pkg . Path ( ) == "runtime_test" && filepath . Base ( pass . Fset . Position ( node . Pos ( ) ) . Filename ) == "map_test.go" {
// special case of runtime tests testing map creation
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
if m , ok := Match ( pass , checkMakeLenCapQ1 , node ) ; ok {
T := m . State [ "typ" ] . ( ast . Expr )
size := m . State [ "size" ] . ( ast . Node )
if _ , ok := pass . TypesInfo . TypeOf ( T ) . Underlying ( ) . ( * types . Slice ) ; ok {
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
report . Report ( pass , size , fmt . Sprintf ( "should use make(%s) instead" , report . Render ( pass , T ) ) , report . FilterGenerated ( ) )
} else if m , ok := Match ( pass , checkMakeLenCapQ2 , node ) ; ok {
// TODO(dh): don't consider sizes identical if they're
// dynamic. for example: make(T, <-ch, <-ch).
T := m . State [ "typ" ] . ( ast . Expr )
size := m . State [ "size" ] . ( ast . Node )
report . Report ( pass , size ,
fmt . Sprintf ( "should use make(%s, %s) instead" , report . Render ( pass , T ) , report . Render ( pass , size ) ) ,
report . FilterGenerated ( ) )
2018-12-28 23:15:05 +01:00
}
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . CallExpr ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
var (
checkAssertNotNilFn1Q = pattern . MustParse ( `
( IfStmt
( AssignStmt [ ( Ident "_" ) ok @ ( Object _ ) ] _ [ ( TypeAssertExpr assert @ ( Object _ ) _ ) ] )
( Or
( BinaryExpr ok "&&" ( BinaryExpr assert "!=" ( Builtin "nil" ) ) )
( BinaryExpr ( BinaryExpr assert "!=" ( Builtin "nil" ) ) "&&" ok ) )
_
_ ) ` )
checkAssertNotNilFn2Q = pattern . MustParse ( `
( IfStmt
nil
( BinaryExpr lhs @ ( Object _ ) "!=" ( Builtin "nil" ) )
[
ifstmt @ ( IfStmt
( AssignStmt [ ( Ident "_" ) ok @ ( Object _ ) ] _ [ ( TypeAssertExpr lhs _ ) ] )
ok
_
_ )
]
nil ) ` )
)
func CheckAssertNotNil ( pass * analysis . Pass ) ( interface { } , error ) {
2019-04-22 12:59:42 +02:00
fn1 := func ( node ast . Node ) {
2020-05-29 22:15:21 +02:00
m , ok := Match ( pass , checkAssertNotNilFn1Q , node )
2018-12-28 23:15:05 +01:00
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
assert := m . State [ "assert" ] . ( types . Object )
assign := m . State [ "ok" ] . ( types . Object )
report . Report ( pass , node , fmt . Sprintf ( "when %s is true, %s can't be nil" , assign . Name ( ) , assert . Name ( ) ) ,
report . ShortRange ( ) ,
report . FilterGenerated ( ) )
2018-12-28 23:15:05 +01:00
}
2019-04-22 12:59:42 +02:00
fn2 := func ( node ast . Node ) {
2020-05-29 22:15:21 +02:00
m , ok := Match ( pass , checkAssertNotNilFn2Q , node )
2019-02-18 20:32:41 +01:00
if ! ok {
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
}
2020-05-29 22:15:21 +02:00
ifstmt := m . State [ "ifstmt" ] . ( * ast . IfStmt )
lhs := m . State [ "lhs" ] . ( types . Object )
assignIdent := m . State [ "ok" ] . ( types . Object )
report . Report ( pass , ifstmt , fmt . Sprintf ( "when %s is true, %s can't be nil" , assignIdent . Name ( ) , lhs . Name ( ) ) ,
report . ShortRange ( ) ,
report . FilterGenerated ( ) )
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn1 , ( * ast . IfStmt ) ( nil ) )
code . Preorder ( pass , fn2 , ( * ast . IfStmt ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
func CheckDeclareAssign ( pass * analysis . Pass ) ( interface { } , error ) {
2019-04-22 12:59:42 +02:00
hasMultipleAssignments := func ( root ast . Node , ident * ast . Ident ) bool {
num := 0
ast . Inspect ( root , func ( node ast . Node ) bool {
if num >= 2 {
return false
}
assign , ok := node . ( * ast . AssignStmt )
if ! ok {
return true
}
for _ , lhs := range assign . Lhs {
if oident , ok := lhs . ( * ast . Ident ) ; ok {
if oident . Obj == ident . Obj {
num ++
}
}
}
2018-12-28 23:15:05 +01:00
return true
2019-04-22 12:59:42 +02:00
} )
return num >= 2
}
fn := func ( node ast . Node ) {
block := node . ( * ast . BlockStmt )
2018-12-28 23:15:05 +01:00
if len ( block . List ) < 2 {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
for i , stmt := range block . List [ : len ( block . List ) - 1 ] {
_ = i
decl , ok := stmt . ( * ast . DeclStmt )
if ! ok {
continue
}
gdecl , ok := decl . Decl . ( * ast . GenDecl )
if ! ok || gdecl . Tok != token . VAR || len ( gdecl . Specs ) != 1 {
continue
}
vspec , ok := gdecl . Specs [ 0 ] . ( * ast . ValueSpec )
if ! ok || len ( vspec . Names ) != 1 || len ( vspec . Values ) != 0 {
continue
}
assign , ok := block . List [ i + 1 ] . ( * ast . AssignStmt )
if ! ok || assign . Tok != token . ASSIGN {
continue
}
if len ( assign . Lhs ) != 1 || len ( assign . Rhs ) != 1 {
continue
}
ident , ok := assign . Lhs [ 0 ] . ( * ast . Ident )
if ! ok {
continue
}
if vspec . Names [ 0 ] . Obj != ident . Obj {
continue
}
2020-05-29 22:15:21 +02:00
if refersTo ( pass , assign . Rhs [ 0 ] , pass . TypesInfo . ObjectOf ( ident ) ) {
2018-12-28 23:15:05 +01:00
continue
}
2019-04-22 12:59:42 +02:00
if hasMultipleAssignments ( block , ident ) {
continue
}
2020-05-29 22:15:21 +02:00
r := & ast . GenDecl {
Specs : [ ] ast . Spec {
& ast . ValueSpec {
Names : vspec . Names ,
Values : [ ] ast . Expr { assign . Rhs [ 0 ] } ,
Type : vspec . Type ,
} ,
} ,
Tok : gdecl . Tok ,
}
report . Report ( pass , decl , "should merge variable declaration with assignment on next line" ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "merge declaration with assignment" , edit . ReplaceWithNode ( pass . Fset , edit . Range { decl . Pos ( ) , assign . End ( ) } , r ) ) ) )
2018-12-28 23:15:05 +01:00
}
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . BlockStmt ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
func CheckRedundantBreak ( pass * analysis . Pass ) ( interface { } , error ) {
2018-12-28 23:15:05 +01:00
fn1 := func ( node ast . Node ) {
2019-04-22 12:59:42 +02:00
clause := node . ( * ast . CaseClause )
2018-12-28 23:15:05 +01:00
if len ( clause . Body ) < 2 {
return
}
branch , ok := clause . Body [ len ( clause . Body ) - 1 ] . ( * ast . BranchStmt )
if ! ok || branch . Tok != token . BREAK || branch . Label != nil {
return
}
2020-05-29 22:15:21 +02:00
report . Report ( pass , branch , "redundant break statement" , report . FilterGenerated ( ) )
2018-12-28 23:15:05 +01:00
}
fn2 := func ( node ast . Node ) {
var ret * ast . FieldList
var body * ast . BlockStmt
switch x := node . ( type ) {
case * ast . FuncDecl :
ret = x . Type . Results
body = x . Body
case * ast . FuncLit :
ret = x . Type . Results
body = x . Body
default :
2020-05-29 22:15:21 +02:00
ExhaustiveTypeSwitch ( node )
2018-12-28 23:15:05 +01:00
}
// if the func has results, a return can't be redundant.
// similarly, if there are no statements, there can be
// no return.
if ret != nil || body == nil || len ( body . List ) < 1 {
return
}
rst , ok := body . List [ len ( body . List ) - 1 ] . ( * ast . ReturnStmt )
if ! ok {
return
}
// we don't need to check rst.Results as we already
// checked x.Type.Results to be nil.
2020-05-29 22:15:21 +02:00
report . Report ( pass , rst , "redundant return statement" , report . FilterGenerated ( ) )
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn1 , ( * ast . CaseClause ) ( nil ) )
code . Preorder ( pass , fn2 , ( * ast . FuncDecl ) ( nil ) , ( * ast . FuncLit ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-09 15:44:17 +02:00
func isStringer ( T types . Type , msCache * typeutil . MethodSetCache ) bool {
ms := msCache . MethodSet ( T )
2019-04-22 12:59:42 +02:00
sel := ms . Lookup ( nil , "String" )
if sel == nil {
2018-12-28 23:15:05 +01:00
return false
}
2019-04-22 12:59:42 +02:00
fn , ok := sel . Obj ( ) . ( * types . Func )
2018-12-28 23:15:05 +01:00
if ! ok {
2019-04-22 12:59:42 +02:00
// should be unreachable
return false
}
sig := fn . Type ( ) . ( * types . Signature )
if sig . Params ( ) . Len ( ) != 0 {
2018-12-28 23:15:05 +01:00
return false
}
2019-04-22 12:59:42 +02:00
if sig . Results ( ) . Len ( ) != 1 {
return false
}
2020-05-29 22:15:21 +02:00
if ! code . IsType ( sig . Results ( ) . At ( 0 ) . Type ( ) , "string" ) {
2019-04-22 12:59:42 +02:00
return false
}
return true
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
var checkRedundantSprintfQ = pattern . MustParse ( ` (CallExpr (Function "fmt.Sprintf") [format arg]) ` )
func CheckRedundantSprintf ( pass * analysis . Pass ) ( interface { } , error ) {
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
2020-05-29 22:15:21 +02:00
m , ok := Match ( pass , checkRedundantSprintfQ , node )
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
format := m . State [ "format" ] . ( ast . Expr )
arg := m . State [ "arg" ] . ( ast . Expr )
if s , ok := code . ExprToString ( pass , format ) ; ! ok || s != "%s" {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-09 15:44:17 +02:00
typ := pass . TypesInfo . TypeOf ( arg )
2018-12-28 23:15:05 +01:00
2020-05-29 22:15:21 +02:00
irpkg := pass . ResultOf [ buildir . Analyzer ] . ( * buildir . IR ) . Pkg
if types . TypeString ( typ , nil ) != "reflect.Value" && isStringer ( typ , & irpkg . Prog . MethodSets ) {
replacement := & ast . CallExpr {
Fun : & ast . SelectorExpr {
X : arg ,
Sel : & ast . Ident { Name : "String" } ,
} ,
}
report . Report ( pass , node , "should use String() instead of fmt.Sprintf" ,
report . Fixes ( edit . Fix ( "replace with call to String method" , edit . ReplaceWithNode ( pass . Fset , node , replacement ) ) ) )
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
if typ . Underlying ( ) == types . Universe . Lookup ( "string" ) . Type ( ) {
if typ == types . Universe . Lookup ( "string" ) . Type ( ) {
2020-05-29 22:15:21 +02:00
report . Report ( pass , node , "the argument is already a string, there's no need to use fmt.Sprintf" ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "remove unnecessary call to fmt.Sprintf" , edit . ReplaceWithNode ( pass . Fset , node , arg ) ) ) )
2018-12-28 23:15:05 +01:00
} else {
2020-05-29 22:15:21 +02:00
replacement := & ast . CallExpr {
Fun : & ast . Ident { Name : "string" } ,
Args : [ ] ast . Expr { arg } ,
}
report . Report ( pass , node , "the argument's underlying type is a string, should use a simple conversion instead of fmt.Sprintf" ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "replace with conversion to string" , edit . ReplaceWithNode ( pass . Fset , node , replacement ) ) ) )
2018-12-28 23:15:05 +01:00
}
}
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . CallExpr ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
var (
checkErrorsNewSprintfQ = pattern . MustParse ( ` (CallExpr (Function "errors.New") [(CallExpr (Function "fmt.Sprintf") args)]) ` )
checkErrorsNewSprintfR = pattern . MustParse ( ` (CallExpr (SelectorExpr (Ident "fmt") (Ident "Errorf")) args) ` )
)
func CheckErrorsNewSprintf ( pass * analysis . Pass ) ( interface { } , error ) {
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
2020-05-29 22:15:21 +02:00
if _ , edits , ok := MatchAndEdit ( pass , checkErrorsNewSprintfQ , checkErrorsNewSprintfR , node ) ; ok {
// TODO(dh): the suggested fix may leave an unused import behind
report . Report ( pass , node , "should use fmt.Errorf(...) instead of errors.New(fmt.Sprintf(...))" ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "use fmt.Errorf" , edits ... ) ) )
2018-12-28 23:15:05 +01:00
}
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . CallExpr ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
func CheckRangeStringRunes ( pass * analysis . Pass ) ( interface { } , error ) {
2020-05-09 15:44:17 +02:00
return sharedcheck . CheckRangeStringRunes ( pass )
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
var checkNilCheckAroundRangeQ = pattern . MustParse ( `
( IfStmt
nil
( BinaryExpr x @ ( Object _ ) "!=" ( Builtin "nil" ) )
[ ( RangeStmt _ _ _ x _ ) ]
nil ) ` )
2018-12-28 23:15:05 +01:00
2020-05-29 22:15:21 +02:00
func CheckNilCheckAroundRange ( pass * analysis . Pass ) ( interface { } , error ) {
fn := func ( node ast . Node ) {
m , ok := Match ( pass , checkNilCheckAroundRangeQ , node )
2018-12-28 23:15:05 +01:00
if ! ok {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-29 22:15:21 +02:00
switch m . State [ "x" ] . ( types . Object ) . Type ( ) . Underlying ( ) . ( type ) {
2018-12-28 23:15:05 +01:00
case * types . Slice , * types . Map :
2020-05-29 22:15:21 +02:00
report . Report ( pass , node , "unnecessary nil check around range" ,
report . ShortRange ( ) ,
report . FilterGenerated ( ) )
2018-12-28 23:15:05 +01:00
}
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . IfStmt ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2020-05-09 15:44:17 +02:00
func isPermissibleSort ( pass * analysis . Pass , node ast . Node ) bool {
2018-12-28 23:15:05 +01:00
call := node . ( * ast . CallExpr )
typeconv , ok := call . Args [ 0 ] . ( * ast . CallExpr )
if ! ok {
return true
}
sel , ok := typeconv . Fun . ( * ast . SelectorExpr )
if ! ok {
return true
}
2020-05-29 22:15:21 +02:00
name := code . SelectorName ( pass , sel )
2018-12-28 23:15:05 +01:00
switch name {
case "sort.IntSlice" , "sort.Float64Slice" , "sort.StringSlice" :
default :
return true
}
return false
}
2020-05-29 22:15:21 +02:00
func CheckSortHelpers ( pass * analysis . Pass ) ( interface { } , error ) {
2020-05-09 15:44:17 +02:00
type Error struct {
node ast . Node
msg string
}
var allErrors [ ] Error
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
2018-12-28 23:15:05 +01:00
var body * ast . BlockStmt
switch node := node . ( type ) {
case * ast . FuncLit :
body = node . Body
case * ast . FuncDecl :
body = node . Body
default :
2020-05-29 22:15:21 +02:00
ExhaustiveTypeSwitch ( node )
2018-12-28 23:15:05 +01:00
}
if body == nil {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
var errors [ ] Error
permissible := false
fnSorts := func ( node ast . Node ) bool {
if permissible {
return false
}
2020-05-29 22:15:21 +02:00
if ! code . IsCallToAST ( pass , node , "sort.Sort" ) {
2018-12-28 23:15:05 +01:00
return true
}
2020-05-09 15:44:17 +02:00
if isPermissibleSort ( pass , node ) {
2018-12-28 23:15:05 +01:00
permissible = true
return false
}
call := node . ( * ast . CallExpr )
2019-02-18 20:32:41 +01:00
typeconv := call . Args [ Arg ( "sort.Sort.data" ) ] . ( * ast . CallExpr )
2018-12-28 23:15:05 +01:00
sel := typeconv . Fun . ( * ast . SelectorExpr )
2020-05-29 22:15:21 +02:00
name := code . SelectorName ( pass , sel )
2018-12-28 23:15:05 +01:00
switch name {
case "sort.IntSlice" :
errors = append ( errors , Error { node , "should use sort.Ints(...) instead of sort.Sort(sort.IntSlice(...))" } )
case "sort.Float64Slice" :
errors = append ( errors , Error { node , "should use sort.Float64s(...) instead of sort.Sort(sort.Float64Slice(...))" } )
case "sort.StringSlice" :
errors = append ( errors , Error { node , "should use sort.Strings(...) instead of sort.Sort(sort.StringSlice(...))" } )
}
return true
}
ast . Inspect ( body , fnSorts )
if permissible {
2019-04-22 12:59:42 +02:00
return
2018-12-28 23:15:05 +01:00
}
2020-05-09 15:44:17 +02:00
allErrors = append ( allErrors , errors ... )
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . FuncLit ) ( nil ) , ( * ast . FuncDecl ) ( nil ) )
2020-05-09 15:44:17 +02:00
sort . Slice ( allErrors , func ( i , j int ) bool {
return allErrors [ i ] . node . Pos ( ) < allErrors [ j ] . node . Pos ( )
} )
var prev token . Pos
for _ , err := range allErrors {
if err . node . Pos ( ) == prev {
continue
}
prev = err . node . Pos ( )
2020-05-29 22:15:21 +02:00
report . Report ( pass , err . node , err . msg , report . FilterGenerated ( ) )
2018-12-28 23:15:05 +01:00
}
2020-05-09 15:44:17 +02:00
return nil , nil
2018-12-28 23:15:05 +01:00
}
2019-02-18 20:32:41 +01:00
2020-05-29 22:15:21 +02:00
var checkGuardedDeleteQ = pattern . MustParse ( `
( IfStmt
( AssignStmt
[ ( Ident "_" ) ok @ ( Ident _ ) ]
":="
( IndexExpr m key ) )
ok
[ call @ ( CallExpr ( Builtin "delete" ) [ m key ] ) ]
nil ) ` )
func CheckGuardedDelete ( pass * analysis . Pass ) ( interface { } , error ) {
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
2020-05-29 22:15:21 +02:00
if m , ok := Match ( pass , checkGuardedDeleteQ , node ) ; ok {
report . Report ( pass , node , "unnecessary guard around call to delete" ,
report . ShortRange ( ) ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "remove guard" , edit . ReplaceWithNode ( pass . Fset , node , m . State [ "call" ] . ( ast . Node ) ) ) ) )
2019-02-18 20:32:41 +01:00
}
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . IfStmt ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2019-02-18 20:32:41 +01:00
}
2020-05-29 22:15:21 +02:00
var (
checkSimplifyTypeSwitchQ = pattern . MustParse ( `
( TypeSwitchStmt
nil
expr @ ( TypeAssertExpr ident @ ( Ident _ ) _ )
body ) ` )
checkSimplifyTypeSwitchR = pattern . MustParse ( ` (AssignStmt ident ":=" expr) ` )
)
func CheckSimplifyTypeSwitch ( pass * analysis . Pass ) ( interface { } , error ) {
2019-04-22 12:59:42 +02:00
fn := func ( node ast . Node ) {
2020-05-29 22:15:21 +02:00
m , ok := Match ( pass , checkSimplifyTypeSwitchQ , node )
2019-02-18 20:32:41 +01:00
if ! ok {
2019-04-22 12:59:42 +02:00
return
2019-02-18 20:32:41 +01:00
}
2020-05-29 22:15:21 +02:00
stmt := node . ( * ast . TypeSwitchStmt )
expr := m . State [ "expr" ] . ( ast . Node )
ident := m . State [ "ident" ] . ( * ast . Ident )
2020-05-09 15:44:17 +02:00
x := pass . TypesInfo . ObjectOf ( ident )
2020-05-29 22:15:21 +02:00
var allOffenders [ ] * ast . TypeAssertExpr
canSuggestFix := true
2019-02-18 20:32:41 +01:00
for _ , clause := range stmt . Body . List {
clause := clause . ( * ast . CaseClause )
if len ( clause . List ) != 1 {
continue
}
hasUnrelatedAssertion := false
2020-05-29 22:15:21 +02:00
var offenders [ ] * ast . TypeAssertExpr
2019-02-18 20:32:41 +01:00
ast . Inspect ( clause , func ( node ast . Node ) bool {
assert2 , ok := node . ( * ast . TypeAssertExpr )
if ! ok {
return true
}
ident , ok := assert2 . X . ( * ast . Ident )
if ! ok {
hasUnrelatedAssertion = true
return false
}
2020-05-09 15:44:17 +02:00
if pass . TypesInfo . ObjectOf ( ident ) != x {
2019-02-18 20:32:41 +01:00
hasUnrelatedAssertion = true
return false
}
2020-05-09 15:44:17 +02:00
if ! types . Identical ( pass . TypesInfo . TypeOf ( clause . List [ 0 ] ) , pass . TypesInfo . TypeOf ( assert2 . Type ) ) {
2019-02-18 20:32:41 +01:00
hasUnrelatedAssertion = true
return false
}
offenders = append ( offenders , assert2 )
return true
} )
if ! hasUnrelatedAssertion {
// don't flag cases that have other type assertions
// unrelated to the one in the case clause. often
// times, this is done for symmetry, when two
// different values have to be asserted to the same
// type.
allOffenders = append ( allOffenders , offenders ... )
}
2020-05-29 22:15:21 +02:00
canSuggestFix = canSuggestFix && ! hasUnrelatedAssertion
2019-02-18 20:32:41 +01:00
}
if len ( allOffenders ) != 0 {
2020-05-29 22:15:21 +02:00
var opts [ ] report . Option
2019-02-18 20:32:41 +01:00
for _ , offender := range allOffenders {
2020-05-29 22:15:21 +02:00
opts = append ( opts , report . Related ( offender , "could eliminate this type assertion" ) )
}
opts = append ( opts , report . FilterGenerated ( ) )
msg := fmt . Sprintf ( "assigning the result of this type assertion to a variable (switch %s := %s.(type)) could eliminate type assertions in switch cases" ,
report . Render ( pass , ident ) , report . Render ( pass , ident ) )
if canSuggestFix {
var edits [ ] analysis . TextEdit
edits = append ( edits , edit . ReplaceWithPattern ( pass , checkSimplifyTypeSwitchR , m . State , expr ) )
for _ , offender := range allOffenders {
edits = append ( edits , edit . ReplaceWithNode ( pass . Fset , offender , offender . X ) )
}
opts = append ( opts , report . Fixes ( edit . Fix ( "simplify type switch" , edits ... ) ) )
report . Report ( pass , expr , msg , opts ... )
} else {
report . Report ( pass , expr , msg , opts ... )
}
}
}
code . Preorder ( pass , fn , ( * ast . TypeSwitchStmt ) ( nil ) )
return nil , nil
}
func CheckRedundantCanonicalHeaderKey ( pass * analysis . Pass ) ( interface { } , error ) {
fn := func ( node ast . Node ) {
call := node . ( * ast . CallExpr )
callName := code . CallNameAST ( pass , call )
switch callName {
case "(net/http.Header).Add" , "(net/http.Header).Del" , "(net/http.Header).Get" , "(net/http.Header).Set" :
default :
return
}
if ! code . IsCallToAST ( pass , call . Args [ 0 ] , "net/http.CanonicalHeaderKey" ) {
return
}
report . Report ( pass , call ,
fmt . Sprintf ( "calling net/http.CanonicalHeaderKey on the 'key' argument of %s is redundant" , callName ) ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "remove call to CanonicalHeaderKey" , edit . ReplaceWithNode ( pass . Fset , call . Args [ 0 ] , call . Args [ 0 ] . ( * ast . CallExpr ) . Args [ 0 ] ) ) ) )
}
code . Preorder ( pass , fn , ( * ast . CallExpr ) ( nil ) )
return nil , nil
}
var checkUnnecessaryGuardQ = pattern . MustParse ( `
( Or
( IfStmt
( AssignStmt [ ( Ident "_" ) ok @ ( Ident _ ) ] ":=" indexexpr @ ( IndexExpr _ _ ) )
ok
set @ ( AssignStmt indexexpr "=" ( CallExpr ( Builtin "append" ) indexexpr : values ) )
( AssignStmt indexexpr "=" ( CompositeLit _ values ) ) )
( IfStmt
( AssignStmt [ ( Ident "_" ) ok ] ":=" indexexpr @ ( IndexExpr _ _ ) )
ok
set @ ( AssignStmt indexexpr "+=" value )
( AssignStmt indexexpr "=" value ) )
( IfStmt
( AssignStmt [ ( Ident "_" ) ok ] ":=" indexexpr @ ( IndexExpr _ _ ) )
ok
set @ ( IncDecStmt indexexpr "++" )
( AssignStmt indexexpr "=" ( BasicLit "INT" "1" ) ) ) ) ` )
func CheckUnnecessaryGuard ( pass * analysis . Pass ) ( interface { } , error ) {
fn := func ( node ast . Node ) {
if m , ok := Match ( pass , checkUnnecessaryGuardQ , node ) ; ok {
if code . MayHaveSideEffects ( pass , m . State [ "indexexpr" ] . ( ast . Expr ) , nil ) {
return
}
report . Report ( pass , node , "unnecessary guard around map access" ,
report . ShortRange ( ) ,
report . Fixes ( edit . Fix ( "simplify map access" , edit . ReplaceWithNode ( pass . Fset , node , m . State [ "set" ] . ( ast . Node ) ) ) ) )
}
}
code . Preorder ( pass , fn , ( * ast . IfStmt ) ( nil ) )
return nil , nil
}
var (
checkElaborateSleepQ = pattern . MustParse ( ` (SelectStmt (CommClause (UnaryExpr "<-" (CallExpr (Function "time.After") [arg])) body)) ` )
checkElaborateSleepR = pattern . MustParse ( ` (CallExpr (SelectorExpr (Ident "time") (Ident "Sleep")) [arg]) ` )
)
func CheckElaborateSleep ( pass * analysis . Pass ) ( interface { } , error ) {
fn := func ( node ast . Node ) {
if m , ok := Match ( pass , checkElaborateSleepQ , node ) ; ok {
if body , ok := m . State [ "body" ] . ( [ ] ast . Stmt ) ; ok && len ( body ) == 0 {
report . Report ( pass , node , "should use time.Sleep instead of elaborate way of sleeping" ,
report . ShortRange ( ) ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "Use time.Sleep" , edit . ReplaceWithPattern ( pass , checkElaborateSleepR , m . State , node ) ) ) )
} else {
// TODO(dh): we could make a suggested fix if the body
// doesn't declare or shadow any identifiers
report . Report ( pass , node , "should use time.Sleep instead of elaborate way of sleeping" ,
report . ShortRange ( ) ,
report . FilterGenerated ( ) )
}
}
}
code . Preorder ( pass , fn , ( * ast . SelectStmt ) ( nil ) )
return nil , nil
}
var checkPrintSprintQ = pattern . MustParse ( `
( Or
( CallExpr
fn @ ( Or
( Function "fmt.Print" )
( Function "fmt.Sprint" )
( Function "fmt.Println" )
( Function "fmt.Sprintln" ) )
[ ( CallExpr ( Function "fmt.Sprintf" ) f : _ ) ] )
( CallExpr
fn @ ( Or
( Function "fmt.Fprint" )
( Function "fmt.Fprintln" ) )
[ _ ( CallExpr ( Function "fmt.Sprintf" ) f : _ ) ] ) ) ` )
func CheckPrintSprintf ( pass * analysis . Pass ) ( interface { } , error ) {
fn := func ( node ast . Node ) {
m , ok := Match ( pass , checkPrintSprintQ , node )
if ! ok {
return
}
name := m . State [ "fn" ] . ( * types . Func ) . Name ( )
var msg string
switch name {
case "Print" , "Fprint" , "Sprint" :
newname := name + "f"
msg = fmt . Sprintf ( "should use fmt.%s instead of fmt.%s(fmt.Sprintf(...))" , newname , name )
case "Println" , "Fprintln" , "Sprintln" :
if _ , ok := m . State [ "f" ] . ( * ast . BasicLit ) ; ! ok {
// This may be an instance of
// fmt.Println(fmt.Sprintf(arg, ...)) where arg is an
// externally provided format string and the caller
// cannot guarantee that the format string ends with a
// newline.
return
}
newname := name [ : len ( name ) - 2 ] + "f"
msg = fmt . Sprintf ( "should use fmt.%s instead of fmt.%s(fmt.Sprintf(...)) (but don't forget the newline)" , newname , name )
}
report . Report ( pass , node , msg ,
report . FilterGenerated ( ) )
}
code . Preorder ( pass , fn , ( * ast . CallExpr ) ( nil ) )
return nil , nil
}
var checkSprintLiteralQ = pattern . MustParse ( `
( CallExpr
fn @ ( Or
( Function "fmt.Sprint" )
( Function "fmt.Sprintf" ) )
[ lit @ ( BasicLit "STRING" _ ) ] ) ` )
func CheckSprintLiteral ( pass * analysis . Pass ) ( interface { } , error ) {
// We only flag calls with string literals, not expressions of
// type string, because some people use fmt.Sprint(s) as a pattern
// for copying strings, which may be useful when extracing a small
// substring from a large string.
fn := func ( node ast . Node ) {
m , ok := Match ( pass , checkSprintLiteralQ , node )
if ! ok {
return
}
callee := m . State [ "fn" ] . ( * types . Func )
lit := m . State [ "lit" ] . ( * ast . BasicLit )
if callee . Name ( ) == "Sprintf" {
if strings . ContainsRune ( lit . Value , '%' ) {
// This might be a format string
return
2019-02-18 20:32:41 +01:00
}
}
2020-05-29 22:15:21 +02:00
report . Report ( pass , node , fmt . Sprintf ( "unnecessary use of fmt.%s" , callee . Name ( ) ) ,
report . FilterGenerated ( ) ,
report . Fixes ( edit . Fix ( "Replace with string literal" , edit . ReplaceWithNode ( pass . Fset , node , lit ) ) ) )
2019-02-18 20:32:41 +01:00
}
2020-05-29 22:15:21 +02:00
code . Preorder ( pass , fn , ( * ast . CallExpr ) ( nil ) )
2020-05-09 15:44:17 +02:00
return nil , nil
2019-02-18 20:32:41 +01:00
}